test(workflow): Add test of workflow cancellation while update handler is executing (#1482)

Author: dandavison

packages/test/src/test-integration-update.ts

@@ -602,6 +602,28 @@ test('update result poll throws WorkflowUpdateRPCTimeoutOrCancelledError', async
   });
 });
 
+const updateThatShouldFail = wf.defineUpdate('updateThatShouldFail');
+
+export async function workflowThatWillBeCanceled(): Promise<void> {
+  wf.setHandler(updateThatShouldFail, async () => {
+    await wf.condition(() => false);
+  });
+  await wf.condition(() => false);
+}
+
+test('update caller gets update failed error on workflow cancellation', async (t) => {
+  const { createWorker, startWorkflow, assertWorkflowUpdateFailed } = helpers(t);
+  const worker = await createWorker();
+  await worker.runUntil(async () => {
+    const w = await startWorkflow(workflowThatWillBeCanceled);
+    const u = await w.startUpdate(updateThatShouldFail, {
+      waitForStage: WorkflowUpdateStage.ACCEPTED,
+    });
+    await w.cancel();
+    await assertWorkflowUpdateFailed(u.result(), wf.CancelledFailure, 'Workflow cancelled');
+  });
+});
+
 export { signalUpdateOrderingWorkflow };
 
 // Validate that issue #1474 is fixed in 1.11.0+