WAIT_COMPLETED

Author: dandavison

tests/nexus/test_workflow_caller_cancellation_types.py

@@ -405,8 +405,20 @@ async def check_behavior_for_wait_cancellation_completed(
     Check that a cancellation request is sent and the caller workflow nexus operation future is
     unblocked after the operation is canceled.
     """
+    # For WAIT_COMPLETED, wait for the handler workflow to complete
+    try:
+        await handler_wf.result()
+    except WorkflowFailureError as err:
+        assert isinstance(err.__cause__, exceptions.CancelledError)
+    else:
+        pytest.fail("Expected WorkflowFailureError")
+
     handler_status = (await handler_wf.describe()).status
     assert handler_status == WorkflowExecutionStatus.CANCELED
+
+    await caller_wf.signal(CallerWorkflow.release)
+    result = await caller_wf.result()
+
     await _assert_event_subsequence(
         [
             (caller_wf, EventType.EVENT_TYPE_WORKFLOW_EXECUTION_STARTED),
@@ -420,7 +432,6 @@ async def check_behavior_for_wait_cancellation_completed(
             (caller_wf, EventType.EVENT_TYPE_WORKFLOW_EXECUTION_COMPLETED),
         ]
     )
-    result = await caller_wf.result()
     caller_op_future_resolved = test_context.caller_op_future_resolved.result()
     handler_wf_canceled_event_time = await _get_event_time(
         handler_wf,