mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-12 10:11:52 +00:00
Implemented handler flags feature (#728)
* Implemented handler flag feature * Cover tests
This commit is contained in:
parent
5f07cb3d06
commit
3ad16be507
6 changed files with 36 additions and 4 deletions
|
|
@ -287,6 +287,8 @@ class TestTelegramEventObserver:
|
|||
observer.register(pipe_handler, mix_data)
|
||||
|
||||
results = await observer.trigger(42)
|
||||
assert len(results) == 2
|
||||
assert results[1].pop("handler")
|
||||
assert results == ((42,), {"b": 2})
|
||||
|
||||
@pytest.mark.parametrize("middleware_type", ("middleware", "outer_middleware"))
|
||||
|
|
|
|||
|
|
@ -126,3 +126,16 @@ class TestCommandObject:
|
|||
)
|
||||
def test_text(self, obj: CommandObject, result: str):
|
||||
assert obj.text == result
|
||||
|
||||
def test_update_handler_flags(self):
|
||||
cmd = Command(commands=["start"])
|
||||
flags = {}
|
||||
cmd.update_handler_flags(flags)
|
||||
|
||||
assert "commands" in flags
|
||||
assert isinstance(flags["commands"], list)
|
||||
assert len(flags["commands"]) == 1
|
||||
assert flags["commands"][0] is cmd
|
||||
|
||||
cmd.update_handler_flags(flags)
|
||||
assert len(flags["commands"]) == 2
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue