Search in sources :

Example 1 with AppServer

use of org.mobicents.tools.sip.balancer.AppServer in project load-balancer by RestComm.

the class NodeGracefulShutdownTest method testGracefulRemovingNode.

@Test
public void testGracefulRemovingNode() throws Exception {
    int okNumber1 = 0;
    int okNumber2 = 0;
    shootist1.sendInitialInvite();
    Thread.sleep(500);
    for (AppServer server : servers) if (server.sipListener.getInviteRequest() != null)
        server.gracefulShutdown();
    Thread.sleep(5000);
    shootist1.sendBye();
    Thread.sleep(2000);
    shootist2.sendInitialInvite();
    Thread.sleep(5000);
    shootist2.sendBye();
    Thread.sleep(2000);
    for (Response res : shootist1.responses) if (res.getStatusCode() == Response.OK)
        okNumber1++;
    for (Response res : shootist2.responses) if (res.getStatusCode() == Response.OK)
        okNumber2++;
    for (AppServer server : servers) assertNotNull(server.sipListener.getInviteRequest());
    assertEquals(2, okNumber1);
    assertEquals(2, okNumber2);
}
Also used : AppServer(org.mobicents.tools.sip.balancer.AppServer) Response(javax.sip.message.Response) ListeningPoint(javax.sip.ListeningPoint) Test(org.junit.Test)

Example 2 with AppServer

use of org.mobicents.tools.sip.balancer.AppServer in project load-balancer by RestComm.

the class HeaderConsistentHashAlgorithmTest method testOKRingingLandOnDifferentNode.

@Test
public void testOKRingingLandOnDifferentNode() throws Exception {
    EventListener failureEventListener = new EventListener() {

        @Override
        public void uasAfterResponse(int statusCode, AppServer source) {
        }

        @Override
        public void uasAfterRequestReceived(String method, AppServer source) {
        // TODO Auto-generated method stub
        }

        @Override
        public void uacAfterRequestSent(String method, AppServer source) {
        // TODO Auto-generated method stub
        }

        @Override
        public void uacAfterResponse(int statusCode, AppServer source) {
            if (statusCode == 180) {
                ringingAppServer = source;
                // source.sendCleanShutdownToBalancers();
                source.stop();
            } else {
                okAppServer = source;
            }
        }
    };
    for (AppServer as : servers) as.setEventListener(failureEventListener);
    shootist.callerSendsBye = true;
    String fromName = "sender";
    String fromHost = "sip-servlets.com";
    SipURI fromAddress = servers[0].protocolObjects.addressFactory.createSipURI(fromName, fromHost);
    String toUser = "replaces";
    String toHost = "sip-servlets.com";
    SipURI toAddress = servers[0].protocolObjects.addressFactory.createSipURI(toUser, toHost);
    SipURI ruri = servers[0].protocolObjects.addressFactory.createSipURI("usera", "127.0.0.1:5033");
    ruri.setLrParam();
    SipURI route = servers[0].protocolObjects.addressFactory.createSipURI("lbint", "127.0.0.1:5065");
    route.setParameter("node_host", "127.0.0.1");
    route.setParameter("node_port", "4060");
    route.setLrParam();
    shootist.start();
    // servers[0].sipListener.sendSipRequest("INVITE", fromAddress, toAddress, null, null, false);
    servers[0].sipListener.sendSipRequest("INVITE", fromAddress, toAddress, null, route, false, null, null, ruri);
    Thread.sleep(20000);
    assertTrue(shootist.inviteRequest.getHeader(RecordRouteHeader.NAME).toString().contains("node_host"));
    assertNotSame(ringingAppServer, okAppServer);
    assertNotNull(ringingAppServer);
    assertNotNull(okAppServer);
}
Also used : AppServer(org.mobicents.tools.sip.balancer.AppServer) EventListener(org.mobicents.tools.sip.balancer.EventListener) SipURI(javax.sip.address.SipURI) Test(org.junit.Test)

Example 3 with AppServer

use of org.mobicents.tools.sip.balancer.AppServer in project load-balancer by RestComm.

the class HeaderConsistentHashAlgorithmTest method setUp.

