Fix: Examples and Unit Test

Signed-off-by: aceppaluni <aceppaluni@gmail.com>

Author: aceppaluni

examples/query/account_balance_query.py

@@ -9,13 +9,16 @@
 ...
 """
 
+import json
+import os
 import sys
 import time
+from urllib.request import Request, urlopen
 
 from hiero_sdk_python import (
     AccountCreateTransaction,
+    AccountId,
     Client,
-    CryptoGetAccountBalanceQuery,
     Hbar,
     PrivateKey,
     ResponseCode,
@@ -45,6 +48,22 @@ def setup_client():
         sys.exit(1)
 
 
+def get_mirror_node_url():
+    """Return the Mirror Node URL for the configured Hedera network."""
+    network = os.getenv("HEDERA_NETWORK", "testnet").lower()
+
+    mirror_node_urls = {
+        "mainnet": "https://mainnet-public.mirrornode.hedera.com",
+        "testnet": "https://testnet.mirrornode.hedera.com",
+        "previewnet": "https://previewnet.mirrornode.hedera.com",
+    }
+
+    if network not in mirror_node_urls:
+        raise ValueError(f"Unsupported HEDERA_NETWORK: {network}. Expected mainnet, testnet, or previewnet.")
+
+    return mirror_node_urls[network]
+
+
 def create_account(client, operator_key, initial_balance=Hbar(10)):
     """
     Create a new account on the Hedera network with an initial balance.
@@ -80,18 +99,36 @@ def create_account(client, operator_key, initial_balance=Hbar(10)):
     return new_account_id, new_account_private_key
 
 
-def get_balance(client, account_id):
+def get_balance(account_id: AccountId):
     """
-    Demonstrate the deprecated account balance query.
+    Get an account balance using the Mirror Node REST API.
+
+    Args:
+        account_id (AccountId): The account whose balance should be retrieved.
 
-    .. deprecated::
-        CryptoGetAccountBalanceQuery is no longer supported. Use the
-        Mirror Node REST API, for example GET /api/v1/accounts/{accountId}.
+    Returns:
+        float: Account balance in HBAR.
     """
     print(f"Querying balance for account {account_id}...")
 
-    balance_query = CryptoGetAccountBalanceQuery().set_account_id(account_id)
-    return balance_query.execute(client)
+    mirror_node_url = get_mirror_node_url()
+    url = f"{mirror_node_url}/api/v1/accounts/{account_id}"
+
+    request = Request(
+        url,
+        headers={"Accept": "application/json"},
+    )
+
+    with urlopen(request, timeout=10) as response:
+        data = json.load(response)
+
+    balance_tinybars = data["balance"]
+    balance_hbars = balance_tinybars / 100_000_000
+
+    print("✓ Account balance retrieved successfully")
+    print(f"  HBAR balance: {balance_hbars} hbars")
+
+    return balance_hbars
 
 
 def transfer_hbars(client, operator_id, operator_key, recipient_id, amount):
@@ -143,9 +180,7 @@ def main():
         print("=" * 60)
         print("INITIAL BALANCE CHECK")
         print("=" * 60)
-        initial_balance = get_balance(client, new_account_id)
-        if initial_balance is None:
-            print(f"Use the Mirror Node REST API instead, for example GET /api/v1/accounts/{new_account_id}.")
+        initial_balance = get_balance(new_account_id)
         print(f"Initial balance of new account: {initial_balance} hbars")
         print("=" * 60 + "\n")
 
@@ -166,7 +201,7 @@ def main():
         print("=" * 60)
         print("UPDATED BALANCE CHECK")
         print("=" * 60)
-        updated_balance = get_balance(client, new_account_id)
+        updated_balance = get_balance(new_account_id)
         print(f"Updated balance of new account: {updated_balance} hbars")
         print("=" * 60 + "\n")
 

examples/query/account_balance_query_2.py

