[Fix #2165] Adding unit test for executeAfter post call

apache · Dec 20, 2024 · 4bbe5f2 · 4bbe5f2
1 parent 9af8228
commit 4bbe5f2
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 10 deletions.
diff --git a/...ervice-common/src/test/java/org/kie/kogito/index/service/api/KogitoRuntimeClientTest.java b/...ervice-common/src/test/java/org/kie/kogito/index/service/api/KogitoRuntimeClientTest.java
@@ -29,17 +29,22 @@
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
+import org.kie.kogito.index.api.ExecuteArgs;
 import org.kie.kogito.index.api.KogitoRuntimeCommonClient;
 import org.kie.kogito.index.model.Job;
+import org.kie.kogito.index.model.ProcessDefinition;
 import org.kie.kogito.index.model.ProcessInstance;
 import org.kie.kogito.index.model.UserTaskInstance;
 import org.kie.kogito.index.service.DataIndexServiceException;
 import org.kie.kogito.index.test.TestUtils;
+import org.kie.kogito.jackson.utils.ObjectMapperFactory;
 import org.kie.kogito.usertask.model.CommentInfo;
 import org.mockito.ArgumentCaptor;
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
 
+import com.fasterxml.jackson.databind.JsonNode;
+
 import io.quarkus.security.credential.TokenCredential;
 import io.quarkus.security.identity.SecurityIdentity;
 import io.vertx.core.AsyncResult;
@@ -117,6 +122,20 @@ public void setup() {
         client.setIdentity(identityMock);
     }
 
+    @Test
+    public void textExecuteAfter() {
+        when(webClientMock.post(anyString())).thenReturn(httpRequestMock);
+        final String processId = "infra";
+        ProcessDefinition definition = TestUtils.getProcessDefinition(processId);
+        definition.setEndpoint(SERVICE_URL + "/" + processId);
+        JsonNode body = ObjectMapperFactory.get().createObjectNode().put("name", "javierito");
+        client.executeProcessIntance(definition, ExecuteArgs.of(body));
+        verify(webClientMock).post("/" + processId);
+        ArgumentCaptor<Object> jsonCaptor = ArgumentCaptor.forClass(Object.class);
+        verify(httpRequestMock).sendJson(jsonCaptor.capture(), any());
+        assertThat(jsonCaptor.getValue()).isEqualTo(body);
+    }
+
     @Test
     public void testAbortProcessInstance() {
         setupIdentityMock();

diff --git a/...est/java/org/kie/kogito/index/service/graphql/query/AbstractGraphQLRuntimesQueriesIT.java b/...est/java/org/kie/kogito/index/service/graphql/query/AbstractGraphQLRuntimesQueriesIT.java
@@ -39,7 +39,6 @@
 import org.kie.kogito.index.api.ExecuteArgs;
 import org.kie.kogito.index.api.KogitoRuntimeClient;
 import org.kie.kogito.index.event.KogitoJobCloudEvent;
-import org.kie.kogito.index.model.ProcessDefinition;
 import org.kie.kogito.index.model.UserTaskInstance;
 import org.kie.kogito.index.service.AbstractIndexingIT;
 import org.kie.kogito.index.service.graphql.GraphQLSchemaManagerImpl;
@@ -123,21 +122,15 @@ void testProcessExecuteInstance() {
         indexProcessCloudEvent(definitionEvent);
         checkOkResponse("{ \"query\" : \"mutation{ ExecuteAfter ( " + fragment("completedInstanceId", assesmentInstanceId) + "," + fragment("processId", infraProcessId) +
                 "," + fragment("processVersion", TestUtils.PROCESS_VERSION) + "," + "input: {" + fragment(infraVarName, infraVarValue) + "})}\"}");
-        verify(dataIndexApiClient).executeProcessIntance(getProcessDefinition(infraProcessId), ExecuteArgs.of(ObjectMapperFactory.get().createObjectNode().put(assesmentVarName, assesmentVarValue)
-                .put(infraVarName, infraVarValue)));
+        verify(dataIndexApiClient).executeProcessIntance(TestUtils.getProcessDefinition(infraProcessId),
+                ExecuteArgs.of(ObjectMapperFactory.get().createObjectNode().put(assesmentVarName, assesmentVarValue)
+                        .put(infraVarName, infraVarValue)));
     }
 
     private String fragment(String name, String value) {
         return name + ": \\\"" + value + "\\\"";
     }
 
-    private ProcessDefinition getProcessDefinition(String processId) {
-        ProcessDefinition def = new ProcessDefinition();
-        def.setId(processId);
-        def.setVersion(TestUtils.PROCESS_VERSION);
-        return def;
-    }
-
     @Test
     void testProcessInstanceRetry() {
         String processInstanceId = UUID.randomUUID().toString();

diff --git a/data-index/data-index-test-utils/src/main/java/org/kie/kogito/index/test/TestUtils.java b/data-index/data-index-test-utils/src/main/java/org/kie/kogito/index/test/TestUtils.java
@@ -59,6 +59,7 @@
 import org.kie.kogito.index.model.Milestone;
 import org.kie.kogito.index.model.MilestoneStatus;
 import org.kie.kogito.index.model.NodeInstance;
+import org.kie.kogito.index.model.ProcessDefinition;
 import org.kie.kogito.index.model.ProcessInstance;
 import org.kie.kogito.index.model.ProcessInstanceError;
 import org.kie.kogito.index.model.ProcessInstanceState;
@@ -121,6 +122,13 @@ public static ProcessDefinitionDataEvent getProcessDefinitionDataEvent(String pr
         return new ProcessDefinitionDataEvent(body);
     }
 
+    public static ProcessDefinition getProcessDefinition(String processId) {
+        ProcessDefinition def = new ProcessDefinition();
+        def.setId(processId);
+        def.setVersion(TestUtils.PROCESS_VERSION);
+        return def;
+    }
+
     public static ProcessInstanceStateDataEvent getProcessCloudEvent(String processId, String processInstanceId, ProcessInstanceState status, String rootProcessInstanceId, String rootProcessId,
             String parentProcessInstanceId, String identity) {