chore: updated e2e test

Signed-off-by: Manish Dait <daitmanish88@gmail.com>

Author: manishdait

src/hiero_sdk_python/query/fee_estimate_query.py

@@ -137,7 +137,7 @@ def execute(self, client) -> FeeEstimateResponse:
         self._ensure_frozen(self._transaction, client)
 
         if self._is_chunked():
-            return self._execute_chunked(client, url, mode)
+            return self._execute_chunked(url, mode)
 
         return self._execute_single(url, mode)
 

tests/integration/transaction_freeze_e2e_test.py

@@ -24,8 +24,10 @@ def test_transaction_executes_successfully(env):
     receipt = tx.execute(executor_client)
 
     # Verify that the transaction_bodys are generated for all nodes present in client network
-    assert len(tx._transaction_body_bytes) == len(env.client.network.nodes)
-    assert set(tx._transaction_body_bytes.keys()) == {node._account_id for node in env.client.network.nodes}
+    assert len(tx._transaction_body_bytes) == len(tx._transaction_ids)
+    assert set(tx._transaction_body_bytes.keys()) == set(tx._transaction_ids)
+
+    # TODO: check for node_ids
 
     assert receipt.status == ResponseCode.SUCCESS, "Transaction must execute successfully"
 
@@ -43,8 +45,10 @@ def test_transaction_executes_successfully_with_node_account_ids(env):
     tx.sign(executor_key)
 
     # Verify that the transaction_bodys are generated for the provided node_account_ids only
-    assert len(tx._transaction_body_bytes) == 2
-    assert set(tx._transaction_body_bytes.keys()) == set(node_account_ids)
+    assert len(tx._transaction_body_bytes) == len(tx._transaction_ids)
+    assert set(tx._transaction_body_bytes.keys()) == set(tx._transaction_ids)
+
+    # TODO: check for the node_ids
 
     receipt = tx.execute(executor_client)
     assert receipt.status == ResponseCode.SUCCESS, "Transaction must execute successfully"
@@ -63,8 +67,10 @@ def test_transaction_executes_successfully_with_single_node_account_id(env):
     tx.sign(executor_key)
 
     # Verify that the transaction_bodys are generated for the provided node_account_id only
-    assert len(tx._transaction_body_bytes) == 1
-    assert set(tx._transaction_body_bytes.keys()) == {node_account_id}
+    assert len(tx._transaction_body_bytes) == len(tx._transaction_ids)
+    assert set(tx._transaction_body_bytes.keys()) == set(tx._transaction_ids)
+
+    # TODO: check for node_ids
 
     receipt = tx.execute(executor_client)
     assert receipt.status == ResponseCode.SUCCESS, "Transaction must execute successfully"

tests/unit/fee_estimate_query_test.py

@@ -380,7 +380,7 @@ def test_port_replacement_for_localhost_execute_multiple():
     with patch.object(query, "_execute_chunked", return_value=MagicMock()) as mock_execute_chunked:
         query.execute(client_1)
 
-        called_url = mock_execute_chunked.call_args[0][1]
+        called_url = mock_execute_chunked.call_args[0][0]
         assert ":8084" in called_url
         assert ":38081" not in called_url
 
@@ -389,6 +389,6 @@ def test_port_replacement_for_localhost_execute_multiple():
 
     with patch.object(query, "_execute_chunked", return_value=MagicMock()) as mock_execute_chunked:
         query.execute(client_2)
-        called_url = mock_execute_chunked.call_args[0][1]
+        called_url = mock_execute_chunked.call_args[0][0]
         assert ":8084" in called_url
         assert ":38081" not in called_url