Skip to content

Commit 2158a56

Browse files
committed
chore: updated unit test
Signed-off-by: Manish Dait <daitmanish88@gmail.com>
1 parent 481796f commit 2158a56

35 files changed

Lines changed: 156 additions & 46 deletions

tests/integration/file_append_transaction_e2e_test.py

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,9 @@
77
from hiero_sdk_python.file.file_append_transaction import FileAppendTransaction
88
from hiero_sdk_python.file.file_contents_query import FileContentsQuery
99
from hiero_sdk_python.file.file_create_transaction import FileCreateTransaction
10+
from hiero_sdk_python.hbar import Hbar
1011
from hiero_sdk_python.response_code import ResponseCode
12+
from hiero_sdk_python.transaction.transaction import Transaction
1113
from hiero_sdk_python.transaction.transaction_id import TransactionId
1214

1315

@@ -324,3 +326,42 @@ def test_file_append_chunk_transaction_can_execute_with_manual_freeze(env):
324326

325327
file_contents = FileContentsQuery().set_file_id(file_id).execute(env.client)
326328
assert file_contents == bytes(content, "utf-8")
329+
330+
331+
@pytest.mark.integration
332+
def test(env):
333+
create_receipt = (
334+
FileCreateTransaction()
335+
.set_keys(env.client.operator_private_key.public_key())
336+
.set_contents(b"")
337+
.execute(env.client)
338+
)
339+
340+
assert create_receipt.status == ResponseCode.SUCCESS
341+
file_id = create_receipt.file_id
342+
343+
file_contents = FileContentsQuery().set_file_id(file_id).execute(env.client)
344+
assert file_contents == b""
345+
346+
content = "A" * (8192) # content with (4000/1024) bytes ie approx 4 chunks
347+
348+
tx = (
349+
FileAppendTransaction()
350+
.set_file_id(file_id)
351+
.set_contents(content)
352+
.set_transaction_id(TransactionId.generate(AccountId(0, 0, 101)))
353+
.set_node_account_ids([AccountId(0, 0, 3), AccountId(0, 0, 4)])
354+
)
355+
356+
tx._default_transaction_fee = Hbar.from_hbars(10).to_tinybars()
357+
358+
print(tx.get_required_chunks())
359+
print(tx.transaction_id)
360+
print(tx.node_account_ids)
361+
362+
bytestx = tx.to_bytes()
363+
ntx = Transaction.from_bytes(bytestx)
364+
365+
print(ntx.get_required_chunks())
366+
print(ntx.transaction_id)
367+
print(ntx.node_account_ids)

tests/integration/topic_message_submit_transaction_e2e_test.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from hiero_sdk_python.response_code import ResponseCode
2020
from hiero_sdk_python.tokens.custom_fixed_fee import CustomFixedFee
2121
from hiero_sdk_python.transaction.custom_fee_limit import CustomFeeLimit
22+
from hiero_sdk_python.transaction.transaction import Transaction
2223
from hiero_sdk_python.transaction.transaction_id import TransactionId
2324

2425

@@ -328,3 +329,33 @@ def test_topic_message_submit_transaction_can_submit_a_large_message_manual_free
328329
assert info.sequence_number == 14
329330

330331
delete_topic(env.client, topic_id)
332+
333+
334+
@pytest.mark.integration
335+
def test(env):
336+
topic_id = create_topic(client=env.client, admin_key=env.operator_key)
337+
338+
info = TopicInfoQuery().set_topic_id(topic_id).execute(env.client)
339+
assert info.sequence_number == 0
340+
341+
message = "ABCD" # message with (1024 * 14) bytes ie 14 chunks
342+
343+
message_tx = (
344+
TopicMessageSubmitTransaction()
345+
.set_topic_id(topic_id)
346+
.set_chunk_size(1)
347+
.set_message(message)
348+
.freeze_with(env.client)
349+
)
350+
print(len(message_tx._transaction_ids))
351+
print(len(message_tx._transaction_body_bytes.keys()))
352+
353+
bytestx = message_tx.to_bytes()
354+
tx = Transaction.from_bytes(bytestx)
355+
356+
print(isinstance(tx, TopicMessageSubmitTransaction))
357+
print(len(tx._transaction_ids))
358+
print(len(tx._transaction_body_bytes.keys()))
359+
tx.execute(env.client)
360+
361+
print(tx.message)

tests/integration/transaction_freeze_e2e_test.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,11 @@
22

33
import pytest
44

5+
from hiero_sdk_python.account.account_create_transaction import AccountCreateTransaction
56
from hiero_sdk_python.account.account_id import AccountId
67
from hiero_sdk_python.client.client import Client
78
from hiero_sdk_python.consensus.topic_create_transaction import TopicCreateTransaction
9+
from hiero_sdk_python.crypto.private_key import PrivateKey
810
from hiero_sdk_python.response_code import ResponseCode
911
from hiero_sdk_python.transaction.transaction import Transaction
1012
from hiero_sdk_python.transaction.transaction_id import TransactionId
@@ -151,3 +153,22 @@ def test_transaction_with_secondary_client_without_operator_can_execute_successf
151153
receipt = tx2.execute(executor_client)
152154

153155
assert receipt.status == ResponseCode.SUCCESS, "Transaction must execute successfully"
156+
157+
158+
@pytest.mark.integration
159+
def test(env):
160+
key = PrivateKey.generate_ecdsa()
161+
162+
tx = (
163+
AccountCreateTransaction()
164+
.set_key_without_alias(key)
165+
.set_initial_balance(1)
166+
.set_account_memo("test account")
167+
.freeze_with(env.client)
168+
)
169+
print(tx._transaction_body_bytes)
170+
171+
tx_bytes = tx.to_bytes()
172+
173+
tx2 = Transaction.from_bytes(tx_bytes)
174+
tx2.execute(env.client)

