Examples with TestingMetadata io.prestosql.testing.TestingMetadata used on opensource projects

Example 1 with TestingMetadata

use of io.prestosql.testing.TestingMetadata in project hetu-core by openlookeng.

the class TestCachedSqlQueryExecution method testRenamedTable.

@Test
public void testRenamedTable() throws Exception {
    setupWithExecutionPlanCacheEnabled(DEFAULT_SESSION);
    TestingMetadata metadata = new TestingMetadata();
    queryRunner.installPlugin(new TestPlugin(metadata));
    queryRunner.createCatalog("test", "test");
    SqlQueryManager manager = (SqlQueryManager) queryRunner.getCoordinator().getQueryManager();
    String query1 = "CREATE TABLE orders AS SELECT ORDERKEY, CUSTKEY, TOTALPRICE FROM tpch.tiny.orders";
    String query2 = "INSERT INTO orders SELECT * FROM orders";
    String query3 = "SELECT * FROM orders";
    Plan plan1 = getPlan(query1, manager);
    Plan plan2 = getPlan(query2, manager);
    Plan plan3 = getPlan(query3, manager);
    metadata.dropTable(DEFAULT_SESSION.toConnectorSession(), metadata.getTableHandle(DEFAULT_SESSION.toConnectorSession(), new SchemaTableName("default", "orders")));
    String query4 = "CREATE TABLE nation AS SELECT * FROM tpch.tiny.nation";
    // Create table again
    Plan plan4 = getPlan(query4, manager);
    // different table with the same name
    metadata.renameTable(DEFAULT_SESSION.toConnectorSession(), metadata.getTableHandle(DEFAULT_SESSION.toConnectorSession(), new SchemaTableName("default", "nation")), new SchemaTableName("default", "orders"));
    // Insert into new table
    Plan plan5 = getPlan(query2, manager);
    Plan plan6 = getPlan(query3, manager);
    assertNotNull(plan1);
    assertNotNull(plan2);
    assertNotNull(plan3);
    assertNotNull(plan4);
    assertNotNull(plan6);
    // Underlying table schema changed
    assertNotSame(plan2.getStatsAndCosts(), plan5.getStatsAndCosts());
    // new plan should not be the same as the old plan
    assertNotSame(plan3.getStatsAndCosts(), plan6.getStatsAndCosts());
}

Example 2 with TestingMetadata

use of io.prestosql.testing.TestingMetadata in project hetu-core by openlookeng.

the class TestCachedSqlQueryExecution method testAlterTable.

@Test
public void testAlterTable() throws Exception {
    setupWithExecutionPlanCacheEnabled(DEFAULT_SESSION);
    TestingMetadata metadata = new TestingMetadata();
    queryRunner.installPlugin(new TestPlugin(metadata));
    queryRunner.createCatalog("test", "test");
    SqlQueryManager manager = (SqlQueryManager) queryRunner.getCoordinator().getQueryManager();
    String query1 = "CREATE TABLE orders AS SELECT ORDERKEY, CUSTKEY, TOTALPRICE FROM tpch.tiny.orders";
    String query2 = "INSERT INTO orders SELECT * FROM orders";
    String query3 = "SELECT * FROM orders";
    Plan plan1 = getPlan(query1, manager);
    Plan plan2 = getPlan(query2, manager);
    Plan plan3 = getPlan(query3, manager);
    metadata.dropTable(DEFAULT_SESSION.toConnectorSession(), metadata.getTableHandle(DEFAULT_SESSION.toConnectorSession(), new SchemaTableName("default", "orders")));
    String query4 = "CREATE TABLE orders AS SELECT * FROM tpch.tiny.orders";
    // Create table again
    Plan plan4 = getPlan(query4, manager);
    // Insert into new table
    Plan plan5 = getPlan(query2, manager);
    Plan plan6 = getPlan(query3, manager);
    assertNotNull(plan1);
    assertNotNull(plan2);
    assertNotNull(plan3);
    assertNotNull(plan4);
    assertNotNull(plan6);
    // Schema changed
    assertNotSame(plan2.getStatsAndCosts(), plan5.getStatsAndCosts());
    // Schema changed
    assertNotSame(plan3.getStatsAndCosts(), plan6.getStatsAndCosts());
}

Example 3 with TestingMetadata

use of io.prestosql.testing.TestingMetadata in project hetu-core by openlookeng.

the class TestCachedSqlQueryExecution method testQueriesWithCurrentPathFunction.

