Skip to content

Commit c2b6287

Browse files
committed
Fix incorrect signatures of late-registered handlers
1 parent 3867507 commit c2b6287

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/worker/test_workflow.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5426,26 +5426,22 @@ async def run(
54265426
if handler_registration == "-late-registered-":
54275427
if handler_dynamism == "-dynamic-":
54285428

5429-
async def my_late_registered_dynamic_update(
5430-
self, name: str, args: Sequence[RawValue]
5431-
) -> str:
5429+
async def my_late_registered_dynamic_update(*args, **kwargs) -> str:
54325430
await workflow.wait_condition(lambda: self.handlers_may_finish)
54335431
return "my-late-registered-dynamic-update-result"
54345432

5435-
async def my_late_registered_dynamic_signal(
5436-
self, name: str, args: Sequence[RawValue]
5437-
) -> None:
5433+
async def my_late_registered_dynamic_signal(*args, **kwargs) -> None:
54385434
await workflow.wait_condition(lambda: self.handlers_may_finish)
54395435

54405436
workflow.set_dynamic_update_handler(my_late_registered_dynamic_update)
54415437
workflow.set_dynamic_signal_handler(my_late_registered_dynamic_signal)
54425438
else:
54435439

5444-
async def my_late_registered_update(self) -> str:
5440+
async def my_late_registered_update() -> str:
54455441
await workflow.wait_condition(lambda: self.handlers_may_finish)
54465442
return "my-late-registered-update-result"
54475443

5448-
async def my_late_registered_signal(self) -> None:
5444+
async def my_late_registered_signal() -> None:
54495445
await workflow.wait_condition(lambda: self.handlers_may_finish)
54505446

54515447
workflow.set_update_handler(

0 commit comments

Comments
 (0)