test: increase initial balances of accounts (#124)

Signed-off-by: skurzyp-blockydevs <stanislaw.kurzyp@blockydevs.com>

Author: skurzyp-blockydevs

python/hedera_agent_kit/plugins/core_account_plugin/create_account.py

@@ -57,6 +57,8 @@ def create_account_prompt(context: Context = {}) -> str:
 
 This tool will create a new Hedera account with a passed public key. If not passed, the tool will use operator's public key.
 
+IMPORTANT: All parameters are optional. If the user does not explicitly provide optional parameters, proceed immediately using the default values. Do NOT ask the user for optional parameters.
+
 Parameters:
 - public_key (str, optional): Public key to use for the account. If not provided, the tool will use the operator's public key.
 - account_memo (str, optional): Optional memo for the account. Can be up to 100 characters long. Too long memos will be truncated in params normalization

python/hedera_agent_kit/plugins/core_consensus_plugin/create_topic.py

@@ -52,7 +52,7 @@ def create_topic_prompt(context: Context = {}) -> str:
 - topic_memo (str, optional): A memo for the topic.
 - transaction_memo (str, optional): An optional memo to include on the submitted transaction.
 - submit_key (bool or str, optional): Submit key for the topic. Pass boolean `true` to use the operator/user key,
-  or provide a Hedera-compatible public key string. Defaults to None (no submit key). If set, restricts who can submit messages to the topic.
+  or provide a Hedera-compatible public key string. Defaults to false (no submit key). If set, restricts who can submit messages to the topic.
 - admin_key (bool or str, optional): Admin key for the topic. Pass boolean `true` to use the operator/user key,
   or provide a Hedera-compatible public key string. Defaults to true (operator key).
 

python/hedera_agent_kit/shared/parameter_schemas/consensus_schema.py

@@ -32,7 +32,7 @@ class CreateTopicParameters(BaseModelWithArbitraryTypes):
                 "or provide a Hedera-compatible public key string (optional). If set, restricts who can submit messages to the topic."
             ),
         ),
