From f195a1e7396b7eaa23ef6f14ffebb700cf741c75 Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Mon, 4 Mar 2024 17:56:12 +0800
Subject: test(workflow): add tests for workflow feature

---
 tests/test_workflow.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 tests/test_workflow.py

(limited to 'tests/test_workflow.py')

diff --git a/tests/test_workflow.py b/tests/test_workflow.py
new file mode 100644
index 00000000..18dab9d5
--- /dev/null
+++ b/tests/test_workflow.py
@@ -0,0 +1,47 @@
+from infini.core import Core
+from infini.generator import Generator
+from infini.handler import Handler
+from infini.injector import Injector
+from infini.input import Input
+from infini.interceptor import Interceptor
+from infini.output import Output
+from infini.router import Startswith
+
+
+def test_workflow():
+    def func_workflow(input: Input):
+        yield input.output("workflow", "test.workflow", block=True)
+
+    input = Input("testmsg")
+
+    handler = Handler()
+    handler.handlers = [
+        {
+            "priority": 0,
+            "router": Startswith(""),
+            "handler": func_workflow,
+        }
+    ]
+
+    interceptor = Interceptor()
+    interceptor.interceptors = []
+
+    generator = Generator()
+    generator.events = {
+        "test.cmd": "cmd",
+        "test.add": "{{ result }}",
+        "block.jianlvchun": "检测到违禁词",
+    }
+    generator.global_variables = {}
+
+    core = Core()
+    core.handler = handler
+    core.interceptor = interceptor
+    core.pre_interceptor = interceptor
+    core.generator = generator
+    core.injector = Injector()
+
+    for output in core.input(input):
+        assert isinstance(output, Output)
+        assert output.type == "workflow"
+        output.status = 0
-- 
cgit v1.2.3-70-g09d2