tests/unit/account_create_transaction_test.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,7 @@ def test_account_create_transaction_build_scheduled_body(mock_account_ids):
108108
def test_account_create_transaction_sign(mock_account_ids, mock_client):
109109
"""Test signing the account create transaction."""
110110
operator_id, node_account_id = mock_account_ids
111+
transaction_id = generate_transaction_id(operator_id)
111112

112113
new_private_key = PrivateKey.generate()
113114
new_public_key = new_private_key.public_key()
@@ -119,20 +120,20 @@ def test_account_create_transaction_sign(mock_account_ids, mock_client):
119120
.set_initial_balance(100000000)
120121
.set_account_memo("Test account")
121122
)
122-
account_tx.transaction_id = generate_transaction_id(operator_id)
123+
account_tx.transaction_id = transaction_id
123124
account_tx.node_account_id = node_account_id
124125
account_tx.freeze_with(mock_client)
125126

126127
# Add first signiture
127128
account_tx.sign(mock_client.operator_private_key)
128-
body_bytes = account_tx._transaction_body_bytes[node_account_id]
129+
body_bytes = account_tx._transaction_body_bytes[transaction_id][node_account_id]
129130

130131
assert body_bytes in account_tx._signature_map, "Body bytes should be a key in the signature map dictionary"
131132
assert len(account_tx._signature_map[body_bytes].sigPair) == 1, "Transaction should have exactly one signature"
132133

133134
# Add second signiture
134135
account_tx.sign(operator_private_key)
135-
body_bytes = account_tx._transaction_body_bytes[node_account_id]
136+
body_bytes = account_tx._transaction_body_bytes[transaction_id][node_account_id]
136137

137138
assert body_bytes in account_tx._signature_map, "Body bytes should be a key in the signature map dictionary"
138139
assert len(account_tx._signature_map[body_bytes].sigPair) == 2, "Transaction should have exactly two signatures"

tests/unit/account_delete_transaction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def test_sign_transaction(mock_client, delete_params):
188188
delete_tx.sign(private_key)
189189

190190
node_id = mock_client.network.current_node._account_id
191-
body_bytes = delete_tx._transaction_body_bytes[node_id]
191+
body_bytes = delete_tx._transaction_body_bytes[delete_tx.transaction_id][node_id]
192192

193193
assert len(delete_tx._signature_map[body_bytes].sigPair) == 1
194194
sig_pair = delete_tx._signature_map[body_bytes].sigPair[0]

tests/unit/account_update_transaction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ def test_sign_transaction(mock_client):
354354
account_tx.sign(private_key)
355355

356356
node_id = mock_client.network.current_node._account_id
357-
body_bytes = account_tx._transaction_body_bytes[node_id]
357+
body_bytes = account_tx._transaction_body_bytes[account_tx.transaction_id][node_id]
358358

359359
assert len(account_tx._signature_map[body_bytes].sigPair) == 1
360360
sig_pair = account_tx._signature_map[body_bytes].sigPair[0]

tests/unit/batch_transaction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def test_sign_transaction(mock_client, mock_tx):
312312
batch_tx.sign(private_key)
313313

314314
node_id = mock_client.network.current_node._account_id
315-
body_bytes = batch_tx._transaction_body_bytes[node_id]
315+
body_bytes = batch_tx._transaction_body_bytes[batch_tx.transaction_id][node_id]
316316

317317
assert body_bytes in batch_tx._signature_map, "signature map must contain an entry for the tx body bytes"
318318
sig_pairs = batch_tx._signature_map[body_bytes].sigPair

tests/unit/contract_delete_transaction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,7 @@ def test_sign_transaction(mock_client, delete_params):
296296
delete_tx.sign(private_key)
297297

298298
node_id = mock_client.network.current_node._account_id
299-
body_bytes = delete_tx._transaction_body_bytes[node_id]
299+
body_bytes = delete_tx._transaction_body_bytes[delete_tx.transaction_id][node_id]
300300

301301
assert len(delete_tx._signature_map[body_bytes].sigPair) == 1
302302
sig_pair = delete_tx._signature_map[body_bytes].sigPair[0]

tests/unit/contract_execute_transaction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ def test_sign_transaction(mock_client, execute_params):
325325
execute_tx.sign(private_key)
326326

327327
node_id = mock_client.network.current_node._account_id
328-
body_bytes = execute_tx._transaction_body_bytes[node_id]
328+
body_bytes = execute_tx._transaction_body_bytes[execute_tx.transaction_id][node_id]
329329

330330
assert len(execute_tx._signature_map[body_bytes].sigPair) == 1
331331
sig_pair = execute_tx._signature_map[body_bytes].sigPair[0]

tests/unit/ethereum_transaction_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def test_sign_transaction(mock_client, ethereum_params):
170170
ethereum_tx.sign(private_key)
171171

172172
node_id = mock_client.network.current_node._account_id
173-
body_bytes = ethereum_tx._transaction_body_bytes[node_id]
173+
body_bytes = ethereum_tx._transaction_body_bytes[ethereum_tx.transaction_id][node_id]
174174

175175
assert len(ethereum_tx._signature_map[body_bytes].sigPair) == 1
176176
sig_pair = ethereum_tx._signature_map[body_bytes].sigPair[0]

0 commit comments

Comments
 (0)