test(injector): fix tests after injector added

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"