@@ -5841,7 +5841,7 @@ async def test_workflow_id_conflict(client: Client):
5841
5841
5842
5842
5843
5843
@workflow .defn
5844
- class TestUpdateCompletionIsHonoredWhenAfterWorkflowReturn1 :
5844
+ class UpdateCompletionIsHonoredWhenAfterWorkflowReturn1Workflow :
5845
5845
def __init__ (self ) -> None :
5846
5846
self .workflow_returned = False
5847
5847
@@ -5867,13 +5867,13 @@ async def test_update_completion_is_honored_when_after_workflow_return_1(
5867
5867
update_id = "my-update"
5868
5868
task_queue = "tq"
5869
5869
wf_handle = await client .start_workflow (
5870
- TestUpdateCompletionIsHonoredWhenAfterWorkflowReturn1 .run ,
5870
+ UpdateCompletionIsHonoredWhenAfterWorkflowReturn1Workflow .run ,
5871
5871
id = f"wf-{ uuid .uuid4 ()} " ,
5872
5872
task_queue = task_queue ,
5873
5873
)
5874
5874
update_result_task = asyncio .create_task (
5875
5875
wf_handle .execute_update (
5876
- TestUpdateCompletionIsHonoredWhenAfterWorkflowReturn1 .my_update ,
5876
+ UpdateCompletionIsHonoredWhenAfterWorkflowReturn1Workflow .my_update ,
5877
5877
id = update_id ,
5878
5878
)
5879
5879
)
@@ -5882,14 +5882,14 @@ async def test_update_completion_is_honored_when_after_workflow_return_1(
5882
5882
async with Worker (
5883
5883
client ,
5884
5884
task_queue = task_queue ,
5885
- workflows = [TestUpdateCompletionIsHonoredWhenAfterWorkflowReturn1 ],
5885
+ workflows = [UpdateCompletionIsHonoredWhenAfterWorkflowReturn1Workflow ],
5886
5886
):
5887
5887
assert await wf_handle .result () == "workflow-result"
5888
5888
assert await update_result_task == "update-result"
5889
5889
5890
5890
5891
5891
@workflow .defn
5892
- class TestUpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 :
5892
+ class UpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 :
5893
5893
def __init__ (self ):
5894
5894
self .received_update = False
5895
5895
self .update_result : asyncio .Future [str ] = asyncio .Future ()
@@ -5923,15 +5923,15 @@ async def test_update_completion_is_honored_when_after_workflow_return_2(
5923
5923
async with Worker (
5924
5924
client ,
5925
5925
task_queue = "tq" ,
5926
- workflows = [TestUpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 ],
5926
+ workflows = [UpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 ],
5927
5927
) as worker :
5928
5928
handle = await client .start_workflow (
5929
- TestUpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 .run ,
5929
+ UpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 .run ,
5930
5930
id = f"wf-{ uuid .uuid4 ()} " ,
5931
5931
task_queue = worker .task_queue ,
5932
5932
)
5933
5933
update_result = await handle .execute_update (
5934
- TestUpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 .my_update
5934
+ UpdateCompletionIsHonoredWhenAfterWorkflowReturnWorkflow2 .my_update
5935
5935
)
5936
5936
assert update_result == "update-result"
5937
5937
assert await handle .result () == "workflow-result"
0 commit comments