@Override public PCollection<Row> buildIOReader(PBegin begin) { assert begin.getPipeline() == upstream.getPipeline(); return upstream.apply(Convert.toRows()); }
@Test @Category(NeedsRunner.class) public void testToRows() { PCollection<Row> rows = pipeline.apply(Create.of(new POJO1())).apply(Convert.toRows()); PAssert.that(rows).containsInAnyOrder(EXPECTED_ROW1); pipeline.run(); }