From 20893c0cbe2e379dd750be83e3204ca9162a3cd0 Mon Sep 17 00:00:00 2001
From: Jackie Han <jkhanjob@gmail.com>
Date: Fri, 22 Dec 2023 14:25:57 -0800
Subject: [PATCH] change boolean validation parameter to a string parameter

Signed-off-by: Jackie Han <jkhanjob@gmail.com>
---
 .../rest/RestCreateWorkflowAction.java             |  2 +-
 .../transport/CreateWorkflowTransportAction.java   |  4 +++-
 .../flowframework/transport/WorkflowRequest.java   | 14 +++++++-------
 .../CreateWorkflowTransportActionTests.java        | 14 +++++++-------
 4 files changed, 18 insertions(+), 16 deletions(-)

diff --git a/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java b/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
index d618e2183..5ad0220fd 100644
--- a/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
+++ b/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
@@ -94,7 +94,7 @@ 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 validation = request.paramAsBoolean(VALIDATION, true);
+            String[] validation = request.paramAsStringArray(VALIDATION, new String[] { "all" });
             boolean provision = request.paramAsBoolean(PROVISION_WORKFLOW, false);
 
             WorkflowRequest workflowRequest = new WorkflowRequest(
diff --git a/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java b/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
index 83bd586bf..eed35f1ac 100644
--- a/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
+++ b/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
@@ -37,6 +37,7 @@
 import org.opensearch.tasks.Task;
 import org.opensearch.transport.TransportService;
 
+import java.util.Arrays;
 import java.util.List;
 
 import static org.opensearch.flowframework.common.CommonValue.PROVISIONING_PROGRESS_FIELD;
@@ -95,7 +96,8 @@ protected void doExecute(Task task, WorkflowRequest request, ActionListener<Work
             user
         );
 
-        if (request.isValidation()) {
+        String[] validateAll = { "all" };
+        if (Arrays.equals(request.getValidation(), validateAll)) {
             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 7b6f60f7a..ab730b2e7 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 validation;
+    private String[] validation;
 
     /**
      * Provision flag
@@ -59,7 +59,7 @@ public class WorkflowRequest extends ActionRequest {
      * @param template the use case template which describes the workflow
      */
     public WorkflowRequest(@Nullable String workflowId, @Nullable Template template) {
-        this(workflowId, template, false, false, null, null);
+        this(workflowId, template, null, false, null, null);
     }
 
     /**
@@ -75,7 +75,7 @@ public WorkflowRequest(
         @Nullable TimeValue requestTimeout,
         @Nullable Integer maxWorkflows
     ) {
-        this(workflowId, template, false, false, requestTimeout, maxWorkflows);
+        this(workflowId, template, null, false, requestTimeout, maxWorkflows);
     }
 
     /**
@@ -90,7 +90,7 @@ public WorkflowRequest(
     public WorkflowRequest(
         @Nullable String workflowId,
         @Nullable Template template,
-        boolean validation,
+        String[] validation,
         boolean provision,
         @Nullable TimeValue requestTimeout,
         @Nullable Integer 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.validation = in.readBoolean();
+        this.validation = in.readStringArray();
         this.provision = in.readBoolean();
         this.requestTimeout = in.readOptionalTimeValue();
         this.maxWorkflows = in.readOptionalInt();
@@ -141,7 +141,7 @@ public Template getTemplate() {
      * Gets the validation flag
      * @return the validation boolean
      */
-    public boolean isValidation() {
+    public String[] getValidation() {
         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(validation);
+        out.writeStringArray(validation);
         out.writeBoolean(provision);
         out.writeOptionalTimeValue(requestTimeout);
         out.writeOptionalInt(maxWorkflows);
diff --git a/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java b/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
index 9e3f1c409..7cf49a8c5 100644
--- a/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
+++ b/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
@@ -143,7 +143,7 @@ public void testValidation_withoutProvision_Success() {
 
         @SuppressWarnings("unchecked")
         ActionListener<WorkflowResponse> listener = mock(ActionListener.class);
-        WorkflowRequest createNewWorkflow = new WorkflowRequest(null, validTemplate, true, false, null, null);
+        WorkflowRequest createNewWorkflow = new WorkflowRequest(null, validTemplate, new String[] { "all" }, false, null, null);
         createWorkflowTransportAction.doExecute(mock(Task.class), createNewWorkflow, listener);
     }
 
@@ -203,7 +203,7 @@ public void testValidation_Failed() throws Exception {
         ActionListener<WorkflowResponse> listener = mock(ActionListener.class);
         // Stub validation failure
         doThrow(Exception.class).when(workflowProcessSorter).validate(any());
-        WorkflowRequest createNewWorkflow = new WorkflowRequest(null, cyclicalTemplate, true, false, null, null);
+        WorkflowRequest createNewWorkflow = new WorkflowRequest(null, cyclicalTemplate, new String[] { "all" }, false, null, null);
 
         createWorkflowTransportAction.doExecute(mock(Task.class), createNewWorkflow, listener);
         verify(listener, times(1)).onFailure(any());
@@ -215,7 +215,7 @@ public void testMaxWorkflow() {
         WorkflowRequest workflowRequest = new WorkflowRequest(
             null,
             template,
-            false,
+            new String[] { "off" },
             false,
             WORKFLOW_REQUEST_TIMEOUT.get(settings),
             MAX_WORKFLOWS.get(settings)
@@ -252,7 +252,7 @@ public void testFailedToCreateNewWorkflow() {
         WorkflowRequest workflowRequest = new WorkflowRequest(
             null,
             template,
-            false,
+            new String[] { "off" },
             false,
             WORKFLOW_REQUEST_TIMEOUT.get(settings),
             MAX_WORKFLOWS.get(settings)
@@ -290,7 +290,7 @@ public void testCreateNewWorkflow() {
         WorkflowRequest workflowRequest = new WorkflowRequest(
             null,
             template,
-            false,
+            new String[] { "off" },
             false,
             WORKFLOW_REQUEST_TIMEOUT.get(settings),
             MAX_WORKFLOWS.get(settings)
@@ -385,7 +385,7 @@ public void testCreateWorkflow_withValidation_withProvision_Success() throws Exc
         WorkflowRequest workflowRequest = new WorkflowRequest(
             null,
             validTemplate,
-            true,
+            new String[] { "all" },
             true,
             WORKFLOW_REQUEST_TIMEOUT.get(settings),
             MAX_WORKFLOWS.get(settings)
@@ -445,7 +445,7 @@ public void testCreateWorkflow_withValidation_withProvision_FailedProvisioning()
         WorkflowRequest workflowRequest = new WorkflowRequest(
             null,
             validTemplate,
-            true,
+            new String[] { "all" },
             true,
             WORKFLOW_REQUEST_TIMEOUT.get(settings),
             MAX_WORKFLOWS.get(settings)