refactor SubWorkflowTest

Author: matheusandre1

impl/test/src/test/java/io/serverlessworkflow/impl/test/SubWorkflowTest.java

@@ -16,13 +16,14 @@
 package io.serverlessworkflow.impl.test;
 
 import static io.serverlessworkflow.api.WorkflowReader.readWorkflowFromClasspath;
-import static org.junit.Assert.assertEquals;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.equalTo;
+import static org.hamcrest.Matchers.is;
 
 import io.serverlessworkflow.api.types.Workflow;
 import io.serverlessworkflow.fluent.spec.WorkflowBuilder;
 import io.serverlessworkflow.fluent.spec.dsl.DSL;
 import io.serverlessworkflow.impl.WorkflowApplication;
-import java.io.IOException;
 import java.util.Map;
 import org.junit.Test;
 
@@ -31,7 +32,7 @@ public class SubWorkflowTest {
   private static final String WORKFLOW_TEST_PATH = "workflows-samples/sub-workflow/";
 
   @Test
-  public void setTest() throws IOException {
+  public void setTest() throws Exception {
     Workflow workflowParent =
         readWorkflowFromClasspath(WORKFLOW_TEST_PATH + "sub-workflow-parent.yaml");
     Workflow workflowChild =
@@ -43,18 +44,16 @@ public void setTest() throws IOException {
           app.workflowDefinition(workflowParent)
               .instance(Map.of())
               .start()
-              .get()
+              .join()
               .asMap()
               .orElseThrow();
-      assertEquals("1", result.get("counter").toString());
-      assertEquals("helloWorld", result.get("greeting").toString());
-    } catch (Exception e) {
-      throw new RuntimeException("Workflow execution failed", e);
+      assertThat(result.get("counter"), equalTo(1));
+      assertThat(result.get("greeting"), equalTo("helloWorld"));
     }
   }
 
   @Test
-  public void setBlankInputTest() throws IOException {
+  public void setBlankInputTest() throws Exception {
     Workflow workflowParent =
         readWorkflowFromClasspath(WORKFLOW_TEST_PATH + "sub-workflow-parent.yaml");
     Workflow workflowChild =
@@ -66,18 +65,16 @@ public void setBlankInputTest() throws IOException {
           app.workflowDefinition(workflowParent)
               .instance(Map.of())
               .start()
-              .get()
+              .join()
               .asMap()
               .orElseThrow();
-      assertEquals("1", result.get("counter").toString());
-      assertEquals("helloWorld", result.get("greeting").toString());
-    } catch (Exception e) {
-      throw new RuntimeException("Workflow execution failed", e);
+      assertThat(result.get("counter"), is(equalTo(1)));
+      assertThat(result.get("greeting"), is(equalTo("helloWorld")));
     }
   }
 
   @Test
-  public void setStringInputTest() throws IOException {
+  public void setStringInputTest() throws Exception {
     Workflow workflowParent =
         readWorkflowFromClasspath(WORKFLOW_TEST_PATH + "sub-workflow-parent.yaml");
     Workflow workflowChild =
@@ -89,18 +86,16 @@ public void setStringInputTest() throws IOException {
           app.workflowDefinition(workflowParent)
               .instance("Tested")
               .start()
-              .get()
+              .join()
               .asMap()
               .orElseThrow();
-      assertEquals("1", result.get("counter").toString());
-      assertEquals("helloWorld", result.get("greeting").toString());
-    } catch (Exception e) {
-      throw new RuntimeException("Workflow execution failed", e);
+      assertThat(result.get("counter"), is(equalTo(1)));
+      assertThat(result.get("greeting"), is(equalTo("helloWorld")));
     }
   }
 
   @Test
-  public void readContextAndSetTest() throws IOException {
+  public void readContextAndSetTest() throws Exception {
     Workflow workflowParent =
         readWorkflowFromClasspath(
             WORKFLOW_TEST_PATH + "read-context-and-set-sub-workflow-parent.yaml");
@@ -112,20 +107,19 @@ public void readContextAndSetTest() throws IOException {
     try (WorkflowApplication app = WorkflowApplication.builder().build()) {
       app.workflowDefinition(workflowChild);
       Map<String, Object> result =
-          app.workflowDefinition(workflowParent).instance(map).start().get().asMap().orElseThrow();
+          app.workflowDefinition(workflowParent).instance(map).start().join().asMap().orElseThrow();
       Map<String, String> updated = (Map<String, String>) result.get("updated");
-      assertEquals("userId_1_tested", updated.get("userId"));
-      assertEquals("test_tested", updated.get("username"));
-      assertEquals("test_tested", updated.get("password"));
-      assertEquals(
-          "The workflow set-into-context:1.0.0 updated user in context", result.get("detail"));
-    } catch (Exception e) {
-      throw new RuntimeException("Workflow execution failed", e);
+      assertThat(updated.get("userId"), is(equalTo("userId_1_tested")));
+      assertThat(updated.get("username"), is(equalTo("test_tested")));
+      assertThat(updated.get("password"), is(equalTo("test_tested")));
+      assertThat(
+          result.get("detail"),
+          is(equalTo("The workflow set-into-context:1.0.0 updated user in context")));
     }
   }
 
   @Test
-  public void outputExportContextAndSetTest() throws IOException {
+  public void outputExportContextAndSetTest() throws Exception {
     Workflow workflowParent =
         readWorkflowFromClasspath(
             WORKFLOW_TEST_PATH + "output-export-and-set-sub-workflow-parent.yaml");
@@ -137,17 +131,15 @@ public void outputExportContextAndSetTest() throws IOException {
     try (WorkflowApplication app = WorkflowApplication.builder().build()) {
       app.workflowDefinition(workflowChild);
       Map<String, Object> result =
-          app.workflowDefinition(workflowParent).instance(map).start().get().asMap().orElseThrow();
-      assertEquals("userId_1_tested", result.get("userId"));
-      assertEquals("test_tested", result.get("username"));
-      assertEquals("test_tested", result.get("password"));
-    } catch (Exception e) {
-      throw new RuntimeException("Workflow execution failed", e);
+          app.workflowDefinition(workflowParent).instance(map).start().join().asMap().orElseThrow();
+      assertThat(result.get("userId"), is(equalTo("userId_1_tested")));
+      assertThat(result.get("username"), is(equalTo("test_tested")));
+      assertThat(result.get("password"), is(equalTo("test_tested")));
     }
   }
 
   @Test
-  public void runSubWorkflowFromDslTest() {
+  public void runSubWorkflowFromDslTest() throws Exception {
     Workflow child =
         WorkflowBuilder.workflow("childFlow", "org.acme", "1.0.0")
             .tasks(d -> d.set("update", s -> s.put("counter", 1).put("greeting", "helloWorld")))
@@ -164,11 +156,9 @@ public void runSubWorkflowFromDslTest() {
     try (WorkflowApplication app = WorkflowApplication.builder().build()) {
       app.workflowDefinition(child);
       Map<String, Object> result =
-          app.workflowDefinition(parent).instance(Map.of()).start().get().asMap().orElseThrow();
-      assertEquals("1", result.get("counter").toString());
-      assertEquals("helloWorld", result.get("greeting").toString());
-    } catch (Exception e) {
-      throw new RuntimeException("Workflow execution failed", e);
+          app.workflowDefinition(parent).instance(Map.of()).start().join().asMap().orElseThrow();
+      assertThat(result.get("counter"), is(equalTo(1)));
+      assertThat(result.get("greeting"), is(equalTo("helloWorld")));
     }
   }
 }