@@ -8,8 +8,10 @@
    account balances.
 """
 
+import json
 import os
 import sys
+from urllib.request import Request, urlopen
 
 from hiero_sdk_python import (
     AccountCreateTransaction,
@@ -23,7 +25,6 @@
     TokenMintTransaction,
     TokenType,
 )
-from hiero_sdk_python.query.account_balance_query import CryptoGetAccountBalanceQuery
 from hiero_sdk_python.tokens.token_id import TokenId
 
 
@@ -41,6 +42,22 @@ def setup_client():
         sys.exit(1)
 
 
+def get_mirror_node_url():
+    """Return the Mirror Node URL for the configured Hedera network."""
+    network = os.getenv("HEDERA_NETWORK", "testnet").lower()
+
+    mirror_node_urls = {
+        "mainnet": "https://mainnet-public.mirrornode.hedera.com",
+        "testnet": "https://testnet.mirrornode.hedera.com",
+        "previewnet": "https://previewnet.mirrornode.hedera.com",
+    }
+
+    if network not in mirror_node_urls:
+        raise ValueError(f"Unsupported HEDERA_NETWORK: {network}. Expected mainnet, testnet, or previewnet.")
+
+    return mirror_node_urls[network]
+
+
 def create_account(client, name, initial_balance=Hbar(10)):
     """Create a test account with initial balance."""
     account_private_key = PrivateKey.generate(key_type)
@@ -95,22 +112,43 @@ def create_and_mint_token(treasury_account_id, treasury_account_key, client):
         sys.exit(1)
 
 
-def get_account_balance(client: Client, account_id: AccountId):
-    """Get account balance using CryptoGetAccountBalanceQuery."""
-    print(f"Retrieving account balance for account id: {account_id}  ...")
+def get_account_balance(account_id: AccountId):
+    """Get account balance using the Mirror Node REST API."""
+    print(f"Retrieving account balance for account id: {account_id} ...")
+
+    mirror_node_url = get_mirror_node_url()
+    url = f"{mirror_node_url}/api/v1/accounts/{account_id}"
+
+    request = Request(
+        url,
+        headers={"Accept": "application/json"},
+    )
+
     try:
-        # Use CryptoGetAccountBalanceQuery to get the account balance
-        account_balance = CryptoGetAccountBalanceQuery().set_account_id(account_id).execute(client)
+        with urlopen(request, timeout=10) as response:
+            data = json.load(response)
+
+        hbar_balance = data["balance"] / 100_000_000
+
         print("✅ Account balance retrieved successfully!")
-        # Print account balance with account_id context
-        print(f"💰 HBAR Balance for {account_id}: {account_balance.hbars} hbars")
-        # Alternatively, you can use: print(account_balance)
-        return account_balance
-    except (ValueError, TypeError, RuntimeError, ConnectionError) as error:
-        print(f"Error retrieving account balance: {error}")
+        print(f"💰 HBAR Balance for {account_id}: {hbar_balance} hbars")
+
+        return data
+
+    except Exception as error:
+        print(f"❌ Error retrieving account balance: {error}")
         sys.exit(1)
 
 
+def get_token_balance(balance_data, token_id: TokenId):
+    """Get a token balance from a Mirror Node account response."""
+    for token in balance_data.get("balance", {}).get("tokens", []):
+        if token["token_id"] == str(token_id):
+            return token["balance"]
+
+    return 0
+
+
 # OPTIONAL comparison function
 def compare_token_balances(client, treasury_id: AccountId, receiver_id: AccountId, token_id: TokenId):
     """Compare token balances between two accounts."""

tests/unit/executable_test.py

@@ -16,7 +16,6 @@
 )
 from hiero_sdk_python.hapi.services import (
     basic_types_pb2,
-    crypto_get_account_balance_pb2,
     query_pb2,
     response_header_pb2,
     response_pb2,
@@ -427,58 +426,20 @@ def test_transaction_node_switching_body_bytes():
         )
 
 
-def test_query_retry_on_busy():
+def test_query_raises_when_account_balance_query_is_unsupported():
     """
-    Test query retry behavior when receiving BUSY response.
-
-    This test simulates two scenarios:
-    1. First node returns BUSY response
-    2. Second node returns OK response with the balance
-
-    Verifies that the query successfully retries on a different node after receiving BUSY,
-    that the balance is returned correctly and that time.sleep was called once for the retry delay.
+    Test that account balance queries raise an error because the
+    CryptoGetBalance endpoint is no longer supported.
     """
-    # Create a BUSY response to simulate a node being temporarily unavailable
-    # This response indicates the node cannot process the request at this time
-    busy_response = response_pb2.Response(
-        cryptogetAccountBalance=crypto_get_account_balance_pb2.CryptoGetAccountBalanceResponse(
-            header=response_header_pb2.ResponseHeader(nodeTransactionPrecheckCode=ResponseCode.BUSY)
-        )
-    )
-
-    # Create a successful OK response with a balance of 1 Hbar
-    # This simulates a successful account balance query response
-    ok_response = response_pb2.Response(
-        cryptogetAccountBalance=crypto_get_account_balance_pb2.CryptoGetAccountBalanceResponse(
-            header=response_header_pb2.ResponseHeader(nodeTransactionPrecheckCode=ResponseCode.OK),
-            balance=100000000,  # Balance in tinybars
-        )
-    )
-
-    # Set up response sequences for multiple nodes:
-    # First node returns BUSY, forcing a retry
-    # Second node returns OK with the balance
-    response_sequences = [
-        [busy_response, ok_response],
-        [ok_response],  # additional response to mimic additional node
-    ]
-
-    with (
-        mock_hedera_servers(response_sequences) as client,
-        patch("hiero_sdk_python.executable.time.sleep") as mock_sleep,
-    ):
+    with mock_hedera_servers([]) as client:
         query = CryptoGetAccountBalanceQuery()
         query.set_account_id(AccountId(0, 0, 1234))
 
-        balance = query.execute(client)
-
-        # Verify we slept once for the retry
-        assert mock_sleep.call_count == 1, "Should have retried once"
-
-        assert balance.hbars.to_tinybars() == 100000000
-        # Verify we switched to the second node
-        assert query._node_account_ids.index == 0
-        assert query._node_account_ids.current == AccountId(0, 0, 3), "Client should have switched to the second node"
+        with pytest.raises(
+            RuntimeError,
+            match="AccountBalanceQuery is no longer supported",
+        ):
+            query.execute(client)
 
 
 # Set max_attempts