diff options
| -rw-r--r-- | tests/test_core.py | 2 | ||||
| -rw-r--r-- | tests/test_generator.py | 22 | ||||
| -rw-r--r-- | tests/test_injector.py | 11 | ||||
| -rw-r--r-- | tests/test_loader.py | 2 | ||||
| -rw-r--r-- | tests/test_register.py | 4 |
5 files changed, 22 insertions, 19 deletions
diff --git a/tests/test_core.py b/tests/test_core.py index 11a8c57d..5f3f12f1 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -1,6 +1,7 @@ from infini.core import Core from infini.generator import TextGenerator from infini.handler import Handler +from infini.injector import Injector from infini.input import Input from infini.interceptor import Interceptor from infini.output import Output @@ -60,6 +61,7 @@ def test_core(): core.interceptor = interceptor core.pre_interceptor = interceptor core.generator = generator + core.injector = Injector() outputs = set() for output in core.input(command_input): diff --git a/tests/test_generator.py b/tests/test_generator.py index 0eed5756..e796efc7 100644 --- a/tests/test_generator.py +++ b/tests/test_generator.py @@ -1,4 +1,5 @@ from infini.generator import TextGenerator +from infini.injector import Injector from infini.output import Output @@ -8,7 +9,7 @@ def test_generator(): "test.event1": "Event1 文本", } generator.match(Output("text", "test.event1")) - assert generator.output(Output("text", "test.event1")) == "Event1 文本" + assert generator.output(Output("text", "test.event1"), Injector()) == "Event1 文本" def test_generator_with_var(): @@ -18,21 +19,8 @@ def test_generator_with_var(): } assert ( - generator.output(Output("text", "test.event1", variables={"var": "变量测试"})) + generator.output( + Output("text", "test.event1", variables={"var": "变量测试"}), Injector() + ) == "Event1 文本: 变量测试" ) - - -def test_generator_with_function(): - def add(a, b): - return a + b - - generator = TextGenerator() - generator.events = { - "test.event": "{{ func(1, 2) }}", - } - - assert ( - generator.output(Output("text", "test.event", variables={"func": add})) - == "3" - ) diff --git a/tests/test_injector.py b/tests/test_injector.py new file mode 100644 index 00000000..0d320a2c --- /dev/null +++ b/tests/test_injector.py @@ -0,0 +1,11 @@ +from infini.injector import Injector + + +def test_injector(): + def add(a: int, b: int = 0): + return a + b + + injector = Injector() + injector.parameters = {"a": 12, "b": 20, "c": 0} + assert injector.inject(add)() == 32 + assert injector.output(add) == 32 diff --git a/tests/test_loader.py b/tests/test_loader.py index 439700e5..05716b60 100644 --- a/tests/test_loader.py +++ b/tests/test_loader.py @@ -19,7 +19,7 @@ def test_loader(): return Output("text", "block.snh", block=True) register.regist_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID") - register.regist_textevent("block.snh", "不许撅{{ get_snh_id() }}") + register.regist_textevent("block.snh", "不许撅{{ get_snh_id }}") register.regist_variable("sxy_id", "苏向夜") diff --git a/tests/test_register.py b/tests/test_register.py index 0ce8351b..ea857a15 100644 --- a/tests/test_register.py +++ b/tests/test_register.py @@ -1,6 +1,7 @@ from infini.core import Core from infini.generator import TextGenerator from infini.handler import Handler +from infini.injector import Injector from infini.input import Input from infini.interceptor import Interceptor from infini.output import Output @@ -22,7 +23,7 @@ def test_register(): return Output("text", "block.snh", block=True) register.regist_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID") - register.regist_textevent("block.snh", "不许撅{{ get_snh_id() }}") + register.regist_textevent("block.snh", "不许撅{{ get_snh_id }}") register.regist_variable("sxy_id", "苏向夜") @@ -49,6 +50,7 @@ def test_register(): core.handler = handler core.generator = generator core.interceptor = interceptor + core.injector = Injector() for output in core.input(blocked_god_input): assert output == "不可直呼苏向夜的ID" |