OrStateTest.java

package org.oddjob.state;


import java.io.IOException;

import junit.framework.TestCase;

import org.oddjob.framework.SimpleJob;
import org.oddjob.scheduling.DefaultExecutors;
import org.oddjob.scheduling.MockScheduledExecutorService;
import org.oddjob.tools.OddjobTestHelper;
import org.oddjob.tools.StateSteps;

public class OrStateTest extends TestCase {

	private class Result implements StateListener {
		State result;
		
		public void jobStateChange(StateEvent event) {
			result = event.getState();
		}
	}
	
	private class UnusedServices extends MockScheduledExecutorService {
	}
	
	
	public void testComplete() {
		
		OrState test = new OrState();
		test.setExecutorService(new UnusedServices());
		test.run();
		
		Result listener = new Result();
		test.addStateListener(listener);

		assertEquals(ParentState.READY, listener.result);
		
		FlagState j1 = new FlagState(JobState.COMPLETE);

		test.setJobs(0, j1);
		
		assertEquals(ParentState.READY, listener.result);
		
		j1.run();
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
		FlagState j2 = new FlagState(JobState.COMPLETE);

		test.setJobs(0, j2);
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
		j2.run();
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
		test.setJobs(1, null);
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
		test.setJobs(0, null);
		
		assertEquals(ParentState.READY, listener.result);
	}
	
	public void testException() {
		
		OrState test = new OrState();
		test.setExecutorService(new UnusedServices());
		test.run();
		
		Result listener = new Result();
		test.addStateListener(listener);

		assertEquals(ParentState.READY, listener.result);
		
		FlagState j1 = new FlagState(JobState.COMPLETE);

		test.setJobs(0, j1);
		
		assertEquals(ParentState.READY, listener.result);
		
		j1.run();
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
		FlagState j2 = new FlagState(JobState.EXCEPTION);

		test.setJobs(0, j2);
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
		j2.run();
		
		assertEquals(ParentState.EXCEPTION, listener.result);
		
		test.setJobs(1, null);
		
		assertEquals(ParentState.EXCEPTION, listener.result);
		
		test.setJobs(0, null);
		
		assertEquals(ParentState.READY, listener.result);
	}
	
	public void testManyComplete() {
		
		OrState test = new OrState();
		test.setExecutorService(new UnusedServices());
		test.run();
		
		Result listener = new Result();
		test.addStateListener(listener);

		assertEquals(ParentState.READY, listener.result);
		
		FlagState j1 = new FlagState(JobState.INCOMPLETE);
		FlagState j2 = new FlagState(JobState.INCOMPLETE);
		FlagState j3 = new FlagState(JobState.COMPLETE);
		FlagState j4 = new FlagState(JobState.INCOMPLETE);

		j1.run();
		j2.run();
		j3.run();
		j4.run();
		
		test.setJobs(0, j1);
		test.setJobs(1, j2);
		test.setJobs(2, j3);
		test.setJobs(3, j4);
		
		assertEquals(ParentState.COMPLETE, listener.result);
		
	}
	
	public void testSerialize() throws IOException, ClassNotFoundException, InterruptedException {
		
		DefaultExecutors services = new DefaultExecutors();
		SimpleJob notSerializable = new SimpleJob() {
			@Override
			protected int execute() throws Throwable {
				return 0;
			}
		};
		
		OrState test = new OrState();
		test.setExecutorService(services.getPoolExecutor());
		test.setJobs(0, notSerializable);

		StateSteps state = new StateSteps(test);
		state.startCheck(ParentState.READY, 
				ParentState.EXECUTING,
				ParentState.COMPLETE);
		
		test.run();
		
		state.checkWait();
		
		services.stop();
		
		OrState copy = (OrState) OddjobTestHelper.copy(test);
		
		assertEquals(ParentState.COMPLETE, copy.lastStateEvent().getState());
	}
}