Skip to content

Commit 4281b20

Browse files
committed
Split tests
1 parent 13b4bc2 commit 4281b20

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

tests/worker/test_update_with_start.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)