Examples with AbstractQueryTest org.teiid.jdbc.AbstractQueryTest used on opensource projects

Example 1 with AbstractQueryTest

use of org.teiid.jdbc.AbstractQueryTest in project teiid by teiid.

the class OffWrapTransactionTests method testMultipleSourceUpdate.

/**
 * Sources = 2
 * Commands = 1, Update
 * Batching = Full Processing, Single Connector Batch
 * result = commit
 */
@Test
public void testMultipleSourceUpdate() throws Exception {
    AbstractQueryTransactionTest userTxn = new AbstractQueryTransactionTest("testMultipleSourceUpdate") {

        public void testCase() throws Exception {
            execute("insert into vm.g1 (pm1e1, pm1e2, pm2e1, pm2e2) values(500, '500', 500, '500')");
        }
    };
    // run test
    getTransactionContainter().runTransaction(userTxn);
    // now verify the results
    AbstractQueryTest test = new QueryExecution(userTxn.getSource("pm1"));
    test.execute("select * from g1 where e1 = 500");
    test.assertRowCount(1);
    test = new QueryExecution(userTxn.getSource("pm2"));
    test.execute("select * from g1 where e1 = 500");
    test.assertRowCount(1);
}

Example 2 with AbstractQueryTest

use of org.teiid.jdbc.AbstractQueryTest in project teiid by teiid.

the class OffWrapTransactionTests method testSingleSourceMultipleCommands.

/**
 * Sources = 1
 * Commands = multiple - Success
 * Batching = Full Processing, Single Connector Batch
 * result = commit
 */
@Test
public void testSingleSourceMultipleCommands() throws Exception {
    AbstractQueryTransactionTest userTxn = new AbstractQueryTransactionTest("testSingleSourceMultipleCommands") {

        public void testCase() throws Exception {
            execute("delete from pm1.g1 where pm1.g1.e1 >= ?", new Object[] { new Integer(100) });
            execute("select * from pm1.g1");
            assertRowCount(100);
            for (int i = 100; i < 110; i++) {
                Integer val = new Integer(i);
                execute("insert into pm1.g1 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
                execute("insert into pm1.g2 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
            }
        }
    };
    // run test
    getTransactionContainter().runTransaction(userTxn);
    // now verify the results
    AbstractQueryTest test = new QueryExecution(userTxn.getSource("pm1"));
    test.execute("select * from g1 where e1 >= 100");
    test.assertRowCount(10);
    test.execute("select * from g2 where e1 >= 100");
    test.assertRowCount(10);
}

Example 3 with AbstractQueryTest

use of org.teiid.jdbc.AbstractQueryTest in project teiid by teiid.

the class OffWrapTransactionTests method testMultipleSourceMultipleCommandsReferentialIntegrityRollback.

/**
 * Sources = 2
 * Commands = multiple - Success
 * Batching = Full Processing, Single Connector Batch
 * result = rollback
 */
@Test
public void testMultipleSourceMultipleCommandsReferentialIntegrityRollback() throws Exception {
    AbstractQueryTransactionTest userTxn = new AbstractQueryTransactionTest("testMultipleSourceMultipleCommandsReferentialIntegrityRollback") {

        public void testCase() throws Exception {
            for (int i = 700; i < 710; i++) {
                Integer val = new Integer(i);
                execute("insert into pm1.g1 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
                execute("insert into pm1.g2 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
                execute("insert into pm2.g1 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
                execute("insert into pm2.g2 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
            }
            // force the rollback by trying to insert an invalid row.
            execute("insert into pm1.g2 (e1, e2) values(?,?)", new Object[] { new Integer(9999), "9999" });
        }

        public boolean exceptionExpected() {
            return true;
        }
    };
    // run test
    getTransactionContainter().runTransaction(userTxn);
    // now verify the results
    AbstractQueryTest test = new QueryExecution(userTxn.getSource("pm1"));
    test.execute("select * from g1 where e1 >= 700 and e1 < 710");
    test.assertRowCount(10);
    test = new QueryExecution(userTxn.getSource("pm2"));
    test.execute("select * from g1 where e1 >= 700 and e1 < 710");
    test.assertRowCount(10);
}

Example 4 with AbstractQueryTest

use of org.teiid.jdbc.AbstractQueryTest in project teiid by teiid.

the class OnWrapTransactionTests method testSingleSourceBatchCommandReferentialIntegrityRollback.

/**
 * Sources = 1 Commands = multiple Success Batching = Full Processing,
 * Single Connector Batch result = rollback
 */
@Test
public void testSingleSourceBatchCommandReferentialIntegrityRollback() throws Exception {
    AbstractQueryTransactionTest userTxn = new AbstractQueryTransactionTest("testSingleSourceBatchCommandReferentialIntegrityRollback") {

        public void testCase() throws Exception {
            ArrayList list = new ArrayList();
            for (int i = 200; i < 210; i++) {
                list.add("insert into pm1.g1 (e1, e2) values(" + i + ",'" + i + "')");
            }
            // try to rollback, since we are in single batch it must
            // rollback
            list.add("insert into pm1.g2 (e1, e2) values(9999,'9999')");
            executeBatch((String[]) list.toArray(new String[list.size()]));
        }

        public boolean exceptionExpected() {
            return true;
        }
    };
    // run test
    getTransactionContainter().runTransaction(userTxn);
    // now verify the results
    AbstractQueryTest test = new QueryExecution(userTxn.getSource("pm1"));
    test.execute("select * from g1 where e1 >= 200 and e1 < 210");
    test.assertRowCount(0);
    test.execute("select * from g2 where e1 = 9999");
    test.assertRowCount(0);
}

Example 5 with AbstractQueryTest

use of org.teiid.jdbc.AbstractQueryTest in project teiid by teiid.

the class AutoWrapTransactionTests method testSingleSourceMultipleCommandsReferentialIntegrityRollback.

/**
 * Sources = 1
 * Commands = multiple - Success
 * Batching = Full Processing, Single Connector Batch
 * result = rollback
 */
@Test
public void testSingleSourceMultipleCommandsReferentialIntegrityRollback() throws Exception {
    AbstractQueryTransactionTest userTxn = new AbstractQueryTransactionTest("testSingleSourceMultipleCommandsReferentialIntegrityRollback") {

        public void testCase() throws Exception {
            for (int i = 200; i < 210; i++) {
                Integer val = new Integer(i);
                execute("insert into pm1.g1 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
                execute("insert into pm1.g2 (e1, e2) values(?,?)", new Object[] { val, val.toString() });
            }
            // try to rollback, however since this pessimistic above two are already commited
            execute("insert into pm1.g2 (e1, e2) values(?,?)", new Object[] { new Integer(9999), "9999" });
        }

        public boolean exceptionExpected() {
            return true;
        }
    };
    // run test
    getTransactionContainter().runTransaction(userTxn);
    // now verify the results
    AbstractQueryTest test = new QueryExecution(userTxn.getSource("pm1"));
    test.execute("select * from g1 where e1 >= 200 and e1 < 210");
    test.assertRowCount(10);
    test.execute("select * from g2 where e1 = 9999");
    test.assertRowCount(0);
}