From fb9ef51e7236676c99e491cb27336cc9a67775d9 Mon Sep 17 00:00:00 2001
From: Takashi Matsuo <tmatsuo@google.com>
Date: Mon, 20 Apr 2020 18:10:15 -0700
Subject: [PATCH] [vision] fix: longer timeout (#3447)

fixes #2962
---
 vision/cloud-client/detect/detect.py             | 2 +-
 vision/cloud-client/detect/detect_test.py        | 2 ++
 vision/cloud-client/detect/requirements-test.txt | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/vision/cloud-client/detect/detect.py b/vision/cloud-client/detect/detect.py
index a7d3eca8759b..a3976ffb7b32 100644
--- a/vision/cloud-client/detect/detect.py
+++ b/vision/cloud-client/detect/detect.py
@@ -836,7 +836,7 @@ def async_detect_document(gcs_source_uri, gcs_destination_uri):
         requests=[async_request])
 
     print('Waiting for the operation to finish.')
-    operation.result(timeout=180)
+    operation.result(timeout=300)
 
     # Once the request has completed and the output has been
     # written to GCS, we can list all the output files.
diff --git a/vision/cloud-client/detect/detect_test.py b/vision/cloud-client/detect/detect_test.py
index 5a8774c12fbd..e577df758b7a 100644
--- a/vision/cloud-client/detect/detect_test.py
+++ b/vision/cloud-client/detect/detect_test.py
@@ -16,6 +16,7 @@
 import uuid
 
 from google.cloud import storage
+import pytest
 
 import detect
 
@@ -208,6 +209,7 @@ def test_detect_crop_hints_uri(capsys):
     assert 'bounds: ' in out
 
 
+@pytest.mark.flaky
 def test_async_detect_document(capsys):
     storage_client = storage.Client()
     bucket = storage_client.get_bucket(BUCKET)
diff --git a/vision/cloud-client/detect/requirements-test.txt b/vision/cloud-client/detect/requirements-test.txt
index 781d4326c947..1b569cb4f2c7 100644
--- a/vision/cloud-client/detect/requirements-test.txt
+++ b/vision/cloud-client/detect/requirements-test.txt
@@ -1 +1,2 @@
 pytest==5.3.2
+flaky==3.6.1