diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_handlers.py | 40 | ||||
| -rw-r--r-- | tests/test_interceptor.py | 7 |
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 = [ |