MultipleAWSNodesCreationTest.java

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

package org.ow2.choreos.integration.ee.nodes;

import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.List;

import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.ow2.choreos.ee.config.CloudConfiguration;
import org.ow2.choreos.ee.nodes.NPMFactory;
import org.ow2.choreos.ee.nodes.cm.BootstrapChecker;
import org.ow2.choreos.nodes.NodeNotCreatedException;
import org.ow2.choreos.nodes.NodePoolManager;
import org.ow2.choreos.nodes.datamodel.CloudNode;
import org.ow2.choreos.nodes.datamodel.NodeSpec;
import org.ow2.choreos.tests.IntegrationTest;
import org.ow2.choreos.utils.LogConfigurator;

/**
 * Test the creation of a bunch of VMs using Node Pool Manager with AWS cloud
 * provider.
 * 
 * The purpose is to verify if NPM properly handles the failures of underlying
 * AWS cloud provider. The test is successful if all created VMs are properly
 * bootstrapped.
 * 
 * @author leonardo
 * 
 */
@Category(IntegrationTest.class)
public class MultipleAWSNodesCreationTest {

    public static final int BUNCH_SIZE = 25;

    /*
     * You may edit this attr to the actual cloud account you want to use
     */
    private static final String CLOUD_ACCOUNT = CloudConfiguration.DEFAULT;

    private final NodePoolManager npm = NPMFactory.getNewNPMInstance(CLOUD_ACCOUNT);

    @BeforeClass
    public static void setUpClass() {
	LogConfigurator.configLog();
    }

    @Test
    public void shouldLeaveNodesBootstraped() throws Exception {

	List<Thread> trds = new ArrayList<Thread>();
	List<SingleTest> tests = new ArrayList<SingleTest>();
	for (int i = 0; i < BUNCH_SIZE; i++) {
	    SingleTest singleTest = new SingleTest(i);
	    Thread t = new Thread(singleTest);
	    trds.add(t);
	    tests.add(singleTest);
	    t.start();
	}

	for (Thread t : trds) {
	    t.join();
	}

	for (SingleTest test : tests) {
	    assertTrue("Node not created in test " + test.index, test.created);
	    assertTrue("Node not bootstrapped " + test.index, test.bootstrapped);
	    System.out.println("Test " + test.index + " OK.");
	}
    }

    private class SingleTest implements Runnable {

	int index;
	boolean created;
	boolean bootstrapped;

	public SingleTest(int index) {
	    this.index = index;
	}

	@Override
	public void run() {

	    try {
		CloudNode node = npm.createNode(new NodeSpec());
		created = true;
		BootstrapChecker checker = new BootstrapChecker();
		bootstrapped = checker.isBootstrapped(node);
	    } catch (NodeNotCreatedException e) {
		System.out.println("Node not created in " + index + " because " + e.getMessage());
	    }
	}

    }

}