Skip to content

Commit 4fedd9c

Browse files
committed
fix(token): allow unset token_id in TokenUnpauseTransaction
Signed-off-by: anchit-goel <anchitgoel5@gmail.com>
1 parent 6911de5 commit 4fedd9c

3 files changed

Lines changed: 6 additions & 12 deletions

File tree

src/hiero_sdk_python/tokens/token_unpause_transaction.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -91,14 +91,8 @@ def _build_proto_body(self) -> TokenUnpauseTransactionBody:
9191
9292
Returns:
9393
TokenUnpauseTransactionBody: The protobuf body for this transaction.
94-
95-
Raises:
96-
ValueError: If account ID or token ID is not set.
9794
"""
98-
if self.token_id is None:
99-
raise ValueError("Missing token ID")
100-
101-
return TokenUnpauseTransactionBody(token=self.token_id._to_proto())
95+
return TokenUnpauseTransactionBody(token=self.token_id._to_proto() if self.token_id is not None else None)
10296

10397
def build_transaction_body(self) -> TransactionBody:
10498
"""

tests/integration/token_unpause_transaction_e2e_test.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from __future__ import annotations
22

3-
import pytest
43
from pytest import fixture
54

65
from hiero_sdk_python.crypto.private_key import PrivateKey
@@ -79,8 +78,8 @@ def test_unpause_token_with_invalid_pasue_key(env, pausable_token):
7978
def test_unpause_token_when_token_id_not_set(env):
8079
"""Test unpause transaction when token_id is not provided."""
8180
unpause_tx = TokenUnpauseTransaction()
82-
with pytest.raises(ValueError, match="Missing token ID"):
83-
unpause_tx.freeze_with(env.client)
81+
unpause_receipt = unpause_tx.execute(env.client)
82+
assert unpause_receipt.status == ResponseCode.INVALID_TOKEN_ID
8483

8584

8685
def test_unpause_token_with_invalid_token_id(env):

tests/unit/token_unpause_transaction_test.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,10 @@ def test_build_transaction_body_when_token_id_not_set(mock_account_ids):
6262

6363
unpause_tx = TokenUnpauseTransaction()
6464
unpause_tx.set_node_account_ids([node_account_id])
65+
unpause_tx.operator_account_id = account_id
6566

66-
with pytest.raises(ValueError, match="Missing token ID"):
67-
unpause_tx.build_transaction_body()
67+
transaction_body = unpause_tx.build_transaction_body()
68+
assert not transaction_body.token_unpause.HasField("token")
6869

6970

7071
def test_set_method(mock_account_ids):

0 commit comments

Comments
 (0)