Skip to content

Commit 72fdcc1

Browse files
committed
Rename workflow
1 parent ed50954 commit 72fdcc1

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
@@ -5604,7 +5604,7 @@ class UseLockOrSemaphoreWorkflowParameters:
56045604

56055605

56065606
@workflow.defn
5607-
class CoroutinesUseLockWorkflow:
5607+
class CoroutinesUseLockOrSemaphoreWorkflow:
56085608
def __init__(self) -> None:
56095609
self.params: UseLockOrSemaphoreWorkflowParameters
56105610
self.lock_or_semaphore: Union[asyncio.Lock, asyncio.Semaphore]
@@ -5674,7 +5674,7 @@ def _exits_critical_section(self, id: str) -> None:
56745674

56755675

56765676
@workflow.defn
5677-
class HandlerCoroutinesUseLockWorkflow(CoroutinesUseLockWorkflow):
5677+
class HandlerCoroutinesUseLockOrSemaphoreWorkflow(CoroutinesUseLockOrSemaphoreWorkflow):
56785678
def __init__(self) -> None:
56795679
super().__init__()
56805680
self.workflow_may_exit = False
@@ -5710,11 +5710,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
57105710
):
57115711
async with new_worker(
57125712
client,
5713-
CoroutinesUseLockWorkflow,
5713+
CoroutinesUseLockOrSemaphoreWorkflow,
57145714
activities=[noop_activity_for_lock_or_semaphore_tests],
57155715
) as worker:
57165716
summary = await client.execute_workflow(
5717-
CoroutinesUseLockWorkflow.run,
5717+
CoroutinesUseLockOrSemaphoreWorkflow.run,
57185718
arg=params,
57195719
id=str(uuid.uuid4()),
57205720
task_queue=worker.task_queue,
@@ -5736,7 +5736,7 @@ async def _do_update_handler_lock_or_semaphore_test(
57365736

57375737
task_queue = "tq"
57385738
handle = await client.start_workflow(
5739-
HandlerCoroutinesUseLockWorkflow.run,
5739+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.run,
57405740
id=f"wf-{str(uuid.uuid4())}",
57415741
task_queue=task_queue,
57425742
)
@@ -5745,21 +5745,21 @@ async def _do_update_handler_lock_or_semaphore_test(
57455745
await admitted_update_task(
57465746
client,
57475747
handle,
5748-
HandlerCoroutinesUseLockWorkflow.my_update,
5748+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.my_update,
57495749
arg=params,
57505750
id=f"update-{i}",
57515751
)
57525752
for i in range(n_updates)
57535753
]
57545754
async with new_worker(
57555755
client,
5756-
HandlerCoroutinesUseLockWorkflow,
5756+
HandlerCoroutinesUseLockOrSemaphoreWorkflow,
57575757
activities=[noop_activity_for_lock_or_semaphore_tests],
57585758
task_queue=task_queue,
57595759
):
57605760
for update_task in admitted_updates:
57615761
await update_task
5762-
await handle.signal(HandlerCoroutinesUseLockWorkflow.finish)
5762+
await handle.signal(HandlerCoroutinesUseLockOrSemaphoreWorkflow.finish)
57635763
summary = await handle.result()
57645764
assert summary == expectation
57655765

0 commit comments

Comments
 (0)