From 51c8873076d517243ab75ba4563fe53129d22042 Mon Sep 17 00:00:00 2001 From: Cade Markegard Date: Wed, 3 Jul 2019 23:00:43 -0700 Subject: [PATCH] [BEAM-7690] Port WordCountTest off DoFnTester --- .../apache/beam/examples/WordCountTest.java | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/examples/java/src/test/java/org/apache/beam/examples/WordCountTest.java b/examples/java/src/test/java/org/apache/beam/examples/WordCountTest.java index ce81aa78eab3b..6b9916e87271d 100644 --- a/examples/java/src/test/java/org/apache/beam/examples/WordCountTest.java +++ b/examples/java/src/test/java/org/apache/beam/examples/WordCountTest.java @@ -28,11 +28,9 @@ import org.apache.beam.sdk.testing.ValidatesRunner; import org.apache.beam.sdk.transforms.Create; import org.apache.beam.sdk.transforms.DoFn; -import org.apache.beam.sdk.transforms.DoFnTester; import org.apache.beam.sdk.transforms.MapElements; +import org.apache.beam.sdk.transforms.ParDo; import org.apache.beam.sdk.values.PCollection; -import org.hamcrest.CoreMatchers; -import org.junit.Assert; import org.junit.Rule; import org.junit.Test; import org.junit.experimental.categories.Category; @@ -46,15 +44,12 @@ public class WordCountTest { /** Example test that tests a specific {@link DoFn}. */ @Test public void testExtractWordsFn() throws Exception { - DoFnTester extractWordsFn = DoFnTester.of(new ExtractWordsFn()); - - Assert.assertThat( - extractWordsFn.processBundle(" some input words "), - CoreMatchers.hasItems("some", "input", "words")); - Assert.assertThat(extractWordsFn.processBundle(" "), CoreMatchers.hasItems()); - Assert.assertThat( - extractWordsFn.processBundle(" some ", " input", " words"), - CoreMatchers.hasItems("some", "input", "words")); + List words = Arrays.asList(" some input words ", " ", " cool ", " foo", " bar"); + PCollection output = + p.apply(Create.of(words).withCoder(StringUtf8Coder.of())) + .apply(ParDo.of(new ExtractWordsFn())); + PAssert.that(output).containsInAnyOrder("some", "input", "words", "cool", "foo", "bar"); + p.run().waitUntilFinish(); } static final String[] WORDS_ARRAY =