@@ -5889,7 +5889,7 @@ class UseLockOrSemaphoreWorkflowParameters:
5889
5889
5890
5890
5891
5891
@workflow .defn
5892
- class CoroutinesUseLockWorkflow :
5892
+ class CoroutinesUseLockOrSemaphoreWorkflow :
5893
5893
def __init__ (self ) -> None :
5894
5894
self .params : UseLockOrSemaphoreWorkflowParameters
5895
5895
self .lock_or_semaphore : Union [asyncio .Lock , asyncio .Semaphore ]
@@ -5959,7 +5959,7 @@ def _exits_critical_section(self, id: str) -> None:
5959
5959
5960
5960
5961
5961
@workflow .defn
5962
- class HandlerCoroutinesUseLockWorkflow ( CoroutinesUseLockWorkflow ):
5962
+ class HandlerCoroutinesUseLockOrSemaphoreWorkflow ( CoroutinesUseLockOrSemaphoreWorkflow ):
5963
5963
def __init__ (self ) -> None :
5964
5964
super ().__init__ ()
5965
5965
self .workflow_may_exit = False
@@ -5995,11 +5995,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
5995
5995
):
5996
5996
async with new_worker (
5997
5997
client ,
5998
- CoroutinesUseLockWorkflow ,
5998
+ CoroutinesUseLockOrSemaphoreWorkflow ,
5999
5999
activities = [noop_activity_for_lock_or_semaphore_tests ],
6000
6000
) as worker :
6001
6001
summary = await client .execute_workflow (
6002
- CoroutinesUseLockWorkflow .run ,
6002
+ CoroutinesUseLockOrSemaphoreWorkflow .run ,
6003
6003
arg = params ,
6004
6004
id = str (uuid .uuid4 ()),
6005
6005
task_queue = worker .task_queue ,
@@ -6021,7 +6021,7 @@ async def _do_update_handler_lock_or_semaphore_test(
6021
6021
6022
6022
task_queue = "tq"
6023
6023
handle = await client .start_workflow (
6024
- HandlerCoroutinesUseLockWorkflow .run ,
6024
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .run ,
6025
6025
id = f"wf-{ str (uuid .uuid4 ())} " ,
6026
6026
task_queue = task_queue ,
6027
6027
)
@@ -6030,21 +6030,21 @@ async def _do_update_handler_lock_or_semaphore_test(
6030
6030
await admitted_update_task (
6031
6031
client ,
6032
6032
handle ,
6033
- HandlerCoroutinesUseLockWorkflow .my_update ,
6033
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .my_update ,
6034
6034
arg = params ,
6035
6035
id = f"update-{ i } " ,
6036
6036
)
6037
6037
for i in range (n_updates )
6038
6038
]
6039
6039
async with new_worker (
6040
6040
client ,
6041
- HandlerCoroutinesUseLockWorkflow ,
6041
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow ,
6042
6042
activities = [noop_activity_for_lock_or_semaphore_tests ],
6043
6043
task_queue = task_queue ,
6044
6044
):
6045
6045
for update_task in admitted_updates :
6046
6046
await update_task
6047
- await handle .signal (HandlerCoroutinesUseLockWorkflow .finish )
6047
+ await handle .signal (HandlerCoroutinesUseLockOrSemaphoreWorkflow .finish )
6048
6048
summary = await handle .result ()
6049
6049
assert summary == expectation
6050
6050
0 commit comments