@@ -5604,7 +5604,7 @@ class UseLockOrSemaphoreWorkflowParameters:
56045604
56055605
56065606@workflow .defn
5607- class CoroutinesUseLockWorkflow :
5607+ class CoroutinesUseLockOrSemaphoreWorkflow :
56085608 def __init__ (self ) -> None :
56095609 self .params : UseLockOrSemaphoreWorkflowParameters
56105610 self .lock_or_semaphore : Union [asyncio .Lock , asyncio .Semaphore ]
@@ -5674,7 +5674,7 @@ def _exits_critical_section(self, id: str) -> None:
56745674
56755675
56765676@workflow .defn
5677- class HandlerCoroutinesUseLockWorkflow ( CoroutinesUseLockWorkflow ):
5677+ class HandlerCoroutinesUseLockOrSemaphoreWorkflow ( CoroutinesUseLockOrSemaphoreWorkflow ):
56785678 def __init__ (self ) -> None :
56795679 super ().__init__ ()
56805680 self .workflow_may_exit = False
@@ -5710,11 +5710,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
57105710):
57115711 async with new_worker (
57125712 client ,
5713- CoroutinesUseLockWorkflow ,
5713+ CoroutinesUseLockOrSemaphoreWorkflow ,
57145714 activities = [noop_activity_for_lock_or_semaphore_tests ],
57155715 ) as worker :
57165716 summary = await client .execute_workflow (
5717- CoroutinesUseLockWorkflow .run ,
5717+ CoroutinesUseLockOrSemaphoreWorkflow .run ,
57185718 arg = params ,
57195719 id = str (uuid .uuid4 ()),
57205720 task_queue = worker .task_queue ,
@@ -5736,7 +5736,7 @@ async def _do_update_handler_lock_or_semaphore_test(
57365736
57375737 task_queue = "tq"
57385738 handle = await client .start_workflow (
5739- HandlerCoroutinesUseLockWorkflow .run ,
5739+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .run ,
57405740 id = f"wf-{ str (uuid .uuid4 ())} " ,
57415741 task_queue = task_queue ,
57425742 )
@@ -5745,21 +5745,21 @@ async def _do_update_handler_lock_or_semaphore_test(
57455745 await admitted_update_task (
57465746 client ,
57475747 handle ,
5748- HandlerCoroutinesUseLockWorkflow .my_update ,
5748+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .my_update ,
57495749 arg = params ,
57505750 id = f"update-{ i } " ,
57515751 )
57525752 for i in range (n_updates )
57535753 ]
57545754 async with new_worker (
57555755 client ,
5756- HandlerCoroutinesUseLockWorkflow ,
5756+ HandlerCoroutinesUseLockOrSemaphoreWorkflow ,
57575757 activities = [noop_activity_for_lock_or_semaphore_tests ],
57585758 task_queue = task_queue ,
57595759 ):
57605760 for update_task in admitted_updates :
57615761 await update_task
5762- await handle .signal (HandlerCoroutinesUseLockWorkflow .finish )
5762+ await handle .signal (HandlerCoroutinesUseLockOrSemaphoreWorkflow .finish )
57635763 summary = await handle .result ()
57645764 assert summary == expectation
57655765
0 commit comments