diff --git a/src/test/java/org/opensearch/flowframework/FlowFrameworkPluginIT.java b/src/test/java/org/opensearch/flowframework/FlowFrameworkPluginIT.java index e85a79b17..fe0af8ba0 100644 --- 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> 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")); } }