opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-07-21 10:38:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: use sqlite3 session in test_app_dsl_service (#38715)

Browse Source
This commit is contained in:
Asuka Minato 2026-07-14 12:51:04 +09:00 committed by GitHub
parent 747773d4b6
commit e044292518
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
api/tests/unit_tests/services/test_app_dsl_service.py
View File

@ -1,29 +1,37 @@
from unittest.mock import Mock
import pytest
from sqlalchemy.orm import Session
from services.app_dsl_service import AppDslService
from services.entities.dsl_entities import ImportStatus
def test_import_app_rejects_oversized_yaml_content_before_parsing(monkeypatch: pytest.MonkeyPatch) -> None:
@pytest.mark.parametrize("sqlite_session", [()], indirect=True)
def test_import_app_rejects_oversized_yaml_content_before_parsing(
monkeypatch: pytest.MonkeyPatch, sqlite_session: Session
) -> None:
monkeypatch.setattr("services.app_dsl_service.DSL_MAX_SIZE", 3)
service = AppDslService(session=Mock())
service = AppDslService(session=sqlite_session)
account = Mock(current_tenant_id="tenant-1")
result = service.import_app(account=account, import_mode="yaml-content", yaml_content="你你")
assert result.status == ImportStatus.FAILED
assert result.error == "File size exceeds the limit of 10MB"
assert not sqlite_session.in_transaction()
def test_import_app_rejects_oversized_yaml_url_bytes_before_decode(monkeypatch: pytest.MonkeyPatch) -> None:
@pytest.mark.parametrize("sqlite_session", [()], indirect=True)
def test_import_app_rejects_oversized_yaml_url_bytes_before_decode(
monkeypatch: pytest.MonkeyPatch, sqlite_session: Session
) -> None:
monkeypatch.setattr("services.app_dsl_service.DSL_MAX_SIZE", 1)
response = Mock()
response.raise_for_status.return_value = None
response.content = b"\xff\xff"
monkeypatch.setattr("services.app_dsl_service.remote_fetcher.make_request", Mock(return_value=response))
service = AppDslService(session=Mock())
service = AppDslService(session=sqlite_session)
result = service.import_app(
account=Mock(current_tenant_id="tenant-1"),
@ -33,14 +41,18 @@ def test_import_app_rejects_oversized_yaml_url_bytes_before_decode(monkeypatch:
assert result.status == ImportStatus.FAILED
assert result.error == "File size exceeds the limit of 10MB"
assert not sqlite_session.in_transaction()
def test_import_app_returns_decode_error_for_invalid_yaml_url_bytes(monkeypatch: pytest.MonkeyPatch) -> None:
@pytest.mark.parametrize("sqlite_session", [()], indirect=True)
def test_import_app_returns_decode_error_for_invalid_yaml_url_bytes(
monkeypatch: pytest.MonkeyPatch, sqlite_session: Session
) -> None:
response = Mock()
response.raise_for_status.return_value = None
response.content = b"\xff"
monkeypatch.setattr("services.app_dsl_service.remote_fetcher.make_request", Mock(return_value=response))
service = AppDslService(session=Mock())
service = AppDslService(session=sqlite_session)
result = service.import_app(
account=Mock(current_tenant_id="tenant-1"),
@ -50,3 +62,4 @@ def test_import_app_returns_decode_error_for_invalid_yaml_url_bytes(monkeypatch:
assert result.status == ImportStatus.FAILED
assert "utf-8" in result.error
assert not sqlite_session.in_transaction()