From 95f131f9fdaec9f0c081d0c1dcc4eedc41be64ac Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Mon, 4 Mar 2024 12:35:39 +0800
Subject: test(generator): sync changes with generator

---
 tests/test_generator.py | 25 ++++++++++++++++++++++++-
 tests/test_loader.py    |  6 +++---
 tests/test_register.py  | 10 +++++-----
 3 files changed, 32 insertions(+), 9 deletions(-)

(limited to 'tests')

diff --git a/tests/test_generator.py b/tests/test_generator.py
index 5994e768..ce33f4ff 100644
--- a/tests/test_generator.py
+++ b/tests/test_generator.py
@@ -1,4 +1,4 @@
-from infini.generator import TextGenerator
+from infini.generator import Generator, TextGenerator
 from infini.injector import Injector
 from infini.output import Output
 
@@ -44,3 +44,26 @@ def test_generator_injector():
         )
         == "[苏向夜]Event1 文本: 变量测试"
     )
+
+
+def test_register_generator():
+    def name(nickname: str = "苏向夜"):
+        return nickname
+
+    custom = TextGenerator()
+    custom.type = "custom_text"
+
+    generator = Generator()
+    generator.events = {
+        "test.event1": "[{{ card_name }}]Event1 文本: {{ var }}",
+    }
+    generator.generators.update({"custom_text": custom})
+
+    generator.global_variables = {"card_name": name}
+    assert (
+        generator.output(
+            Output("custom_text", "test.event1", variables={"var": "变量测试"}),
+            Injector(),
+        )
+        == "[苏向夜]Event1 文本: 变量测试"
+    )
diff --git a/tests/test_loader.py b/tests/test_loader.py
index 05716b60..82c1dc7b 100644
--- a/tests/test_loader.py
+++ b/tests/test_loader.py
@@ -18,10 +18,10 @@ def test_loader():
     def test_handler(_: Input):
         return Output("text", "block.snh", block=True)
 
-    register.regist_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID")
-    register.regist_textevent("block.snh", "不许撅{{ get_snh_id }}")
+    register.register_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID")
+    register.register_textevent("block.snh", "不许撅{{ get_snh_id }}")
 
-    register.regist_variable("sxy_id", "苏向夜")
+    register.register_variable("sxy_id", "苏向夜")
 
     @register.dynamic_variable()
     def get_snh_id():
diff --git a/tests/test_register.py b/tests/test_register.py
index ea857a15..654836f1 100644
--- a/tests/test_register.py
+++ b/tests/test_register.py
@@ -1,5 +1,5 @@
 from infini.core import Core
-from infini.generator import TextGenerator
+from infini.generator import Generator
 from infini.handler import Handler
 from infini.injector import Injector
 from infini.input import Input
@@ -22,10 +22,10 @@ def test_register():
     def test_handler(_: Input):
         return Output("text", "block.snh", block=True)
 
-    register.regist_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID")
-    register.regist_textevent("block.snh", "不许撅{{ get_snh_id }}")
+    register.register_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID")
+    register.register_textevent("block.snh", "不许撅{{ get_snh_id }}")
 
-    register.regist_variable("sxy_id", "苏向夜")
+    register.register_variable("sxy_id", "苏向夜")
 
     @register.dynamic_variable()
     def get_snh_id():
@@ -39,7 +39,7 @@ def test_register():
     pre_interceptor.interceptors = register.pre_interceptors
     handler = Handler()
     handler.handlers = register.handlers
-    generator = TextGenerator()
+    generator = Generator()
     generator.events = register.events
     generator.global_variables = register.global_variables
     interceptor = Interceptor()
-- 
cgit v1.2.3-70-g09d2