Add better support for wait timeout with subStatus (#177)

Author: longquanzheng

iwf-idl

@@ -1,9 +1,7 @@
 package io.iworkflow.integ;
 
-import io.iworkflow.core.Client;
-import io.iworkflow.core.ClientOptions;
-import io.iworkflow.core.ImmutableStopWorkflowOptions;
-import io.iworkflow.core.WorkflowUncompletedException;
+import io.iworkflow.core.*;
+import io.iworkflow.gen.models.ErrorSubStatus;
 import io.iworkflow.gen.models.WorkflowErrorType;
 import io.iworkflow.gen.models.WorkflowStatus;
 import io.iworkflow.gen.models.WorkflowStopType;
@@ -28,6 +26,34 @@ public void setup() throws ExecutionException, InterruptedException {
         TestSingletonWorkerService.startWorkerIfNotUp();
     }
 
+    @Test
+    public void testWorkflowWaitTimeout() throws InterruptedException {
+        final Client client = new Client(WorkflowRegistry.registry, ClientOptions.localDefault);
+        final String wfId = "testWorkflowTimeout" + System.currentTimeMillis() / 1000;
+        final Integer input = 1;
+
+
+        client.startWorkflow(BasicSignalWorkflow.class, wfId, 100, input);
+
+
+        long startMs = System.currentTimeMillis();
+        long elapsedMs;
+        try {
+            client.getSimpleWorkflowResultWithWait(Integer.class, wfId);
+        } catch (ClientSideException e) {
+            Assertions.assertEquals(ErrorSubStatus.LONG_POLL_TIME_OUT_SUB_STATUS, e.getErrorSubStatus());
+            Assertions.assertEquals(420, e.getStatusCode());
+
+            elapsedMs = (System.currentTimeMillis() - startMs) / 1000;
+            // NOTE: because the default poll timeout is 60-2 = 28s in iWF service
+            Assertions.assertTrue(elapsedMs >= 55 && elapsedMs <= 62, "expect to poll for 58 seconds, actual is %d " + elapsedMs);
+            return;
+        } catch (Exception e) {
+            Assertions.fail("expected to catch ClientSideException");
+        }
+        Assertions.fail("expected to catch ClientSideException");
+    }
+
     @Test
     public void testWorkflowTimeout() throws InterruptedException {
         final Client client = new Client(WorkflowRegistry.registry, ClientOptions.localDefault);