[Backport 2.x] Fixing ingest pipeline integ test (#616)

opensearch-project · Mar 26, 2024 · 7faa889 · 7faa889
1 parent e592b17
commit 7faa889
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java b/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java
@@ -667,8 +667,8 @@ protected Response deleteUser(String user) throws IOException {
         return TestHelpers.makeRequest(client(), "DELETE", "/_opendistro/_security/api/internalusers/" + user, null, "", null);
     }
 
-    protected GetPipelineResponse getPipelines() throws IOException {
-        Response getPipelinesResponse = TestHelpers.makeRequest(client(), "GET", "_ingest/pipeline", null, "", null);
+    protected GetPipelineResponse getPipelines(String pipelineId) throws IOException {
+        Response getPipelinesResponse = TestHelpers.makeRequest(client(), "GET", "_ingest/pipeline/" + pipelineId, null, "", null);
 
         // Parse entity content into SearchResponse
         MediaType mediaType = MediaType.fromMediaType(getPipelinesResponse.getEntity().getContentType().getValue());

diff --git a/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java b/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
@@ -395,7 +395,7 @@ public void testCreateAndProvisionIngestAndSearchPipeline() throws Exception {
         assertEquals(5, resourcesCreated.size());
         String modelId = resourcesCreated.get(2).resourceId();
 
-        GetPipelineResponse getPipelinesResponse = getPipelines();
+        GetPipelineResponse getPipelinesResponse = getPipelines("append-1");
 
         assertTrue(getPipelinesResponse.pipelines().get(0).toString().contains(modelId));