use of org.apache.drill.exec.physical.rowSet.DirectRowSet in project drill by apache.
the class TestJdbcWriterWithMySQL method testBasicCTASWithSpacesInFieldNames.
@Test
public void testBasicCTASWithSpacesInFieldNames() throws Exception {
String query = "CREATE TABLE mysql.`drill_mysql_test`.`test table` (`My id`, `My name`) AS SELECT * FROM (VALUES(1,2), (3,4))";
// Create the table and insert the values
QuerySummary insertResults = queryBuilder().sql(query).run();
assertTrue(insertResults.succeeded());
// Query the table to see if the insertion was successful
String testQuery = "SELECT * FROM mysql.`drill_mysql_test`.`test table`";
DirectRowSet results = queryBuilder().sql(testQuery).rowSet();
TupleMetadata expectedSchema = new SchemaBuilder().add("My id", MinorType.BIGINT, DataMode.OPTIONAL).add("My name", MinorType.BIGINT, DataMode.OPTIONAL).buildSchema();
RowSet expected = new RowSetBuilder(client.allocator(), expectedSchema).addRow(1L, 2L).addRow(3L, 4L).build();
RowSetUtilities.verify(expected, results);
// Now drop the table
String dropQuery = "DROP TABLE mysql.`drill_mysql_test`.`test table`";
QuerySummary dropResults = queryBuilder().sql(dropQuery).run();
assertTrue(dropResults.succeeded());
}
use of org.apache.drill.exec.physical.rowSet.DirectRowSet in project drill by apache.
the class TestJdbcPluginWithClickhouse method pushDownAggWithDecimal.
@Test
public void pushDownAggWithDecimal() throws Exception {
String query = "SELECT sum(decimal_field * smallint_field) AS `order_total`\n" + "FROM clickhouse.`default`.person e";
DirectRowSet results = queryBuilder().sql(query).rowSet();
TupleMetadata expectedSchema = new SchemaBuilder().addNullable("order_total", TypeProtos.MinorType.VARDECIMAL, 38, 2).buildSchema();
RowSet expected = client.rowSetBuilder(expectedSchema).addRow(123.32).build();
RowSetUtilities.verify(expected, results);
}
use of org.apache.drill.exec.physical.rowSet.DirectRowSet in project drill by apache.
the class TestJdbcPluginWithMySQLIT method pushDownAggWithDecimal.
@Test
public void pushDownAggWithDecimal() throws Exception {
String query = "SELECT sum(decimal_field * smallint_field) AS `order_total`\n" + "FROM mysql.`drill_mysql_test`.person e";
DirectRowSet results = queryBuilder().sql(query).rowSet();
TupleMetadata expectedSchema = new SchemaBuilder().addNullable("order_total", TypeProtos.MinorType.VARDECIMAL, 38, 2).buildSchema();
RowSet expected = client.rowSetBuilder(expectedSchema).addRow(123.32).build();
RowSetUtilities.verify(expected, results);
}
use of org.apache.drill.exec.physical.rowSet.DirectRowSet in project drill by apache.
the class TestJdbcWriterWithH2 method testBasicCTASWithSpacesInTableName.
@Test
public void testBasicCTASWithSpacesInTableName() throws Exception {
String query = "CREATE TABLE h2.tmp.`drill_h2_test`.`test table` (ID, NAME) AS SELECT * FROM (VALUES(1,2), (3,4))";
// Create the table and insert the values
QuerySummary insertResults = queryBuilder().sql(query).run();
assertTrue(insertResults.succeeded());
try {
// Query the table to see if the insertion was successful
String testQuery = "SELECT * FROM h2.tmp.`drill_h2_test`.`test table`";
DirectRowSet results = queryBuilder().sql(testQuery).rowSet();
TupleMetadata expectedSchema = new SchemaBuilder().add("ID", MinorType.BIGINT, DataMode.OPTIONAL).add("NAME", MinorType.BIGINT, DataMode.OPTIONAL).buildSchema();
RowSet expected = new RowSetBuilder(client.allocator(), expectedSchema).addRow(1L, 2L).addRow(3L, 4L).build();
RowSetUtilities.verify(expected, results);
} finally {
String dropQuery = "DROP TABLE h2.tmp.`drill_h2_test`.`test table`";
QuerySummary dropResults = queryBuilder().sql(dropQuery).run();
assertTrue(dropResults.succeeded());
}
}
use of org.apache.drill.exec.physical.rowSet.DirectRowSet in project drill by apache.
the class TestJdbcWriterWithH2 method testCTASWithDuplicateTable.
@Test
public void testCTASWithDuplicateTable() throws Exception {
String query = String.format("CREATE TABLE %s (ID, NAME) AS SELECT * FROM (VALUES(1,2), (3,4))", TEST_TABLE);
// Create the table and insert the values
QuerySummary insertResults = queryBuilder().sql(query).run();
assertTrue(insertResults.succeeded());
try {
// Run the query again, should fail.
try {
queryBuilder().sql(query).run();
fail();
} catch (UserRemoteException e) {
assertTrue(e.getMessage().contains("VALIDATION ERROR"));
}
// Try again with IF NOT EXISTS, Should not do anything, but not throw an exception
query = String.format("CREATE TABLE IF NOT EXISTS %s (ID, NAME) AS SELECT * FROM (VALUES(1,2), (3,4))", TEST_TABLE);
DirectRowSet results = queryBuilder().sql(query).rowSet();
TupleMetadata expectedSchema = new SchemaBuilder().add("ok", MinorType.BIT).add("summary", MinorType.VARCHAR, DataMode.OPTIONAL).buildSchema();
RowSet expected = new RowSetBuilder(client.allocator(), expectedSchema).addRow(false, "A table or view with given name [test_table] already exists in schema [h2.tmp.drill_h2_test]").build();
RowSetUtilities.verify(expected, results);
} finally {
QuerySummary dropResults = queryBuilder().sql(DROP_TEST_TABLE).run();
assertTrue(dropResults.succeeded());
}
}
Aggregations