From c5943be36fcdf442a92e2a4a4fc7fa03fbefc268 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Mon, 4 Mar 2024 11:48:35 +0800 Subject: feat(test): update tests for injector changes --- tests/test_injector.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) (limited to 'tests/test_injector.py') diff --git a/tests/test_injector.py b/tests/test_injector.py index 426e5684..ccd040ba 100644 --- a/tests/test_injector.py +++ b/tests/test_injector.py @@ -1,4 +1,9 @@ +from infini.handler import Handler from infini.injector import Injector +from infini.input import Input +from infini.loader import Loader +from infini.output import Output +from infini.router import Startswith def test_injector(): @@ -12,3 +17,30 @@ def test_injector(): injector.parameters = {"a": 12, "b": 20, "c": 0, "card_name": name} assert injector.inject(add)() == 32 assert injector.output(add) == 32 + + +def test_handler_injector(): + input = Input("test_message") + + def absolute(input: Input, plain_text: str) -> Output: + return input.output( + "text", + plain_text, + status=0, + block=False, + ) + + handler = Handler() + handler.handlers = [ + { + "priority": 2, + "router": Startswith(".add"), + "handler": absolute, + }, + ] + + core = Loader().into_core() + core.handler = handler + + for output in core.input(input): + assert output == "test_message" -- cgit v1.2.3-70-g09d2