Implemented handler flags feature (#728)

* Implemented handler flag feature

* Cover tests
This commit is contained in:
Alex Root Junior 2021-10-25 23:37:14 +03:00 committed by GitHub
parent 5f07cb3d06
commit 3ad16be507
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 36 additions and 4 deletions

View file

@ -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"))

View file

@ -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