CVS update: JGroups/tests/junit-functional/org/jgroups/tests ProgrammaticApiTest.java

  User: belaban 
  Date: 10/10/20 14:09:57

  Modified:    tests/junit-functional/org/jgroups/tests
                        ProgrammaticApiTest.java
  Log:
  https://jira.jboss.org/browse/JGRP-1245
  
  Revision  Changes    Path
  1.3       +90 -11    JGroups/tests/junit-functional/org/jgroups/tests/ProgrammaticApiTest.java
  
  Index: ProgrammaticApiTest.java
  ===================================================================
  RCS file: /cvsroot/javagroups/JGroups/tests/junit-functional/org/jgroups/tests/ProgrammaticApiTest.java,v
  retrieving revision 1.2
  retrieving revision 1.3
  diff -u -r1.2 -r1.3
  --- ProgrammaticApiTest.java	20 Oct 2010 12:19:02 -0000	1.2
  +++ ProgrammaticApiTest.java	20 Oct 2010 14:09:57 -0000	1.3
  @@ -1,36 +1,91 @@
   package org.jgroups.tests;
   
   import org.jgroups.*;
  -import org.jgroups.protocols.SHARED_LOOPBACK;
  +import org.jgroups.protocols.*;
  +import org.jgroups.protocols.pbcast.GMS;
  +import org.jgroups.protocols.pbcast.NAKACK;
  +import org.jgroups.protocols.pbcast.STABLE;
   import org.jgroups.stack.Protocol;
   import org.jgroups.stack.ProtocolStack;
   import org.jgroups.util.Util;
  +import org.testng.annotations.AfterMethod;
  +import org.testng.annotations.BeforeMethod;
   import org.testng.annotations.Test;
   
  +import java.net.InetAddress;
  +
   /**
    * @author Bela Ban
  - * @version $Id: ProgrammaticApiTest.java,v 1.2 2010/10/20 12:19:02 belaban Exp $
  + * @version $Id: ProgrammaticApiTest.java,v 1.3 2010/10/20 14:09:57 belaban Exp $
    */
  -@Test(groups=Global.FUNCTIONAL,sequential=false)
  +@Test(groups=Global.FUNCTIONAL,sequential=true)
   public class ProgrammaticApiTest {
  +    JChannel c1, c2;
  +
  +    @BeforeMethod
  +    void init() {
  +        c1=new JChannel(false); c1.setName("A");
  +        c2=new JChannel(false); c2.setName("B");
  +    }
  +
  +    @AfterMethod
  +    void destroy() {
  +        Util.close(c2, c1);
  +    }
   
       public void testChannelCreation() throws Exception {
  -        JChannel ch=new JChannel(false);
  -        MyReceiver receiver=new MyReceiver();
  -        ch.setReceiver(receiver);
  -        ProtocolStack stack=ch.createProtocolStack();
  +        MyReceiver receiver=new MyReceiver(null);
  +        c1.setReceiver(receiver);
  +        ProtocolStack stack=new ProtocolStack();
  +        c1.setProtocolStack(stack);
           stack.addProtocol(new SHARED_LOOPBACK()).addProtocol(new MockProtocol1()).addProtocol(new MockProtocol2());
           stack.init();
  -        ch.connect("demo");
  +        c1.connect("demo");
   
           Protocol transport=stack.getTransport();
           transport.up(new Event(Event.MSG, new Message(null, Util.createRandomAddress(), "hello world")));
  -
  -        Util.close(ch);
           assert receiver.getNumMsgsReceived() == 1;
       }
   
   
  +    public void testSharedTransport() throws Exception {
  +        ProtocolStack stack1=new ProtocolStack(), stack2=new ProtocolStack();
  +        c1.setProtocolStack(stack1);
  +        c2.setProtocolStack(stack2);
  +
  +        MyReceiver receiver1=new MyReceiver("A"), receiver2=new MyReceiver("B");
  +
  +        UDP shared_transport=(UDP)new UDP().setValue("bind_addr", InetAddress.getByName("192.168.1.5"))
  +                .setValue("singleton_name", "shared");
  +
  +        stack1.addProtocol(shared_transport).addProtocols(createProtocols());
  +        stack2.addProtocol(shared_transport).addProtocols(createProtocols());
  +
  +        stack1.init();
  +        stack2.init();
  +
  +        c1.setReceiver(receiver1);
  +        c2.setReceiver(receiver2);
  +
  +        c1.connect("cluster-one");
  +        c2.connect("cluster-two");
  +
  +        for(int i=0; i < 10; i++)
  +            c1.send(new Message(null, null, "hello-" + i));
  +
  +        for(int i=0; i < 5; i++)
  +            c2.send(new Message(null, null, "hello-" + i));
  +
  +        for(int i =0; i < 20; i++) {
  +            if(receiver1.getNumMsgsReceived() == 10 && receiver2.getNumMsgsReceived() == 5)
  +                break;
  +            Util.sleep(500);
  +        }
  +        assert receiver1.getNumMsgsReceived() == 10 : "num msgs for A: " + receiver1.getNumMsgsReceived() + " (expected=10)";
  +        assert receiver2.getNumMsgsReceived() == 5 : "num msgs for B: " + receiver1.getNumMsgsReceived() + " (expected=5)";
  +    }
  +
  +
   
       protected static class MockProtocol1 extends Protocol {
   
  @@ -40,16 +95,40 @@
           
       }
   
  +    static Protocol[] createProtocols() {
  +        return new Protocol[] {
  +                new PING(),
  +                new MERGE2(),
  +                new FD_SOCK(),
  +                new FD_ALL().setValue("timeout", 12000).setValue("interval", 3000),
  +                new VERIFY_SUSPECT(),
  +                new BARRIER(),
  +                new NAKACK(),
  +                new UNICAST2(),
  +                new STABLE(),
  +                new GMS(),
  +                new UFC(),
  +                new MFC(),
  +                new FRAG2()
  +        };
  +    }
  +
  +
   
       static class MyReceiver extends ReceiverAdapter {
           int num_msgs_received=0;
  +        final String name;
  +
  +        public MyReceiver(String name) {
  +            this.name=name;
  +        }
   
           public int getNumMsgsReceived() {
               return num_msgs_received;
           }
   
           public void receive(Message msg) {
  -            System.out.println("<< " + msg);
  +            System.out.println((name != null? "[" + name + "]" : "") + "<< " + msg.getObject());
               num_msgs_received++;
           }
       }
  
  
  

------------------------------------------------------------------------------
Download new Adobe(R) Flash(R) Builder(TM) 4
The new Adobe(R) Flex(R) 4 and Flash(R) Builder(TM) 4 (formerly 
Flex(R) Builder(TM)) enable the development of rich applications that run
across multiple browsers and platforms. Download your free trials today!
http://p.sf.net/sfu/adobe-dev2dev