Skip to content

Commit 7f28a0a

Browse files
committed
fix: updated query unit test
Signed-off-by: Manish Dait <daitmanish88@gmail.com>
1 parent eab5ea7 commit 7f28a0a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

tests/unit/test_query.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,9 @@ def test_before_execute_payment_not_required(query, mock_client):
4343
# payment_amount is None, should not set payment_amount
4444
query._before_execute(mock_client)
4545

46-
assert query.node_account_ids == mock_client.get_node_account_ids()
46+
# since node_account_ids is not set it will be empty
47+
# query internally use node form client
48+
assert query.node_account_ids == []
4749
assert query.operator == mock_client.operator
4850
assert query.payment_amount is None
4951

@@ -56,8 +58,10 @@ def test_before_execute_payment_required(query_requires_payment, mock_client):
5658

5759
# payment_amount is None, should set payment_amount to 2 Hbars
5860
query_requires_payment._before_execute(mock_client)
59-
60-
assert query_requires_payment.node_account_ids == mock_client.get_node_account_ids()
61+
62+
# since node_account_ids is not set it will be empty
63+
# query internally use node form client
64+
assert query_requires_payment.node_account_ids == []
6165
assert query_requires_payment.operator == mock_client.operator
6266
assert query_requires_payment.payment_amount.to_tinybars() == Hbar(2).to_tinybars()
6367

0 commit comments

Comments
 (0)