Skip to content

Commit cf0fb36

Browse files
committed
Respond to upstream changes
1 parent 7665bf5 commit cf0fb36

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/worker/test_workflow.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -5736,11 +5736,12 @@ async def _run_workflow_and_get_warning(self) -> bool:
57365736
if self.handler_waiting == "-wait-all-handlers-finish-":
57375737
await update_task
57385738
else:
5739-
with pytest.raises(RPCError) as update_err:
5739+
with pytest.raises(WorkflowUpdateFailedError) as err_info:
57405740
await update_task
5741-
assert update_err.value.status == RPCStatusCode.NOT_FOUND and (
5742-
str(update_err.value).lower()
5743-
== "workflow execution already completed"
5741+
update_err = err_info.value
5742+
assert isinstance(update_err.cause, ApplicationError)
5743+
assert (
5744+
update_err.cause.type == "AcceptedUpdateCompletedWorkflow"
57445745
)
57455746

57465747
with pytest.raises(WorkflowFailureError) as err:

0 commit comments

Comments
 (0)