Search in sources :

Example 1 with ExtractEventDataFn

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));
}
Also used : ExtractEventDataFn(org.apache.beam.examples.cookbook.JoinExamples.ExtractEventDataFn) TableRow(com.google.api.services.bigquery.model.TableRow) KV(org.apache.beam.sdk.values.KV) Test(org.junit.Test)

Aggregations

TableRow (com.google.api.services.bigquery.model.TableRow)1 ExtractEventDataFn (org.apache.beam.examples.cookbook.JoinExamples.ExtractEventDataFn)1 KV (org.apache.beam.sdk.values.KV)1 Test (org.junit.Test)1