FollowUp: Retry annotation should work on class level (opensearch-project#4091)

* FollowUp: Retry annotation should work on class level

Signed-off-by: Lantao Jin <ltjin@amazon.com>

* Ignore testQ7 in macOS

Signed-off-by: Lantao Jin <ltjin@amazon.com>

---------

Signed-off-by: Lantao Jin <ltjin@amazon.com>

Author: LantaoJin

integ-test/src/test/java/org/opensearch/sql/calcite/tpch/CalcitePPLTpchIT.java

@@ -14,7 +14,9 @@
 import static org.opensearch.sql.util.MatcherUtils.verifySchemaInOrder;
 
 import java.io.IOException;
+import java.util.Locale;
 import org.json.JSONObject;
+import org.junit.Assume;
 import org.junit.Ignore;
 import org.junit.Test;
 import org.opensearch.sql.ppl.PPLIntegTestCase;
@@ -176,6 +178,8 @@ public void testQ6() throws IOException {
 
   @Test
   public void testQ7() throws IOException {
+    String osName = System.getProperty("os.name").toLowerCase(Locale.ROOT);
+    Assume.assumeFalse("testQ7 on macOS CI could socket timeout", osName.contains("mac"));
     String ppl = sanitize(loadFromFile("tpch/queries/q7.ppl"));
     JSONObject actual = executeQuery(ppl);
     verifySchemaInOrder(

integ-test/src/test/java/org/opensearch/sql/util/RetryProcessor.java

@@ -5,6 +5,7 @@
 
 package org.opensearch.sql.util;
 
+import java.util.Optional;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.junit.rules.TestWatcher;
@@ -17,7 +18,9 @@ public class RetryProcessor extends TestWatcher {
 
   @Override
   public Statement apply(Statement base, Description description) {
-    Retry retry = description.getAnnotation(Retry.class);
+    Retry retry =
+        Optional.ofNullable(description.getAnnotation(Retry.class))
+            .orElseGet(() -> description.getTestClass().getAnnotation(Retry.class));
     if (retry == null) {
       return base;
     }