Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions tests/worker/test_workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -6845,6 +6845,14 @@ async def _run_workflow_and_get_warning(self) -> bool:
id=workflow_id,
task_queue=task_queue,
)
# Keep the update on the run that receives it so continue-as-new cannot
# make a completion poll look for the update on the successor run.
assert handle.result_run_id
handle = self.client.get_workflow_handle(
workflow_id,
run_id=handle.result_run_id,
first_execution_run_id=handle.first_execution_run_id,
)
if self.workflow_termination_type == "-cancellation-":
await handle.cancel()

Expand Down