Skip to content

Commit c99a4b2

Browse files
committed
Fix: Mocking test
Solutions: Use AsyncMock instead of Magic Mock
1 parent 404455a commit c99a4b2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/unit/test_upload.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import hashlib
2-
from unittest.mock import MagicMock, patch
2+
from unittest.mock import AsyncMock, patch
33

44
import pytest
55
from aleph_message.models import AlephMessage, StoreMessage
@@ -18,7 +18,7 @@ def mock_authenticated_aleph_http_client():
1818
pkey = get_fallback_private_key()
1919
account = ETHAccount(private_key=pkey)
2020

21-
http_session = MagicMock()
21+
http_session = AsyncMock()
2222
mock_client = MockHttpClient.return_value
2323
mock_client.http_session = http_session
2424
mock_client.account = account
@@ -30,7 +30,7 @@ async def test_upload_with_message(mock_authenticated_aleph_http_client):
3030
content = b"Test pyaleph upload\n"
3131
file_hash = hashlib.sha256(content).hexdigest()
3232

33-
message = MagicMock()
33+
message = AsyncMock()
3434
message.content.item_hash = file_hash
3535
status = MessageStatus.PROCESSED
3636
mock_authenticated_aleph_http_client.create_store.return_value = (message, status)

0 commit comments

Comments
 (0)