apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-25 16:56:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: enhance file classification tests for Azure DI configuration

Browse source
This commit is contained in:
Anish Sarkar 2026-04-08 05:13:17 +05:30
parent 6038f6dfc0
commit d072ca60bb
1 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

17
surfsense_backend/tests/unit/etl_pipeline/test_etl_pipeline_service.py
View file

@ -609,7 +609,9 @@ async def test_extract_zip_raises_unsupported_error(tmp_path):
("file.heif", "UNSTRUCTURED", True),
],
)
def test_should_skip_for_service(filename, etl_service, expected_skip):
def test_should_skip_for_service(filename, etl_service, expected_skip, monkeypatch):
monkeypatch.setattr("app.config.config.AZURE_DI_ENDPOINT", None, raising=False)
monkeypatch.setattr("app.config.config.AZURE_DI_KEY", None, raising=False)
from app.etl_pipeline.file_classifier import should_skip_for_service
assert should_skip_for_service(filename, etl_service) is expected_skip, (
@ -617,6 +619,19 @@ def test_should_skip_for_service(filename, etl_service, expected_skip):
)
def test_heif_not_skipped_for_llamacloud_when_azure_di_configured(monkeypatch):
"""With Azure DI credentials, .heif is accepted by LLAMACLOUD."""
monkeypatch.setattr(
"app.config.config.AZURE_DI_ENDPOINT",
"https://fake.cognitiveservices.azure.com/",
raising=False,
)
monkeypatch.setattr("app.config.config.AZURE_DI_KEY", "fake-key", raising=False)
from app.etl_pipeline.file_classifier import should_skip_for_service
assert should_skip_for_service("file.heif", "LLAMACLOUD") is False
# ---------------------------------------------------------------------------
# Slice 14b - ETL pipeline rejects per-parser incompatible documents
# ---------------------------------------------------------------------------