Skip to content

Commit 1322b5a

Browse files
committed
Rename workflow
1 parent 65995d5 commit 1322b5a

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
@@ -5889,7 +5889,7 @@ class UseLockOrSemaphoreWorkflowParameters:
58895889

58905890

58915891
@workflow.defn
5892-
class CoroutinesUseLockWorkflow:
5892+
class CoroutinesUseLockOrSemaphoreWorkflow:
58935893
def __init__(self) -> None:
58945894
self.params: UseLockOrSemaphoreWorkflowParameters
58955895
self.lock_or_semaphore: Union[asyncio.Lock, asyncio.Semaphore]
@@ -5959,7 +5959,7 @@ def _exits_critical_section(self, id: str) -> None:
59595959

59605960

59615961
@workflow.defn
5962-
class HandlerCoroutinesUseLockWorkflow(CoroutinesUseLockWorkflow):
5962+
class HandlerCoroutinesUseLockOrSemaphoreWorkflow(CoroutinesUseLockOrSemaphoreWorkflow):
59635963
def __init__(self) -> None:
59645964
super().__init__()
59655965
self.workflow_may_exit = False
@@ -5995,11 +5995,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
59955995
):
59965996
async with new_worker(
59975997
client,
5998-
CoroutinesUseLockWorkflow,
5998+
CoroutinesUseLockOrSemaphoreWorkflow,
59995999
activities=[noop_activity_for_lock_or_semaphore_tests],
60006000
) as worker:
60016001
summary = await client.execute_workflow(
6002-
CoroutinesUseLockWorkflow.run,
6002+
CoroutinesUseLockOrSemaphoreWorkflow.run,
60036003
arg=params,
60046004
id=str(uuid.uuid4()),
60056005
task_queue=worker.task_queue,
@@ -6021,7 +6021,7 @@ async def _do_update_handler_lock_or_semaphore_test(
60216021

60226022
task_queue = "tq"
60236023
handle = await client.start_workflow(
6024-
HandlerCoroutinesUseLockWorkflow.run,
6024+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.run,
60256025
id=f"wf-{str(uuid.uuid4())}",
60266026
task_queue=task_queue,
60276027
)
@@ -6030,21 +6030,21 @@ async def _do_update_handler_lock_or_semaphore_test(
60306030
await admitted_update_task(
60316031
client,
60326032
handle,
6033-
HandlerCoroutinesUseLockWorkflow.my_update,
6033+
HandlerCoroutinesUseLockOrSemaphoreWorkflow.my_update,
60346034
arg=params,
60356035
id=f"update-{i}",
60366036
)
60376037
for i in range(n_updates)
60386038
]
60396039
async with new_worker(
60406040
client,
6041-
HandlerCoroutinesUseLockWorkflow,
6041+
HandlerCoroutinesUseLockOrSemaphoreWorkflow,
60426042
activities=[noop_activity_for_lock_or_semaphore_tests],
60436043
task_queue=task_queue,
60446044
):
60456045
for update_task in admitted_updates:
60466046
await update_task
6047-
await handle.signal(HandlerCoroutinesUseLockWorkflow.finish)
6047+
await handle.signal(HandlerCoroutinesUseLockOrSemaphoreWorkflow.finish)
60486048
summary = await handle.result()
60496049
assert summary == expectation
60506050

0 commit comments

Comments
 (0)