@@ -6639,7 +6639,7 @@ class UseLockOrSemaphoreWorkflowParameters:
6639
6639
6640
6640
6641
6641
@workflow .defn
6642
- class CoroutinesUseLockWorkflow :
6642
+ class CoroutinesUseLockOrSemaphoreWorkflow :
6643
6643
def __init__ (self ) -> None :
6644
6644
self .params : UseLockOrSemaphoreWorkflowParameters
6645
6645
self .lock_or_semaphore : Union [asyncio .Lock , asyncio .Semaphore ]
@@ -6709,7 +6709,7 @@ def _exits_critical_section(self, id: str) -> None:
6709
6709
6710
6710
6711
6711
@workflow .defn
6712
- class HandlerCoroutinesUseLockWorkflow ( CoroutinesUseLockWorkflow ):
6712
+ class HandlerCoroutinesUseLockOrSemaphoreWorkflow ( CoroutinesUseLockOrSemaphoreWorkflow ):
6713
6713
def __init__ (self ) -> None :
6714
6714
super ().__init__ ()
6715
6715
self .workflow_may_exit = False
@@ -6745,11 +6745,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
6745
6745
):
6746
6746
async with new_worker (
6747
6747
client ,
6748
- CoroutinesUseLockWorkflow ,
6748
+ CoroutinesUseLockOrSemaphoreWorkflow ,
6749
6749
activities = [noop_activity_for_lock_or_semaphore_tests ],
6750
6750
) as worker :
6751
6751
summary = await client .execute_workflow (
6752
- CoroutinesUseLockWorkflow .run ,
6752
+ CoroutinesUseLockOrSemaphoreWorkflow .run ,
6753
6753
arg = params ,
6754
6754
id = str (uuid .uuid4 ()),
6755
6755
task_queue = worker .task_queue ,
@@ -6771,7 +6771,7 @@ async def _do_update_handler_lock_or_semaphore_test(
6771
6771
6772
6772
task_queue = "tq"
6773
6773
handle = await client .start_workflow (
6774
- HandlerCoroutinesUseLockWorkflow .run ,
6774
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .run ,
6775
6775
id = f"wf-{ str (uuid .uuid4 ())} " ,
6776
6776
task_queue = task_queue ,
6777
6777
)
@@ -6780,21 +6780,21 @@ async def _do_update_handler_lock_or_semaphore_test(
6780
6780
await admitted_update_task (
6781
6781
client ,
6782
6782
handle ,
6783
- HandlerCoroutinesUseLockWorkflow .my_update ,
6783
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .my_update ,
6784
6784
arg = params ,
6785
6785
id = f"update-{ i } " ,
6786
6786
)
6787
6787
for i in range (n_updates )
6788
6788
]
6789
6789
async with new_worker (
6790
6790
client ,
6791
- HandlerCoroutinesUseLockWorkflow ,
6791
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow ,
6792
6792
activities = [noop_activity_for_lock_or_semaphore_tests ],
6793
6793
task_queue = task_queue ,
6794
6794
):
6795
6795
for update_task in admitted_updates :
6796
6796
await update_task
6797
- await handle .signal (HandlerCoroutinesUseLockWorkflow .finish )
6797
+ await handle .signal (HandlerCoroutinesUseLockOrSemaphoreWorkflow .finish )
6798
6798
summary = await handle .result ()
6799
6799
assert summary == expectation
6800
6800
0 commit comments