From 7ad77e9e774b912d130b6afb4924149eb2d25fb2 Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 18 Jul 2024 08:19:58 +0800 Subject: [PATCH] fix test --- api/tests/unit_tests/conftest.py | 17 +++++++++ .../graph_engine/test_graph_engine.py | 36 ++++++++----------- 2 files changed, 31 insertions(+), 22 deletions(-) diff --git a/api/tests/unit_tests/conftest.py b/api/tests/unit_tests/conftest.py index afc9802cf1..ca3082953a 100644 --- a/api/tests/unit_tests/conftest.py +++ b/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 diff --git a/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py b/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py index b59f4af6a1..69362b7e1b 100644 --- a/api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py +++ b/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'