[Backport 2.x] Fixed testPluginInstalled failure (#92) (#93)

Fixed testPluginInstalled failure (#92) (cherry picked from commit 1b856de) Signed-off-by: Owais Kazi <[email protected]>
opensearch-project · Oct 14, 2023 · 8c37b2d · 8c37b2d
1 parent 922d36c
commit 8c37b2d
Showing 1 changed file with 5 additions and 16 deletions.
diff --git a/src/test/java/org/opensearch/flowframework/FlowFrameworkPluginIT.java b/src/test/java/org/opensearch/flowframework/FlowFrameworkPluginIT.java
@@ -8,33 +8,22 @@
  */
 package org.opensearch.flowframework;
 
-import com.carrotsearch.randomizedtesting.annotations.ThreadLeakScope;
-
 import org.apache.http.util.EntityUtils;
 import org.opensearch.client.Request;
 import org.opensearch.client.Response;
-import org.opensearch.plugins.Plugin;
-import org.opensearch.test.OpenSearchIntegTestCase;
+import org.opensearch.test.rest.OpenSearchRestTestCase;
 
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
-import java.util.Collection;
-import java.util.Collections;
-
-@ThreadLeakScope(ThreadLeakScope.Scope.NONE)
-@OpenSearchIntegTestCase.ClusterScope(scope = OpenSearchIntegTestCase.Scope.SUITE)
-public class FlowFrameworkPluginIT extends OpenSearchIntegTestCase {
 
-    @Override
-    protected Collection<Class<? extends Plugin>> nodePlugins() {
-        return Collections.singletonList(FlowFrameworkPlugin.class);
-    }
+public class FlowFrameworkPluginIT extends OpenSearchRestTestCase {
 
     public void testPluginInstalled() throws IOException {
-        Response response = createRestClient().performRequest(new Request("GET", "/_cat/plugins"));
+        Response response = client().performRequest(new Request("GET", "/_cat/plugins"));
         String body = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
 
         logger.info("response body: {}", body);
-        assertTrue(body.contains("flowframework"));
+        assertNotNull(body);
+        assertTrue(body.contains("opensearch-flow-framework"));
     }
 }