@Before
public void setUp() throws Exception {
    shootist = new Shootist();
    balancer = new BalancerRunner();
    LoadBalancerConfiguration lbConfig = new LoadBalancerConfiguration();
    lbConfig.getSipConfiguration().getInternalLegConfiguration().setUdpPort(5065);
    lbConfig.getSipConfiguration().getAlgorithmConfiguration().setAlgorithmClass(HeaderConsistentHashBalancerAlgorithm.class.getName());
    balancer.start(lbConfig);
    for (int q = 0; q < servers.length; q++) {
        servers[q] = new AppServer("node" + q, 4060 + q, "127.0.0.1", 2000, 5060, 5065, "0", ListeningPoint.UDP, 2222 + q);
        servers[q].start();
    }
    Thread.sleep(5000);
}
Also used : AppServer(org.mobicents.tools.sip.balancer.AppServer) BalancerRunner(org.mobicents.tools.sip.balancer.BalancerRunner) LoadBalancerConfiguration(org.mobicents.tools.configuration.LoadBalancerConfiguration) HeaderConsistentHashBalancerAlgorithm(org.mobicents.tools.sip.balancer.HeaderConsistentHashBalancerAlgorithm) Shootist(org.mobicents.tools.sip.balancer.operation.Shootist) ListeningPoint(javax.sip.ListeningPoint) Before(org.junit.Before)

Example 4 with AppServer

use of org.mobicents.tools.sip.balancer.AppServer in project load-balancer by RestComm.

the class PureConsistentHashNodeDeathTest method testRegister.

@Test
public void testRegister() throws Exception {
    EventListener failureEventListener = new EventListener() {

        @Override
        public void uasAfterResponse(int statusCode, AppServer source) {
        }

        @Override
        public void uasAfterRequestReceived(String method, AppServer source) {
            if (method.equals("REGISTER"))
                register = source;
        }

        @Override
        public void uacAfterRequestSent(String method, AppServer source) {
        // TODO Auto-generated method stub
        }

        @Override
        public void uacAfterResponse(int statusCode, AppServer source) {
        // TODO Auto-generated method stub
        }
    };
    for (AppServer as : servers) as.setEventListener(failureEventListener);
    shootist.sendInitial("REGISTER");
    Thread.sleep(5000);
    assertNotNull(register);
}
Also used : AppServer(org.mobicents.tools.sip.balancer.AppServer) EventListener(org.mobicents.tools.sip.balancer.EventListener) Test(org.junit.Test)

Example 5 with AppServer

use of org.mobicents.tools.sip.balancer.AppServer in project load-balancer by RestComm.

the class PureConsistentHashNodeDeathTest method testAllNodesDead.

@Test
public void testAllNodesDead() throws Exception {
    for (AppServer as : servers) {
        // as.sendCleanShutdownToBalancers();
        as.stop();
    // as.sendHeartbeat=false;
    }
    Thread.sleep(3000);
    assertTrue(balancer.getNodes().isEmpty());
    shootist.callerSendsBye = true;
    shootist.sendInitialInvite();
    Thread.sleep(15000);
    assertEquals(500, shootist.responses.get(0).getStatusCode());
}
Also used : AppServer(org.mobicents.tools.sip.balancer.AppServer) Test(org.junit.Test)

Aggregations

AppServer (org.mobicents.tools.sip.balancer.AppServer)72 Test (org.junit.Test)41 EventListener (org.mobicents.tools.sip.balancer.EventListener)37 ListeningPoint (javax.sip.ListeningPoint)27 Before (org.junit.Before)26 BalancerRunner (org.mobicents.tools.sip.balancer.BalancerRunner)26 Shootist (org.mobicents.tools.sip.balancer.operation.Shootist)26 LoadBalancerConfiguration (org.mobicents.tools.configuration.LoadBalancerConfiguration)24 SipURI (javax.sip.address.SipURI)16 SinglePointTest (org.mobicents.tools.sip.balancer.SinglePointTest)11 WorstCaseUdpTestAffinityAlgorithm (org.mobicents.tools.sip.balancer.WorstCaseUdpTestAffinityAlgorithm)7 UDPPacketForwarder (org.mobicents.tools.sip.balancer.UDPPacketForwarder)5 UserBasedAlgorithm (org.mobicents.tools.sip.balancer.UserBasedAlgorithm)5 ArrayList (java.util.ArrayList)2 SipException (javax.sip.SipException)2 Response (javax.sip.message.Response)2 CongestionControlMessageValve (org.mobicents.ext.javax.sip.congestion.CongestionControlMessageValve)1 Node (org.mobicents.tools.heartbeat.api.Node)1 ActiveStandbyAlgorithm (org.mobicents.tools.sip.balancer.ActiveStandbyAlgorithm)1 HeaderConsistentHashBalancerAlgorithm (org.mobicents.tools.sip.balancer.HeaderConsistentHashBalancerAlgorithm)1