aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test_interceptor.py
diff options
context:
space:
mode:
authorHsiangNianian <i@jyunko.cn>2025-08-02 11:30:27 +0800
committerHsiangNianian <i@jyunko.cn>2025-08-02 11:30:27 +0800
commitec4566c3d17370c778a1e6cd6b22ed85263731a0 (patch)
tree469258084d3ae0e72db37dd98593731ff20131e4 /tests/test_interceptor.py
parentc68a18ca6440f6460b3b5fce311901229f8f50a9 (diff)
downloadinfini-ec4566c3d17370c778a1e6cd6b22ed85263731a0.tar.gz
infini-ec4566c3d17370c778a1e6cd6b22ed85263731a0.zip
refactor(v3): publish v3 branch
Diffstat (limited to 'tests/test_interceptor.py')
-rw-r--r--tests/test_interceptor.py38
1 files changed, 0 insertions, 38 deletions
diff --git a/tests/test_interceptor.py b/tests/test_interceptor.py
deleted file mode 100644
index ded63471..00000000
--- a/tests/test_interceptor.py
+++ /dev/null
@@ -1,38 +0,0 @@
-from infini.input import Input
-from infini.interceptor import Interceptor
-from infini.output import Output
-from infini.router import Contains
-
-
-def test_interceptor():
- input = Input("这个人叫简律纯.")
- valid_input = Input("这个叫苏向夜.")
-
- def intercept(_: Input) -> Input | Output:
- return Output("text", "block.jianlvchun", block=True)
-
- interceptor = Interceptor()
- interceptor.interceptors = [
- {
- "priority": 1,
- "router": Contains("简律纯"),
- "handler": intercept,
- }
- ]
- for output in interceptor.input(input):
- assert isinstance(output, Output)
- assert output.name == "block.jianlvchun"
-
- for valid_output in interceptor.input(valid_input):
- assert isinstance(valid_output, Input)
- assert valid_output.get_plain_text() == "这个叫苏向夜."
-
- for output in interceptor.output(Output("text", "none", block=True), "简律纯"):
- assert isinstance(output, Output)
- assert output.name == "block.jianlvchun"
-
- for output in interceptor.output(
- Output("text", "none", block=True), "这个叫苏向夜."
- ):
- assert isinstance(output, str)
- assert output == "这个叫苏向夜."