[jboss-cvs] JBoss Messaging SVN: r3494 - branches/Branch_Stable/src/main/org/jboss/jms/server/connectionmanager.

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

Author: clebert.suconic at jboss.com
Date: 2007-12-12 22:11:03 -0500 (Wed, 12 Dec 2007)
New Revision: 3494

Modified:
   branches/Branch_Stable/src/main/org/jboss/jms/server/connectionmanager/SimpleConnectionManager.java
Log:
Fixing SimpleConnectionManagerTest::testWithMock -- small tweak

Modified: branches/Branch_Stable/src/main/org/jboss/jms/server/connectionmanager/SimpleConnectionManager.java
===================================================================
--- branches/Branch_Stable/src/main/org/jboss/jms/server/connectionmanager/SimpleConnectionManager.java	2007-12-12 23:09:38 UTC (rev 3493)
+++ branches/Branch_Stable/src/main/org/jboss/jms/server/connectionmanager/SimpleConnectionManager.java	2007-12-13 03:11:03 UTC (rev 3494)
@@ -365,10 +365,14 @@
             try
             {
       			log.debug("clearing up state for connection " + sce);
+
+               // sce could also be a mock test.. so this test is required
+               if (sce instanceof ServerConnectionEndpoint)
+               {
+                  //Remoting is dumb and doesn't clean up it's state after itself - so we have to do it.
+                  ((ServerConnectionEndpoint)sce).closeCallbackClient();
+               }
       			
-      			//Remoting is dumb and doesn't clean up it's state after itself - so we have to do it.
-      			((ServerConnectionEndpoint)sce).closeCallbackClient();
-      			
                sce.closing(-1);
                sce.close();
                log.debug("cleared up state for connection " + sce);