@Test public void testCheckEdge() throws Exception { HiveToEdge hiveToEdge = new HiveIntNullEdge(); HiveInputDescription inputDesc = new HiveInputDescription(); HiveTableSchema schema = TestSchema.builder().addColumn("foo", HiveType.INT).addColumn("bar", HiveType.INT).build(); hiveToEdge.checkInput(inputDesc, schema); schema = TestSchema.builder().addColumn("foo", HiveType.INT).addColumn("bar", HiveType.LONG).build(); checkEdgeThrows(hiveToEdge, inputDesc, schema); }
@Test public void testCheckVertex() throws Exception { HiveToVertex hiveToVertex = new HiveIntNullNullVertex(); HiveInputDescription inputDesc = new HiveInputDescription(); HiveTableSchema schema = TestSchema.builder().addColumn("foo", HiveType.INT).addColumn("bar", HiveType.LIST).build(); hiveToVertex.checkInput(inputDesc, schema); schema = TestSchema.builder() .addColumn("foo", HiveType.INT) .addColumn("bar", HiveType.STRING) .build(); checkVertexThrows(hiveToVertex, inputDesc, schema); }