diff --git a/surfsense_backend/tests/integration/conftest.py b/surfsense_backend/tests/integration/conftest.py index 6b8aa3cdb..e67a025cc 100644 --- a/surfsense_backend/tests/integration/conftest.py +++ b/surfsense_backend/tests/integration/conftest.py @@ -158,13 +158,12 @@ def patched_embed_texts_raises(monkeypatch) -> MagicMock: @pytest.fixture def patched_chunk_text(monkeypatch) -> MagicMock: - mock = MagicMock(return_value=["Test chunk content."]) + from app.indexing_pipeline.document_chunker import ChunkSlice + + text = "Test chunk content." + mock = MagicMock(return_value=[ChunkSlice(text, 0, len(text))]) monkeypatch.setattr( - "app.indexing_pipeline.cache.cached_indexing.chunk_text", - mock, - ) - monkeypatch.setattr( - "app.indexing_pipeline.cache.cached_indexing.chunk_text_hybrid", + "app.indexing_pipeline.cache.cached_indexing.chunk_markdown_with_spans", mock, ) return mock