MessageFlowHarness.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2011, Red Hat, Inc. and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.restcomm.media.client.mgcp.test;

import java.net.InetAddress;

import org.apache.log4j.Logger;
import org.restcomm.media.client.mgcp.stack.JainMgcpStackImpl;
import org.restcomm.media.client.mgcp.stack.JainMgcpStackProviderImpl;

public abstract class MessageFlowHarness extends TestHarness {

	private static Logger logger = Logger.getLogger("mgcp.test");

	protected static long TRANSACTION_TIMES_OUT_FOR = 31000;

	protected static long STACKS_START_FOR = 1000;

	protected static long STACKS_SHUT_DOWN_FOR = 500;

	// timeout values depend on pc
	protected static long MESSAGES_ARRIVE_FOR = 3000;

	protected static long RETRANSMISSION_TRANSACTION_TIMES_OUT_FOR = 5000;

	protected InetAddress caIPAddress = null;
	protected InetAddress mgIPAddress = null;

	protected JainMgcpStackImpl caStack = null;
	protected JainMgcpStackImpl mgStack = null;

	protected JainMgcpStackProviderImpl caProvider = null;
	protected JainMgcpStackProviderImpl mgProvider = null;

	public MessageFlowHarness(String name) {
		super(name);
	}

	public void setUp() throws java.lang.Exception {
		caIPAddress = InetAddress.getByName(LOCAL_ADDRESS);
		mgIPAddress = InetAddress.getByName(LOCAL_ADDRESS);

		caStack = new JainMgcpStackImpl(caIPAddress, CA_PORT);
		mgStack = new JainMgcpStackImpl(mgIPAddress, MGW_PORT);

		caProvider = (JainMgcpStackProviderImpl) caStack.createProvider();
		mgProvider = (JainMgcpStackProviderImpl) mgStack.createProvider();

	}

	public void tearDown() throws java.lang.Exception {

		System.out.println("CLOSE THE STACK");

		if (caStack != null) {
			caStack.close();
			caStack = null;
		}

		if (mgStack != null) {
			mgStack.close();
			mgStack = null;
		}

		// Wait for stack threads to release resources (e.g. port)
		sleep(STACKS_SHUT_DOWN_FOR);
	}

	protected static void waitForTimeout() {
		sleep(TRANSACTION_TIMES_OUT_FOR);
	}

	protected static void waitForRetransmissionTimeout() {
		sleep(RETRANSMISSION_TRANSACTION_TIMES_OUT_FOR);
	}

	public static void waitForMessage() {
		sleep(MESSAGES_ARRIVE_FOR);
	}

	protected static void sleep(long sleepFor) {
		try {
			Thread.sleep(sleepFor);
		} catch (InterruptedException ex) {
			// Ignore
		}
	}
}