From 38b2d1005a73fcdd472d5cb42090a8e15d0208cf Mon Sep 17 00:00:00 2001 From: CuriousHet Date: Sat, 18 Apr 2026 10:12:26 +0530 Subject: [PATCH] test: fix librarian mocks after BlobStore async conversion --- tests/unit/test_librarian/test_chunked_upload.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/unit/test_librarian/test_chunked_upload.py b/tests/unit/test_librarian/test_chunked_upload.py index d0b73d48..eef83e1e 100644 --- a/tests/unit/test_librarian/test_chunked_upload.py +++ b/tests/unit/test_librarian/test_chunked_upload.py @@ -22,6 +22,10 @@ def _make_librarian(min_chunk_size=1): """Create a Librarian with mocked blob_store and table_store.""" lib = Librarian.__new__(Librarian) 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.load_document = AsyncMock() lib.min_chunk_size = min_chunk_size