@@ -5533,7 +5533,7 @@ class UseLockOrSemaphoreWorkflowParameters:
55335533
55345534
55355535@workflow .defn
5536- class CoroutinesUseLockWorkflow :
5536+ class CoroutinesUseLockOrSemaphoreWorkflow :
55375537 def __init__ (self ) -> None :
55385538 self .params : UseLockOrSemaphoreWorkflowParameters
55395539 self .lock_or_semaphore : Union [asyncio .Lock , asyncio .Semaphore ]
@@ -5603,7 +5603,7 @@ def _exits_critical_section(self, id: str) -> None:
56035603
56045604
56055605@workflow .defn
5606- class HandlerCoroutinesUseLockWorkflow ( CoroutinesUseLockWorkflow ):
5606+ class HandlerCoroutinesUseLockOrSemaphoreWorkflow ( CoroutinesUseLockOrSemaphoreWorkflow ):
56075607 def __init__ (self ) -> None :
56085608 super ().__init__ ()
56095609 self .workflow_may_exit = False
@@ -5639,11 +5639,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
56395639):
56405640 async with new_worker (
56415641 client ,
5642- CoroutinesUseLockWorkflow ,
5642+ CoroutinesUseLockOrSemaphoreWorkflow ,
56435643 activities = [noop_activity_for_lock_or_semaphore_tests ],
56445644 ) as worker :
56455645 summary = await client .execute_workflow (
5646- CoroutinesUseLockWorkflow .run ,
5646+ CoroutinesUseLockOrSemaphoreWorkflow .run ,
56475647 arg = params ,
56485648 id = str (uuid .uuid4 ()),
56495649 task_queue = worker .task_queue ,
@@ -5665,7 +5665,7 @@ async def _do_update_handler_lock_or_semaphore_test(
56655665
56665666 task_queue = "tq"
56675667 handle = await client .start_workflow (
5668- HandlerCoroutinesUseLockWorkflow .run ,
5668+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .run ,
56695669 id = f"wf-{ str (uuid .uuid4 ())} " ,
56705670 task_queue = task_queue ,
56715671 )
@@ -5674,21 +5674,21 @@ async def _do_update_handler_lock_or_semaphore_test(
56745674 await admitted_update_task (
56755675 client ,
56765676 handle ,
5677- HandlerCoroutinesUseLockWorkflow .my_update ,
5677+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .my_update ,
56785678 arg = params ,
56795679 id = f"update-{ i } " ,
56805680 )
56815681 for i in range (n_updates )
56825682 ]
56835683 async with new_worker (
56845684 client ,
5685- HandlerCoroutinesUseLockWorkflow ,
5685+ HandlerCoroutinesUseLockOrSemaphoreWorkflow ,
56865686 activities = [noop_activity_for_lock_or_semaphore_tests ],
56875687 task_queue = task_queue ,
56885688 ):
56895689 for update_task in admitted_updates :
56905690 await update_task
5691- await handle .signal (HandlerCoroutinesUseLockWorkflow .finish )
5691+ await handle .signal (HandlerCoroutinesUseLockOrSemaphoreWorkflow .finish )
56925692 summary = await handle .result ()
56935693 assert summary == expectation
56945694
0 commit comments