From 912a856d366d352b2efd329e9279214ebf2880af Mon Sep 17 00:00:00 2001 From: Daniel Widdis Date: Mon, 30 Oct 2023 20:25:23 -0700 Subject: [PATCH] Fix parameter key in tests Signed-off-by: Daniel Widdis --- .../workflow/CreateConnectorStepTests.java | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/src/test/java/org/opensearch/flowframework/workflow/CreateConnectorStepTests.java b/src/test/java/org/opensearch/flowframework/workflow/CreateConnectorStepTests.java index cee480bf6..63855f7bd 100644 --- a/src/test/java/org/opensearch/flowframework/workflow/CreateConnectorStepTests.java +++ b/src/test/java/org/opensearch/flowframework/workflow/CreateConnectorStepTests.java @@ -10,6 +10,7 @@ import org.opensearch.core.action.ActionListener; import org.opensearch.core.rest.RestStatus; +import org.opensearch.flowframework.common.CommonValue; import org.opensearch.flowframework.exception.FlowFrameworkException; import org.opensearch.ml.client.MachineLearningNodeClient; import org.opensearch.ml.common.connector.ConnectorAction; @@ -58,13 +59,13 @@ public void setUp() throws Exception { inputData = new WorkflowData( Map.ofEntries( - Map.entry("name", "test"), - Map.entry("description", "description"), - Map.entry("version", "1"), - Map.entry("protocol", "test"), - Map.entry("params", params), - Map.entry("credentials", credentials), - Map.entry("actions", actions) + Map.entry(CommonValue.NAME_FIELD, "test"), + Map.entry(CommonValue.DESCRIPTION_FIELD, "description"), + Map.entry(CommonValue.VERSION_FIELD, "1"), + Map.entry(CommonValue.PROTOCOL_FIELD, "test"), + Map.entry(CommonValue.PARAMETERS_FIELD, params), + Map.entry(CommonValue.CREDENTIALS_FIELD, credentials), + Map.entry(CommonValue.ACTIONS_FIELD, actions) ) ); }