@@ -6639,7 +6639,7 @@ class UseLockOrSemaphoreWorkflowParameters:
66396639
66406640
66416641@workflow .defn
6642- class CoroutinesUseLockWorkflow :
6642+ class CoroutinesUseLockOrSemaphoreWorkflow :
66436643 def __init__ (self ) -> None :
66446644 self .params : UseLockOrSemaphoreWorkflowParameters
66456645 self .lock_or_semaphore : Union [asyncio .Lock , asyncio .Semaphore ]
@@ -6709,7 +6709,7 @@ def _exits_critical_section(self, id: str) -> None:
67096709
67106710
67116711@workflow .defn
6712- class HandlerCoroutinesUseLockWorkflow ( CoroutinesUseLockWorkflow ):
6712+ class HandlerCoroutinesUseLockOrSemaphoreWorkflow ( CoroutinesUseLockOrSemaphoreWorkflow ):
67136713 def __init__ (self ) -> None :
67146714 super ().__init__ ()
67156715 self .workflow_may_exit = False
@@ -6745,11 +6745,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
67456745):
67466746 async with new_worker (
67476747 client ,
6748- CoroutinesUseLockWorkflow ,
6748+ CoroutinesUseLockOrSemaphoreWorkflow ,
67496749 activities = [noop_activity_for_lock_or_semaphore_tests ],
67506750 ) as worker :
67516751 summary = await client .execute_workflow (
6752- CoroutinesUseLockWorkflow .run ,
6752+ CoroutinesUseLockOrSemaphoreWorkflow .run ,
67536753 arg = params ,
67546754 id = str (uuid .uuid4 ()),
67556755 task_queue = worker .task_queue ,
@@ -6771,7 +6771,7 @@ async def _do_update_handler_lock_or_semaphore_test(
67716771
67726772 task_queue = "tq"
67736773 handle = await client .start_workflow (
6774- HandlerCoroutinesUseLockWorkflow .run ,
6774+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .run ,
67756775 id = f"wf-{ str (uuid .uuid4 ())} " ,
67766776 task_queue = task_queue ,
67776777 )
@@ -6780,21 +6780,21 @@ async def _do_update_handler_lock_or_semaphore_test(
67806780 await admitted_update_task (
67816781 client ,
67826782 handle ,
6783- HandlerCoroutinesUseLockWorkflow .my_update ,
6783+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .my_update ,
67846784 arg = params ,
67856785 id = f"update-{ i } " ,
67866786 )
67876787 for i in range (n_updates )
67886788 ]
67896789 async with new_worker (
67906790 client ,
6791- HandlerCoroutinesUseLockWorkflow ,
6791+ HandlerCoroutinesUseLockOrSemaphoreWorkflow ,
67926792 activities = [noop_activity_for_lock_or_semaphore_tests ],
67936793 task_queue = task_queue ,
67946794 ):
67956795 for update_task in admitted_updates :
67966796 await update_task
6797- await handle .signal (HandlerCoroutinesUseLockWorkflow .finish )
6797+ await handle .signal (HandlerCoroutinesUseLockOrSemaphoreWorkflow .finish )
67986798 summary = await handle .result ()
67996799 assert summary == expectation
68006800
0 commit comments