apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-14 07:42:11 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

test: fix librarian mocks after BlobStore async conversion

Browse source
This commit is contained in:
CuriousHet 2026-04-18 10:12:26 +05:30
parent f09b78aa21
commit 38b2d1005a
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/unit/test_librarian/test_chunked_upload.py
View file

@ -22,6 +22,10 @@ def _make_librarian(min_chunk_size=1):
"""Create a Librarian with mocked blob_store and table_store.""" """Create a Librarian with mocked blob_store and table_store."""
lib = Librarian.__new__(Librarian) lib = Librarian.__new__(Librarian)
lib.blob_store = MagicMock() lib.blob_store = MagicMock()
lib.blob_store.create_multipart_upload = AsyncMock()
lib.blob_store.upload_part = AsyncMock()
lib.blob_store.complete_multipart_upload = AsyncMock()
lib.blob_store.abort_multipart_upload = AsyncMock()
lib.table_store = AsyncMock() lib.table_store = AsyncMock()
lib.load_document = AsyncMock() lib.load_document = AsyncMock()
lib.min_chunk_size = min_chunk_size lib.min_chunk_size = min_chunk_size