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

Compatibility...

Browse source
This commit is contained in:
JRoot Junior 2023-11-18 01:43:03 +02:00
parent 2c1c2fcc03
commit b5f7867451
No known key found for this signature in database
GPG key ID: 738964250D5FF6E2
2 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
aiogram/fsm/scene.py
View file

@ -269,7 +269,7 @@ class SceneHandlerWrapper:
return self
def __str__(self) -> str:
result = f"SceneHandlerWrapper({self.handler.callback}"
result = f"SceneHandlerWrapper({self.scene}, {self.handler.callback}"
if self.after:
result += f", after={self.after}"
result += ")"

19
tests/test_fsm/test_scene.py
View file

@ -299,21 +299,26 @@ class TestSceneHandlerWrapper:
assert result == 42
def test_scene_handler_wrapper_str(self):
# Mock objects
scene_mock = AsyncMock(spec=Scene)
handler_mock = AsyncMock()
after_mock = AsyncMock() # Implement this according to your After type
class MyScene(Scene):
pass
scene_handler_wrapper = SceneHandlerWrapper(scene_mock, handler_mock, after=after_mock)
async def handler_mock(*args, **kwargs):
pass
after = After.back()
scene_handler_wrapper = SceneHandlerWrapper(MyScene, handler_mock, after=after)
result = str(scene_handler_wrapper)
assert result == f"SceneHandlerWrapper({handler_mock}, after={after_mock})"
assert result == f"SceneHandlerWrapper({MyScene}, {handler_mock}, after={after})"
def test_await(self):
class MyScene(Scene):
pass
handler_mock = AsyncMock()
async def handler_mock(*args, **kwargs):
pass
scene_handler_wrapper = SceneHandlerWrapper(MyScene, handler_mock)
assert inspect.isawaitable(scene_handler_wrapper)