Fix test case

Signed-off-by: Gagan Juneja <[email protected]>
Gaganjuneja · Sep 8, 2023 · 518b39f · 518b39f
1 parent c1ec431
commit 518b39f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/server/src/internalClusterTest/java/org/opensearch/search/aggregations/bucket/IpRangeIT.java b/server/src/internalClusterTest/java/org/opensearch/search/aggregations/bucket/IpRangeIT.java
@@ -56,6 +56,11 @@
 @OpenSearchIntegTestCase.SuiteScopeTestCase
 public class IpRangeIT extends OpenSearchIntegTestCase {
 
+    @Override
+    protected boolean addMockTelemetryPlugin() {
+        return true;
+    }
+
     public static class DummyScriptPlugin extends MockScriptPlugin {
         @Override
         public Map<String, Function<Map<String, Object>, Object>> pluginScripts() {

diff --git a/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java b/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java
@@ -2328,6 +2328,7 @@ private static void initializeSuiteScope() throws Exception {
                 INSTANCE.beforeInternal();
                 INSTANCE.setupSuiteScopeCluster();
                 success = true;
+                StrictCheckSpanProcessor.restartTheProcessor();
             } finally {
                 if (!success) {
                     afterClass();

diff --git a/...lemetry/src/main/java/org/opensearch/test/telemetry/tracing/StrictCheckSpanProcessor.java b/...lemetry/src/main/java/org/opensearch/test/telemetry/tracing/StrictCheckSpanProcessor.java
@@ -97,9 +97,7 @@ public static synchronized void shutdown() {
      * restartTheProcessor
      */
     public static synchronized void restartTheProcessor() {
-        if (shutdown.get() == true) {
-            shutdown.set(false);
-        }
+        spanMap.clear();
     }
 
     /**