Create ChunkedTransaction class

Signed-off-by: MonaaEid <monaa_eid@hotmail.com>

Author: MonaaEid

src/hiero_sdk_python/transaction/chunked_transaction.py

@@ -12,10 +12,6 @@
 from hiero_sdk_python.transaction.transaction_response import TransactionResponse
 
 
-if TYPE_CHECKING:
-    pass
-
-
 class ChunkedTransaction(Transaction, ABC):
     """
     Abstract base class for transactions that support chunking.
@@ -43,17 +39,6 @@ def __init__(self) -> None:
         self.chunk_size: int = 1024
         self.max_chunks: int = 20
 
-    @abstractmethod
-    def get_required_chunks(self) -> int:
-        """
-        Returns the number of chunks required for the current content.
-
-        Subclasses must implement this based on their content type.
-
-        Returns:
-            int: Number of chunks required.
-        """
-        pass
 
     @abstractmethod
     def _build_proto_body(self):
@@ -148,10 +133,9 @@ def freeze_with(self, client: Client) -> ChunkedTransaction:
         if self._transaction_body_bytes:
             return self
 
-        self._validate_chunking()
         self._resolve_transaction_id(client)
 
-        if self.transaction_id is None or self.transaction_id.valid_start is None:
+        if self.transaction_id.valid_start is None:
             raise ValueError("Transaction ID with valid_start must be set before freezing chunked transaction.")
 
         # Generate transaction IDs for all chunks if not already done
@@ -265,11 +249,10 @@ def execute_all(
             List[TransactionReceipt]: If wait_for_receipt is True (default)
             List[TransactionResponse]: If wait_for_receipt is False
         """
-        self._validate_chunking()
-
-        required_chunks = self.get_required_chunks()
+        self._validate_chunking() # Moved here
 
-        if required_chunks == 1:
+        # For single-chunk transactions, delegate to the standard execution flow.
+        if self.get_required_chunks() == 1:
             return [
                 super().execute(
                     client,
@@ -279,13 +262,14 @@ def execute_all(
                 )
             ]
 
-        # Ensure the initial transaction ID and chunk transaction IDs exist.
-        if not self._transaction_ids:
+        # For multi-chunk transactions, ensure we are frozen before proceeding.
+        if not self._transaction_body_bytes:
             self.freeze_with(client)
 
         responses = []
+        required_chunks = self.get_required_chunks()
 
-        for chunk_index in range(required_chunks):
+        for chunk_index in range(self.get_required_chunks()):
             self._current_chunk_index = chunk_index
 
             if chunk_index < len(self._transaction_ids):

tests/integration/topic_message_submit_transaction_e2e_test.py

@@ -108,7 +108,7 @@ def test_topic_message_submit_transaction_fails_if_max_chunks_less_than_requied(
     message_tx = TopicMessageSubmitTransaction().set_topic_id(topic_id).set_message(message).set_max_chunks(2)
 
     with pytest.raises(ValueError):
-        message_tx.freeze_with(env.client)
+        message_tx.execute(env.client)
 
     delete_topic(env.client, topic_id)
 
@@ -289,7 +289,7 @@ def test_integration_topic_message_submit_transaction_fails_if_required_chunk_gr
     with pytest.raises(
         ValueError, match="Message requires 4 chunks but max_chunks=2. Increase limit with set_max_chunks()."
     ):
-        message_transaction.freeze_with(env.client)
+        message_transaction.execute(env.client)
 
     delete_topic(env.client, topic_id)