Feature: missing get_balances from sdk (with new field)

Author: 1yam

src/aleph/sdk/client/http.py

@@ -49,10 +49,11 @@
     RemovedMessageError,
     ResourceNotFoundError,
 )
-from ..query.filters import CreditsFilter, MessageFilter, PostFilter
+from ..query.filters import BalanceFilter, MessageFilter, PostFilter
 from ..query.responses import (
     AddressCreditResponse,
-    CreditsResponse,
+    BalanceResponse,
+    CreditsHistoryResponse,
     MessagesResponse,
     Post,
     PostsResponse,
@@ -629,3 +630,17 @@ async def get_credit_history(
             resp.raise_for_status()
             result = await resp.json()
             return CreditsHistoryResponse.model_validate(result)
+
+    async def get_balances(
+        self,
+        address: str,
+        filter: Optional[BalanceFilter] = None,
+    ) -> BalanceResponse:
+
+        async with self.http_session.get(
+            f"/api/v0/addresses/{address}/balance",
+            params=filter.as_http_params() if filter else None,
+        ) as resp:
+            resp.raise_for_status()
+            result = await resp.json()
+            return BalanceResponse.model_validate(result)

src/aleph/sdk/query/responses.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import datetime as dt
+from decimal import Decimal
 from typing import Any, Dict, List, Optional, Union
 
 from aleph_message.models import (
@@ -112,4 +114,9 @@ class CreditHistoryResponseItem(BaseModel):
     message_timestamp: dt.datetime
 
 
-
+class BalanceResponse(BaseModel):
+    address: str
+    balance: Decimal
+    details: Optional[Dict[str, Decimal]] = None
+    locked_amount: Decimal
+    credit_balance: int = 0

tests/unit/test_balance.py

@@ -0,0 +1,39 @@
+from unittest.mock import patch
+
+import pytest
+
+from aleph.sdk.query.responses import BalanceResponse
+from tests.unit.conftest import make_mock_get_session
+
+
+@pytest.mark.asyncio
+async def test_get_balances():
+    """
+    Test that the get_balances method returns the correct BalanceResponse
+    for a specific address when called on the AlephHttpClient.
+    """
+    address = "0xd463495a6FEaC9921FD0C3a595B81E7B2C02B24d"
+
+    balance_data = {
+        "address": address,
+        "balance": 351.25,
+        "details": {"ETH": 100.5, "SOL": 250.75},
+        "locked_amount": 50.0,
+        "credit_balance": 1000,
+    }
+
+    mock_client = make_mock_get_session(balance_data)
+
+    expected_url = f"/api/v0/addresses/{address}/balance"
+    # Adding type assertion to handle None case
+    assert mock_client._http_session is not None
+    with patch.object(
+        mock_client._http_session, "get", wraps=mock_client._http_session.get
+    ) as spy:
+        async with mock_client:
+            response = await mock_client.get_balances(address)
+
+            # Verify the response
+            assert isinstance(response, BalanceResponse)
+            # Verify the balances command calls the correct URL
+            spy.assert_called_once_with(expected_url, params=None)