From eeb88a02dee6e42d5fa709761649037568a173a5 Mon Sep 17 00:00:00 2001
From: Jackie Han <jkhanjob@gmail.com>
Date: Wed, 20 Dec 2023 14:07:08 -0800
Subject: [PATCH] Change dryrun param in create workflow to validation

Signed-off-by: Jackie Han <jkhanjob@gmail.com>
---
 .../flowframework/common/CommonValue.java     |  4 ++--
 .../rest/RestCreateWorkflowAction.java        |  6 ++---
 .../CreateWorkflowTransportAction.java        |  2 +-
 .../transport/WorkflowRequest.java            | 24 +++++++++----------
 .../FlowFrameworkRestTestCase.java            |  6 ++---
 .../rest/FlowFrameworkRestApiIT.java          |  2 +-
 .../CreateWorkflowTransportActionTests.java   |  8 +++----
 7 files changed, 26 insertions(+), 26 deletions(-)

diff --git a/src/main/java/org/opensearch/flowframework/common/CommonValue.java b/src/main/java/org/opensearch/flowframework/common/CommonValue.java
index bb9eaf108..ffe45c87e 100644
--- a/src/main/java/org/opensearch/flowframework/common/CommonValue.java
+++ b/src/main/java/org/opensearch/flowframework/common/CommonValue.java
@@ -63,8 +63,8 @@ private CommonValue() {}
     public static final String WORKFLOW_URI = FLOW_FRAMEWORK_BASE_URI + "/workflow";
     /** Field name for workflow Id, the document Id of the indexed use case template */
     public static final String WORKFLOW_ID = "workflow_id";
-    /** Field name for dry run, the flag to indicate if validation is necessary */
-    public static final String DRY_RUN = "dryrun";
+    /** Field name for template validation, the flag to indicate if validation is necessary */
+    public static final String VALIDATION = "validation";
     /** The field name for provision workflow within a use case template*/
     public static final String PROVISION_WORKFLOW = "provision";
 
diff --git a/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java b/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
index e254b66f7..a70b0104b 100644
--- a/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
+++ b/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
@@ -32,8 +32,8 @@
 import java.util.Locale;
 
 import static org.opensearch.core.xcontent.XContentParserUtils.ensureExpectedToken;
-import static org.opensearch.flowframework.common.CommonValue.DRY_RUN;
 import static org.opensearch.flowframework.common.CommonValue.PROVISION_WORKFLOW;
+import static org.opensearch.flowframework.common.CommonValue.VALIDATION;
 import static org.opensearch.flowframework.common.CommonValue.WORKFLOW_ID;
 import static org.opensearch.flowframework.common.CommonValue.WORKFLOW_URI;
 import static org.opensearch.flowframework.common.FlowFrameworkSettings.FLOW_FRAMEWORK_ENABLED;
@@ -94,10 +94,10 @@ protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient cli
             XContentParser parser = request.contentParser();
             ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.nextToken(), parser);
             Template template = Template.parse(parser);
-            boolean dryRun = request.paramAsBoolean(DRY_RUN, false);
+            boolean validation = request.paramAsBoolean(VALIDATION, true);
             boolean provision = request.paramAsBoolean(PROVISION_WORKFLOW, false);
 
