cleanup

Signed-off-by: Jackie Han <[email protected]>
jackiehanyang · Dec 12, 2023 · dbf5fc8 · dbf5fc8
1 parent 771da9f
commit dbf5fc8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 12 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java b/src/main/java/org/opensearch/flowframework/transport/CreateWorkflowTransportAction.java
@@ -51,8 +51,6 @@ public class CreateWorkflowTransportAction extends HandledTransportAction<Workfl
     private final Client client;
     private final Settings settings;
 
-    private final TransportService transportService;
-
     /**
      * Intantiates a new CreateWorkflowTransportAction
      * @param transportService the TransportService
@@ -72,7 +70,6 @@ public CreateWorkflowTransportAction(
         Client client
     ) {
         super(CreateWorkflowAction.NAME, transportService, actionFilters, WorkflowRequest::new);
-        this.transportService = transportService;
         this.workflowProcessSorter = workflowProcessSorter;
         this.flowFrameworkIndicesHandler = flowFrameworkIndicesHandler;
         this.settings = settings;

diff --git a/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java b/src/test/java/org/opensearch/flowframework/transport/CreateWorkflowTransportActionTests.java
@@ -195,7 +195,7 @@ public void testCreateNewWorkflow() {
         WorkflowRequest workflowRequest = new WorkflowRequest(
             null,
             template,
-            true,
+            false,
             WORKFLOW_REQUEST_TIMEOUT.get(settings),
             MAX_WORKFLOWS.get(settings)
         );
@@ -228,14 +228,6 @@ public void testCreateNewWorkflow() {
             return null;
         }).when(flowFrameworkIndicesHandler).putInitialStateToWorkflowState(any(), any(), any());
 
-        doAnswer(invocation -> {
-            ActionListener<WorkflowResponse> responseListener = invocation.getArgument(2);
-            WorkflowResponse response = mock(WorkflowResponse.class);
-            when(response.getWorkflowId()).thenReturn("1");
-            responseListener.onResponse(response);
-            return null;
-        }).when(client).execute(eq(ProvisionWorkflowAction.INSTANCE), any(WorkflowRequest.class), any(ActionListener.class));
-
         ArgumentCaptor<WorkflowResponse> workflowResponseCaptor = ArgumentCaptor.forClass(WorkflowResponse.class);
 
         createWorkflowTransportAction.doExecute(mock(Task.class), workflowRequest, listener);