test(injector): update injector tests

2 files changed, 24 insertions, 1 deletions

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