diff --git a/tests/test_dispatcher/test_event/test_telegram.py b/tests/test_dispatcher/test_event/test_telegram.py index 4d0e11e1..df9e128d 100644 --- a/tests/test_dispatcher/test_event/test_telegram.py +++ b/tests/test_dispatcher/test_event/test_telegram.py @@ -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")) diff --git a/tests/test_dispatcher/test_filters/test_command.py b/tests/test_dispatcher/test_filters/test_command.py index 18888fe1..d7a0ef55 100644 --- a/tests/test_dispatcher/test_filters/test_command.py +++ b/tests/test_dispatcher/test_filters/test_command.py @@ -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