@Test
public void testQueriesWithCurrentPathFunction() throws Exception {
    setupWithExecutionPlanCacheEnabled(DEFAULT_SESSION);
    TestingMetadata metadata = new TestingMetadata();
    queryRunner.installPlugin(new TestPlugin(metadata));
    queryRunner.createCatalog("test", "test");
    SqlQueryManager manager = (SqlQueryManager) queryRunner.getCoordinator().getQueryManager();
    String query1 = "SELECT CURRENT_PATH";
    Plan plan1 = getPlan(query1, manager);
    setupWithExecutionPlanCacheEnabled(TPCH_SESSION);
    TestingMetadata metadata2 = new TestingMetadata();
    queryRunner.installPlugin(new TestPlugin(metadata2));
    queryRunner.createCatalog("test", "test");
    SqlQueryManager manager2 = (SqlQueryManager) queryRunner.getCoordinator().getQueryManager();
    Plan plan2 = getPlan(query1, manager2);
    assertNotNull(plan1);
    assertNotNull(plan2);
    // Should not have the same plan
    assertNotSame(plan1.getStatsAndCosts(), plan2.getStatsAndCosts());
    MaterializedResult rows1 = queryRunner.execute(DEFAULT_SESSION, query1);
    MaterializedResult rows2 = queryRunner.execute(TPCH_SESSION, query1);
    assertEquals(rows1.getRowCount(), rows2.getRowCount());
    MaterializedRow row1 = Iterables.getOnlyElement(rows1);
    MaterializedRow row2 = Iterables.getOnlyElement(rows2);
    assertEquals(row1.getFieldCount(), row2.getFieldCount());
    assertNotSame(row1.getFields(), row2.getFields());
}

Example 4 with TestingMetadata

use of io.prestosql.testing.TestingMetadata in project hetu-core by openlookeng.

the class TestCachedSqlQueryExecution method testCreateTableRepeated.

@Test
public void testCreateTableRepeated() throws Exception {
    setupWithExecutionPlanCacheEnabled(DEFAULT_SESSION);
    TestingMetadata metadata = new TestingMetadata();
    queryRunner.installPlugin(new TestPlugin(metadata));
    queryRunner.createCatalog("test", "test");
    SqlQueryManager manager = (SqlQueryManager) queryRunner.getCoordinator().getQueryManager();
    String query1 = "CREATE TABLE orders AS SELECT * FROM tpch.tiny.orders";
    String query3 = "INSERT INTO orders SELECT * FROM orders";
    String query4 = "INSERT INTO orders SELECT * FROM test.default.orders";
    Plan plan1 = getPlan(query1, manager);
    metadata.dropTable(DEFAULT_SESSION.toConnectorSession(), metadata.getTableHandle(DEFAULT_SESSION.toConnectorSession(), new SchemaTableName("default", "orders")));
    Plan plan2 = getPlan(query1, manager);
    Plan plan3 = getPlan(query3, manager);
    Plan plan4 = getPlan(query4, manager);
    assertNotNull(plan1);
    assertNotNull(plan2);
    assertNotNull(plan3);
    assertNotNull(plan4);
    // Create statements should not be cached
    assertNotSame(plan1.getStatsAndCosts(), plan2.getStatsAndCosts());
    // Statement does not include fully qualified name
    assertNotSame(plan3.getStatsAndCosts(), plan4.getStatsAndCosts());
}

Example 5 with TestingMetadata

use of io.prestosql.testing.TestingMetadata in project hetu-core by openlookeng.

the class TestCachedSqlQueryExecution method testQueriesWithCurrentTimeFunction.

@Test
public void testQueriesWithCurrentTimeFunction() throws Exception {
    setupWithExecutionPlanCacheEnabled(DEFAULT_SESSION);
    TestingMetadata metadata = new TestingMetadata();
    queryRunner.installPlugin(new TestPlugin(metadata));
    queryRunner.createCatalog("test", "test");
    SqlQueryManager manager = (SqlQueryManager) queryRunner.getCoordinator().getQueryManager();
    String query1 = "SELECT CURRENT_TIME";
    Plan plan1 = getPlan(query1, manager);
    Plan plan2 = getPlan(query1, manager);
    assertNotNull(plan1);
    assertNotNull(plan2);
    // Should not have the same plan
    assertNotSame(plan1.getStatsAndCosts(), plan2.getStatsAndCosts());
    MaterializedResult rows1 = queryRunner.execute(DEFAULT_SESSION, query1);
    MaterializedResult rows2 = queryRunner.execute(DEFAULT_SESSION, query1);
    assertEquals(rows1.getRowCount(), rows2.getRowCount());
    MaterializedRow row1 = Iterables.getOnlyElement(rows1);
    MaterializedRow row2 = Iterables.getOnlyElement(rows2);
    assertEquals(row1.getFieldCount(), row2.getFieldCount());
    assertNotSame(row1.getFields(), row2.getFields());
}