feat: added integratiion test

Signed-off-by: Manish Dait <[email protected]>

Author: manishdait

tests/integration/query_e2e_test.py

@@ -6,9 +6,10 @@
 from hiero_sdk_python.exceptions import PrecheckError
 from hiero_sdk_python.hbar import Hbar
 from hiero_sdk_python.query.account_balance_query import CryptoGetAccountBalanceQuery
+from hiero_sdk_python.query.account_info_query import AccountInfoQuery
 from hiero_sdk_python.query.token_info_query import TokenInfoQuery
 from hiero_sdk_python.response_code import ResponseCode
-from tests.integration.utils import IntegrationTestEnv, create_fungible_token
+from tests.integration.utils import IntegrationTestEnv, env, create_fungible_token
 
 @pytest.mark.integration
 def test_integration_free_query_no_cost():
@@ -236,4 +237,24 @@ def test_integration_paid_query_payment_too_high_fails():
         with pytest.raises(PrecheckError, match="failed precheck with status: INSUFFICIENT_PAYER_BALANCE"):
             query.execute(env.client)
     finally:
-        env.close()
+        env.close()
+
+@pytest.mark.integration
+def test_integration_query_exceeds_max_payment(env):
+    """Test that Query fails when cost exceeds max_query_payment."""
+    receipt = env.create_account(1)
+    account_id = receipt.id
+    
+    # Set max payment below actual cost
+    query = (
+        AccountInfoQuery()
+        .set_account_id(account_id)
+        .set_max_query_payment(Hbar.from_tinybars(1))  # Intentionally too low to fail
+    )
+    
+    with pytest.raises(ValueError) as e:
+        query.execute(env.client)
+
+    msg = str(e.value)
+    assert "Query cost" in msg and "exceeds max set query payment:" in msg
+    

tests/unit/query_test.py

@@ -328,7 +328,6 @@ def test_set_max_payment_override_client_max_payment_and_error(query_requires_pa
     # Mock get_cost to return 2 Hbar, exceeding the query.max_query_payment
     mock_get_cost = MagicMock(return_value=Hbar(2))
     query_requires_payment.get_cost = mock_get_cost
-    query_requires_payment.get_cost = mock_get_cost
 
     # Execution should raise ValueError because 2 > query.max_query_payment (1)
     expected_msg = "Query cost ℏ2.0 HBAR exceeds max set query payment: ℏ1.0 HBAR"
@@ -367,7 +366,6 @@ def test_payment_query_use_client_max_payment_and_error(query_requires_payment,
     # Mock get_cost to return 2 Hbar, exceeding the client.default_max_query_payment
     mock_get_cost = MagicMock(return_value=Hbar(2))
     query_requires_payment.get_cost = mock_get_cost
-    query_requires_payment.get_cost = mock_get_cost
 
     # Execution should raise ValueError because 2 > client.default_max_query_payment
     expected_msg = "Query cost ℏ2.0 HBAR exceeds max set query payment: ℏ1.0 HBAR"