vision: increase timeout for tests (#3383)

Fix: #2955
Fix: #2992

Author: nnegrey

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]

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]

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,

vision/cloud-client/product_search/requirements-test.txt

@@ -1 +1,2 @@
 pytest==5.3.2
+flaky==3.6.1