Skip to content

Commit a524593

Browse files
committed
Rename workflow
1 parent 0a3dcb7 commit a524593

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
@@ -5533,7 +5533,7 @@ class UseLockOrSemaphoreWorkflowParameters:
55335533

55345534

55355535
@workflow.defn
5536-
class CoroutinesUseLockWorkflow:
5536+
class CoroutinesUseLockOrSemaphoreWorkflow:
55375537
def __init__(self) -> None:
55385538
self.params: UseLockOrSemaphoreWorkflowParameters
55395539
self.lock_or_semaphore: Union[asyncio.Lock, asyncio.Semaphore]
@@ -5603,7 +5603,7 @@ def _exits_critical_section(self, id: str) -> None:
56035603

56045604

56055605
@workflow.defn
5606-
class HandlerCoroutinesUseLockWorkflow(CoroutinesUseLockWorkflow):
5606+
class HandlerCoroutinesUseLockOrSemaphoreWorkflow(CoroutinesUseLockOrSemaphoreWorkflow):
56075607
def __init__(self) -> None:
56085608
super().__init__()
56095609
self.workflow_may_exit = False
@@ -5639,11 +5639,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
56395639
):
56405640
async with new_worker(
56415641
client,
5642-
CoroutinesUseLockWorkflow,
5642+
CoroutinesUseLockOrSemaphoreWorkflow,
56435643
activities=[noop_activity_for_lock_or_semaphore_tests],
56445644
) as worker:
56455645
summary = await client.execute_workflow(
5646-
CoroutinesUseLockWorkflow.run,
5646+
CoroutinesUseLockOrSemaphoreWorkflow.run,
56475647
arg=params,
56485648
id=str(uuid.uuid4()),
56495649
task_queue=worker.task_queue,
@@ -5665,7 +5665,7 @@ async def _do_update_handler_lock_or_semaphore_test(
56655665

56665666
task_queue = "tq"
56675667
handle = await client.start_workflow(
5668-
HandlerCoroutinesUseLockWorkflow.run,
5668+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.run,
56695669
id=f"wf-{str(uuid.uuid4())}",
56705670
task_queue=task_queue,
56715671
)
@@ -5674,21 +5674,21 @@ async def _do_update_handler_lock_or_semaphore_test(
56745674
await admitted_update_task(
56755675
client,
56765676
handle,
5677-
HandlerCoroutinesUseLockWorkflow.my_update,
5677+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.my_update,
56785678
arg=params,
56795679
id=f"update-{i}",
56805680
)
56815681
for i in range(n_updates)
56825682
]
56835683
async with new_worker(
56845684
client,
5685-
HandlerCoroutinesUseLockWorkflow,
5685+
HandlerCoroutinesUseLockOrSemaphoreWorkflow,
56865686
activities=[noop_activity_for_lock_or_semaphore_tests],
56875687
task_queue=task_queue,
56885688
):
56895689
for update_task in admitted_updates:
56905690
await update_task
5691-
await handle.signal(HandlerCoroutinesUseLockWorkflow.finish)
5691+
await handle.signal(HandlerCoroutinesUseLockOrSemaphoreWorkflow.finish)
56925692
summary = await handle.result()
56935693
assert summary == expectation
56945694

0 commit comments

Comments
 (0)