chore(vision): add retry to flakey test (googleapis#7842)

Graham Paye · web-flow · commit f9b6bf8fbc9f · 2020-09-28T18:09:51.000-06:00
diff --git a/google-cloud-vision/samples/product_search/acceptance/product_search_import_product_images_test.rb b/google-cloud-vision/samples/product_search/acceptance/product_search_import_product_images_test.rb
@@ -27,16 +27,31 @@
       "fake_product_id_for_testing_2" => "shoes_2.jpg"
     }
 
-    output = `ruby #{snippet_filepath} #{@project_id} #{@location}`
+    product_set = nil
+    error = nil
+    5.times do
+      break if product_set
+      begin
+        output = `ruby #{snippet_filepath} #{@project_id} #{@location}`
 
-    # Verify console output
-    product_images.values.each do |image_uri|
-      _(output).must_include image_uri
+        # Verify console output
+        product_images.values.each do |image_uri|
+          _(output).must_include image_uri
+        end
+
+        # Verify project set existence
+        product_set_path = @client.product_set_path project:     @project_id,
+                                                    location:    @location,
+                                                    product_set: @product_set_id
+        product_set = @client.get_product_set name: product_set_path
+      rescue Google::Cloud::NotFoundError => e
+        error = e
+        sleep rand(10..60)
+      end
     end
 
-    # Verify project set existence
-    product_set_path = @client.product_set_path project: @project_id, location: @location, product_set: @product_set_id
-    product_set = @client.get_product_set name: product_set_path
+    raise error if error && !product_set
+
     assert product_set
 
     # Verify product reference image URIs
