use of org.apache.beam.examples.cookbook.JoinExamples.ExtractEventDataFn in project beam by apache.
the class JoinExamplesTest method testExtractEventDataFn.
@Test
public void testExtractEventDataFn() throws Exception {
DoFnTester<TableRow, KV<String, String>> extractEventDataFn = DoFnTester.of(new ExtractEventDataFn());
List<KV<String, String>> results = extractEventDataFn.processBundle(EVENTS);
Assert.assertThat(results, CoreMatchers.hasItem(kv1));
Assert.assertThat(results, CoreMatchers.hasItem(kv2));
}
Aggregations