/*
* JBoss, Home of Professional Open Source
* Copyright 2011, Red Hat, Inc. and/or its affiliates, and individual
* contributors as indicated by the @authors tag. All rights reserved.
* See the copyright.txt in the distribution for a full listing
* of individual contributors.
*
* This copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU General Public License, v. 2.0.
*
* This program 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
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License,
* v. 2.0 along with this distribution; if not, write to the Free
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
package org.mobicents.protocols.sctp;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import java.util.Arrays;
import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.mobicents.protocols.api.Association;
import org.mobicents.protocols.api.AssociationListener;
import org.mobicents.protocols.api.IpChannelType;
import org.mobicents.protocols.api.PayloadData;
import org.testng.annotations.*;
/**
* @author amit bhayani
*
*/
public class ClientAssociationTest {
private static final String SERVER_NAME = "testserver";
private static final String SERVER_HOST = "127.0.0.1";
private static final int SERVER_PORT = 2345;
private static final String SERVER_ASSOCIATION_NAME = "serverAsscoiation";
private static final String CLIENT_ASSOCIATION_NAME = "clientAsscoiation";
private static final String CLIENT_HOST = "127.0.0.1";
private static final int CLIENT_PORT = 2346;
private final byte[] CLIENT_MESSAGE = "Client says Hi".getBytes();
private final byte[] SERVER_MESSAGE = "Server says Hi".getBytes();
private ManagementImpl management = null;
// private Management managementClient = null;
private ServerImpl server = null;
private AssociationImpl serverAssociation = null;
private AssociationImpl clientAssociation = null;
private volatile boolean clientAssocUp = false;
private volatile boolean serverAssocUp = false;
private volatile boolean clientAssocDown = false;
private volatile boolean serverAssocDown = false;
private byte[] clientMessage;
private byte[] serverMessage;
@BeforeClass
public static void setUpClass() throws Exception {
}
@AfterClass
public static void tearDownClass() throws Exception {
}
public void setUp(IpChannelType ipChannelType) throws Exception {
this.clientAssocUp = false;
this.serverAssocUp = false;
this.clientAssocDown = false;
this.serverAssocDown = false;
this.clientMessage = null;
this.serverMessage = null;
this.management = new ManagementImpl("ClientAssociationTest");
this.management.setConnectDelay(10000);// Try connecting every 10 secs
this.management.setSingleThread(true);
this.management.start();
this.management.removeAllResourses();
this.server = this.management.addServer(SERVER_NAME, SERVER_HOST, SERVER_PORT, ipChannelType, false, 0, null);
this.serverAssociation = this.management.addServerAssociation(CLIENT_HOST, CLIENT_PORT, SERVER_NAME, SERVER_ASSOCIATION_NAME, ipChannelType);
this.clientAssociation = this.management.addAssociation(CLIENT_HOST, CLIENT_PORT, SERVER_HOST, SERVER_PORT, CLIENT_ASSOCIATION_NAME, ipChannelType, null);
}
public void tearDown() throws Exception {
this.management.removeAssociation(CLIENT_ASSOCIATION_NAME);
this.management.removeAssociation(SERVER_ASSOCIATION_NAME);
this.management.removeServer(SERVER_NAME);
this.management.stop();
}
/**
* In this test client association is started without server being started.
* Client keeps attempting to connect to server, till server comes up
*
* @throws Exception
*/
@Test(groups = { "functional", "sctp" })
public void testConnectAttemptsSctp() throws Exception {
if (SctpTransferTest.checkSctpEnabled())
this.testConnectAttemptsByProtoclol(IpChannelType.SCTP);
}
/**
* In this test client association is started without server being started.
* Client keeps attempting to connect to server, till server comes up
*
* @throws Exception
*/
@Test(groups = { "functional", "tcp" })
public void testConnectAttemptsTcp() throws Exception {
this.testConnectAttemptsByProtoclol(IpChannelType.TCP);
}
private void testConnectAttemptsByProtoclol(IpChannelType ipChannelType) throws Exception {
// BasicConfigurator.configure();
// Logger logger = Logger.getLogger(ServerImpl.class.getName());
// logger.setLevel(Level.ALL);
this.setUp(ipChannelType);
this.serverAssociation.setAssociationListener(new ServerAssociationListener());
this.management.startAssociation(SERVER_ASSOCIATION_NAME);
this.clientAssociation.setAssociationListener(new ClientAssociationListenerImpl());
this.management.startAssociation(CLIENT_ASSOCIATION_NAME);
Thread.sleep(1000 * 20);
assertFalse(clientAssocUp);
assertFalse(serverAssocUp);
// Lets start the server now
this.management.startServer(SERVER_NAME);
Thread.sleep(1000 * 40);
assertTrue(clientAssocUp);
assertTrue(serverAssocUp);
this.management.stopAssociation(CLIENT_ASSOCIATION_NAME);
Thread.sleep(1000);
this.management.stopAssociation(SERVER_ASSOCIATION_NAME);
this.management.stopServer(SERVER_NAME);
Thread.sleep(1000 * 2);
assertTrue(Arrays.equals(SERVER_MESSAGE, clientMessage));
assertTrue(Arrays.equals(CLIENT_MESSAGE, serverMessage));
assertTrue(clientAssocDown);
assertTrue(serverAssocDown);
Runtime runtime = Runtime.getRuntime();
this.tearDown();
}
private class ClientAssociationListenerImpl implements AssociationListener {
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationUp
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationUp(Association association, int maxInboundStreams, int maxOutboundStreams) {
System.out.println(this + " onCommunicationUp");
clientAssocUp = true;
PayloadData payloadData = new PayloadData(CLIENT_MESSAGE.length, CLIENT_MESSAGE, true, false, 3, 1);
try {
association.send(payloadData);
} catch (Exception e) {
e.printStackTrace();
}
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationShutdown
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationShutdown(Association association) {
System.out.println(this + " onCommunicationShutdown");
clientAssocDown = true;
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationLost
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationLost(Association association) {
System.out.println(this + " onCommunicationLost");
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationRestart
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationRestart(Association association) {
System.out.println(this + " onCommunicationRestart");
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onPayload(org.mobicents
* .protocols.sctp.Association,
* org.mobicents.protocols.sctp.PayloadData)
*/
@Override
public void onPayload(Association association, PayloadData payloadData) {
System.out.println(this + " onPayload");
clientMessage = new byte[payloadData.getDataLength()];
System.arraycopy(payloadData.getData(), 0, clientMessage, 0, payloadData.getDataLength());
System.out.println(this + "received " + new String(clientMessage));
}
/* (non-Javadoc)
* @see org.mobicents.protocols.api.AssociationListener#inValidStreamId(org.mobicents.protocols.api.PayloadData)
*/
@Override
public void inValidStreamId(PayloadData payloadData) {
// TODO Auto-generated method stub
}
}
private class ServerAssociationListener implements AssociationListener {
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationUp
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationUp(Association association, int maxInboundStreams, int maxOutboundStreams) {
System.out.println(this + " onCommunicationUp");
serverAssocUp = true;
PayloadData payloadData = new PayloadData(SERVER_MESSAGE.length, SERVER_MESSAGE, true, false, 3, 1);
try {
association.send(payloadData);
} catch (Exception e) {
e.printStackTrace();
}
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationShutdown
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationShutdown(Association association) {
System.out.println(this + " onCommunicationShutdown");
serverAssocDown = true;
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationLost
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationLost(Association association) {
System.out.println(this + " onCommunicationLost");
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onCommunicationRestart
* (org.mobicents.protocols.sctp.Association)
*/
@Override
public void onCommunicationRestart(Association association) {
System.out.println(this + " onCommunicationRestart");
}
/*
* (non-Javadoc)
*
* @see
* org.mobicents.protocols.sctp.AssociationListener#onPayload(org.mobicents
* .protocols.sctp.Association,
* org.mobicents.protocols.sctp.PayloadData)
*/
@Override
public void onPayload(Association association, PayloadData payloadData) {
System.out.println(this + " onPayload");
serverMessage = new byte[payloadData.getDataLength()];
System.arraycopy(payloadData.getData(), 0, serverMessage, 0, payloadData.getDataLength());
System.out.println(this + "received " + new String(serverMessage));
}
/* (non-Javadoc)
* @see org.mobicents.protocols.api.AssociationListener#inValidStreamId(org.mobicents.protocols.api.PayloadData)
*/
@Override
public void inValidStreamId(PayloadData payloadData) {
// TODO Auto-generated method stub
}
}
}