[jboss-cvs] JBoss Messaging SVN: r4483 - trunk/tests/src/org/jboss/messaging/tests/unit/jms/client.

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

Author: jmesnil
Date: 2008-06-16 10:13:25 -0400 (Mon, 16 Jun 2008)
New Revision: 4483

Modified:
   trunk/tests/src/org/jboss/messaging/tests/unit/jms/client/JBossSessionTest.java
Log:
added unit tests for JBossSession

Modified: trunk/tests/src/org/jboss/messaging/tests/unit/jms/client/JBossSessionTest.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/unit/jms/client/JBossSessionTest.java	2008-06-16 13:29:58 UTC (rev 4482)
+++ trunk/tests/src/org/jboss/messaging/tests/unit/jms/client/JBossSessionTest.java	2008-06-16 14:13:25 UTC (rev 4483)
@@ -45,6 +45,7 @@
 import javax.jms.TopicPublisher;
 import javax.jms.TopicSession;
 import javax.jms.TransactionInProgressException;
+import javax.transaction.xa.XAResource;
 
 import junit.framework.TestCase;
 
@@ -1869,6 +1870,52 @@
       }
    }
    
+   public void testGetXAResource() throws Exception
+   {
+      expect(mockClientSession.getXAResource()).andReturn(mockClientSession);
+      replay(mockClientConn, mockClientSession);
+
+      JBossConnection connection = new JBossConnection(mockClientConn,
+            JBossConnection.TYPE_QUEUE_CONNECTION, null, -1);
+      JBossSession session = new JBossSession(connection, false, true,
+            Session.AUTO_ACKNOWLEDGE, mockClientSession,
+            JBossSession.TYPE_QUEUE_SESSION);
+
+      XAResource xares = session.getXAResource();
+      assertNotNull(xares);
+      assertSame(mockClientSession, xares);
+   }
+   
+   public void testGetQueueSession() throws Exception
+   {
+      replay(mockClientConn, mockClientSession);
+
+      JBossConnection connection = new JBossConnection(mockClientConn,
+            JBossConnection.TYPE_QUEUE_CONNECTION, null, -1);
+      JBossSession session = new JBossSession(connection, false, true,
+            Session.AUTO_ACKNOWLEDGE, mockClientSession,
+            JBossSession.TYPE_QUEUE_SESSION);
+
+      QueueSession queueSess = session.getQueueSession();
+      assertNotNull(queueSess);
+      assertSame(session, queueSess);
+   }
+   
+   public void testGetCoreSession() throws Exception
+   {
+      replay(mockClientConn, mockClientSession);
+
+      JBossConnection connection = new JBossConnection(mockClientConn,
+            JBossConnection.TYPE_QUEUE_CONNECTION, null, -1);
+      JBossSession session = new JBossSession(connection, false, true,
+            Session.AUTO_ACKNOWLEDGE, mockClientSession,
+            JBossSession.TYPE_QUEUE_SESSION);
+
+      ClientSession clientSession = session.getCoreSession();
+      assertNotNull(clientSession);
+      assertSame(mockClientSession, clientSession);
+   }
+   
    public void testUnsubscribe() throws Exception
    {
       String subName = randomString();