fix test

api/tests/unit_tests/conftest.py

@@ -1,7 +1,24 @@
 import os
 
+import pytest
+from flask import Flask
+
 # Getting the absolute path of the current file's directory
 ABS_PATH = os.path.dirname(os.path.abspath(__file__))
 
 # Getting the absolute path of the project's root directory
 PROJECT_DIR = os.path.abspath(os.path.join(ABS_PATH, os.pardir, os.pardir))
+
+CACHED_APP = Flask(__name__)
+CACHED_APP.config.update({"TESTING": True})
+
+
+@pytest.fixture()
+def app() -> Flask:
+    return CACHED_APP
+
+
+@pytest.fixture(autouse=True)
+def _provide_app_context(app: Flask):
+    with app.app_context():
+        yield

api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py

@@ -1,7 +1,5 @@
 from unittest.mock import patch
 
-from flask import Flask
-
 from core.app.entities.app_invoke_entities import InvokeFrom
 from core.workflow.entities.node_entities import SystemVariable, UserFrom
 from core.workflow.entities.variable_pool import VariablePool
@@ -130,22 +128,19 @@ def test_run_parallel(mock_close, mock_remove):
 
     # print("")
 
-    app = Flask('test')
-
     items = []
-    with app.app_context():
-        generator = graph_engine.run()
-        for item in generator:
-            # print(type(item), item)
-            items.append(item)
-            if isinstance(item, NodeRunSucceededEvent):
-                assert item.route_node_state.status == RouteNodeState.Status.SUCCESS
+    generator = graph_engine.run()
+    for item in generator:
+        # print(type(item), item)
+        items.append(item)
+        if isinstance(item, NodeRunSucceededEvent):
+            assert item.route_node_state.status == RouteNodeState.Status.SUCCESS
 
-            assert not isinstance(item, NodeRunFailedEvent)
-            assert not isinstance(item, GraphRunFailedEvent)
+        assert not isinstance(item, NodeRunFailedEvent)
+        assert not isinstance(item, GraphRunFailedEvent)
 
-            if isinstance(item, BaseNodeEvent) and item.route_node_state.node_id in ['answer2', 'answer3']:
-                assert item.parallel_id is not None
+        if isinstance(item, BaseNodeEvent) and item.route_node_state.node_id in ['answer2', 'answer3']:
+            assert item.parallel_id is not None
 
     assert len(items) == 12
     assert isinstance(items[0], GraphRunStartedEvent)
@@ -290,14 +285,11 @@ def test_run_branch(mock_close, mock_remove):
 
     # print("")
 
-    app = Flask('test')
-
     items = []
-    with app.app_context():
-        generator = graph_engine.run()
-        for item in generator:
-            print(type(item), item)
-            items.append(item)
+    generator = graph_engine.run()
+    for item in generator:
+        print(type(item), item)
+        items.append(item)
 
     assert len(items) == 8
     assert items[3].route_node_state.node_id == 'if-else-1'