MasterWorkerAPIImplementationTest.java

package uk.ac.imperial.lsds.seepmaster.comm;

import java.util.Map;
import java.util.concurrent.Executors;

import org.junit.Test;

import uk.ac.imperial.lsds.seep.comm.Comm;
import uk.ac.imperial.lsds.seep.comm.IOComm;
import uk.ac.imperial.lsds.seep.comm.protocol.BootstrapCommand;
import uk.ac.imperial.lsds.seep.comm.serialization.JavaSerializer;
import uk.ac.imperial.lsds.seep.infrastructure.ControlEndPoint;
import uk.ac.imperial.lsds.seepmaster.LifecycleManager;
import uk.ac.imperial.lsds.seepmaster.infrastructure.master.InfrastructureManager;
import uk.ac.imperial.lsds.seepmaster.infrastructure.master.InfrastructureManagerFactory;
import uk.ac.imperial.lsds.seepmaster.query.GenericQueryManager;

public class MasterWorkerAPIImplementationTest {

	@Test
	public void testBootstrap() {
		InfrastructureManager inf = InfrastructureManagerFactory.createInfrastructureManager(0);
		Map<Integer, ControlEndPoint> mapOperatorToEndPoint = null;
		Comm cu = new IOComm(new JavaSerializer(), Executors.newCachedThreadPool());
		LifecycleManager lifeManager = LifecycleManager.getInstance();
		GenericQueryManager qm = GenericQueryManager.getInstance(inf, mapOperatorToEndPoint, cu, lifeManager, null);
		
		MasterWorkerAPIImplementation api = new MasterWorkerAPIImplementation(qm, inf);
		
		int avail = inf.executionUnitsAvailable();
		assert(avail == 0);

		BootstrapCommand bc = new BootstrapCommand("10.0.0.1", 3500, 4500);
		api.bootstrapCommand(bc);
		
		avail = inf.executionUnitsAvailable();
		assert(avail == 1);
	}

}