-            WorkflowRequest workflowRequest = new WorkflowRequest(workflowId, template, dryRun, provision, requestTimeout, maxWorkflows);
+            WorkflowRequest workflowRequest = new WorkflowRequest(workflowId, template, validation, provision, requestTimeout, maxWorkflows);
 
             return channel -> client.execute(CreateWorkflowAction.INSTANCE, workflowRequest, ActionListener.wrap(response -> {
                 XContentBuilder builder = response.toXContent(channel.newBuilder(), ToXContent.EMPTY_PARAMS);
diff --git a/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java b/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
index 92f89c082..83bd586bf 100644
--- a/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
+++ b/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
@@ -95,7 +95,7 @@ protected void doExecute(Task task, WorkflowRequest request, ActionListener<Work
             user
         );
 
-        if (request.isDryRun()) {
+        if (request.isValidation()) {
             try {
                 validateWorkflows(templateWithUser);
             } catch (Exception e) {
diff --git a/src/main/java/org/opensearch/flowframework/transport/WorkflowRequest.java b/src/main/java/org/opensearch/flowframework/transport/WorkflowRequest.java
index a030dccfa..063cdbac2 100644
--- a/src/main/java/org/opensearch/flowframework/transport/WorkflowRequest.java
+++ b/src/main/java/org/opensearch/flowframework/transport/WorkflowRequest.java
@@ -36,7 +36,7 @@ public class WorkflowRequest extends ActionRequest {
     /**
      * Validation flag
      */
-    private boolean dryRun;
+    private boolean validation;
 
     /**
      * Provision flag
@@ -54,7 +54,7 @@ public class WorkflowRequest extends ActionRequest {
     private Integer maxWorkflows;
 
     /**
-     * Instantiates a new WorkflowRequest, defaults dry run to false and set requestTimeout and maxWorkflows to null
+     * Instantiates a new WorkflowRequest, defaults validation to false and set requestTimeout and maxWorkflows to null
      * @param workflowId the documentId of the workflow
      * @param template the use case template which describes the workflow
      */
@@ -63,7 +63,7 @@ public WorkflowRequest(@Nullable String workflowId, @Nullable Template template)
     }
 
     /**
-     * Instantiates a new WorkflowRequest and defaults dry run to false
+     * Instantiates a new WorkflowRequest and defaults validation to false
      * @param workflowId the documentId of the workflow
      * @param template the use case template which describes the workflow
      * @param requestTimeout timeout of the request
@@ -82,7 +82,7 @@ public WorkflowRequest(
      * Instantiates a new WorkflowRequest
      * @param workflowId the documentId of the workflow
      * @param template the use case template which describes the workflow
-     * @param dryRun flag to indicate if validation is necessary
+     * @param validation flag to indicate if validation is necessary
      * @param provision flag to indicate if provision is necessary
      * @param requestTimeout timeout of the request
      * @param maxWorkflows max number of workflows
@@ -90,14 +90,14 @@ public WorkflowRequest(
     public WorkflowRequest(
         @Nullable String workflowId,
         @Nullable Template template,
-        boolean dryRun,
+        boolean validation,
         boolean provision,
         @Nullable TimeValue requestTimeout,
         @Nullable Integer maxWorkflows
     ) {
         this.workflowId = workflowId;
         this.template = template;
-        this.dryRun = dryRun;
+        this.validation = validation;
         this.provision = provision;
         this.requestTimeout = requestTimeout;
         this.maxWorkflows = maxWorkflows;
@@ -113,7 +113,7 @@ public WorkflowRequest(StreamInput in) throws IOException {
         this.workflowId = in.readOptionalString();
         String templateJson = in.readOptionalString();
         this.template = templateJson == null ? null : Template.parse(templateJson);
-        this.dryRun = in.readBoolean();
+        this.validation = in.readBoolean();
         this.provision = in.readBoolean();
         this.requestTimeout = in.readOptionalTimeValue();
         this.maxWorkflows = in.readOptionalInt();
@@ -138,11 +138,11 @@ public Template getTemplate() {
     }
 
     /**
-     * Gets the dry run validation flag
-     * @return the dry run boolean
+     * Gets the validation flag
+     * @return the validation boolean
      */
-    public boolean isDryRun() {
-        return this.dryRun;
+    public boolean isValidation() {
+        return this.validation;
     }
 
     /**
@@ -174,7 +174,7 @@ public void writeTo(StreamOutput out) throws IOException {
         super.writeTo(out);
         out.writeOptionalString(workflowId);
         out.writeOptionalString(template == null ? null : template.toJson());
-        out.writeBoolean(dryRun);
+        out.writeBoolean(validation);
         out.writeBoolean(provision);
         out.writeOptionalTimeValue(requestTimeout);
         out.writeOptionalInt(maxWorkflows);
diff --git a/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java b/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java
index ac537047f..dcb860a17 100644
--- a/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java
+++ b/src/test/java/org/opensearch/flowframework/FlowFrameworkRestTestCase.java
@@ -321,13 +321,13 @@ protected Response createWorkflow(Template template) throws Exception {
     }
 
     /**
-     * Helper method to invoke the Create Workflow Rest Action with dry run validation
+     * Helper method to invoke the Create Workflow Rest Action with validation
      * @param template the template to create
      * @throws Exception if the request fails
      * @return a rest response
      */
-    protected Response createWorkflowDryRun(Template template) throws Exception {
-        return TestHelpers.makeRequest(client(), "POST", WORKFLOW_URI + "?dryrun=true", ImmutableMap.of(), template.toJson(), null);
+    protected Response createWorkflowValidation(Template template) throws Exception {
+        return TestHelpers.makeRequest(client(), "POST", WORKFLOW_URI, ImmutableMap.of(), template.toJson(), null);
     }
 
     /**
diff --git a/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java b/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
index e8a99946e..9e01d2ca1 100644
--- a/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
+++ b/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
@@ -151,7 +151,7 @@ public void testCreateAndProvisionRemoteModelWorkflow() throws Exception {
             .build();
 
         // Hit dry run
-        ResponseException exception = expectThrows(ResponseException.class, () -> createWorkflowDryRun(cyclicalTemplate));
+        ResponseException exception = expectThrows(ResponseException.class, () -> createWorkflowValidation(cyclicalTemplate));
         assertTrue(exception.getMessage().contains("Cycle detected: [workflow_step_2->workflow_step_1, workflow_step_1->workflow_step_2]"));
 
         // Hit Create Workflow API with original template
diff --git a/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java b/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
index 0addb7f1f..9e3f1c409 100644
--- a/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
+++ b/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
@@ -138,7 +138,7 @@ public void setUp() throws Exception {
         );
     }
 
-    public void testDryRunValidation_withoutProvision_Success() {
+    public void testValidation_withoutProvision_Success() {
         Template validTemplate = generateValidTemplate();
 
         @SuppressWarnings("unchecked")
@@ -147,7 +147,7 @@ public void testDryRunValidation_withoutProvision_Success() {
         createWorkflowTransportAction.doExecute(mock(Task.class), createNewWorkflow, listener);
     }
 
-    public void testDryRunValidation_Failed() throws Exception {
+    public void testValidation_Failed() throws Exception {
 
         WorkflowNode createConnector = new WorkflowNode(
             "workflow_step_1",
@@ -374,7 +374,7 @@ public void testUpdateWorkflow() {
         assertEquals("1", responseCaptor.getValue().getWorkflowId());
     }
 
-    public void testCreateWorkflow_withDryRun_withProvision_Success() throws Exception {
+    public void testCreateWorkflow_withValidation_withProvision_Success() throws Exception {
 
         Template validTemplate = generateValidTemplate();
 
@@ -435,7 +435,7 @@ public void testCreateWorkflow_withDryRun_withProvision_Success() throws Excepti
         assertEquals("1", workflowResponseCaptor.getValue().getWorkflowId());
     }
 
-    public void testCreateWorkflow_withDryRun_withProvision_FailedProvisioning() throws Exception {
+    public void testCreateWorkflow_withValidation_withProvision_FailedProvisioning() throws Exception {
 
         Template validTemplate = generateValidTemplate();