commit 3b1a0773477124170f2196a4739297e68528cc80
author Yifan Zou <yifanzou@google.com> Wed Oct 03 21:40:09 2018 -0700
committer Yifan Zou <yifanzou@google.com> Wed Oct 03 23:07:27 2018 -0700
tree 3e138e00fb12dde7c0042f599578d35c17357b64
parent e4b7f096bb77b5bdadb0b157ab1c0392ba66beff

[BEAM-5643] fix broken python it test

sdks/python/apache_beam/examples/complete/game/user_score_it_test.py

diff --git a/sdks/python/apache_beam/examples/complete/game/user_score_it_test.py b/sdks/python/apache_beam/examples/complete/game/user_score_it_test.py
index 1c754a9..a29c99e 100644
--- a/sdks/python/apache_beam/examples/complete/game/user_score_it_test.py
+++ b/sdks/python/apache_beam/examples/complete/game/user_score_it_test.py
@@ -64,8 +64,11 @@
   def test_user_score_it(self):
 
     state_verifier = PipelineStateMatcher(PipelineState.DONE)
-    file_verifier = FileChecksumMatcher(self.output + '*-of-*',
-                                        self.DEFAULT_EXPECTED_CHECKSUM)
+    arg_sleep_secs = self.test_pipeline.get_option('sleep_secs')
+    sleep_secs = int(arg_sleep_secs) if arg_sleep_secs is not None else None
+    file_verifier = FileChecksumMatcher(self.output + '/*-of-*',
+                                        self.DEFAULT_EXPECTED_CHECKSUM,
+                                        sleep_secs)
 
     extra_opts = {'input': self.DEFAULT_INPUT_FILE,
                   'output': self.output + '/user-score',