[jboss-svn-commits] JBL Code SVN: r31883 - labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/org/jboss/jbossts/performance/jdbc.

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

Author: whitingjr
Date: 2010-03-01 10:37:49 -0500 (Mon, 01 Mar 2010)
New Revision: 31883

Modified:
   labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/org/jboss/jbossts/performance/jdbc/JDBCTest.java
Log:
Added abstract connection handler implementation to test construction.


Modified: labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/org/jboss/jbossts/performance/jdbc/JDBCTest.java
===================================================================
--- labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/org/jboss/jbossts/performance/jdbc/JDBCTest.java	2010-03-01 15:36:42 UTC (rev 31882)
+++ labs/jbosstm/workspace/whitingjr/trunk/performance/src/test/java/org/jboss/jbossts/performance/jdbc/JDBCTest.java	2010-03-01 15:37:49 UTC (rev 31883)
@@ -22,11 +22,15 @@
 
 package org.jboss.jbossts.performance.jdbc;
 
+import java.lang.reflect.Constructor;
+
+import javax.sql.DataSource;
+
 import org.apache.log4j.NDC;
 import org.jboss.jbossts.performance.MultiThreadedTest;
 import org.jboss.jbossts.performance.configuration.JPAConfiguration;
 import org.jboss.jbossts.performance.configuration.MultithreadedTestConfiguration;
-import org.jboss.jbossts.performance.task.pooling.RawConnection;
+import org.jboss.jbossts.performance.task.pooling.ConnectionHandler;
 
 /**
  * This class creates an instance of the underlying task.
@@ -39,7 +43,8 @@
    @Override
    public Runnable getTask(final MultithreadedTestConfiguration taskConfiguration) throws Exception
    {
-      return new JDBCTask(new JPAConfiguration(new RawConnection(getDataSourceA(), getDataSourceB()), getUserTransactionA(), getEntityManagerFactory(), getEntityManagerFactoryB(), taskConfiguration.getIterationCount(), this.isOptionalWriteEnabled, NDC.cloneStack(), taskConfiguration));
+      Constructor<ConnectionHandler> handlerConstructor =  taskConfiguration.getConnectionHandlerClass().getConstructor(DataSource.class, DataSource.class );
+      return new JDBCTask(new JPAConfiguration(handlerConstructor.newInstance (getDataSourceA(), getDataSourceB()), getUserTransactionA(), getEntityManagerFactory(), getEntityManagerFactoryB(), taskConfiguration.getIterationCount(), this.isOptionalWriteEnabled, NDC.cloneStack(), taskConfiguration));
    }
    @Override
    protected boolean isMultiResourceTest()