fix: wait up to 20 seconds for blob to appear (GoogleCloudPlatform#3061)

crwilcox · web-flow · commit dd8984f61783 · 2020-03-09T22:12:03.000Z
This test fails sometimes with a 3 second wait. Extend this wait, but poll to avoid taking all the time.
diff --git a/video/cloud-client/analyze/beta_snippets_test.py b/video/cloud-client/analyze/beta_snippets_test.py
@@ -101,11 +101,15 @@ def test_annotation_to_storage_streaming(capsys, video_path, bucket):
     assert 'Storage' in out
 
     # It takes a few seconds before the results show up on GCS.
-    time.sleep(3)
+    for _ in range(10):
+        time.sleep(3)
+
+        blobs_iterator = bucket.list_blobs()
+        blobs = [blob for blob in blobs_iterator]
+        if len(blobs):
+            break
 
     # Confirm that one output blob had been written to GCS.
-    blobs_iterator = bucket.list_blobs()
-    blobs = [blob for blob in blobs_iterator]
     assert len(blobs) == 1
 
 
