Examples with TempTableDataManager org.teiid.query.tempdata.TempTableDataManager used on opensource projects

Example 6 with TempTableDataManager

use of org.teiid.query.tempdata.TempTableDataManager in project teiid by teiid.

the class TestTempTables method testForeignTemp.

@Test
public void testForeignTemp() throws Exception {
    HardcodedDataManager hdm = new HardcodedDataManager(metadata);
    BufferManager bm = BufferManagerFactory.getStandaloneBufferManager();
    SessionAwareCache<CachedResults> cache = new SessionAwareCache<CachedResults>("resultset", DefaultCacheFactory.INSTANCE, SessionAwareCache.Type.RESULTSET, 0);
    cache.setTupleBufferCache(bm);
    dataManager = new TempTableDataManager(hdm, bm, cache);
    // $NON-NLS-1$
    execute("create foreign temporary table x (e1 string options (nameinsource 'a'), e2 integer, e3 string, primary key (e1)) options (cardinality 1000, updatable true, \"other\" 'prop') on pm1", new List[] { Arrays.asList(0) });
    TempMetadataID id = this.tempStore.getMetadataStore().getData().get("x");
    // ensure that we're using the actual metadata
    assertNotNull(id);
    assertNotNull(this.metadata.getPrimaryKey(id));
    assertEquals(1000, this.metadata.getCardinality(id), 0);
    assertEquals("pm1", this.metadata.getName(this.metadata.getModelID(id)));
    assertEquals("prop", this.metadata.getExtensionProperty(id, "other", false));
    hdm.addData("SELECT x.a, x.e2, x.e3 FROM x", new List[] { Arrays.asList(1, 2, "3") });
    // $NON-NLS-1$
    execute("select * from x", new List[] { Arrays.asList(1, 2, "3") });
    hdm.addData("SELECT g_0.e2 AS c_0, g_0.e3 AS c_1, g_0.a AS c_2 FROM x AS g_0 ORDER BY c_1, c_0", new List[] { Arrays.asList(2, "3", "1") });
    hdm.addData("SELECT g_0.e3, g_0.e2 FROM x AS g_0", new List[] { Arrays.asList("3", 2) });
    hdm.addData("DELETE FROM x WHERE x.a = '1'", new List[] { Arrays.asList(1) });
    // ensure compensation behaves as if physical - not temp
    // $NON-NLS-1$
    execute("delete from x where e2 = (select max(e2) from x as z where e3 = x.e3)", new List[] { Arrays.asList(1) }, TestOptimizer.getGenericFinder());
    hdm.addData("SELECT g_0.e1 FROM g1 AS g_0, x AS g_1 WHERE g_1.a = g_0.e1", new List[] { Arrays.asList(1) });
    // ensure pushdown support
    // $NON-NLS-1$
    execute("select g1.e1 from pm1.g1 g1, x where x.e1 = g1.e1", new List[] { Arrays.asList(1) }, TestOptimizer.getGenericFinder());
    try {
        // $NON-NLS-1$
        execute("create local temporary table x (e1 string)", new List[] { Arrays.asList(0) });
        fail();
    } catch (QueryResolverException e) {
    }
    // ensure that drop works
    execute("drop table x", new List[] { Arrays.asList(0) });
    try {
        execute("drop table x", new List[] { Arrays.asList(0) });
        fail();
    } catch (QueryResolverException e) {
    }
}