Repository: incubator-beam Updated Branches: refs/heads/master 0b0a1b797 -> 9fbf429b2
Fix NPE in StatefulParDoEvaluatorFactoryTest mocking Project: http://git-wip-us.apache.org/repos/asf/incubator-beam/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-beam/commit/00b961df Tree: http://git-wip-us.apache.org/repos/asf/incubator-beam/tree/00b961df Diff: http://git-wip-us.apache.org/repos/asf/incubator-beam/diff/00b961df Branch: refs/heads/master Commit: 00b961df2e6086ea9f5cb8b9b8fb747739d33670 Parents: 4927cc1 Author: Kenneth Knowles <k...@google.com> Authored: Wed Dec 14 13:12:43 2016 -0800 Committer: Kenneth Knowles <k...@google.com> Committed: Wed Dec 14 13:12:54 2016 -0800 ---------------------------------------------------------------------- .../beam/runners/direct/StatefulParDoEvaluatorFactoryTest.java | 6 ++++++ 1 file changed, 6 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-beam/blob/00b961df/runners/direct-java/src/test/java/org/apache/beam/runners/direct/StatefulParDoEvaluatorFactoryTest.java ---------------------------------------------------------------------- diff --git a/runners/direct-java/src/test/java/org/apache/beam/runners/direct/StatefulParDoEvaluatorFactoryTest.java b/runners/direct-java/src/test/java/org/apache/beam/runners/direct/StatefulParDoEvaluatorFactoryTest.java index 7c086a1..326310b 100644 --- a/runners/direct-java/src/test/java/org/apache/beam/runners/direct/StatefulParDoEvaluatorFactoryTest.java +++ b/runners/direct-java/src/test/java/org/apache/beam/runners/direct/StatefulParDoEvaluatorFactoryTest.java @@ -31,6 +31,7 @@ import com.google.common.collect.Iterables; import com.google.common.collect.Lists; import java.io.Serializable; import java.util.ArrayList; +import java.util.Collections; import java.util.List; import org.apache.beam.runners.direct.DirectRunner.CommittedBundle; import org.apache.beam.runners.direct.DirectRunner.UncommittedBundle; @@ -94,6 +95,11 @@ public class StatefulParDoEvaluatorFactoryTest implements Serializable { public void setup() { MockitoAnnotations.initMocks(this); when((StateInternals<Object>) mockStepContext.stateInternals()).thenReturn(stateInternals); + when(mockEvaluationContext.createSideInputReader(anyList())) + .thenReturn( + SideInputContainer.create( + mockEvaluationContext, Collections.<PCollectionView<?>>emptyList()) + .createReaderForViews(Collections.<PCollectionView<?>>emptyList())); } @Test
