[teiid-commits] teiid SVN: r1555 - trunk/test-integration/db/src/main/java/org/teiid/test/framework/datasource.

[teiid-commits at lists.jboss.org]

Author: vhalbert at redhat.com
Date: 2009-11-16 15:57:50 -0500 (Mon, 16 Nov 2009)
New Revision: 1555

Added:
   trunk/test-integration/db/src/main/java/org/teiid/test/framework/datasource/DataStore.java
Log:
Teiid 773 - cleanup and refactoring so that the assumption is there will be 2 datasources required in order to run the test.    

Added: trunk/test-integration/db/src/main/java/org/teiid/test/framework/datasource/DataStore.java
===================================================================
--- trunk/test-integration/db/src/main/java/org/teiid/test/framework/datasource/DataStore.java	                        (rev 0)
+++ trunk/test-integration/db/src/main/java/org/teiid/test/framework/datasource/DataStore.java	2009-11-16 20:57:50 UTC (rev 1555)
@@ -0,0 +1,111 @@
+/*
+ * Copyright (c) 2000-2007 MetaMatrix, Inc.
+ * All rights reserved.
+ */
+package org.teiid.test.framework.datasource;
+
+import java.sql.Connection;
+import java.sql.Statement;
+
+import org.teiid.test.framework.connection.ConnectionStrategy;
+import org.teiid.test.framework.exception.QueryTestFailedException;
+
+/** 
+ * This class loads the data in the databases specified, to a known state
+ */
+public class DataStore {
+
+    /**
+     * Called at the start of all the tests to initialize the database to ensure
+     * it's in the proper state.
+     * 
+     * @param connStrategy
+     */
+    public static void initialize(ConnectionStrategy connStrategy) {
+	try {
+	    load(getConnection("pm1", connStrategy));
+	    
+	    load(getConnection("pm2", connStrategy));
+
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        } 
+    }
+    
+    private static Connection getConnection(String identifier, ConnectionStrategy connStrategy) throws QueryTestFailedException {
+	Connection conn = connStrategy.createDriverConnection(identifier);
+	// force autocommit back to true, just in case the last user didnt
+	try {
+		conn.setAutoCommit(true);
+	} catch (Exception sqle) {
+		throw new QueryTestFailedException(sqle);
+	}
+	
+	return conn;
+    }
+    
+    private static void load(Connection c) throws Exception {
+        // DDL
+        // drop table g1;
+        // drop table g2;
+        
+        // oracle
+        // create Table g1 (e1 number(5) PRIMARY KEY,   e2 varchar2(50));
+        // create Table g2 (e1 number(5) REFERENCES g1, e2 varchar2(50));
+        
+        // SQL Server
+        // create Table g1 (e1 int PRIMARY KEY,   e2 varchar(50));
+        // create Table g2 (e1 int references g1, e2 varchar(50));   
+	            
+
+            Statement stmt = c.createStatement();
+            stmt.addBatch("delete from g2");
+            stmt.addBatch("delete from g1");
+            
+            for (int i = 0; i < 100; i++) {
+                stmt.addBatch("insert into g1 (e1, e2) values("+i+",'"+i+"')");
+            }
+            
+            
+            for (int i = 0; i < 50; i++) {
+                stmt.addBatch("insert into g2 (e1, e2) values("+i+",'"+i+"')");
+            }
+            
+            stmt.executeBatch();
+            stmt.close();
+
+    }
+    
+    /**
+     * Called as part of the setup for each test.  
+     * This will set the database state as if {@link #initialize(ConnectionStrategy)} was called.
+     * However, for performance reasons, the process goes about removing what's not needed instead of cleaning out everything
+     * and reinstalling. 
+     * @param connStrategy
+     */
+    public static void setup(ConnectionStrategy connStrategy) {
+	try {
+	    setUpTest(getConnection("pm1", connStrategy));
+	    
+	    setUpTest(getConnection("pm2", connStrategy));
+
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        } 
+
+	
+    }
+    
+    private static void setUpTest(Connection c) throws Exception {
+	
+            Statement stmt = c.createStatement();
+            stmt.addBatch("delete from g2 where e1 > 50"); //$NON-NLS-1$
+            stmt.addBatch("delete from g1 where e1 > 100"); 
+    
+            stmt.executeBatch();
+            stmt.close();
+            
+
+    }
+
+}


Property changes on: trunk/test-integration/db/src/main/java/org/teiid/test/framework/datasource/DataStore.java
___________________________________________________________________
Name: svn:mime-type
   + text/plain