python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added more tests for non-function handlers

Browse source
This commit is contained in:
JRoot Junior 2024-10-05 22:46:32 +03:00
parent 15776ba164
commit 4ecdca1438
No known key found for this signature in database
GPG key ID: 738964250D5FF6E2
1 changed files with 34 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tests/test_fsm/test_scene.py
View file

@ -1628,3 +1628,37 @@ class TestSceneInheritance:
assert parent_enter_action.callback is ParentScene.on_enter
assert child_enter_action.callback is not ParentScene.on_enter
assert child_enter_action.callback is ChildScene.on_enter
def test_override_non_function_handler_by_function(self):
class ParentScene(Scene):
do_exit = on.message(Command("exit"), after=After.exit)
class ChildScene1(ParentScene):
pass
class ChildScene2(ParentScene):
do_exit = on.message(Command("exit"), after=After.back)
class ChildScene3(ParentScene):
@on.message(Command("exit"), after=After.back)
async def do_exit(self, message: Message) -> None:
pass
assert len(ParentScene.__scene_config__.handlers) == 1
assert len(ChildScene1.__scene_config__.handlers) == 1
assert len(ChildScene2.__scene_config__.handlers) == 1
assert len(ChildScene3.__scene_config__.handlers) == 1
parent_handler = ParentScene.__scene_config__.handlers[0]
child_1_handler = ChildScene1.__scene_config__.handlers[0]
child_2_handler = ChildScene2.__scene_config__.handlers[0]
child_3_handler = ChildScene3.__scene_config__.handlers[0]
assert child_1_handler.handler is parent_handler.handler
assert child_1_handler.after == parent_handler.after
assert child_1_handler.handler is _empty_handler
assert child_2_handler.after != parent_handler.after
assert child_2_handler.handler is _empty_handler
assert child_3_handler.handler is not _empty_handler