[jboss-svn-commits] JBL Code SVN: r31880 - labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/auction/test/basic.

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

Author: whitingjr
Date: 2010-03-01 10:33:36 -0500 (Mon, 01 Mar 2010)
New Revision: 31880

Modified:
   labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/auction/test/basic/ProfiledStateTransitions.java
Log:
Added configuration of the connection handler type.


Modified: labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/auction/test/basic/ProfiledStateTransitions.java
===================================================================
--- labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/auction/test/basic/ProfiledStateTransitions.java	2010-03-01 14:46:00 UTC (rev 31879)
+++ labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/auction/test/basic/ProfiledStateTransitions.java	2010-03-01 15:33:36 UTC (rev 31880)
@@ -65,6 +65,7 @@
 import org.jboss.ejb3.embedded.EJB3StandaloneBootstrap;
 import org.jboss.jbossts.performance.persistence.vendor.AbstractVendorControl;
 import org.jboss.jbossts.performance.profiler.Profiler;
+import org.jboss.jbossts.performance.task.pooling.ConnectionHandler;
 import org.testng.Assert;
 import org.testng.annotations.AfterClass;
 import org.testng.annotations.AfterGroups;
@@ -123,6 +124,7 @@
    private DefaultDataTypeFactory dataTypeFactoryB;
    protected Profiler profiler;
    private int concurrencyCount;
+   private Class<ConnectionHandler> connectionHandlerClass;
    
    @Override
    public void prepareDataSet(String location, String factory) throws Exception
@@ -262,11 +264,12 @@
          String jndiNameUserTx) throws Exception
    {/* Replace implementation to configure Hibernate Session Factory JNDI name. */}
    
+   @SuppressWarnings("unchecked")
    @BeforeTest(groups = "integration-warmup")
    @Parameters(
-   {"deploy_beans_xml", "scan_classpath", "jndi_datasource_A", "jndi_datasource_B", "jndi_name_emf_A", "jndi_name_emf_B", "jndi_name_usertx_A", "jndi_name_usertx_B", "session_jndi_A", "session_jndi_B", "invoke_optional_write", "transaction_timeout", "use_lrco", "concurrency_count"})
+   {"deploy_beans_xml", "scan_classpath", "jndi_datasource_A", "jndi_datasource_B", "jndi_name_emf_A", "jndi_name_emf_B", "jndi_name_usertx_A", "jndi_name_usertx_B", "session_jndi_A", "session_jndi_B", "invoke_optional_write", "transaction_timeout", "use_lrco", "concurrency_count", "connection_handler_fqcn"})
    public void startContainer(String deployBeansXml, String scanClasspath, String jndiDatasourceA, String jndiDatasourceB, String jndiNameEMF_A, String jndiNameEMF_B,
-         String jndiNameUserTxA, String jndiNameUserTxB, String jndiNameHSF_A, String jndiNameHSF_B, String flag, String transactionTimeout, String useLRCO, String concurrencyCountConfig) throws Exception
+         String jndiNameUserTxA, String jndiNameUserTxB, String jndiNameHSF_A, String jndiNameHSF_B, String flag, String transactionTimeout, String useLRCO, String concurrencyCountConfig, String connectionHandlerConfig) throws Exception
    {
       // Set configuration options from TestNG parameters
       JNDI_DATASOURCE = jndiDatasourceA;
@@ -298,6 +301,11 @@
          this.concurrencyCount = Integer.parseInt(concurrencyCountConfig);
       }
       
+      if (StringUtils.isNotBlank(connectionHandlerConfig))
+      {
+         this.connectionHandlerClass  = (Class<ConnectionHandler>)Class.forName(connectionHandlerConfig);
+      }
+      
       // Boot the JBoss Microcontainer with EJB3 settings, automatically
       // loads ejb3-interceptors-aop.xml and embedded-jboss-beans.xml
       EJB3StandaloneBootstrap.boot(null);
@@ -606,4 +614,8 @@
    {
       return getDataSource(this.JNDI_DATASOURCE_B);
    }
+   public Class<ConnectionHandler> getConnectionHandlerClass()
+   {
+      return connectionHandlerClass;
+   }
 }