private void simpleAvroTestHelper(AvroTestUtil.AvroTestRecordWriter testSetup, final String sql) throws Exception { final String file = testSetup.getFilePath(); final String sqlWithTable = String.format(sql, file); testBuilder() .sqlQuery(sqlWithTable) .unOrdered() .baselineRecords(testSetup.getExpectedRecords()) .go(); }
@Test public void testSimplePrimitiveSchema_SelectColumnSubset() throws Exception { final AvroTestUtil.AvroTestRecordWriter testSetup = AvroTestUtil.generateSimplePrimitiveSchema_NoNullValues(); final String file = testSetup.getFilePath(); final String sql = "select h_boolean, e_double from dfs_root.\"" + file + "\""; List<String> projectList = Lists.newArrayList("`h_boolean`", "`e_double`"); testBuilder() .sqlQuery(sql) .unOrdered() .baselineRecords(project(testSetup.getExpectedRecords(), projectList)) .go(); }
@Test public void testSimpleEnumSchema_NoNullValues() throws Exception { final AvroTestUtil.AvroTestRecordWriter testSetup = AvroTestUtil.generateSimpleEnumSchema_NoNullValues(); final String file = testSetup.getFilePath(); final String sql = "select a_string, b_enum from dfs_root.\"" + file + "\""; List<String> projectList = Lists.newArrayList("`a_string`", "`b_enum`"); testBuilder() .sqlQuery(sql) .unOrdered() .baselineRecords(project(testSetup.getExpectedRecords(), projectList)) .go(); }
@Test public void testSimpleNestedSchema_StarQuery() throws Exception { final AvroTestUtil.AvroTestRecordWriter testSetup = AvroTestUtil.generateSimpleNestedSchema_NoNullValues(); final String file = testSetup.getFilePath(); final String sql = "select * from dfs_root.\"" + file + "\""; testBuilder() .sqlQuery(sql) .unOrdered() .baselineRecords(testSetup.getExpectedRecords()) .go(); } @Test
@Test @Ignore public void testSimplePrimitiveSchema_NoNullValues() throws Exception { final AvroTestUtil.AvroTestRecordWriter testSetup = AvroTestUtil.generateSimplePrimitiveSchema_NoNullValues(); final String file = testSetup.getFilePath(); final String sql = "select a_string, b_int, c_long, d_float, e_double, f_bytes, h_boolean, g_null " + "from dfs_root.\"" + file + "\""; test(sql); testBuilder() .sqlQuery(sql) .unOrdered() .baselineRecords(testSetup.getExpectedRecords()) .go(); }
@Test @Ignore public void testBatchCutoff() throws Exception { final AvroTestUtil.AvroTestRecordWriter testSetup = AvroTestUtil.generateSimplePrimitiveSchema_NoNullValues(5000); final String file = testSetup.getFilePath(); final String sql = "select a_string, b_int, c_long, d_float, e_double, f_bytes, h_boolean, g_null " + "from dfs_root.\"" + file + "\""; test(sql); testBuilder() .sqlQuery(sql) .unOrdered() .expectsNumBatches(2) .baselineRecords(testSetup.getExpectedRecords()) .go(); }