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,10 +128,7 @@ 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)
@@ -290,10 +285,7 @@ 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)