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

refactor: enhance Google Drive indexer to support file extension filtering, improving file handling and error reporting

Browse source
This commit is contained in:
Anish Sarkar 2026-04-06 22:34:49 +05:30
parent 0fb92b7c56
commit f03bf05aaa
3 changed files with 46 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -272,6 +272,23 @@ def full_scan_mocks(mock_dropbox_client, monkeypatch):
download_and_index_mock = AsyncMock(return_value=(0, 0))
monkeypatch.setattr(_mod, "_download_and_index", download_and_index_mock)
from app.services.page_limit_service import PageLimitService as _RealPLS
mock_page_limit_instance = MagicMock()
mock_page_limit_instance.get_page_usage = AsyncMock(return_value=(0, 999_999))
mock_page_limit_instance.update_page_usage = AsyncMock()
class _MockPageLimitService:
estimate_pages_from_metadata = staticmethod(
_RealPLS.estimate_pages_from_metadata
)
def __init__(self, session):
self.get_page_usage = mock_page_limit_instance.get_page_usage
self.update_page_usage = mock_page_limit_instance.update_page_usage
monkeypatch.setattr(_mod, "PageLimitService", _MockPageLimitService)
return {
"dropbox_client": mock_dropbox_client,
"session": mock_session,
@ -377,6 +394,23 @@ def selected_files_mocks(mock_dropbox_client, monkeypatch):
download_and_index_mock = AsyncMock(return_value=(0, 0))
monkeypatch.setattr(_mod, "_download_and_index", download_and_index_mock)
from app.services.page_limit_service import PageLimitService as _RealPLS
mock_page_limit_instance = MagicMock()
mock_page_limit_instance.get_page_usage = AsyncMock(return_value=(0, 999_999))
mock_page_limit_instance.update_page_usage = AsyncMock()
class _MockPageLimitService:
estimate_pages_from_metadata = staticmethod(
_RealPLS.estimate_pages_from_metadata
)
def __init__(self, session):
self.get_page_usage = mock_page_limit_instance.get_page_usage
self.update_page_usage = mock_page_limit_instance.update_page_usage
monkeypatch.setattr(_mod, "PageLimitService", _MockPageLimitService)
return {
"dropbox_client": mock_dropbox_client,
"session": mock_session,