apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-28 18:36:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fix test mocks for vision_llm kwarg

Browse source
This commit is contained in:
CREDO23 2026-04-10 18:20:49 +02:00
parent a95bf58c8f
commit f556446d07
4 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
surfsense_backend/tests/unit/connector_indexers/test_dropbox_parallel.py
View file

@ -168,7 +168,7 @@ async def test_concurrency_bounded_by_semaphore(
active = 0
peak = 0
async def _slow_extract(client, file):
async def _slow_extract(client, file, **kwargs):
nonlocal active, peak
async with lock:
active += 1
@ -209,7 +209,7 @@ async def test_heartbeat_fires_during_parallel_downloads(
monkeypatch.setattr(_mod, "HEARTBEAT_INTERVAL_SECONDS", 0)
async def _slow_extract(client, file):
async def _slow_extract(client, file, **kwargs):
await asyncio.sleep(0.05)
return _mock_extract_ok(file["id"], file["name"])

4
surfsense_backend/tests/unit/connector_indexers/test_google_drive_parallel.py
View file

@ -162,7 +162,7 @@ async def test_concurrency_bounded_by_semaphore(
active = 0
peak = 0
async def _slow_extract(client, file):
async def _slow_extract(client, file, **kwargs):
nonlocal active, peak
async with lock:
active += 1
@ -204,7 +204,7 @@ async def test_heartbeat_fires_during_parallel_downloads(
monkeypatch.setattr(_mod, "HEARTBEAT_INTERVAL_SECONDS", 0)
async def _slow_extract(client, file):
async def _slow_extract(client, file, **kwargs):
await asyncio.sleep(0.05)
return _mock_extract_ok(file["id"], file["name"])

4
surfsense_backend/tests/unit/connector_indexers/test_onedrive_parallel.py
View file

@ -162,7 +162,7 @@ async def test_concurrency_bounded_by_semaphore(
active = 0
peak = 0
async def _slow_extract(client, file):
async def _slow_extract(client, file, **kwargs):
nonlocal active, peak
async with lock:
active += 1
@ -203,7 +203,7 @@ async def test_heartbeat_fires_during_parallel_downloads(
monkeypatch.setattr(_mod, "HEARTBEAT_INTERVAL_SECONDS", 0)
async def _slow_extract(client, file):
async def _slow_extract(client, file, **kwargs):
await asyncio.sleep(0.05)
return _mock_extract_ok(file["id"], file["name"])

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

@ -431,7 +431,7 @@ async def test_llamacloud_heif_accepted_only_with_azure_di(tmp_path, mocker):
mocker.patch("app.config.config.AZURE_DI_ENDPOINT", None, create=True)
mocker.patch("app.config.config.AZURE_DI_KEY", None, create=True)
with pytest.raises(EtlUnsupportedFileError, match="not supported by LLAMACLOUD"):
with pytest.raises(EtlUnsupportedFileError, match="document parser does not support this format"):
await EtlPipelineService().extract(
EtlRequest(file_path=str(heif_file), filename="photo.heif")
)