TestFixture.java

/**
Copyright (c) 2007-2013 Alysson Bessani, Eduardo Alchieri, Paulo Sousa, and the authors indicated in the @author tags

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package bftsmart;

import java.io.IOException;

import org.junit.AfterClass;
import org.junit.BeforeClass;

/**
 * 
 * @author Marcel Santos
 *
 */
public class TestFixture {

	private static Process replica0;
	private static Process replica1;
	private static Process replica2;
	private static Process replica3;
	
	private static ConsoleLogger log0;
	private static ConsoleLogger log1;
	private static ConsoleLogger log2;
	private static ConsoleLogger log3;

	private static String[] command = new String[5];
	
	@BeforeClass
	public static void startServers() {
		try {
			System.out.println("Stoping running servers, if any");
			String stopCmd = "kill -9 $(ps aux | grep '[b]ftsmart.demo.bftmap.BFTMapServer' | awk '{print $ 2}')";
			try {
				new ProcessBuilder(stopCmd).redirectErrorStream(true).start();
			} catch (IOException e) {
				System.out.println("Exception stoping remaining replicas");
				e.printStackTrace();
			}
			
			System.out.println("Starting the servers");
			command[0] = "java";
			command[1] = "-cp";
			command[2] = "bin/:lib/*";
			command[3] = "bftsmart.demo.bftmap.BFTMapServer";
			command[4] = "0";
			
			startServer(0);

			Thread.sleep(2000);
			startServer(1);

			Thread.sleep(2000);
			startServer(2);
			
			Thread.sleep(2000);
			startServer(3);

		} catch(InterruptedException ie) {
			System.out.println("Exception during Thread sleep: " + ie.getMessage());
		}
	}

	@AfterClass
	public static void stopServers() throws IllegalArgumentException, IllegalAccessException, NoSuchFieldException, IOException  {
		System.out.println("Stopping servers");
		stopServer(0);
		stopServer(1);
		stopServer(2);
		stopServer(3);
		System.out.println("Servers stopped");
	}
	
	public static void stopServer(int id) {
		switch(id) {
		case 0:
			replica0.destroy();
			break;
		case 1:
			replica1.destroy();
			break;
		case 2:
			replica2.destroy();
			break;
		case 3:
			replica3.destroy();
			break;
		default:
			System.out.println("### Couldn't stop server. Server not found ###");
			break;
		}
	}

	public static void startServer(int id) {
		command[4] = String.valueOf(id);
		try {
			switch(id) {
			case 0:
				replica0 = new ProcessBuilder(command).redirectErrorStream(true).start();
				log0 = new ConsoleLogger();
				log0.setIn(replica0.getInputStream());
				log0.setOut(System.out);
				log0.setIndex(String.valueOf(id));
				log0.start();
				break;
			case 1:
				replica1 = new ProcessBuilder(command).redirectErrorStream(true).start();
				log1 = new ConsoleLogger();
				log1.setIn(replica1.getInputStream());
				log1.setOut(System.out);
				log1.setIndex(String.valueOf(id));
				log1.start();
				break;
			case 2:
				replica2 = new ProcessBuilder(command).redirectErrorStream(true).start();
				log2 = new ConsoleLogger();
				log2.setIn(replica2.getInputStream());
				log2.setOut(System.out);
				log2.setIndex(String.valueOf(id));
				log2.start();
				break;
			case 3:
				replica3 = new ProcessBuilder(command).redirectErrorStream(true).start();
				log3 = new ConsoleLogger();
				log3.setIn(replica3.getInputStream());
				log3.setOut(System.out);
				log3.setIndex(String.valueOf(id));
				log3.start();
				break;
			default:
				System.out.println("Id not supported");
				break;
			}
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
}