@@ -47,11 +47,24 @@ public void registerValidWorkflowClass() {
4747 assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerWorkflow (TestWorkflow .class ));
4848 }
4949
50+ @ Test
51+ public void registerValidVersionWorkflowClass () {
52+ assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerWorkflow (TestWorkflow .class ,"testWorkflowV1" , false ));
53+ assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerWorkflow (TestWorkflow .class ,"testWorkflowV2" , true ));
54+ }
55+
5056 @ Test
5157 public void registerValidWorkflowInstance () {
5258 assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerWorkflow (new TestWorkflow ()));
5359 }
5460
61+ @ Test
62+ public void registerValidVersionWorkflowInstance () {
63+ assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerWorkflow (new TestWorkflow (),"testWorkflowV1" , false ));
64+ assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerWorkflow (new TestWorkflow (),"testWorkflowV2" , true ));
65+ }
66+
67+
5568 @ Test
5669 public void registerValidWorkflowActivityClass () {
5770 assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerActivity (TestActivity .class ));
@@ -62,12 +75,15 @@ public void registerValidWorkflowActivityInstance() {
6275 assertDoesNotThrow (() -> new WorkflowRuntimeBuilder ().registerActivity (new TestActivity ()));
6376 }
6477
78+
79+
6580 @ Test
6681 public void buildTest () {
6782 assertDoesNotThrow (() -> {
6883 try {
6984 WorkflowRuntime runtime = new WorkflowRuntimeBuilder ().build ();
7085 System .out .println ("WorkflowRuntime created" );
86+ runtime .close ();
7187 } catch (Exception e ) {
7288 throw new RuntimeException (e );
7389 }
@@ -82,16 +98,18 @@ public void loggingOutputTest() {
8298
8399 Logger testLogger = mock (Logger .class );
84100
85- assertDoesNotThrow (() -> new WorkflowRuntimeBuilder (testLogger ).registerWorkflow (TestWorkflow .class ));
86- assertDoesNotThrow (() -> new WorkflowRuntimeBuilder (testLogger ).registerActivity (TestActivity .class ));
101+ var runtimeBuilder = new WorkflowRuntimeBuilder (testLogger );
102+ assertDoesNotThrow (() -> runtimeBuilder .registerWorkflow (TestWorkflow .class ));
103+ assertDoesNotThrow (() -> runtimeBuilder .registerActivity (TestActivity .class ));
87104
88- WorkflowRuntimeBuilder workflowRuntimeBuilder = new WorkflowRuntimeBuilder ();
105+ var runtime = runtimeBuilder . build ();
89106
90- WorkflowRuntime runtime = workflowRuntimeBuilder .build ();
91107 verify (testLogger , times (1 ))
92108 .info (eq ("Registered Workflow: {}" ), eq ("TestWorkflow" ));
93109
94110 verify (testLogger , times (1 ))
95111 .info (eq ("Registered Activity: {}" ), eq ("TestActivity" ));
112+
113+ runtime .close ();
96114 }
97115}
0 commit comments