Skip to content

Commit f299ffe

Browse files
committed
Rename workflow
1 parent c728e2d commit f299ffe

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/worker/test_workflow.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6639,7 +6639,7 @@ class UseLockOrSemaphoreWorkflowParameters:
66396639

66406640

66416641
@workflow.defn
6642-
class CoroutinesUseLockWorkflow:
6642+
class CoroutinesUseLockOrSemaphoreWorkflow:
66436643
def __init__(self) -> None:
66446644
self.params: UseLockOrSemaphoreWorkflowParameters
66456645
self.lock_or_semaphore: Union[asyncio.Lock, asyncio.Semaphore]
@@ -6709,7 +6709,7 @@ def _exits_critical_section(self, id: str) -> None:
67096709

67106710

67116711
@workflow.defn
6712-
class HandlerCoroutinesUseLockWorkflow(CoroutinesUseLockWorkflow):
6712+
class HandlerCoroutinesUseLockOrSemaphoreWorkflow(CoroutinesUseLockOrSemaphoreWorkflow):
67136713
def __init__(self) -> None:
67146714
super().__init__()
67156715
self.workflow_may_exit = False
@@ -6745,11 +6745,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
67456745
):
67466746
async with new_worker(
67476747
client,
6748-
CoroutinesUseLockWorkflow,
6748+
CoroutinesUseLockOrSemaphoreWorkflow,
67496749
activities=[noop_activity_for_lock_or_semaphore_tests],
67506750
) as worker:
67516751
summary = await client.execute_workflow(
6752-
CoroutinesUseLockWorkflow.run,
6752+
CoroutinesUseLockOrSemaphoreWorkflow.run,
67536753
arg=params,
67546754
id=str(uuid.uuid4()),
67556755
task_queue=worker.task_queue,
@@ -6771,7 +6771,7 @@ async def _do_update_handler_lock_or_semaphore_test(
67716771

67726772
task_queue = "tq"
67736773
handle = await client.start_workflow(
6774-
HandlerCoroutinesUseLockWorkflow.run,
6774+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.run,
67756775
id=f"wf-{str(uuid.uuid4())}",
67766776
task_queue=task_queue,
67776777
)
@@ -6780,21 +6780,21 @@ async def _do_update_handler_lock_or_semaphore_test(
67806780
await admitted_update_task(
67816781
client,
67826782
handle,
6783-
HandlerCoroutinesUseLockWorkflow.my_update,
6783+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.my_update,
67846784
arg=params,
67856785
id=f"update-{i}",
67866786
)
67876787
for i in range(n_updates)
67886788
]
67896789
async with new_worker(
67906790
client,
6791-
HandlerCoroutinesUseLockWorkflow,
6791+
HandlerCoroutinesUseLockOrSemaphoreWorkflow,
67926792
activities=[noop_activity_for_lock_or_semaphore_tests],
67936793
task_queue=task_queue,
67946794
):
67956795
for update_task in admitted_updates:
67966796
await update_task
6797-
await handle.signal(HandlerCoroutinesUseLockWorkflow.finish)
6797+
await handle.signal(HandlerCoroutinesUseLockOrSemaphoreWorkflow.finish)
67986798
summary = await handle.result()
67996799
assert summary == expectation
68006800

0 commit comments

Comments
 (0)