From 766dc74a04bfd8f789b42d892d71ea33cc4c5a42 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Sat, 27 Jan 2024 19:08:18 +0800 Subject: :white_check_mark: test(refact): sync tests with infini changes --- tests/test_interceptor.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) (limited to 'tests/test_interceptor.py') diff --git a/tests/test_interceptor.py b/tests/test_interceptor.py index fb12beac..afb135e9 100644 --- a/tests/test_interceptor.py +++ b/tests/test_interceptor.py @@ -9,7 +9,7 @@ def test_interceptor(): valid_input = Input("这个叫苏向夜.") def intercept(_: Input) -> Input | Output: - return Output("text", "block.jianlvchun", block=True) # TODO 拦截器阻塞标识 + return Output("text", "block.jianlvchun", block=True) interceptor = Interceptor() interceptor.interceptors = [ @@ -19,10 +19,18 @@ def test_interceptor(): "handler": intercept, } ] - output = interceptor.input(input) - assert isinstance(output, Output) - assert output.name == "block.jianlvchun" + for output in interceptor.input(input): + assert isinstance(output, Output) + assert output.name == "block.jianlvchun" - valid_output = interceptor.input(valid_input) - assert isinstance(valid_output, Input) - assert valid_output.get_plain_text() == "这个叫苏向夜." + for valid_output in interceptor.input(valid_input): + assert isinstance(valid_output, Input) + assert valid_output.get_plain_text() == "这个叫苏向夜." + + for output in interceptor.output("简律纯"): + assert isinstance(output, Output) + assert output.name == "block.jianlvchun" + + for output in interceptor.output("这个叫苏向夜."): + assert isinstance(output, str) + assert output == "这个叫苏向夜." -- cgit v1.2.3-70-g09d2