opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-08-02 18:56:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: use sqlite3 session in test_pipeline_template_base (#38730)

Browse Source
This commit is contained in:
Asuka Minato 2026-07-25 23:30:01 +09:00 committed by GitHub
parent a45f862fb2
commit 99ed826a55
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/tests/unit_tests/services/rag_pipeline/pipeline_template/test_pipeline_template_base.py
View File

@ -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()