LoadBalancerSession3Test.java

/* Copyright 2012 predic8 GmbH, www.predic8.com

   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 com.predic8.membrane.examples.tests;

import java.io.File;
import java.io.IOException;
import java.util.regex.Pattern;

import org.apache.commons.io.FileUtils;
import org.junit.Assert;
import org.junit.Test;

import com.predic8.membrane.test.AssertUtils;
import com.predic8.membrane.examples.DistributionExtractingTestcase;
import com.predic8.membrane.examples.Process2;
import com.predic8.membrane.examples.util.BufferLogger;

public class LoadBalancerSession3Test extends DistributionExtractingTestcase {

	/**
	 * The test as described in README.txt, but "wsimport" (previously called by ant)
	 * was removed and is run directly from this test before everything else. Thereby
	 * we can use a Maven dependency on wsimport and do not have to download it ourselves.
	 */
	@Test
	public void test() throws IOException, InterruptedException {
		File base = getExampleDir("loadbalancer-session-3");

		AssertUtils.replaceInFile(new File(base, "proxies.xml"), "8080", "3023");
		AssertUtils.replaceInFile(new File(base, "src/com/predic8/chat/Client.java"), "8080", "3023");
		AssertUtils.replaceInFile(new File(base, "data/ChatService.wsdl"), "8080", "3023");

		Process2 sl = new Process2.Builder().in(base).script("service-proxy").waitForMembrane().start();
		try {

			File buildXML = new File(base, "build.xml");

			// remove <exec...</exec> from build.xml
			String s = Pattern.compile("<exec.*</exec>", Pattern.DOTALL).matcher(FileUtils.readFileToString(buildXML)).replaceAll("");
			FileUtils.writeStringToFile(buildXML, s);

			File classes = new File(base, "build" + File.separator + "classes");
			classes.mkdirs();
			File source = new File(base, "src");
			source.mkdirs();

			// run "wsimport" generating java sources
			Assert.assertTrue(new com.sun.tools.ws.wscompile.WsimportTool(System.out).run(new String[] {
					"-quiet",
					"-Xnocompile",
					new File(base, "data" + File.separator + "ChatService.wsdl").getAbsolutePath(),
					"-s",
					source.getAbsolutePath()
			}));

			// call "ant compile" now so that both antNodeX processes do call it at the same time
			BufferLogger loggerCompile = new BufferLogger();
			Process2 antCompile = new Process2.Builder().in(base).withWatcher(loggerCompile).executable("ant compile").start();
			try {
				int result = antCompile.waitFor(60000);
				if (result != 0)
					throw new AssertionError("'ant compile' returned non-zero " + result + ":\r\n" + loggerCompile.toString());
			} finally {
				antCompile.killScript();
			}


			BufferLogger loggerNode1 = new BufferLogger();
			BufferLogger loggerNode2 = new BufferLogger();
			Process2 antNode1 = new Process2.Builder().in(base).withWatcher(loggerNode1).executable("ant run-node1").start();
			try {
				Process2 antNode2 = new Process2.Builder().in(base).withWatcher(loggerNode2).executable("ant run-node2").start();
				try {

					LoadBalancerUtil.addLBNodeViaHTML("http://localhost:9000/admin/", "localhost", 4000);
					LoadBalancerUtil.addLBNodeViaHTML("http://localhost:9000/admin/", "localhost", 4001);

					Thread.sleep(1000); // wait for nodes to come up

					Process2 antClient = new Process2.Builder().in(base).executable("ant run-client -Dlogin=jim").start();
					try {
						antClient.waitFor(60000);
					} finally {
						antClient.killScript();
					}

				} finally {
					antNode2.killScript();
				}
			} finally {
				antNode1.killScript();
			}

			AssertUtils.assertContains("Hallo World", loggerNode1.toString());
			AssertUtils.assertContainsNot("Hallo World", loggerNode2.toString());
		} finally {
			sl.killScript();
		}

	}

}