Cover tests

This commit is contained in:
Alex Root Junior 2021-10-13 01:53:21 +03:00
parent 7564b3e2db
commit a3dd01bdb7
2 changed files with 15 additions and 0 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