-    ] = None
+    ] = False
 
     admin_key: Annotated[
         Optional[Union[bool, str]],

python/test/e2e/create_non_fungible_token_e2e_test.py

@@ -187,8 +187,9 @@ async def test_create_nft_minimal_params(
     assert token_info.name == "MyNFT"
     assert token_info.symbol == "MNFT"
     assert token_info.token_type == TokenType.NON_FUNGIBLE_UNIQUE
-    # Default max supply is infinite (0)
-    assert token_info.max_supply == 0
+    # Default supply type is finite
+    assert token_info.supply_type == SupplyType.FINITE
+    assert token_info.max_supply == 100 # the default max supply is 100
 
 
 @pytest.mark.asyncio

python/test/e2e/delete_token_allowance_e2e_test.py

@@ -39,7 +39,7 @@
 from test.utils.teardown import return_hbars_and_delete_account
 
 # Constants
-DEFAULT_EXECUTOR_BALANCE = Hbar(50, in_tinybars=False)
+DEFAULT_EXECUTOR_BALANCE = Hbar(100, in_tinybars=False)
 MIRROR_NODE_WAITING_TIME_SEC = 10
 
 
@@ -153,7 +153,7 @@ async def spender_account(
     # Executor creates spender funded with a small balance
     spender_resp = await executor_wrapper.create_account(
         CreateAccountParametersNormalised(
-            initial_balance=Hbar(50),
+            initial_balance=Hbar(100),
             key=spender_key.public_key(),
         )
     )

python/test/e2e/get_hbar_balance_e2e_test.py

@@ -65,7 +65,7 @@ async def executor_account(
 
     await wait(MIRROR_NODE_WAITING_TIME)
 
-    yield executor_account_id, executor_key, executor_client, executor_wrapper
+    yield executor_account_id, executor_key, executor_client, executor_wrapper, operator_wrapper
 
     await return_hbars_and_delete_account(
         executor_wrapper,
@@ -77,7 +77,7 @@ async def executor_account(
 @pytest.fixture
 async def langchain_test_setup(executor_account):
     """Initialize LangChain agent and toolkit using executor client as operator."""
-    _, _, executor_client, _ = executor_account
+    _, _, executor_client, _, _ = executor_account
     setup = await create_langchain_test_setup(custom_client=executor_client)
     yield setup
     setup.cleanup()
@@ -139,36 +139,6 @@ async def execute_get_hbar_balance(
 # TEST CASES
 # ============================================================================
 
-
-@pytest.mark.asyncio
-async def test_get_hbar_balance_for_executor_account(
-    agent_executor,
-    executor_account,
-    langchain_config,
-    response_parser: ResponseParserService,
-):
-    """Test fetching HBAR balance for executor (default) account."""
-    executor_account_id, _, executor_client, executor_wrapper = executor_account
-    executor_id_str = str(executor_account_id)
-
-    # Get expected balance directly (before agent call)
-    expected_balance = executor_wrapper.get_account_hbar_balance(executor_id_str)
-
-    input_text = f"What is the HBAR balance of {executor_id_str}?"
-    parsed_data = await execute_get_hbar_balance(
-        agent_executor, input_text, langchain_config, response_parser
-    )
-
-    human_message = parsed_data["humanMessage"]
-    raw_data = parsed_data["raw"]
-
-    assert parsed_data.get("error") is None
-    assert executor_id_str in human_message
-
-    assert str(int(expected_balance)) in raw_data.get("balance")
-    assert "HBAR Balance" in human_message
-
-
 @pytest.mark.asyncio
 async def test_get_hbar_balance_for_specific_account_nonzero(
     agent_executor,

python/test/e2e/get_pending_airdrop_query_e2e_test.py

@@ -81,7 +81,7 @@ async def setup_environment():
     recipient_resp = await operator_wrapper.create_account(
         CreateAccountParametersNormalised(
             key=executor_key.public_key(),  # Using an executor key for simplicity in test control
-            initial_balance=Hbar(0),
+            initial_balance=Hbar(20),
             max_automatic_token_associations=0,
         )
     )

python/test/e2e/mint_erc721_e2e_test.py

@@ -54,7 +54,7 @@ async def setup_environment():
     # Recipient account
     recipient_resp = await operator_wrapper.create_account(
         CreateAccountParametersNormalised(
-            key=executor_key.public_key(), initial_balance=Hbar(5)
+            key=executor_key.public_key(), initial_balance=Hbar(20)
         )
     )
     recipient_account_id = recipient_resp.account_id

python/test/e2e/transfer_non_fungible_token_with_allowance_e2e_test.py

@@ -38,8 +38,8 @@
 from test.utils.teardown import return_hbars_and_delete_account
 
 # Constants
-DEFAULT_OWNER_BALANCE = Hbar(100)
-DEFAULT_SPENDER_BALANCE = Hbar(50)
+DEFAULT_OWNER_BALANCE = Hbar(150)
+DEFAULT_SPENDER_BALANCE = Hbar(100)
 TOOL_NAME = "transfer_non_fungible_token_with_allowance_tool"
 
 

python/test/e2e/update_topic_e2e_test.py

@@ -31,7 +31,7 @@
 from test.utils.teardown import return_hbars_and_delete_account
 
 # Constants
-DEFAULT_EXECUTOR_BALANCE = Hbar(20, in_tinybars=False)
+DEFAULT_EXECUTOR_BALANCE = Hbar(50, in_tinybars=False)
 
 
 # ============================================================================
@@ -309,7 +309,7 @@ async def test_update_autorenew_account(
     secondary_key = executor_client.operator_private_key
     secondary_resp = await executor_wrapper.create_account(
         CreateAccountParametersNormalised(
-            initial_balance=Hbar(0), key=secondary_key.public_key()
+            initial_balance=Hbar(50), key=secondary_key.public_key()
         )
     )
     secondary_account_id = str(secondary_resp.account_id)