Fix unit tests broken with changes to fix integ tests

Signed-off-by: Daniel Widdis <widdis@gmail.com>
opensearch-project · Dec 19, 2023 · 5c69354 · 5c69354
1 parent 673a5f5
commit 5c69354
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/test/java/org/opensearch/flowframework/workflow/RegisterLocalModelStepTests.java b/src/test/java/org/opensearch/flowframework/workflow/RegisterLocalModelStepTests.java
@@ -252,7 +252,6 @@ public void testMissingInputs() {
             "model_type",
             "embedding_dimension",
             "framework_type",
-            "model_group_id",
             "version",
             "url",
             "model_content_hash_value" }) {

diff --git a/src/test/java/org/opensearch/flowframework/workflow/WorkflowProcessSorterTests.java b/src/test/java/org/opensearch/flowframework/workflow/WorkflowProcessSorterTests.java
@@ -340,7 +340,7 @@ public void testFailedGraphValidation() {
             FlowFrameworkException.class,
             () -> workflowProcessSorter.validateGraph(sortedProcessNodes, validator)
         );
-        assertEquals("Invalid graph, missing the following required inputs : [connector_id]", ex.getMessage());
+        assertEquals("Invalid workflow, node [workflow_step_1] missing the following required inputs : [connector_id]", ex.getMessage());
         assertEquals(RestStatus.BAD_REQUEST, ex.getRestStatus());
     }