From 99ed826a55ee35082039975e8ec91a7338d9f286 Mon Sep 17 00:00:00 2001 From: Asuka Minato Date: Sat, 25 Jul 2026 23:30:01 +0900 Subject: [PATCH] test: use sqlite3 session in test_pipeline_template_base (#38730) --- .../pipeline_template/test_pipeline_template_base.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/api/tests/unit_tests/services/rag_pipeline/pipeline_template/test_pipeline_template_base.py b/api/tests/unit_tests/services/rag_pipeline/pipeline_template/test_pipeline_template_base.py index 17cd5db7ab3..25472552e8e 100644 --- a/api/tests/unit_tests/services/rag_pipeline/pipeline_template/test_pipeline_template_base.py +++ b/api/tests/unit_tests/services/rag_pipeline/pipeline_template/test_pipeline_template_base.py @@ -1,4 +1,5 @@ -from unittest.mock import Mock +import pytest +from sqlalchemy.orm import Session from services.rag_pipeline.pipeline_template.pipeline_template_base import PipelineTemplateRetrievalBase @@ -16,10 +17,11 @@ class DummyRetrieval(PipelineTemplateRetrievalBase): return "dummy" -def test_pipeline_template_retrieval_base_concrete_implementation() -> None: +@pytest.mark.parametrize("sqlite_session", [()], indirect=True) +def test_pipeline_template_retrieval_base_concrete_implementation(sqlite_session: Session) -> None: retrieval = DummyRetrieval() - session = Mock() - assert retrieval.get_pipeline_templates("en-US", session=session) == {"language": "en-US"} - assert retrieval.get_pipeline_template_detail("tpl-1", session=session) == {"id": "tpl-1"} + assert retrieval.get_pipeline_templates("en-US", session=sqlite_session) == {"language": "en-US"} + assert retrieval.get_pipeline_template_detail("tpl-1", session=sqlite_session) == {"id": "tpl-1"} assert retrieval.get_type() == "dummy" + assert not sqlite_session.in_transaction()