From 8aaa6f75b2f4b0a0b84f33fabd1bd4f97df982d2 Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Fri, 26 Jan 2024 19:02:26 +0800
Subject: :sparkles: feat(ci): add generator tests and refact core tests

---
 tests/test_core.py      | 23 ++++++++++++++---------
 tests/test_generator.py | 11 +++++++++++
 2 files changed, 25 insertions(+), 9 deletions(-)
 create mode 100644 tests/test_generator.py

diff --git a/tests/test_core.py b/tests/test_core.py
index 91d96a2c..db402dea 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -13,7 +13,7 @@ def test_core():
     valid_input = Input("这个叫苏向夜.")
     command_and_valid_input = Input(".echo 苏向夜打爆了简某人的狗头")
 
-    def intercept(_: Input) -> Input | Output:
+    def intercept_jianlvchun(_: Input) -> Input | Output:
         return Output("text", "block.jianlvchun", block=True)  # TODO 拦截器阻塞标识
 
     interceptor = Interceptor()
@@ -21,20 +21,18 @@ def test_core():
         {
             "priority": 1,
             "router": ContainsRouter("简律纯"),
-            "handler": intercept,
+            "handler": intercept_jianlvchun,
         }
     ]
 
     def add(input: Input) -> Output:
+        result = str(sum(list(map(int, input.get_plain_text().lstrip(".add").split()))))
         return Output(
-            "text",
-            str(sum(list(map(int, input.get_plain_text().lstrip(".add").split())))),
-            status=0,
-            block=False,
+            "text", "test.add", status=0, block=False, variables={"result": result}
         )
 
     def cmd(_: Input) -> Output:
-        return Output("text", "cmd", status=0, block=False)
+        return Output("text", "test.cmd", status=0, block=False)
 
     handler = Handler()
     handler.handlers = [
@@ -50,11 +48,18 @@ def test_core():
         },
     ]
 
+    generator = Generator()
+    generator.events = {
+        "test.cmd": "cmd",
+        "test.add": "{{ result }}",
+        "block.jianlvchun": "检测到违禁词",
+    }
+
     core = Core()
     core.handler = handler
     core.interceptor = interceptor
     core.pre_interceptor = interceptor
-    core.generator = Generator()
+    core.generator = generator
 
     outputs = set()
     for output in core.input(command_input):
@@ -67,7 +72,7 @@ def test_core():
     assert count == 0
 
     for output in core.input(intercepted_input):
-        assert output == "block.jianlvchun"
+        assert output == "检测到违禁词"
 
     outputs = set()
     for output in core.input(command_and_valid_input):
diff --git a/tests/test_generator.py b/tests/test_generator.py
new file mode 100644
index 00000000..32f1d626
--- /dev/null
+++ b/tests/test_generator.py
@@ -0,0 +1,11 @@
+from infini.generator import Generator
+from infini.output import Output
+
+
+def test_generator():
+    generator = Generator()
+    generator.events = {
+        "test.event1": "Event1 文本",
+    }
+    generator.match(Output("text", "test.event1"))
+    assert generator.output(Output("text", "test.event1")) == "Event1 文本"
-- 
cgit v1.2.3-70-g09d2