Update expectation in ProcessBundleHandlerTest

Project: http://git-wip-us.apache.org/repos/asf/beam/repo
Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/040dbd4f
Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/040dbd4f
Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/040dbd4f

Branch: refs/heads/master
Commit: 040dbd4fd21a31dd3eb0f34fb9610ce857111db8
Parents: 3042d76
Author: Thomas Groh <tg...@google.com>
Authored: Thu May 4 17:01:24 2017 -0700
Committer: Thomas Groh <tg...@google.com>
Committed: Thu May 4 17:14:48 2017 -0700

----------------------------------------------------------------------
 .../beam/fn/harness/control/ProcessBundleHandlerTest.java      | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/beam/blob/040dbd4f/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/control/ProcessBundleHandlerTest.java
----------------------------------------------------------------------
diff --git 
a/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/control/ProcessBundleHandlerTest.java
 
b/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/control/ProcessBundleHandlerTest.java
index 3e06369..748ffea 100644
--- 
a/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/control/ProcessBundleHandlerTest.java
+++ 
b/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/control/ProcessBundleHandlerTest.java
@@ -18,6 +18,7 @@
 
 package org.apache.beam.fn.harness.control;
 
+import static 
org.apache.beam.sdk.util.WindowedValue.timestampedValueInGlobalWindow;
 import static org.apache.beam.sdk.util.WindowedValue.valueInGlobalWindow;
 import static org.hamcrest.Matchers.contains;
 import static org.hamcrest.Matchers.empty;
@@ -433,7 +434,10 @@ public class ProcessBundleHandlerTest {
     additionalOutputValues.clear();
 
     Iterables.getOnlyElement(finishFunctions).run();
-    assertThat(mainOutputValues, 
contains(valueInGlobalWindow("FinishBundle")));
+    assertThat(
+        mainOutputValues,
+        contains(
+            timestampedValueInGlobalWindow("FinishBundle", 
GlobalWindow.INSTANCE.maxTimestamp())));
     mainOutputValues.clear();
   }
 

Reply via email to