test: Add multiple attempts to Gen App Builder SearchTest (GoogleCloudPlatform#8299)

holtskinner · web-flow · commit a545e7718db4 · 2023-06-24T11:36:12.000-05:00
- Fixes GoogleCloudPlatform#8003
diff --git a/discoveryengine/src/test/java/discoveryengine/v1beta/SearchTest.java b/discoveryengine/src/test/java/discoveryengine/v1beta/SearchTest.java
@@ -19,13 +19,17 @@
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.assertNotNull;
 
+import com.google.cloud.testing.junit4.MultipleAttemptsRule;
 import java.io.ByteArrayOutputStream;
 import java.io.PrintStream;
 import org.junit.After;
 import org.junit.Before;
+import org.junit.Rule;
 import org.junit.Test;
 
 public class SearchTest {
+  @Rule public final MultipleAttemptsRule multipleAttemptsRule = new MultipleAttemptsRule(3);
+
   private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
   private static final String LOCATION = "global";
   private static final String COLLECTION_ID = "default_collection";
