aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests
diff options
context:
space:
mode:
author苏向夜 <fu050409@163.com>2024-01-26 15:04:21 +0800
committer苏向夜 <fu050409@163.com>2024-01-26 15:04:21 +0800
commit0250c8373c8ea12c6d12cf399a15c57f8690b032 (patch)
treecb988940f3769bf05608cfb03d2a4832c4ce4302 /tests
parent0e8be9de58454de079d772dec6c0ef9c1774a775 (diff)
downloadinfini-0250c8373c8ea12c6d12cf399a15c57f8690b032.tar.gz
infini-0250c8373c8ea12c6d12cf399a15c57f8690b032.zip
:recycle: refactor(output): add __init__ method in Output class in order to generate output mehtod in initialization
Diffstat (limited to 'tests')
-rw-r--r--tests/test_handlers.py40
-rw-r--r--tests/test_interceptor.py7
2 files changed, 15 insertions, 32 deletions
diff --git a/tests/test_handlers.py b/tests/test_handlers.py
index 8d2d2894..bf9dd9a0 100644
--- a/tests/test_handlers.py
+++ b/tests/test_handlers.py
@@ -8,21 +8,15 @@ def test_handler():
input = Input(".add 1 2")
def add(input: Input) -> Output:
- a, b = map(int, input.get_plain_text().lstrip(".add").split())
- output = Output()
- output.block = False
- output.status = 0
- output.type = "text"
- output.name = str(a + b)
- return output
+ return Output(
+ "text",
+ str(sum(list(map(int, input.get_plain_text().lstrip(".add").split())))),
+ status=0,
+ block=False,
+ )
def cmd(_: Input) -> Output:
- output = Output()
- output.block = False
- output.status = 0
- output.type = "text"
- output.name = "cmd"
- return output
+ return Output("text", "cmd", status=0, block=False)
handler = Handler()
handler.handlers = [
@@ -48,21 +42,15 @@ def test_handler_block():
input = Input(".add 1 2")
def add(input: Input) -> Output:
- a, b = map(int, input.get_plain_text().lstrip(".add").split())
- output = Output()
- output.block = False
- output.status = 0
- output.type = "text"
- output.name = str(a + b)
- return output
+ return Output(
+ "text",
+ str(sum(list(map(int, input.get_plain_text().lstrip(".add").split())))),
+ status=0,
+ block=False,
+ )
def cmd(_: Input) -> Output:
- output = Output()
- output.block = True
- output.status = 0
- output.type = "text"
- output.name = "cmd"
- return output
+ return Output("text", "cmd", status=0, block=True)
handler = Handler()
handler.handlers = [
diff --git a/tests/test_interceptor.py b/tests/test_interceptor.py
index df5b6750..fb12beac 100644
--- a/tests/test_interceptor.py
+++ b/tests/test_interceptor.py
@@ -9,12 +9,7 @@ def test_interceptor():
valid_input = Input("这个叫苏向夜.")
def intercept(_: Input) -> Input | Output:
- output = Output()
- output.block = True # TODO 拦截器阻塞标识
- output.name = "block.jianlvchun"
- output.status = 0
- output.type = "text"
- return output
+ return Output("text", "block.jianlvchun", block=True) # TODO 拦截器阻塞标识
interceptor = Interceptor()
interceptor.interceptors = [