Examples with QueryBuilder org.apache.drill.test.QueryBuilder used on opensource projects

Example 21 with QueryBuilder

use of org.apache.drill.test.QueryBuilder in project drill by apache.

the class PhoenixSQLTest method testJoinPushdown.

@Test
public void testJoinPushdown() throws Exception {
    String sql = "select a.n_name, b.r_name from phoenix123.v1.nation a join phoenix123.v1.region b " + "on a.n_regionkey = b.r_regionkey";
    QueryBuilder builder = client.queryBuilder().sql(sql);
    RowSet sets = builder.rowSet();
    builder.planMatcher().exclude("Join").include("Phoenix\\(.* INNER JOIN").match();
    TupleMetadata schema = new SchemaBuilder().addNullable("n_name", MinorType.VARCHAR).addNullable("r_name", MinorType.VARCHAR).build();
    RowSet expected = new RowSetBuilder(client.allocator(), schema).addRow("ALGERIA", "AFRICA").addRow("ARGENTINA", "AMERICA").addRow("BRAZIL", "AMERICA").addRow("CANADA", "AMERICA").addRow("EGYPT", "MIDDLE EAST").addRow("ETHIOPIA", "AFRICA").addRow("FRANCE", "EUROPE").addRow("GERMANY", "EUROPE").addRow("INDIA", "ASIA").addRow("INDONESIA", "ASIA").addRow("IRAN", "MIDDLE EAST").addRow("IRAQ", "MIDDLE EAST").addRow("JAPAN", "ASIA").addRow("JORDAN", "MIDDLE EAST").addRow("KENYA", "AFRICA").addRow("MOROCCO", "AFRICA").addRow("MOZAMBIQUE", "AFRICA").addRow("PERU", "AMERICA").addRow("CHINA", "ASIA").addRow("ROMANIA", "EUROPE").addRow("SAUDI ARABIA", "MIDDLE EAST").addRow("VIETNAM", "ASIA").addRow("RUSSIA", "EUROPE").addRow("UNITED KINGDOM", "EUROPE").addRow("UNITED STATES", "AMERICA").build();
    new RowSetComparison(expected).verifyAndClearAll(sets);
}

Example 22 with QueryBuilder

use of org.apache.drill.test.QueryBuilder in project drill by apache.

the class PhoenixSQLTest method testExplicitQuery.

@Test
public void testExplicitQuery() throws Exception {
    String sql = "select n_nationkey, n_regionkey, n_name from phoenix123.v1.nation";
    QueryBuilder builder = queryBuilder().sql(sql);
    RowSet sets = builder.rowSet();
    TupleMetadata schema = new SchemaBuilder().addNullable("n_nationkey", MinorType.BIGINT).addNullable("n_regionkey", MinorType.BIGINT).addNullable("n_name", MinorType.VARCHAR).build();
    RowSet expected = new RowSetBuilder(client.allocator(), schema).addRow(0, 0, "ALGERIA").addRow(1, 1, "ARGENTINA").addRow(2, 1, "BRAZIL").addRow(3, 1, "CANADA").addRow(4, 4, "EGYPT").addRow(5, 0, "ETHIOPIA").addRow(6, 3, "FRANCE").addRow(7, 3, "GERMANY").addRow(8, 2, "INDIA").addRow(9, 2, "INDONESIA").addRow(10, 4, "IRAN").addRow(11, 4, "IRAQ").addRow(12, 2, "JAPAN").addRow(13, 4, "JORDAN").addRow(14, 0, "KENYA").addRow(15, 0, "MOROCCO").addRow(16, 0, "MOZAMBIQUE").addRow(17, 1, "PERU").addRow(18, 2, "CHINA").addRow(19, 3, "ROMANIA").addRow(20, 4, "SAUDI ARABIA").addRow(21, 2, "VIETNAM").addRow(22, 3, "RUSSIA").addRow(23, 3, "UNITED KINGDOM").addRow(24, 1, "UNITED STATES").build();
    new RowSetComparison(expected).verifyAndClearAll(sets);
}

Example 23 with QueryBuilder

use of org.apache.drill.test.QueryBuilder in project drill by apache.

the class SecuredPhoenixSQLTest method testJoinWithFilterPushdown.

