Examples with QueryExecution org.teiid.test.framework.query.QueryExecution used on opensource projects

Example 6 with QueryExecution

use of org.teiid.test.framework.query.QueryExecution in project teiid by teiid.

the class AutoWrapTransactionTests 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 7 with QueryExecution

use of org.teiid.test.framework.query.QueryExecution in project teiid by teiid.

the class CommonTransactionTests method testMultipleSourcePreparedUpdate.

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

        public void testCase() throws Exception {
            Integer value = new Integer(500);
            execute("insert into vm.g1 (pm1e1, pm1e2, pm2e1, pm2e2) values(?,?,?,?)", new Object[] { value, value.toString(), value, value.toString() });
        }
    };
    // 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 8 with QueryExecution

use of org.teiid.test.framework.query.QueryExecution in project teiid by teiid.

the class CommonTransactionTests method testMultipleSourceTimeout.

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

        public void testCase() throws Exception {
            // time out after 1 sec
            executeBatch(getMultipleSourceBatch(), 1);
        }

        public boolean exceptionExpected() {
            return true;
        }

        public void after() {
            if (!exceptionOccurred()) {
                Assert.assertTrue("should have failed with time out exception", false);
            } else {
                if (getLastException() != null) {
                    String msg = "NA";
                    SQLException s = getLastException();
                    Throwable t = s.getCause();
                    if (t instanceof TimeoutException) {
                        msg = t.getMessage();
                    } else if (s instanceof TeiidSQLException) {
                        TeiidSQLException mm = (TeiidSQLException) t;
                        if (mm.getNextException() != null) {
                            SQLException next = mm.getNextException();
                            msg = next.getMessage();
                        } else {
                            msg = mm.getMessage();
                        }
                    } else {
                        msg = s.getMessage();
                    }
                    boolean isfound = (msg.indexOf("Operation timed out before completion") != -1 ? true : false);
                    Assert.assertTrue("Exception Message didnt match 'Operation timed out before completion' found: " + msg, isfound);
                } else {
                    Assert.assertTrue("Program Error: it indicates exception occured, but no exception is found", false);
                }
            }
        }
    };
    getTransactionContainter().runTransaction(userTxn);
    // now verify the results (this may finish under one second, then this test is not valid)
    AbstractQueryTest test = new QueryExecution(userTxn.getSource("pm1"));
    test.execute("select * from g1 where e1 >= 600 and e1 < 750");
    test.assertRowCount(0);
    test.execute("select * from g2 where e1 >= 600 and e1 < 750");
    test.assertRowCount(0);
    test.execute("select distinct e2 from g1 where e1 >= 600 and e1 < 750");
    test.assertRowCount(0);
}

Example 9 with QueryExecution

use of org.teiid.test.framework.query.QueryExecution in project teiid by teiid.

the class CommonTransactionTests method testMultipleSourceBulkRowInsert.

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

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

Example 10 with QueryExecution

use of org.teiid.test.framework.query.QueryExecution in project teiid by teiid.

the class CommonTransactionTests method testSingleSourceUpdate.

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

        public void testCase() throws Exception {
            execute("insert into pm1.g1 (e1, e2) values(100, '100')");
        }
    };
    // 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(1);
}