aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test_interceptor.py
diff options
context:
space:
mode:
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 == "这个叫苏向夜."