[jboss-cvs] JBoss Messaging SVN: r4543 - trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl.

[jboss-cvs-commits at lists.jboss.org]

Author: clebert.suconic at jboss.com
Date: 2008-06-20 19:40:32 -0400 (Fri, 20 Jun 2008)
New Revision: 4543

Modified:
   trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConnectionImplTest.java
   trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConsumerImplTest.java
Log:
Adding few tests

Modified: trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConnectionImplTest.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConnectionImplTest.java	2008-06-20 23:39:47 UTC (rev 4542)
+++ trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConnectionImplTest.java	2008-06-20 23:40:32 UTC (rev 4543)
@@ -189,6 +189,8 @@
       
       assertTrue(conn.isClosed());
       
+      assertSame(conn.getRemotingConnection(), rc);
+      
       //Close again should do nothing
       EasyMock.reset(rc, pd, sess1, sess2, sess3);
       

Modified: trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConsumerImplTest.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConsumerImplTest.java	2008-06-20 23:39:47 UTC (rev 4542)
+++ trunk/tests/src/org/jboss/messaging/tests/unit/core/client/impl/ClientConsumerImplTest.java	2008-06-20 23:40:32 UTC (rev 4543)
@@ -32,6 +32,7 @@
 import org.jboss.messaging.core.logging.Logger;
 import org.jboss.messaging.core.remoting.PacketDispatcher;
 import org.jboss.messaging.core.remoting.RemotingConnection;
+import org.jboss.messaging.core.remoting.impl.wireformat.PacketImpl;
 import org.jboss.messaging.tests.util.UnitTestCase;
 
 import java.util.ArrayList;
@@ -469,6 +470,28 @@
       EasyMock.verify(msgs.toArray());
    }
 
+   public void testClose() throws Exception
+   {
+      ClientSessionInternal session = EasyMock.createStrictMock(ClientSessionInternal.class);
+      ClientConnectionInternal connection = EasyMock.createStrictMock(ClientConnectionInternal.class);
+      RemotingConnection rc = EasyMock.createStrictMock(RemotingConnection.class);
+      ExecutorService executor = EasyMock.createStrictMock(ExecutorService.class);
+      PacketDispatcher pd = EasyMock.createStrictMock(PacketDispatcher.class);
+
+      final long clientTargetID = 120912;
+      ClientConsumerInternal consumer =
+         new ClientConsumerImpl(session, 675765, clientTargetID, 787, false, rc, pd, executor, 878787);
+
+      pd.unregister(clientTargetID);
+      session.removeConsumer(consumer);
+      EasyMock.expect(rc.sendBlocking(675765, 878787, new PacketImpl(PacketImpl.CLOSE))).andReturn(null);
+      EasyMock.replay(session, connection, rc, executor, pd);
+      
+      consumer.close();
+      EasyMock.verify(session, connection, rc, executor, pd);
+   }
+   
+   
    public void testCleanUp() throws Exception
    {
       ClientSessionInternal session = EasyMock.createStrictMock(ClientSessionInternal.class);