From 318a30667b03a3dce5deedaadd7a71785edd4088 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Wed, 28 Feb 2024 22:45:42 +0800 Subject: test(injector): fix tests after injector added --- tests/test_register.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tests/test_register.py') 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" -- cgit v1.2.3-70-g09d2