vision: increase timeout for tests (#3383)

Fix: #2955 Fix: #2992
GoogleCloudPlatform · Apr 14, 2020 · 385d741 · 385d741
1 parent 2194a07
commit 385d741
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/vision/cloud-client/product_search/product_in_product_set_management.py b/vision/cloud-client/product_search/product_in_product_set_management.py
@@ -147,7 +147,7 @@ def purge_products_in_product_set(
         # If force is not set to True, the service raises an exception.
         force=force)
 
-    operation.result(timeout=120)
+    operation.result(timeout=300)
 
     print('Deleted products in product set.')
 # [END vision_product_search_purge_products_in_product_set]

diff --git a/vision/cloud-client/product_search/product_management.py b/vision/cloud-client/product_search/product_management.py
@@ -205,7 +205,7 @@ def purge_orphan_products(project_id, location, force):
         # If force is not set to True, the service raises an exception.
         force=force)
 
-    operation.result(timeout=120)
+    operation.result(timeout=300)
 
     print('Orphan products deleted.')
 # [END vision_product_search_purge_orphan_products]

diff --git a/vision/cloud-client/product_search/product_search_test.py b/vision/cloud-client/product_search/product_search_test.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import os
+from flaky import flaky
 
 from product_search import get_similar_products_file, get_similar_products_uri
 
@@ -30,6 +31,7 @@
 FILTER = 'style=womens'
 
 
+@flaky(max_runs=5, min_passes=1)
 def test_get_similar_products_file(capsys):
     get_similar_products_file(
         PROJECT_ID, LOCATION, PRODUCT_SET_ID, PRODUCT_CATEGORY, FILE_PATH_1,

diff --git a/vision/cloud-client/product_search/requirements-test.txt b/vision/cloud-client/product_search/requirements-test.txt
@@ -1 +1,2 @@
 pytest==5.3.2
+flaky==3.6.1