Skip to content

Commit ec5e700

Browse files
committed
Updating test to process work
1 parent 850e9de commit ec5e700

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4181,6 +4181,11 @@ public void testSwitchStreamingWorkerHarness() throws Exception {
41814181
assertTrue(
41824182
"Worker should switch back to SingleSourceWorkerHarness",
41834183
worker.getStreamingWorkerHarness() instanceof SingleSourceWorkerHarness);
4184+
// Process some work with CloudPath again.
4185+
server.whenGetWorkCalled().thenReturn(makeInput(2, 2000));
4186+
result = server.waitForAndGetCommits(1);
4187+
assertEquals(2, result.size());
4188+
assertTrue(result.containsKey(2L));
41844189

41854190
worker.stop();
41864191
}

0 commit comments

Comments
 (0)