Skip to content

Commit dfb4bb8

Browse files
committed
fix(token): use HasField to check token presence in TokenUnpauseTransaction._from_proto
Signed-off-by: anchit-goel <anchitgoel5@gmail.com>
1 parent 4fedd9c commit dfb4bb8

2 files changed

Lines changed: 9 additions & 1 deletion

File tree

src/hiero_sdk_python/tokens/token_unpause_transaction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def _from_proto(cls, proto: TokenUnpauseTransactionBody) -> TokenUnpauseTransact
8282
Returns:
8383
TokenUnpauseTransaction: A new instance of TokenUnpauseTransaction
8484
"""
85-
token_id = TokenId._from_proto(proto.token) if proto.token else None
85+
token_id = TokenId._from_proto(proto.token) if proto.HasField("token") else None
8686
return cls(token_id=token_id)
8787

8888
def _build_proto_body(self) -> TokenUnpauseTransactionBody:

tests/unit/token_unpause_transaction_test.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,14 @@ def test_from_proto(mock_account_ids):
170170
assert unpause_tx.token_id.num == token_id.num
171171

172172

173+
def test_from_proto_without_token():
174+
"""Test creating a TokenUnpauseTransaction from protobuf representation without token."""
175+
proto = TokenUnpauseTransactionBody()
176+
unpause_tx = TokenUnpauseTransaction._from_proto(proto)
177+
178+
assert unpause_tx.token_id is None
179+
180+
173181
def test_upause_transaction_can_execute(mock_account_ids):
174182
"""Test that a token upause transaction can be executed successfully."""
175183
_, _, _, token_id, _ = mock_account_ids

0 commit comments

Comments
 (0)