@@ -92,20 +92,31 @@ def test_build_transaction_body(mock_account_ids):
9292
9393def test_transaction_for_invalid_params (mock_account_ids ):
9494 """Test building the token cancel airdrop transaction body with invalid params."""
95- sender_id , receiver_id , _ , token_id , _ = mock_account_ids
96- sample_pending_airdrop = PendingAirdropId (sender_id = sender_id , receiver_id = receiver_id , token_id = token_id )
95+ sender_id , receiver_id , node_account_id , token_id , _ = mock_account_ids
96+ sample_pending_airdrop = PendingAirdropId (
97+ sender_id = sender_id ,
98+ receiver_id = receiver_id ,
99+ token_id = token_id ,
100+ )
97101
98- # With empty pending airdrops list
99102 cancel_airdrop_tx_1 = TokenCancelAirdropTransaction ()
100- with pytest .raises (ValueError , match = "Pending airdrops list must contain mininum 1 and maximum 10 pendingAirdrop." ):
101- cancel_airdrop_tx_1 .build_transaction_body ()
103+ cancel_airdrop_tx_1 .transaction_id = generate_transaction_id (sender_id )
104+ cancel_airdrop_tx_1 .set_node_account_ids ([node_account_id ])
105+
106+ transaction_body = cancel_airdrop_tx_1 .build_transaction_body ()
107+
108+ assert len (transaction_body .tokenCancelAirdrop .pending_airdrops ) == 0
102109
103- # With pending airdrops list containing more than 10 ids
104110 cancel_airdrop_tx_2 = TokenCancelAirdropTransaction ()
105111 for _ in range (11 ):
106112 cancel_airdrop_tx_2 .add_pending_airdrop (sample_pending_airdrop )
107- with pytest .raises (ValueError , match = "Pending airdrops list must contain mininum 1 and maximum 10 pendingAirdrop." ):
108- cancel_airdrop_tx_2 .build_transaction_body ()
113+
114+ cancel_airdrop_tx_2 .transaction_id = generate_transaction_id (sender_id )
115+ cancel_airdrop_tx_2 .set_node_account_ids ([node_account_id ])
116+
117+ transaction_body = cancel_airdrop_tx_2 .build_transaction_body ()
118+
119+ assert len (transaction_body .tokenCancelAirdrop .pending_airdrops ) == 11
109120
110121
111122def test_set_pending_airdrops (mock_account_ids ):
0 commit comments