From 9f17e1ca9544fcffce7d3059100b1e787e10b0b6 Mon Sep 17 00:00:00 2001 From: Asuka Minato Date: Thu, 30 Jul 2026 13:32:23 +0900 Subject: [PATCH] test: move RAG pipeline import coverage to unit tests (#38938) --- .../rag_pipeline/test_rag_pipeline_import.py | 32 ++++++++----------- 1 file changed, 14 insertions(+), 18 deletions(-) rename api/tests/{test_containers_integration_tests => unit_tests}/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py (93%) diff --git a/api/tests/test_containers_integration_tests/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py b/api/tests/unit_tests/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py similarity index 93% rename from api/tests/test_containers_integration_tests/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py rename to api/tests/unit_tests/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py index b9a0029d131..248c6c210a8 100644 --- a/api/tests/test_containers_integration_tests/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py +++ b/api/tests/unit_tests/controllers/console/datasets/rag_pipeline/test_rag_pipeline_import.py @@ -1,7 +1,9 @@ -"""Testcontainers integration tests for rag_pipeline_import controller endpoints.""" +"""Unit tests for rag_pipeline_import controller endpoints.""" from __future__ import annotations +from collections.abc import Iterator +from inspect import unwrap from unittest.mock import MagicMock, patch import pytest @@ -16,16 +18,22 @@ from controllers.console.datasets.rag_pipeline.rag_pipeline_import import ( ) from core.plugin.entities.plugin import PluginDependency, PluginDependencyType from models.dataset import Pipeline +from models.engine import db from services.entities.dsl_entities import CheckDependenciesResult, ImportStatus from services.rag_pipeline.rag_pipeline_dsl_service import RagPipelineImportInfo -from tests.test_containers_integration_tests.controllers.console.helpers import unwrap + + +@pytest.fixture +def app() -> Iterator[Flask]: + app = Flask(__name__) + app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:" + db.init_app(app) + + with app.app_context(): + yield app class TestRagPipelineImportApi: - @pytest.fixture - def app(self, flask_app_with_containers: Flask) -> Flask: - return flask_app_with_containers - def _payload(self, mode: str = "create") -> dict[str, str]: return { "mode": mode, @@ -140,10 +148,6 @@ class TestRagPipelineImportApi: class TestRagPipelineImportConfirmApi: - @pytest.fixture - def app(self, flask_app_with_containers: Flask) -> Flask: - return flask_app_with_containers - def test_confirm_success(self, app: Flask) -> None: api = RagPipelineImportConfirmApi() method = unwrap(api.post) @@ -205,10 +209,6 @@ class TestRagPipelineImportConfirmApi: class TestRagPipelineImportCheckDependenciesApi: - @pytest.fixture - def app(self, flask_app_with_containers: Flask) -> Flask: - return flask_app_with_containers - def test_get_success(self, app: Flask) -> None: api = RagPipelineImportCheckDependenciesApi() method = unwrap(api.get) @@ -272,10 +272,6 @@ class TestRagPipelineImportCheckDependenciesApi: class TestRagPipelineExportApi: - @pytest.fixture - def app(self, flask_app_with_containers: Flask) -> Flask: - return flask_app_with_containers - def test_get_with_include_secret(self, app: Flask) -> None: api = RagPipelineExportApi() method = unwrap(api.get)