Writing Validation Tests with JUnit

package net.roboconf.sample;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Logger;

import net.roboconf.core.model.beans.ApplicationTemplate;
import net.roboconf.core.model.beans.Instance;
import net.roboconf.core.model.beans.Instance.InstanceStatus;
import net.roboconf.core.model.helpers.InstanceHelpers;
import net.roboconf.core.utils.ProgramUtils;
import net.roboconf.dm.management.ManagedApplication;
import net.roboconf.dm.management.Manager;
import net.roboconf.messaging.rabbitmq.internal.RabbitMqClientFactory;
import net.roboconf.target.docker.internal.DockerHandler;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;

public class SampleTest {

	@Rule
	public TemporaryFolder folder = new TemporaryFolder();
	private Manager manager;


	@Before
	public void prepareManager() throws Exception {

		// Create a new manager.
		this.manager = new Manager();

		// Define where the work directory will be located.
		this.manager.configurationMngr().setWorkingDirectory( this.folder.newFolder());

		// Inject target implementations.
		// Normally, this is managed automatically by our OSGi framework.
		// Here, you have to do this by hand.
		this.manager.targetAppears( new DockerHandler());

		// Set the messaging by hand.
		// Normally, this is managed automatically by our OSGi framework.
		// Here, you have to do this by hand.
		
		// This part is a little bit complex.
		// A more simple solution was provided for Roboconf 0.7.
		MessagingClientFactoryRegistry registry = new MessagingClientFactoryRegistry();
		registry.addMessagingClientFactory( new RabbitMqClientFactory());

		ReconfigurableClientDm internalClient = TestUtils.getInternalField(
				this.manager,
				"messagingClient",
				ReconfigurableClientDm.class );

		internalClient.setRegistry( registry );
		
		// Set the messaging type
		manager.setMessagingType( "rabbitmq" );

		// Start the DM.
		this.manager.start();
	}


	@After
	public void stopManager() {

		// Do we need to clean the hosts?
		for( ManagedApplication ma : this.manager.applicationMngr().getManagedApplications()) {
			try {
				this.manager.instancesMngr().undeployAll( ma, null );

			} catch( IOException e ) {
				System.out.println( "Oops, we failed to undeploy application " + ma );
			}
		}

		// Stop the DM.
		this.manager.stop();
	}


	@Test
	public void runScenario() throws Exception {

		// Load an application template.
		File myAppLocation = new File( "whatever.zip" );
		ApplicationTemplate tpl = this.manager.applicationTemplateMngr().loadApplicationTemplate( myAppLocation );

		// Create an application.
		ManagedApplication ma = this.manager.applicationMngr().createApplication( "my-app", "", tpl );

		// Create a deployment target.
		String targetId = this.manager.targetsMngr().createTarget( new File( "my-target.properties" ));

		// Make it the default target for this application.
		this.manager.targetsMngr().associateTargetWithScopedInstance( targetId, ma.getApplication(), null );

		// The application may have inherited target associations from its template.
		// If we want to remove them...
		for( Instance scopedInstance : InstanceHelpers.getAllInstances( ma.getApplication())) {
			String instancePath = InstanceHelpers.computeInstancePath( scopedInstance );
			this.manager.targetsMngr().dissociateTargetFromScopedInstance( ma.getApplication(), instancePath );
		}

		// Perform a full deployment.
		this.manager.instancesMngr().deployAndStartAll( ma, null );

		// This may take time!
		for( int i=0; i<10; i++ ) {

			// Are all the instance deployed?
			for( Instance inst : InstanceHelpers.getAllInstances( ma.getApplication())) {
				if( inst.getStatus() != InstanceStatus.DEPLOYED_STARTED )
					break;
			}

			// Wait 5 seconds
			Thread.sleep( 5000 );
		}

		// By the end of the loop, we assume the deployment SHOULD have completed.
		// Verify assertions...
		Assert.assertTrue( true );

		// Run a verification script located under "src/test/resources".
		URL url = getClass().getResource( "/verif.sh" );
		File scriptFile = new File( url.toURI());

		List<String> cmd = Arrays.asList( "/bin/bash", scriptFile.getAbsolutePath());
		File execDir = new File( "." );
		Map<String,String> env = new HashMap<>( 0 );

		ProgramUtils.executeCommand( Logger.getLogger( getClass().getName()), cmd, execDir, env );
	}
}