From c681a3d355089a91ec83a692664fb4cf6ef43f26 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Sat, 2 Mar 2024 11:35:29 +0800 Subject: test(injector): update injector tests --- tests/test_generator.py | 20 ++++++++++++++++++++ tests/test_injector.py | 5 ++++- 2 files changed, 24 insertions(+), 1 deletion(-) diff --git a/tests/test_generator.py b/tests/test_generator.py index e796efc7..5994e768 100644 --- a/tests/test_generator.py +++ b/tests/test_generator.py @@ -24,3 +24,23 @@ def test_generator_with_var(): ) == "Event1 文本: 变量测试" ) + + +def test_generator_injector(): + def name(nickname: str = "苏向夜"): + return nickname + + injector = Injector() + injector.parameters = {"a": 12, "b": 20, "c": 0} + + generator = TextGenerator() + generator.events = { + "test.event1": "[{{ card_name }}]Event1 文本: {{ var }}", + } + generator.global_variables = {"card_name": name} + assert ( + generator.output( + Output("text", "test.event1", variables={"var": "变量测试"}), injector + ) + == "[苏向夜]Event1 文本: 变量测试" + ) diff --git a/tests/test_injector.py b/tests/test_injector.py index 0d320a2c..426e5684 100644 --- a/tests/test_injector.py +++ b/tests/test_injector.py @@ -2,10 +2,13 @@ from infini.injector import Injector def test_injector(): + def name(name: str): + return name + def add(a: int, b: int = 0): return a + b injector = Injector() - injector.parameters = {"a": 12, "b": 20, "c": 0} + injector.parameters = {"a": 12, "b": 20, "c": 0, "card_name": name} assert injector.inject(add)() == 32 assert injector.output(add) == 32 -- cgit v1.2.3-70-g09d2