@@ -680,13 +680,8 @@ async def update(self, arg1: str, arg2: str) -> UpdateResult:
680680 return UpdateResult (result = arg1 + "-" + arg2 )
681681
682682
683- async def test_update_with_start_overloads (client : Client ):
684- async with new_worker (
685- client ,
686- NoParamWorkflow ,
687- OneParamWorkflow ,
688- TwoParamWorkflow ,
689- ) as worker :
683+ async def test_update_with_start_no_param (client : Client ):
684+ async with new_worker (client , NoParamWorkflow ) as worker :
690685 # No-params typed
691686 no_param_start_op = WithStartWorkflowOperation (
692687 NoParamWorkflow .my_workflow_run ,
@@ -721,6 +716,9 @@ async def test_update_with_start_overloads(client: Client):
721716 wf_handle = await no_param_start_op .workflow_handle ()
722717 assert await wf_handle .result () == WorkflowResult (result = "workflow-result" )
723718
719+
720+ async def test_update_with_start_one_param (client : Client ):
721+ async with new_worker (client , OneParamWorkflow ) as worker :
724722 # One-param typed
725723 one_param_start_op = WithStartWorkflowOperation (
726724 OneParamWorkflow .my_workflow_run ,
@@ -759,6 +757,9 @@ async def test_update_with_start_overloads(client: Client):
759757 wf_handle = await one_param_start_op .workflow_handle ()
760758 assert await wf_handle .result () == WorkflowResult (result = "workflow-arg" )
761759
760+
761+ async def test_update_with_start_two_param (client : Client ):
762+ async with new_worker (client , TwoParamWorkflow ) as worker :
762763 # Two-params typed
763764 two_param_start_op = WithStartWorkflowOperation (
764765 TwoParamWorkflow .my_workflow_run ,
0 commit comments