:recycle: refactor(output): add __init__ method in Output class in order to generate output mehtod in initialization

1 files changed, 14 insertions, 26 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 = [