@@ -5533,7 +5533,7 @@ class UseLockOrSemaphoreWorkflowParameters:
5533
5533
5534
5534
5535
5535
@workflow .defn
5536
- class CoroutinesUseLockWorkflow :
5536
+ class CoroutinesUseLockOrSemaphoreWorkflow :
5537
5537
def __init__ (self ) -> None :
5538
5538
self .params : UseLockOrSemaphoreWorkflowParameters
5539
5539
self .lock_or_semaphore : Union [asyncio .Lock , asyncio .Semaphore ]
@@ -5603,7 +5603,7 @@ def _exits_critical_section(self, id: str) -> None:
5603
5603
5604
5604
5605
5605
@workflow .defn
5606
- class HandlerCoroutinesUseLockWorkflow ( CoroutinesUseLockWorkflow ):
5606
+ class HandlerCoroutinesUseLockOrSemaphoreWorkflow ( CoroutinesUseLockOrSemaphoreWorkflow ):
5607
5607
def __init__ (self ) -> None :
5608
5608
super ().__init__ ()
5609
5609
self .workflow_may_exit = False
@@ -5639,11 +5639,11 @@ async def _do_workflow_coroutines_lock_or_semaphore_test(
5639
5639
):
5640
5640
async with new_worker (
5641
5641
client ,
5642
- CoroutinesUseLockWorkflow ,
5642
+ CoroutinesUseLockOrSemaphoreWorkflow ,
5643
5643
activities = [noop_activity_for_lock_or_semaphore_tests ],
5644
5644
) as worker :
5645
5645
summary = await client .execute_workflow (
5646
- CoroutinesUseLockWorkflow .run ,
5646
+ CoroutinesUseLockOrSemaphoreWorkflow .run ,
5647
5647
arg = params ,
5648
5648
id = str (uuid .uuid4 ()),
5649
5649
task_queue = worker .task_queue ,
@@ -5665,7 +5665,7 @@ async def _do_update_handler_lock_or_semaphore_test(
5665
5665
5666
5666
task_queue = "tq"
5667
5667
handle = await client .start_workflow (
5668
- HandlerCoroutinesUseLockWorkflow .run ,
5668
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .run ,
5669
5669
id = f"wf-{ str (uuid .uuid4 ())} " ,
5670
5670
task_queue = task_queue ,
5671
5671
)
@@ -5674,21 +5674,21 @@ async def _do_update_handler_lock_or_semaphore_test(
5674
5674
await admitted_update_task (
5675
5675
client ,
5676
5676
handle ,
5677
- HandlerCoroutinesUseLockWorkflow .my_update ,
5677
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow .my_update ,
5678
5678
arg = params ,
5679
5679
id = f"update-{ i } " ,
5680
5680
)
5681
5681
for i in range (n_updates )
5682
5682
]
5683
5683
async with new_worker (
5684
5684
client ,
5685
- HandlerCoroutinesUseLockWorkflow ,
5685
+ HandlerCoroutinesUseLockOrSemaphoreWorkflow ,
5686
5686
activities = [noop_activity_for_lock_or_semaphore_tests ],
5687
5687
task_queue = task_queue ,
5688
5688
):
5689
5689
for update_task in admitted_updates :
5690
5690
await update_task
5691
- await handle .signal (HandlerCoroutinesUseLockWorkflow .finish )
5691
+ await handle .signal (HandlerCoroutinesUseLockOrSemaphoreWorkflow .finish )
5692
5692
summary = await handle .result ()
5693
5693
assert summary == expectation
5694
5694
0 commit comments