feat: added new tests to topic_info_query_test.py (hiero-ledger#1163)

Signed-off-by: Antonio Ceppellini <[email protected]>

Author: AntonioCeppellini

CHANGELOG.md

@@ -52,6 +52,7 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - Added __str__ and __repr__ to AccountBalance
 - Added GitHub workflow that makes sure newly added test files follow pytest test files naming conventions (#1054)
 - Added advanced issue template for contributors `.github/ISSUE_TEMPLATE/06_advanced_issue.yml`.
+- Add new tests to `tests/unit/topic_info_query_test.py` (#1124)
 
 ### Changed
 - Reduce office-hours reminder spam by posting only on each user's most recent open PR, grouping by author and sorting by creation time (#1121)

tests/unit/topic_info_query_test.py

@@ -3,12 +3,14 @@
 import pytest
 
 from hiero_sdk_python.consensus.topic_info import TopicInfo
+from hiero_sdk_python.executable import _ExecutionState
 from hiero_sdk_python.hapi.services import (
     basic_types_pb2,
     consensus_get_topic_info_pb2,
     consensus_topic_info_pb2,
     response_header_pb2,
-    response_pb2
+    response_pb2,
+    query_pb2,
 )
 from hiero_sdk_python.hapi.services.query_header_pb2 import ResponseType
 from hiero_sdk_python.query.topic_info_query import TopicInfoQuery
@@ -59,6 +61,15 @@ def create_topic_info_response(status_code=ResponseCode.OK, with_info=True):
     return responses
 
 
+def _response_with_status(status: ResponseCode) -> response_pb2.Response:
+    """Create a Response with only the consensusGetTopicInfo header status set."""
+    return response_pb2.Response(
+        consensusGetTopicInfo=consensus_get_topic_info_pb2.ConsensusGetTopicInfoResponse(
+            header=response_header_pb2.ResponseHeader(nodeTransactionPrecheckCode=status)
+        )
+    )
+
+
 def test_topic_info_query(topic_id):
     """Test basic functionality of TopicInfoQuery with mock server."""
     responses = create_topic_info_response()
@@ -96,3 +107,61 @@ def test_topic_info_query_with_empty_topic_id():
             query.execute(client)
 
         assert "Topic ID must be set" in str(exc_info.value)
+
+
+def test_make_request_builds_expected_protobuf(topic_id):
+    """
+    Covers TopicInfoQuery._make_request() using REAL protobuf classes.
+    This is fast and deterministic (no network).
+    """
+    query = TopicInfoQuery().set_topic_id(topic_id)
+
+    req = query._make_request()
+    assert isinstance(req, query_pb2.Query)
+
+    # Ensure the correct oneof field is populated
+    assert req.HasField("consensusGetTopicInfo")
+    assert req.consensusGetTopicInfo.HasField("topicID")
+
+    # Compare serialized protobuf to avoid equality quirks
+    expected_topic_id_proto = topic_id._to_proto()
+    assert (
+        req.consensusGetTopicInfo.topicID.SerializeToString()
+        == expected_topic_id_proto.SerializeToString()
+    )
+
+
+@pytest.mark.parametrize(
+    "status,expected_state",
+    [
+        (ResponseCode.OK, _ExecutionState.FINISHED),
+        (ResponseCode.UNKNOWN, _ExecutionState.RETRY),
+        (ResponseCode.BUSY, _ExecutionState.RETRY),
+        (ResponseCode.PLATFORM_NOT_ACTIVE, _ExecutionState.RETRY),
+        (ResponseCode.INVALID_TOPIC_ID, _ExecutionState.ERROR),
+    ],
+)
+def test_should_retry_branches(status, expected_state):
+    """Covers OK / RETRY / ERROR branches of TopicInfoQuery._should_retry()."""
+    query = TopicInfoQuery()
+    response = _response_with_status(status)
+
+    assert query._should_retry(response) == expected_state
+
+
+def test_freeze_prevents_set_topic_id_if_available(topic_id):
+    """
+    If TopicInfoQuery implements freeze semantics, ensure it prevents mutation.
+    This is written to not break the suite if freeze() doesn't exist in your version.
+    """
+    query = TopicInfoQuery()
+
+    if not hasattr(query, "freeze"):
+        pytest.skip("TopicInfoQuery.freeze() not available in this SDK version")
+
+    query.freeze()
+
+    with pytest.raises(ValueError) as exc_info:
+        query.set_topic_id(topic_id)
+
+    assert "frozen" in str(exc_info.value).lower()