From 68f2fd356d5aedaf99886eccabf45b5d99a1fca1 Mon Sep 17 00:00:00 2001 From: Dan Davison Date: Thu, 16 Jan 2025 13:29:19 -0500 Subject: [PATCH] Respond to upstream changes --- tests/worker/test_workflow.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py index 59129d5f7..813d5814c 100644 --- a/tests/worker/test_workflow.py +++ b/tests/worker/test_workflow.py @@ -5502,12 +5502,11 @@ async def _get_workflow_result( await handle.signal(handler_name) else: if not wait_all_handlers_finished: - with pytest.raises(RPCError) as err: + with pytest.raises(WorkflowUpdateFailedError) as err_info: await handle.execute_update(handler_name, id="my-update") - assert ( - err.value.status == RPCStatusCode.NOT_FOUND - and "workflow execution already completed" in str(err.value).lower() - ) + update_err = err_info.value + assert isinstance(update_err.cause, ApplicationError) + assert update_err.cause.type == "AcceptedUpdateCompletedWorkflow" else: await handle.execute_update(handler_name, id="my-update") @@ -5736,11 +5735,12 @@ async def _run_workflow_and_get_warning(self) -> bool: if self.handler_waiting == "-wait-all-handlers-finish-": await update_task else: - with pytest.raises(RPCError) as update_err: + with pytest.raises(WorkflowUpdateFailedError) as err_info: await update_task - assert update_err.value.status == RPCStatusCode.NOT_FOUND and ( - str(update_err.value).lower() - == "workflow execution already completed" + update_err = err_info.value + assert isinstance(update_err.cause, ApplicationError) + assert ( + update_err.cause.type == "AcceptedUpdateCompletedWorkflow" ) with pytest.raises(WorkflowFailureError) as err: