Examples with ExtractCountryInfoFn org.apache.beam.examples.cookbook.JoinExamples.ExtractCountryInfoFn used on opensource projects

Example 1 with ExtractCountryInfoFn

use of org.apache.beam.examples.cookbook.JoinExamples.ExtractCountryInfoFn in project beam by apache.

the class JoinExamplesTest method testExtractCountryInfoFn.

@Test
public void testExtractCountryInfoFn() throws Exception {
    DoFnTester<TableRow, KV<String, String>> extractCountryInfoFn = DoFnTester.of(new ExtractCountryInfoFn());
    List<KV<String, String>> results = extractCountryInfoFn.processBundle(CCS);
    Assert.assertThat(results, CoreMatchers.hasItem(kv3));
    Assert.assertThat(results, CoreMatchers.hasItem(kv4));
}