chore: added test for topic message submit tx

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

Author: manishdait

tests/unit/topic_message_submit_transaction_test.py

@@ -639,3 +639,14 @@ def test_topic_submit_execute_returns_failed_receipt_by_default(topic_id):
         receipt = tx.execute(client)
 
         assert receipt.status == ResponseCode.INVALID_SIGNATURE
+
+def test_topic_submit_message_raises_error_on_freeze(topic_id):
+    """Test transaction raises error on freeze when the transaction_id and node_id not set"""
+    tx = (
+        TopicMessageSubmitTransaction()
+        .set_topic_id(topic_id)
+        .set_message("Hello Hiero")
+    )
+
+    with pytest.raises(ValueError):
+        tx.freeze()

tests/unit/transaction_test.py

@@ -2,6 +2,8 @@
 
 from hiero_sdk_python.account.account_create_transaction import AccountCreateTransaction
 from hiero_sdk_python.account.account_id import AccountId
+from hiero_sdk_python.consensus.topic_id import TopicId
+from hiero_sdk_python.consensus.topic_message_submit_transaction import TopicMessageSubmitTransaction
 from hiero_sdk_python.crypto.private_key import PrivateKey
 from hiero_sdk_python.exceptions import ReceiptStatusError
 from hiero_sdk_python.file.file_append_transaction import FileAppendTransaction
@@ -27,7 +29,7 @@
 
 @pytest.fixture
 def file_id():
-    """Returns a file_is for test."""
+    """Returns a file_id for test."""
     return FileId.from_string("0.0.1")
 
 
@@ -277,8 +279,8 @@ def test_tx_without_optional_fields_should_have_smaller_tx_body(
     assert tx1.body_size < tx2.body_size
 
 
-def test_chunk_tx_should_return_list_of_body_sizes(file_id, account_id, transaction_id):
-    """Test should return array of body sizes for multi-chunk transaction."""
+def test_file_append_chunk_tx_should_return_list_of_body_sizes(file_id, account_id, transaction_id):
+    """Test file append tx should return array of body sizes for multi-chunk transaction."""
     chunk_size = 1024
     content = "a" * (chunk_size * 3)
 
@@ -297,8 +299,8 @@ def test_chunk_tx_should_return_list_of_body_sizes(file_id, account_id, transact
     assert len(sizes) == 3
 
 
-def test_single_chunk_tx_return_list_of_len_one(file_id, account_id, transaction_id):
-    """Test should return array of one size for single-chunk transaction."""
+def test_file_append_single_chunk_tx_return_list_of_len_one(file_id, account_id, transaction_id):
+    """Test file append tx should return array of one size for single-chunk transaction."""
     content = "small_content"
     tx = (
         FileAppendTransaction()
@@ -314,6 +316,43 @@ def test_single_chunk_tx_return_list_of_len_one(file_id, account_id, transaction
     assert len(sizes) == 1
 
 
+def test_message_submit_chunk_tx_should_return_list_of_body_sizes(topic_id, account_id, transaction_id):
+    """Test topic message submit tx should return array of body sizes for multi-chunk transaction."""
+    chunk_size = 1024
+    message= "a" * (chunk_size * 3)
+
+    tx = (
+        TopicMessageSubmitTransaction()
+        .set_topic_id(topic_id)
+        .set_chunk_size(chunk_size)
+        .set_message(message)
+        .set_transaction_id(transaction_id)
+        .set_node_account_id(account_id)
+        .freeze()
+    )
+
+    sizes = tx.body_size_all_chunks
+    assert isinstance(sizes, list)
+    assert len(sizes) == 3
+
+
+def test_message_submit_single_chunk_tx_return_list_of_len_one(topic_id, account_id, transaction_id):
+    """Test topic message submit tx should return array of one size for single-chunk transaction."""
+    message = "small_content"
+    tx = (
+        TopicMessageSubmitTransaction()
+        .set_topic_id(topic_id)
+        .set_message(message)
+        .set_transaction_id(transaction_id)
+        .set_node_account_id(account_id)
+        .freeze()
+    )
+
+    sizes = tx.body_size_all_chunks
+    assert isinstance(sizes, list)
+    assert len(sizes) == 1
+
+
 def test_tx_with_no_content_should_return_single_body_chunk(
     file_id, account_id, transaction_id
 ):