@Test
@Disabled("use the remote query server directly without minicluster")
public void testJoinWithFilterPushdown() throws Exception {
    String sql = "select 10 as DRILL, a.n_name, b.r_name from phoenix123.v1.nation a join phoenix123.v1.region b " + "on a.n_regionkey = b.r_regionkey where b.r_name = 'ASIA'";
    QueryBuilder builder = client.queryBuilder().sql(sql);
    RowSet sets = builder.rowSet();
    builder.planMatcher().exclude("Join").exclude("Filter").include("Phoenix\\(.* INNER JOIN .* WHERE").match();
    TupleMetadata schema = new SchemaBuilder().addNullable("DRILL", MinorType.INT).addNullable("n_name", MinorType.VARCHAR).addNullable("r_name", MinorType.VARCHAR).build();
    RowSet expected = new RowSetBuilder(client.allocator(), schema).addRow(10, "INDIA", "ASIA").addRow(10, "INDONESIA", "ASIA").addRow(10, "JAPAN", "ASIA").addRow(10, "CHINA", "ASIA").addRow(10, "VIETNAM", "ASIA").build();
    new RowSetComparison(expected).verifyAndClearAll(sets);
}

Example 24 with QueryBuilder

use of org.apache.drill.test.QueryBuilder in project drill by apache.

the class SecuredPhoenixSQLTest method doTestExplicitQuery.

private void doTestExplicitQuery() throws Exception {
    String sql = "select n_nationkey, n_regionkey, n_name from phoenix123.v1.nation";
    QueryBuilder builder = queryBuilder().sql(sql);
    RowSet sets = builder.rowSet();
    TupleMetadata schema = new SchemaBuilder().addNullable("n_nationkey", MinorType.BIGINT).addNullable("n_regionkey", MinorType.BIGINT).addNullable("n_name", MinorType.VARCHAR).build();
    RowSet expected = new RowSetBuilder(client.allocator(), schema).addRow(0, 0, "ALGERIA").addRow(1, 1, "ARGENTINA").addRow(2, 1, "BRAZIL").addRow(3, 1, "CANADA").addRow(4, 4, "EGYPT").addRow(5, 0, "ETHIOPIA").addRow(6, 3, "FRANCE").addRow(7, 3, "GERMANY").addRow(8, 2, "INDIA").addRow(9, 2, "INDONESIA").addRow(10, 4, "IRAN").addRow(11, 4, "IRAQ").addRow(12, 2, "JAPAN").addRow(13, 4, "JORDAN").addRow(14, 0, "KENYA").addRow(15, 0, "MOROCCO").addRow(16, 0, "MOZAMBIQUE").addRow(17, 1, "PERU").addRow(18, 2, "CHINA").addRow(19, 3, "ROMANIA").addRow(20, 4, "SAUDI ARABIA").addRow(21, 2, "VIETNAM").addRow(22, 3, "RUSSIA").addRow(23, 3, "UNITED KINGDOM").addRow(24, 1, "UNITED STATES").build();
    new RowSetComparison(expected).verifyAndClearAll(sets);
}

Example 25 with QueryBuilder

use of org.apache.drill.test.QueryBuilder in project drill by apache.

the class SecuredPhoenixSQLTest method doTestLimitPushdown.

private void doTestLimitPushdown() throws Exception {
    String sql = "select n_name, n_regionkey from phoenix123.v1.nation limit 20 offset 10";
    QueryBuilder builder = client.queryBuilder().sql(sql);
    RowSet sets = builder.rowSet();
    builder.planMatcher().exclude("Limit").include("OFFSET .* ROWS FETCH NEXT .* ROWS ONLY").match();
    assertEquals(15, sets.rowCount());
    TupleMetadata schema = new SchemaBuilder().addNullable("n_name", MinorType.VARCHAR).addNullable("n_regionkey", MinorType.BIGINT).build();
    RowSet expected = new RowSetBuilder(client.allocator(), schema).addRow("IRAN", 4).addRow("IRAQ", 4).addRow("JAPAN", 2).addRow("JORDAN", 4).addRow("KENYA", 0).addRow("MOROCCO", 0).addRow("MOZAMBIQUE", 0).addRow("PERU", 1).addRow("CHINA", 2).addRow("ROMANIA", 3).addRow("SAUDI ARABIA", 4).addRow("VIETNAM", 2).addRow("RUSSIA", 3).addRow("UNITED KINGDOM", 3).addRow("UNITED STATES", 1).build();
    new RowSetComparison(expected).verifyAndClearAll(sets);
}