From bb54ca26fd42b0e0cbf9cc745c4e2ff7c574cc4f Mon Sep 17 00:00:00 2001 From: Dan Davison Date: Tue, 20 Aug 2024 11:21:50 -0400 Subject: [PATCH] Fix incorrect signatures of late-registered handlers --- tests/worker/test_workflow.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py index d922bafa..930f23ac 100644 --- a/tests/worker/test_workflow.py +++ b/tests/worker/test_workflow.py @@ -5426,26 +5426,22 @@ async def run( if handler_registration == "-late-registered-": if handler_dynamism == "-dynamic-": - async def my_late_registered_dynamic_update( - self, name: str, args: Sequence[RawValue] - ) -> str: + async def my_late_registered_dynamic_update(*args, **kwargs) -> str: await workflow.wait_condition(lambda: self.handlers_may_finish) return "my-late-registered-dynamic-update-result" - async def my_late_registered_dynamic_signal( - self, name: str, args: Sequence[RawValue] - ) -> None: + async def my_late_registered_dynamic_signal(*args, **kwargs) -> None: await workflow.wait_condition(lambda: self.handlers_may_finish) workflow.set_dynamic_update_handler(my_late_registered_dynamic_update) workflow.set_dynamic_signal_handler(my_late_registered_dynamic_signal) else: - async def my_late_registered_update(self) -> str: + async def my_late_registered_update() -> str: await workflow.wait_condition(lambda: self.handlers_may_finish) return "my-late-registered-update-result" - async def my_late_registered_signal(self) -> None: + async def my_late_registered_signal() -> None: await workflow.wait_condition(lambda: self.handlers_may_finish) workflow.set_update_handler(