rms/org.eclipse.ptp.rm.jaxb.tests/src/org/eclipse/ptp/rm/jaxb/tests/RMLaunchTest.java - ptp/org.eclipse.ptp - Git at Google

 /*******************************************************************************
  * Copyright (c) 2011 University of Illinois All rights reserved. This program
  * and the accompanying materials are made available under the terms of the
  * Eclipse Public License v1.0 which accompanies this distribution, and is
  * available at http://www.eclipse.org/legal/epl-v10.html
  *
  * Contributors:
  * 	Albert L. Rossi - design and implementation
  ******************************************************************************/

 package org.eclipse.ptp.rm.jaxb.tests;

 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.TreeMap;

 import junit.framework.TestCase;

 import org.eclipse.core.resources.IFile;
 import org.eclipse.core.resources.IResource;
 import org.eclipse.core.runtime.CoreException;
 import org.eclipse.core.runtime.IPath;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.NullProgressMonitor;
 import org.eclipse.debug.core.ILaunch;
 import org.eclipse.debug.core.ILaunchConfiguration;
 import org.eclipse.debug.core.ILaunchConfigurationType;
 import org.eclipse.debug.core.ILaunchConfigurationWorkingCopy;
 import org.eclipse.debug.core.ILaunchDelegate;
 import org.eclipse.debug.core.ILaunchManager;
 import org.eclipse.debug.core.IStreamListener;
 import org.eclipse.debug.core.model.IStreamMonitor;
 import org.eclipse.ptp.core.IPTPLaunchConfigurationConstants;
 import org.eclipse.ptp.core.PTPCorePlugin;
 import org.eclipse.ptp.remote.core.IRemoteConnection;
 import org.eclipse.ptp.remote.core.IRemoteConnectionManager;
 import org.eclipse.ptp.remote.core.IRemoteServices;
 import org.eclipse.ptp.remote.core.PTPRemoteCorePlugin;
 import org.eclipse.ptp.rm.jaxb.core.IJAXBNonNLSConstants;
 import org.eclipse.ptp.rm.jaxb.core.data.PropertyType;
 import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManager;
 import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManagerConfiguration;
 import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManagerControl;
 import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManagerMonitor;
 import org.eclipse.ptp.rm.jaxb.core.variables.RMVariableMap;
 import org.eclipse.ptp.rmsystem.AbstractResourceManagerConfiguration;
 import org.eclipse.ptp.rmsystem.IJobStatus;
 import org.eclipse.ptp.rmsystem.ResourceManagerServiceProvider;

 public class RMLaunchTest extends TestCase implements IJAXBNonNLSConstants {

 	private class TestLaunchConfiguration implements ILaunchConfiguration {

 		private final Map<String, Object> store = new TreeMap<String, Object>();

 		public boolean contentsEqual(ILaunchConfiguration configuration) {
 			return false;
 		}

 		public ILaunchConfigurationWorkingCopy copy(String name) throws CoreException {
 			return null;
 		}

 		public void delete() throws CoreException {
 		}

 		public boolean exists() {
 			return false;
 		}

 		@SuppressWarnings("rawtypes")
 		public Object getAdapter(Class adapter) {
 			return null;
 		}

 		public boolean getAttribute(String attributeName, boolean defaultValue) throws CoreException {
 			Boolean value = (Boolean) store.get(attributeName);
 			if (value == null) {
 				value = defaultValue;
 			}
 			return value;
 		}

 		public int getAttribute(String attributeName, int defaultValue) throws CoreException {
 			Integer value = (Integer) store.get(attributeName);
 			if (value == null) {
 				value = defaultValue;
 			}
 			return value;
 		}

 		@SuppressWarnings("rawtypes")
 		public List getAttribute(String attributeName, List defaultValue) throws CoreException {
 			List value = (List) store.get(attributeName);
 			if (value == null) {
 				value = defaultValue;
 			}
 			return value;
 		}

 		@SuppressWarnings("rawtypes")
 		public Map getAttribute(String attributeName, Map defaultValue) throws CoreException {
 			Map value = (Map) store.get(attributeName);
 			if (value == null) {
 				value = defaultValue;
 			}
 			return value;
 		}

 		@SuppressWarnings("rawtypes")
 		public Set getAttribute(String attributeName, Set defaultValue) throws CoreException {
 			Set value = (Set) store.get(attributeName);
 			if (value == null) {
 				value = defaultValue;
 			}
 			return value;
 		}

 		public String getAttribute(String attributeName, String defaultValue) throws CoreException {
 			String value = (String) store.get(attributeName);
 			if (value == null) {
 				value = defaultValue;
 			}
 			return value;
 		}

 		@SuppressWarnings("rawtypes")
 		public Map getAttributes() throws CoreException {
 			return store;
 		}

 		public String getCategory() throws CoreException {
 			return null;
 		}

 		public IFile getFile() {
 			return null;
 		}

 		public IPath getLocation() {
 			return null;
 		}

 		public IResource[] getMappedResources() throws CoreException {
 			return null;
 		}

 		public String getMemento() throws CoreException {
 			return null;
 		}

 		@SuppressWarnings("rawtypes")
 		public Set getModes() throws CoreException {
 			return null;
 		}

 		public String getName() {
 			return null;
 		}

 		@SuppressWarnings("rawtypes")
 		public ILaunchDelegate getPreferredDelegate(Set modes) throws CoreException {
 			return null;
 		}

 		public ILaunchConfigurationType getType() throws CoreException {
 			return null;
 		}

 		public ILaunchConfigurationWorkingCopy getWorkingCopy() throws CoreException {
 			return null;
 		}

 		public boolean hasAttribute(String attributeName) throws CoreException {
 			return false;
 		}

 		public boolean isLocal() {
 			return false;
 		}

 		public boolean isMigrationCandidate() throws CoreException {
 			return false;
 		}

 		public boolean isReadOnly() {
 			return false;
 		}

 		public boolean isWorkingCopy() {
 			return false;
 		}

 		public ILaunch launch(String mode, IProgressMonitor monitor) throws CoreException {
 			return null;
 		}

 		public ILaunch launch(String mode, IProgressMonitor monitor, boolean build) throws CoreException {
 			return null;
 		}

 		public ILaunch launch(String mode, IProgressMonitor monitor, boolean build, boolean register) throws CoreException {
 			return null;
 		}

 		public void migrate() throws CoreException {

 		}

 		public boolean supportsMode(String mode) throws CoreException {
 			return false;
 		}
 	}

 	private static final String xml = DATA + "test-pbs.xml"; //$NON-NLS-1$
 	private JAXBResourceManagerConfiguration rmConfig;
 	private JAXBResourceManager rm;
 	private ILaunchConfiguration launchConfig;

 	@Override
 	public void setUp() {
 		/*
 		 * You will need to copy all the executables in the
 		 * org.eclipse.ptp.rm.jaxb.tests/data directory to you home; it seems
 		 * the JUnit plugin runner does not actually execute in the directory
 		 * indicated by the Run Configuration.
 		 */
 	}

 	@Override
 	public void tearDown() {
 	}

 	public void testResourceManager() {
 		try {
 			emulateConfigureWizard();
 			rm = new JAXBResourceManager(rmConfig, new JAXBResourceManagerControl(rmConfig), new JAXBResourceManagerMonitor(
 					rmConfig));
 			PTPCorePlugin.getDefault().getModelManager().addResourceManager(rm);
 			rm.start(new NullProgressMonitor());
 			try {
 				Thread.sleep(2000);
 			} catch (InterruptedException ignored) {
 			}
 			emulateLaunchTab();
 			String jobId = rm.submitJob(launchConfig, ILaunchManager.RUN_MODE, new NullProgressMonitor());
 			System.out.println("SUBMITTED: " + jobId); //$NON-NLS-1$
 			IJobStatus status = rm.getJobStatus(jobId);
 			System.out.println("STATUS: " + status.getState()); //$NON-NLS-1$
 			if (status != null) {
 				status.getStreamsProxy().getOutputStreamMonitor().addListener(new IStreamListener() {
 					public void streamAppended(String text, IStreamMonitor monitor) {
 						System.out.println(text);
 					}
 				});
 			}
 			rm.stop();
 		} catch (Throwable t) {
 			t.printStackTrace();
 			assertNotNull(t);
 		}
 	}

 	/*
 	 * We do here what is done through the wizard.
 	 */
 	private void emulateConfigureWizard() throws Throwable {
 		rmConfig = new JAXBResourceManagerConfiguration(AbstractResourceManagerConfiguration.BASE,
 				new ResourceManagerServiceProvider());
 		// JAXBRMConfigurationSelectionWizardPage
 		rmConfig.setRMConfigurationURL(JAXBTestsPlugin.getURL(xml));
 		// JAXBRMControlConfigurationWizardPage
 		rmConfig.getResourceManagerData();
 		// use remote = local
 		IRemoteServices localServices = PTPRemoteCorePlugin.getDefault().getDefaultServices();
 		assert (localServices != null);
 		IRemoteConnectionManager localConnectionManager = localServices.getConnectionManager();
 		assert (localConnectionManager != null);
 		IRemoteConnection localConnection = localConnectionManager.getConnection(ZEROSTR);
 		assert (localConnection != null);
 		rmConfig.setRemoteServicesId(localServices.getId());
 		rmConfig.setConnectionName(localConnection.getName());
 	}

 	@SuppressWarnings("unchecked")
 	private void emulateLaunchTab() throws Throwable {
 		launchConfig = new TestLaunchConfiguration();
 		Map<Object, Object> env = launchConfig.getAttributes();
 		Map<String, String> live = new HashMap<String, String>();
 		live.put("FOO_VAR_1", "FOO_VALUE_1"); //$NON-NLS-1$ //$NON-NLS-2$
 		live.put("FOO_VAR_2", "FOO_VALUE_2"); //$NON-NLS-1$ //$NON-NLS-2$
 		live.put("FOO_VAR_3", "FOO_VALUE_3"); //$NON-NLS-1$ //$NON-NLS-2$
 		env.put(ILaunchManager.ATTR_ENVIRONMENT_VARIABLES, live);
 		env.put(ILaunchManager.ATTR_APPEND_ENVIRONMENT_VARIABLES, true);
 		env.put(IPTPLaunchConfigurationConstants.ATTR_WORKING_DIR, "/u/ncsa/arossi/test"); //$NON-NLS-1$
 		env.put(IPTPLaunchConfigurationConstants.ATTR_EXECUTABLE_PATH, "/u/ncsa/arossi/test/foo"); //$NON-NLS-1$
 		env.put(IPTPLaunchConfigurationConstants.ATTR_ARGUMENTS, "-v -f /u/ncsa/arossi/test/data"); //$NON-NLS-1$
 		env.put("Job_Name", "TestRMLaunch"); //$NON-NLS-1$ //$NON-NLS-2$
 		env.put("Resource_List.nodes", "1:ppn=8"); //$NON-NLS-1$ //$NON-NLS-2$
 		env.put("Resource_List.walltime", "00:10:00"); //$NON-NLS-1$ //$NON-NLS-2$
 		env.put("export_all", true); //$NON-NLS-1$
 		env.put(MPI_CMD, "mpiexec"); //$NON-NLS-1$
 		env.put(MPI_ARGS, "-machinefile $PBS_NODEFILE -np 8"); //$NON-NLS-1$
 		PropertyType queues = (PropertyType) RMVariableMap.getActiveInstance().getVariables().get("available_queues"); //$NON-NLS-1$
 		if (queues != null) {
 			List<String> q = (List<String>) queues.getValue();
 			env.put("destination", q.get(0)); //$NON-NLS-1$
 		}
 		env.put("directory", "/Users/arossi"); //$NON-NLS-1$//$NON-NLS-2$
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2011 University of Illinois All rights reserved. This program
	* and the accompanying materials are made available under the terms of the
	* Eclipse Public License v1.0 which accompanies this distribution, and is
	* available at http://www.eclipse.org/legal/epl-v10.html
	*
	* Contributors:
	* Albert L. Rossi - design and implementation
	******************************************************************************/

	package org.eclipse.ptp.rm.jaxb.tests;

	import java.util.HashMap;
	import java.util.List;
	import java.util.Map;
	import java.util.Set;
	import java.util.TreeMap;

	import junit.framework.TestCase;

	import org.eclipse.core.resources.IFile;
	import org.eclipse.core.resources.IResource;
	import org.eclipse.core.runtime.CoreException;
	import org.eclipse.core.runtime.IPath;
	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.core.runtime.NullProgressMonitor;
	import org.eclipse.debug.core.ILaunch;
	import org.eclipse.debug.core.ILaunchConfiguration;
	import org.eclipse.debug.core.ILaunchConfigurationType;
	import org.eclipse.debug.core.ILaunchConfigurationWorkingCopy;
	import org.eclipse.debug.core.ILaunchDelegate;
	import org.eclipse.debug.core.ILaunchManager;
	import org.eclipse.debug.core.IStreamListener;
	import org.eclipse.debug.core.model.IStreamMonitor;
	import org.eclipse.ptp.core.IPTPLaunchConfigurationConstants;
	import org.eclipse.ptp.core.PTPCorePlugin;
	import org.eclipse.ptp.remote.core.IRemoteConnection;
	import org.eclipse.ptp.remote.core.IRemoteConnectionManager;
	import org.eclipse.ptp.remote.core.IRemoteServices;
	import org.eclipse.ptp.remote.core.PTPRemoteCorePlugin;
	import org.eclipse.ptp.rm.jaxb.core.IJAXBNonNLSConstants;
	import org.eclipse.ptp.rm.jaxb.core.data.PropertyType;
	import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManager;
	import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManagerConfiguration;
	import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManagerControl;
	import org.eclipse.ptp.rm.jaxb.core.rm.JAXBResourceManagerMonitor;
	import org.eclipse.ptp.rm.jaxb.core.variables.RMVariableMap;
	import org.eclipse.ptp.rmsystem.AbstractResourceManagerConfiguration;
	import org.eclipse.ptp.rmsystem.IJobStatus;
	import org.eclipse.ptp.rmsystem.ResourceManagerServiceProvider;

	public class RMLaunchTest extends TestCase implements IJAXBNonNLSConstants {

	private class TestLaunchConfiguration implements ILaunchConfiguration {

	private final Map<String, Object> store = new TreeMap<String, Object>();

	public boolean contentsEqual(ILaunchConfiguration configuration) {
	return false;
	}

	public ILaunchConfigurationWorkingCopy copy(String name) throws CoreException {
	return null;
	}

	public void delete() throws CoreException {
	}

	public boolean exists() {
	return false;
	}

	@SuppressWarnings("rawtypes")
	public Object getAdapter(Class adapter) {
	return null;
	}

	public boolean getAttribute(String attributeName, boolean defaultValue) throws CoreException {
	Boolean value = (Boolean) store.get(attributeName);
	if (value == null) {
	value = defaultValue;
	}
	return value;
	}

	public int getAttribute(String attributeName, int defaultValue) throws CoreException {
	Integer value = (Integer) store.get(attributeName);
	if (value == null) {
	value = defaultValue;
	}
	return value;
	}

	@SuppressWarnings("rawtypes")
	public List getAttribute(String attributeName, List defaultValue) throws CoreException {
	List value = (List) store.get(attributeName);
	if (value == null) {
	value = defaultValue;
	}
	return value;
	}

	@SuppressWarnings("rawtypes")
	public Map getAttribute(String attributeName, Map defaultValue) throws CoreException {
	Map value = (Map) store.get(attributeName);
	if (value == null) {
	value = defaultValue;
	}
	return value;
	}

	@SuppressWarnings("rawtypes")
	public Set getAttribute(String attributeName, Set defaultValue) throws CoreException {
	Set value = (Set) store.get(attributeName);
	if (value == null) {
	value = defaultValue;
	}
	return value;
	}

	public String getAttribute(String attributeName, String defaultValue) throws CoreException {
	String value = (String) store.get(attributeName);
	if (value == null) {
	value = defaultValue;
	}
	return value;
	}

	@SuppressWarnings("rawtypes")
	public Map getAttributes() throws CoreException {
	return store;
	}

	public String getCategory() throws CoreException {
	return null;
	}

	public IFile getFile() {
	return null;
	}

	public IPath getLocation() {
	return null;
	}

	public IResource[] getMappedResources() throws CoreException {
	return null;
	}

	public String getMemento() throws CoreException {
	return null;
	}

	@SuppressWarnings("rawtypes")
	public Set getModes() throws CoreException {
	return null;
	}

	public String getName() {
	return null;
	}

	@SuppressWarnings("rawtypes")
	public ILaunchDelegate getPreferredDelegate(Set modes) throws CoreException {
	return null;
	}

	public ILaunchConfigurationType getType() throws CoreException {
	return null;
	}

	public ILaunchConfigurationWorkingCopy getWorkingCopy() throws CoreException {
	return null;
	}

	public boolean hasAttribute(String attributeName) throws CoreException {
	return false;
	}

	public boolean isLocal() {
	return false;
	}

	public boolean isMigrationCandidate() throws CoreException {
	return false;
	}

	public boolean isReadOnly() {
	return false;
	}

	public boolean isWorkingCopy() {
	return false;
	}

	public ILaunch launch(String mode, IProgressMonitor monitor) throws CoreException {
	return null;
	}

	public ILaunch launch(String mode, IProgressMonitor monitor, boolean build) throws CoreException {
	return null;
	}

	public ILaunch launch(String mode, IProgressMonitor monitor, boolean build, boolean register) throws CoreException {
	return null;
	}

	public void migrate() throws CoreException {

	}

	public boolean supportsMode(String mode) throws CoreException {
	return false;
	}
	}

	private static final String xml = DATA + "test-pbs.xml"; //$NON-NLS-1$
	private JAXBResourceManagerConfiguration rmConfig;
	private JAXBResourceManager rm;
	private ILaunchConfiguration launchConfig;

	@Override
	public void setUp() {
	/*
	* You will need to copy all the executables in the
	* org.eclipse.ptp.rm.jaxb.tests/data directory to you home; it seems
	* the JUnit plugin runner does not actually execute in the directory
	* indicated by the Run Configuration.
	*/
	}

	@Override
	public void tearDown() {
	}

	public void testResourceManager() {
	try {
	emulateConfigureWizard();
	rm = new JAXBResourceManager(rmConfig, new JAXBResourceManagerControl(rmConfig), new JAXBResourceManagerMonitor(
	rmConfig));
	PTPCorePlugin.getDefault().getModelManager().addResourceManager(rm);
	rm.start(new NullProgressMonitor());
	try {
	Thread.sleep(2000);
	} catch (InterruptedException ignored) {
	}
	emulateLaunchTab();
	String jobId = rm.submitJob(launchConfig, ILaunchManager.RUN_MODE, new NullProgressMonitor());
	System.out.println("SUBMITTED: " + jobId); //$NON-NLS-1$
	IJobStatus status = rm.getJobStatus(jobId);
	System.out.println("STATUS: " + status.getState()); //$NON-NLS-1$
	if (status != null) {
	status.getStreamsProxy().getOutputStreamMonitor().addListener(new IStreamListener() {
	public void streamAppended(String text, IStreamMonitor monitor) {
	System.out.println(text);
	}
	});
	}
	rm.stop();
	} catch (Throwable t) {
	t.printStackTrace();
	assertNotNull(t);
	}
	}

	/*
	* We do here what is done through the wizard.
	*/
	private void emulateConfigureWizard() throws Throwable {
	rmConfig = new JAXBResourceManagerConfiguration(AbstractResourceManagerConfiguration.BASE,
	new ResourceManagerServiceProvider());
	// JAXBRMConfigurationSelectionWizardPage
	rmConfig.setRMConfigurationURL(JAXBTestsPlugin.getURL(xml));
	// JAXBRMControlConfigurationWizardPage
	rmConfig.getResourceManagerData();
	// use remote = local
	IRemoteServices localServices = PTPRemoteCorePlugin.getDefault().getDefaultServices();
	assert (localServices != null);
	IRemoteConnectionManager localConnectionManager = localServices.getConnectionManager();
	assert (localConnectionManager != null);
	IRemoteConnection localConnection = localConnectionManager.getConnection(ZEROSTR);
	assert (localConnection != null);
	rmConfig.setRemoteServicesId(localServices.getId());
	rmConfig.setConnectionName(localConnection.getName());
	}

	@SuppressWarnings("unchecked")
	private void emulateLaunchTab() throws Throwable {
	launchConfig = new TestLaunchConfiguration();
	Map<Object, Object> env = launchConfig.getAttributes();
	Map<String, String> live = new HashMap<String, String>();
	live.put("FOO_VAR_1", "FOO_VALUE_1"); //$NON-NLS-1$ //$NON-NLS-2$
	live.put("FOO_VAR_2", "FOO_VALUE_2"); //$NON-NLS-1$ //$NON-NLS-2$
	live.put("FOO_VAR_3", "FOO_VALUE_3"); //$NON-NLS-1$ //$NON-NLS-2$
	env.put(ILaunchManager.ATTR_ENVIRONMENT_VARIABLES, live);
	env.put(ILaunchManager.ATTR_APPEND_ENVIRONMENT_VARIABLES, true);
	env.put(IPTPLaunchConfigurationConstants.ATTR_WORKING_DIR, "/u/ncsa/arossi/test"); //$NON-NLS-1$
	env.put(IPTPLaunchConfigurationConstants.ATTR_EXECUTABLE_PATH, "/u/ncsa/arossi/test/foo"); //$NON-NLS-1$
	env.put(IPTPLaunchConfigurationConstants.ATTR_ARGUMENTS, "-v -f /u/ncsa/arossi/test/data"); //$NON-NLS-1$
	env.put("Job_Name", "TestRMLaunch"); //$NON-NLS-1$ //$NON-NLS-2$
	env.put("Resource_List.nodes", "1:ppn=8"); //$NON-NLS-1$ //$NON-NLS-2$
	env.put("Resource_List.walltime", "00:10:00"); //$NON-NLS-1$ //$NON-NLS-2$
	env.put("export_all", true); //$NON-NLS-1$
	env.put(MPI_CMD, "mpiexec"); //$NON-NLS-1$
	env.put(MPI_ARGS, "-machinefile $PBS_NODEFILE -np 8"); //$NON-NLS-1$
	PropertyType queues = (PropertyType) RMVariableMap.getActiveInstance().getVariables().get("available_queues"); //$NON-NLS-1$
	if (queues != null) {
	List<String> q = (List<String>) queues.getValue();
	env.put("destination", q.get(0)); //$NON-NLS-1$
	}
	env.put("directory", "/Users/arossi"); //$NON-NLS-1$//$NON-NLS-2$
	}
	}