@@ -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