Add test for handling an InsufficientFundsError while creating an instance.

Author: MHHukiewitz

tests/unit/conftest.py

@@ -57,6 +57,13 @@ def json_messages():
         return json.load(f)
 
 
+@pytest.fixture
+def rejected_message():
+    message_path = Path(__file__).parent / "rejected_message.json"
+    with open(message_path) as f:
+        return json.load(f)
+
+
 @pytest.fixture
 def aleph_messages() -> List[AlephMessage]:
     return [
@@ -213,3 +220,23 @@ def get(self, *_args, **_kwargs):
     client.http_session = http_session
 
     return client
+
+
+@pytest.fixture
+def mock_session_with_rejected_message(ethereum_account, rejected_message) -> AuthenticatedAlephHttpClient:
+    class MockHttpSession(AsyncMock):
+        def get(self, *_args, **_kwargs):
+            return make_custom_mock_response(rejected_message)
+
+        def post(self, *_args, **_kwargs):
+            return make_custom_mock_response({
+                "message_status": "rejected",
+                "publication_status": {"status": "success", "failed": []},
+            }, status=422)
+
+    http_session = MockHttpSession()
+
+    client = AuthenticatedAlephHttpClient(account=ethereum_account, api_server="http://localhost")
+    client.http_session = http_session
+
+    return client

tests/unit/rejected_message.json

@@ -0,0 +1,53 @@
+{
+  "status": "rejected",
+  "item_hash": "7091b61632e0385adb6978dccbec1da40e9c400ffac7233076a3ad0219f49de0",
+  "reception_time": "2023-12-03T16:00:53.432607+00:00",
+  "message": {
+    "time": 1701619253.309467,
+    "type": "INSTANCE",
+    "chain": "ETH",
+    "sender": "0x48A87924135892bEE2F264547a0D69909b3bA5C6",
+    "channel": null,
+    "content": {
+      "time": 1701619253.3092096,
+      "rootfs": {
+        "parent": {
+          "ref": "549ec451d9b099cad112d4aaa2c00ac40fb6729a92ff252ff22eef0b5c3cb613",
+          "use_latest": true
+        },
+        "size_mib": 2000,
+        "persistence": "host"
+      },
+      "address": "0x48A87924135892bEE2F264547a0D69909b3bA5C6",
+      "volumes": [],
+      "resources": {
+        "vcpus": 1,
+        "memory": 256,
+        "seconds": 30
+      },
+      "allow_amend": false,
+      "environment": {
+        "internet": true,
+        "aleph_api": true,
+        "reproducible": false,
+        "shared_cache": false
+      },
+      "authorized_keys": [
+        "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQCdU/B01GaUQ3l8aIt0XR6WajhM/5oTdxsON3b90quwQjH38Ba+Z3z74gf4+f56vxRw7XqtC8pg8zuh2L8+Fo3IHctULJD24YbLr2CagwrPsd9fsK+DsJSrcnTKaBBjvgloQi96hl4hPtlR9jA5bRYked0bFxdbFUMCmF17tpU0AC/dLeH3FwVDfh5dF11SX6ezg8brKd9SHYDRkOBHyOLIOWMtOgIgfEy9OsNdAS9OlOHyKxz4MEl5xkq5lRk9nS1n45A3rrIwi91qamjphHNrEutKaN/ramuW+davCP4xfnEVyu2DmoteDM+lmmf4skN90CwxhlrYpc5xpie5vTTYxgUCSnN0PyYiD0m7YHpZwMg1yHKB+aZghOIYUyKVJ8fIH/uU1DKwqcZ5qmH8sPvB4tCy3QOn7m7xTYVd/R3CECdG/R/55IUfGbg8KoAHBgmQ13/hSJ3IshMYOTWwMenma29D8UqKsyTfiMfof6Yf6NYOeDkAsUSIOEetxcv3MaE= mhh@mhh-TUXEDO"
+      ]
+    },
+    "item_hash": "7091b61632e0385adb6978dccbec1da40e9c400ffac7233076a3ad0219f49de0",
+    "item_type": "inline",
+    "signature": "0xa8e549c5228b379bede875016ed25215c6cd7c3e9131ec1b3bdb49af4859c8840251362c3c421995f520648026621245b4896c0a88d6a5735e5e2620812f22ff1b",
+    "item_content": "{\"address\":\"0x48A87924135892bEE2F264547a0D69909b3bA5C6\",\"time\":1701619253.3092096,\"allow_amend\":false,\"authorized_keys\":[\"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQCdU/B01GaUQ3l8aIt0XR6WajhM/5oTdxsON3b90quwQjH38Ba+Z3z74gf4+f56vxRw7XqtC8pg8zuh2L8+Fo3IHctULJD24YbLr2CagwrPsd9fsK+DsJSrcnTKaBBjvgloQi96hl4hPtlR9jA5bRYked0bFxdbFUMCmF17tpU0AC/dLeH3FwVDfh5dF11SX6ezg8brKd9SHYDRkOBHyOLIOWMtOgIgfEy9OsNdAS9OlOHyKxz4MEl5xkq5lRk9nS1n45A3rrIwi91qamjphHNrEutKaN/ramuW+davCP4xfnEVyu2DmoteDM+lmmf4skN90CwxhlrYpc5xpie5vTTYxgUCSnN0PyYiD0m7YHpZwMg1yHKB+aZghOIYUyKVJ8fIH/uU1DKwqcZ5qmH8sPvB4tCy3QOn7m7xTYVd/R3CECdG/R/55IUfGbg8KoAHBgmQ13/hSJ3IshMYOTWwMenma29D8UqKsyTfiMfof6Yf6NYOeDkAsUSIOEetxcv3MaE= mhh@mhh-TUXEDO\"],\"environment\":{\"reproducible\":false,\"internet\":true,\"aleph_api\":true,\"shared_cache\":false},\"resources\":{\"vcpus\":1,\"memory\":256,\"seconds\":30},\"volumes\":[],\"rootfs\":{\"parent\":{\"ref\":\"549ec451d9b099cad112d4aaa2c00ac40fb6729a92ff252ff22eef0b5c3cb613\",\"use_latest\":true},\"persistence\":\"host\",\"size_mib\":2000}}"
+  },
+  "error_code": 5,
+  "details": {
+    "errors": [
+      {
+        "account_balance": "0",
+        "required_balance": "4213.265726725260407192763523"
+      }
+    ]
+  }
+}

tests/unit/test_asynchronous.py

@@ -1,16 +1,20 @@
+import datetime
 from unittest.mock import AsyncMock
 
 import pytest as pytest
 from aleph_message.models import (
     AggregateMessage,
     ForgetMessage,
     InstanceMessage,
+    MessageType,
     PostMessage,
     ProgramMessage,
     StoreMessage,
 )
+from aleph_message.models.execution.environment import MachineResources
 from aleph_message.status import MessageStatus
 
+from aleph.sdk.exceptions import InsufficientFundsError
 from aleph.sdk.types import StorageEnum
 
 
@@ -121,3 +125,78 @@ async def test_forget(mock_session_with_post_success):
 
     assert mock_session_with_post_success.http_session.post.called_once
     assert isinstance(forget_message, ForgetMessage)
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "message_type, content",
+    [
+        (
+            MessageType.aggregate,
+            {
+                "content": {"Hello": datetime.datetime.now()},
+                "key": "test",
+                "address": "0x1",
+                "time": 1.0,
+            },
+        ),
+        (
+            MessageType.aggregate,
+            {
+                "content": {"Hello": datetime.date.today()},
+                "key": "test",
+                "address": "0x1",
+                "time": 1.0,
+            },
+        ),
+        (
+            MessageType.aggregate,
+            {
+                "content": {"Hello": datetime.time()},
+                "key": "test",
+                "address": "0x1",
+                "time": 1.0,
+            },
+        ),
+        (
+            MessageType.aggregate,
+            {
+                "content": {
+                    "Hello": MachineResources(
+                        vcpus=1,
+                        memory=1024,
+                        seconds=1,
+                    )
+                },
+                "key": "test",
+                "address": "0x1",
+                "time": 1.0,
+            },
+        ),
+    ],
+)
+async def test_prepare_aleph_message(
+    mock_session_with_post_success, message_type, content
+):
+    # Call the function under test
+    async with mock_session_with_post_success as session:
+        await session._prepare_aleph_message(
+            message_type=message_type,
+            content=content,
+            channel="TEST",
+        )
+
+
+@pytest.mark.asyncio
+async def test_create_instance_insufficient_funds_error(
+    mock_session_with_rejected_message
+):
+    async with mock_session_with_rejected_message as session:
+        with pytest.raises(InsufficientFundsError):
+            await session.create_instance(
+                rootfs="cafecafecafecafecafecafecafecafecafecafecafecafecafecafecafecafe",
+                rootfs_size=1,
+                rootfs_name="rootfs",
+                channel="TEST",
+                metadata={"tags": ["test"]},
+            )

tests/unit/test_asynchronous_get.py

@@ -4,7 +4,7 @@
 import pytest
 from aleph_message.models import MessagesResponse, MessageType
 
-from aleph.sdk.exceptions import ForgottenMessageError
+from aleph.sdk.exceptions import ForgottenMessageError, InvalidMessageError
 from aleph.sdk.query.filters import MessageFilter, PostFilter
 from aleph.sdk.query.responses import PostsResponse
 from tests.unit.conftest import make_mock_get_session
@@ -85,5 +85,15 @@ async def test_get_forgotten_message():
             await session.get_message("cafebabe")
 
 
+@pytest.mark.asyncio
+async def test_get_message_error(rejected_message):
+    mock_session = make_mock_get_session(
+        rejected_message
+    )
+    async with mock_session as session:
+        error = await session.get_message_error(rejected_message["item_hash"])
+        assert error["error_code"] == rejected_message["error_code"]
+        assert error["details"] == rejected_message["details"]
+
 if __name__ == "__main __":
     unittest.main()