[teiid-commits] teiid SVN: r1528 - trunk/test-integration/db/src/test/java/org/teiid/test/testcases.

[teiid-commits at lists.jboss.org]

Author: vhalbert at redhat.com
Date: 2009-10-06 11:18:45 -0400 (Tue, 06 Oct 2009)
New Revision: 1528

Modified:
   trunk/test-integration/db/src/test/java/org/teiid/test/testcases/BaseAbstractTransactionTestCase.java
Log:
Teiid 773 -  updated logic to support fine grain control over which datasource type is assigned to which model

Modified: trunk/test-integration/db/src/test/java/org/teiid/test/testcases/BaseAbstractTransactionTestCase.java
===================================================================
--- trunk/test-integration/db/src/test/java/org/teiid/test/testcases/BaseAbstractTransactionTestCase.java	2009-10-06 15:16:35 UTC (rev 1527)
+++ trunk/test-integration/db/src/test/java/org/teiid/test/testcases/BaseAbstractTransactionTestCase.java	2009-10-06 15:18:45 UTC (rev 1528)
@@ -21,8 +21,8 @@
         super(name);
     }
     
+
     
-    
     @Override
 	protected void setUp() throws Exception {
 		// TODO Auto-generated method stub
@@ -32,11 +32,27 @@
 
 	}
 
+    
+    protected ConfigPropertyLoader getConfig() {
+    	return this.config;
+    }
 
-
 	protected void addProperty(String key, String value) {
     	config.setProperty(key, value);
     }
+	
+	/**
+	 * Call to assign a specific database type to the model.   When a datasource is requested for this model,
+	 * a datasource of the specific dbtype will be returned.  See {@link DataSourceFactory} for the logic that
+	 * controls this behavior.
+	 * @param modelName
+	 * @param dbtype
+	 *
+	 * @since
+	 */
+	protected void setAssignModelToDatabaseType(String modelName, String dbtype) {
+		config.setModelAssignedToDatabaseType(modelName, dbtype);
+	}
     
     protected TransactionContainer getTransactionContainter() throws QueryTestFailedException {
     	return TransactionFactory.create(config);