[jboss-cvs] JBossAS SVN: r88367 - branches/Branch_5_x/testsuite/src/main/org/jboss/test/jca/test.

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

Author: jesper.pedersen
Date: 2009-05-07 10:57:24 -0400 (Thu, 07 May 2009)
New Revision: 88367

Added:
   branches/Branch_5_x/testsuite/src/main/org/jboss/test/jca/test/InternalManagedConnectionPoolStatUnitTestCase.java
Log:
unit test

Added: branches/Branch_5_x/testsuite/src/main/org/jboss/test/jca/test/InternalManagedConnectionPoolStatUnitTestCase.java
===================================================================
--- branches/Branch_5_x/testsuite/src/main/org/jboss/test/jca/test/InternalManagedConnectionPoolStatUnitTestCase.java	                        (rev 0)
+++ branches/Branch_5_x/testsuite/src/main/org/jboss/test/jca/test/InternalManagedConnectionPoolStatUnitTestCase.java	2009-05-07 14:57:24 UTC (rev 88367)
@@ -0,0 +1,131 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2008, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.test.jca.test;
+
+import javax.resource.spi.ManagedConnectionFactory;
+
+import junit.framework.TestCase;
+
+import org.jboss.logging.Logger;
+import org.jboss.resource.connectionmanager.BaseConnectionManager2;
+import org.jboss.resource.connectionmanager.CachedConnectionManager;
+import org.jboss.resource.connectionmanager.ConnectionListener;
+import org.jboss.resource.connectionmanager.InternalManagedConnectionPool;
+import org.jboss.resource.connectionmanager.JBossManagedConnectionPool;
+import org.jboss.resource.connectionmanager.ManagedConnectionPool;
+import org.jboss.resource.connectionmanager.NoTxConnectionManager;
+import org.jboss.resource.connectionmanager.JBossManagedConnectionPool.BasePool;
+import org.jboss.test.jca.adapter.TestManagedConnectionFactory;
+
+/**
+ * An InternalManagedConnectionPoolStatUnitTestCase.
+ * 
+ * @author <a href="jesper.pedersen at jboss.org">Jesper Pedersen</a>
+ * @version $Revision: 85731 $
+ */
+public class InternalManagedConnectionPoolStatUnitTestCase extends TestCase
+{
+   private static Logger log = Logger.getLogger(InternalManagedConnectionPoolStatUnitTestCase.class);
+
+   private CachedConnectionManager ccm = new CachedConnectionManager();
+
+   /**
+    * Creates a new <code>InternalManagedConnectionPoolStatUnitTestCase</code> instance.
+    * @param name test name
+    */
+   public InternalManagedConnectionPoolStatUnitTestCase(String name)
+   {
+      super(name);
+   }
+
+   /**
+    * Get a BaseConnectionManager2 object with an OnePool connection pool
+    * @param pp The pool parameters
+    * @return BaseConnectionManager2 instance
+    * @exception Exception Thrown if an error occurs
+    */
+   private BaseConnectionManager2 getCM(InternalManagedConnectionPool.PoolParams pp) throws Exception
+   {
+      ManagedConnectionFactory mcf = new TestManagedConnectionFactory();
+      ManagedConnectionPool poolingStrategy = new JBossManagedConnectionPool.OnePool(mcf, pp, false, log);
+      BaseConnectionManager2 cm = new NoTxConnectionManager(ccm, poolingStrategy);
+      poolingStrategy.setConnectionListenerFactory(cm);
+      
+      if (pp.prefill)
+      {
+         BasePool bp = (BasePool)poolingStrategy;
+         bp.prefill(null, null, false);
+      }
+      
+      return cm;
+   }
+
+   /**
+    * Shutdown the BaseConnectionManager2
+    * @param cm The instance
+    */
+   private void shutdown(BaseConnectionManager2 cm)
+   {
+      ManagedConnectionPool pool = cm.getPoolingStrategy();
+      pool.shutdown();
+   }
+
+   
+   public void testConnectionCount() throws Exception
+   {
+      InternalManagedConnectionPool.PoolParams pp = new InternalManagedConnectionPool.PoolParams();
+      pp.minSize = 5;
+      pp.maxSize = 10;
+      pp.blockingTimeout = 100;
+      pp.idleTimeout = 500;
+
+      BaseConnectionManager2 cm = getCM(pp);
+      try
+      {
+         ManagedConnectionPool ps = cm.getPoolingStrategy();
+
+         assertTrue("0: Initial check", ps.getConnectionCount() == 0);
+
+         // Get a connection
+         ConnectionListener cl = cm.getManagedConnection(null, null);
+         assertTrue("1: Got a null connection!", cl.getManagedConnection() != null);
+         assertTrue("1: One connection", ps.getConnectionCount() == 1);
+
+         // Get another connection
+         ConnectionListener cl2 = cm.getManagedConnection(null, null);
+         assertTrue("2: Got a null connection!", cl2.getManagedConnection() != null);
+         assertTrue("2: Two connections", ps.getConnectionCount() == 2);
+
+         // Return first
+         cm.returnManagedConnection(cl, true);
+         assertTrue("3: One connection", ps.getConnectionCount() == 1);
+
+         // Return second
+         cm.returnManagedConnection(cl2, true);
+         assertTrue("4: Zero connections", ps.getConnectionCount() == 0);
+      }
+      finally
+      {
+         shutdown(cm);
+      }
+   }
+}