test(search): introduce retry for search test (#10206)

metadata-io/src/test/java/com/linkedin/metadata/search/fixtures/GoldenTestBase.java

@@ -22,6 +22,7 @@
 import com.linkedin.metadata.search.SearchResult;
 import com.linkedin.metadata.search.SearchService;
 import io.datahubproject.metadata.context.OperationContext;
+import io.datahubproject.test.SearchRetry;
 import io.datahubproject.test.metadata.context.TestOperationContexts;
 import io.datahubproject.test.search.SearchTestUtils;
 import java.util.Collections;
@@ -95,7 +96,7 @@ public void testNameMatchPetProfile() {
     assertTrue(secondResultUrn.toString().contains("pet_profiles"));
   }
 
-  @Test
+  @Test(retryAnalyzer = SearchRetry.class)
   public void testGlossaryTerms() {
     /*
      Searching for "ReturnRate" should return all tables that have the glossary term applied before

metadata-io/src/test/java/io/datahubproject/test/SearchRetry.java

@@ -0,0 +1,26 @@
+package io.datahubproject.test;
+
+import org.testng.IRetryAnalyzer;
+import org.testng.ITestResult;
+
+/** Used to retry search fixture tests 1 time after 5 seconds */
+public class SearchRetry implements IRetryAnalyzer {
+
+  private int retryCount = 0;
+  private static final int maxRetryCount = 1;
+  private static final int delayMs = 5000;
+
+  @Override
+  public boolean retry(ITestResult result) {
+    if (retryCount < maxRetryCount) {
+      retryCount++;
+      try {
+        Thread.sleep(delayMs);
+      } catch (InterruptedException e) {
+        throw new RuntimeException(e);
+      }
+      return true;
+    }
+    return false;
+  }
+}