JMXExamplesTest.java

package org.oddjob.jmx;

import java.io.File;
import java.util.Properties;

import junit.framework.TestCase;

import org.apache.log4j.Logger;
import org.oddjob.FailedToStopException;
import org.oddjob.Oddjob;
import org.oddjob.OddjobLookup;
import org.oddjob.Stateful;
import org.oddjob.arooa.ArooaParseException;
import org.oddjob.arooa.convert.ArooaConversionException;
import org.oddjob.arooa.reflect.ArooaPropertyException;
import org.oddjob.state.JobState;
import org.oddjob.state.ParentState;
import org.oddjob.state.ServiceState;
import org.oddjob.tools.FragmentHelper;
import org.oddjob.tools.OurDirs;
import org.oddjob.tools.StateSteps;

public class JMXExamplesTest extends TestCase {

	private static final Logger logger = Logger.getLogger(JMXExamplesTest.class);

	Oddjob serverOddjob;
	Oddjob clientOddjob;
	
	@Override
	protected void setUp() throws Exception {
		super.setUp();
		
		logger.info("----------------  " + getName() + "  -----------------");
	}

	@Override
	protected void tearDown() throws Exception {
		super.tearDown();
		
		if (clientOddjob != null) {
			clientOddjob.destroy();
		}
		if (serverOddjob != null) {
			serverOddjob.destroy();
		}
	}
	
	public void testSimpleClientServerExample() throws ArooaParseException, FailedToStopException {
		
		Properties props = new Properties();
		props.setProperty("hosts.freds.pc", "localhost");
		
		OurDirs dirs = new OurDirs();
		
		File testDir = dirs.relative("test/java/org/oddjob/jmx");
		
		serverOddjob = new Oddjob();
		serverOddjob.setFile(new File(testDir, "ServerExample.xml"));

		serverOddjob.run();
		
		assertEquals(ParentState.STARTED, 
				serverOddjob.lastStateEvent().getState());
		
		FragmentHelper helper = new FragmentHelper();
		helper.setProperties(props);
		
		JMXClientJob client = (JMXClientJob) helper.createComponentFromResource(
				"org/oddjob/jmx/ClientExample.xml");

		StateSteps clientSteps = new StateSteps(client);
		clientSteps.startCheck(ServiceState.STARTABLE, ServiceState.STARTING, 
				ServiceState.STARTED);
		
		client.run();
		
		clientSteps.checkNow();
		
		clientSteps.startCheck(ServiceState.STARTED, ServiceState.STOPPED);
		
		client.stop();
		
		clientSteps.checkNow();
	}
	
	public void testClientRunsServerJobExample() throws InterruptedException, ArooaPropertyException, ArooaConversionException {
		
		Properties props = new Properties();
		props.setProperty("hosts.freds.pc", "localhost");
		
		OurDirs dirs = new OurDirs();
		
		File testDir = dirs.relative("test/java/org/oddjob/jmx");
		
		serverOddjob = new Oddjob();
		serverOddjob.setProperties(props);
		serverOddjob.setFile(new File(testDir, "ServerExample.xml"));

		serverOddjob.run();
		
		assertEquals(ParentState.STARTED, 
				serverOddjob.lastStateEvent().getState());
		
		OddjobLookup serverLookup = new OddjobLookup(serverOddjob);
		
		Stateful serverJob = serverLookup.lookup("server-jobs/greeting", 
				Stateful.class);
	
		clientOddjob = new Oddjob();
		clientOddjob.setProperties(props);
		clientOddjob.setFile(new File(testDir, "ClientRunsServerJob.xml"));

		StateSteps serverJobStates = new StateSteps(serverJob);
		serverJobStates.startCheck(JobState.READY, JobState.EXECUTING, 
				JobState.COMPLETE);
		
		StateSteps clientOddjobStates = new StateSteps(clientOddjob);
		clientOddjobStates.startCheck(ParentState.READY, ParentState.EXECUTING, 
				ParentState.COMPLETE);

		logger.info("** running Oddjob Client **");
		clientOddjob.run();
		
		clientOddjobStates.checkWait();
		
		serverJobStates.checkNow();
	}
	
	public void testClientTriggersOnServerJobExample() throws InterruptedException, ArooaPropertyException, ArooaConversionException {
		
		Properties props = new Properties();
		props.setProperty("hosts.freds.pc", "localhost");
		
		OurDirs dirs = new OurDirs();
		
		File testDir = dirs.relative("test/java/org/oddjob/jmx");
		
		serverOddjob = new Oddjob();
		serverOddjob.setProperties(props);
		serverOddjob.setFile(new File(testDir, "ServerExample.xml"));

		serverOddjob.run();
		
		assertEquals(ParentState.STARTED, 
				serverOddjob.lastStateEvent().getState());
		
		OddjobLookup serverLookup = new OddjobLookup(serverOddjob);
		
		Runnable serverJob = serverLookup.lookup("server-jobs/greeting", 
				Runnable.class);
	
		clientOddjob = new Oddjob();
		clientOddjob.setProperties(props);
		clientOddjob.setFile(new File(testDir, "ClientTrigger.xml"));

		clientOddjob.run();
		
		OddjobLookup clientLookup = new OddjobLookup(clientOddjob);
		
		Stateful localJob = clientLookup.lookup("local-job", 
				Stateful.class);
		
		assertEquals(ParentState.STARTED, 
				clientOddjob.lastStateEvent().getState());
		
		assertEquals(JobState.READY, 
				localJob.lastStateEvent().getState());
		
		StateSteps state = new StateSteps(clientOddjob);
		
		state.startCheck(ParentState.STARTED, ParentState.ACTIVE, 
				ParentState.COMPLETE);
		
		serverJob.run();
		
		state.checkWait();

		// Todo Fix this
//		clientOddjob.destroy();
//		serverOddjob.destroy();
	}

}