test: remove TokenBalanceQuery from unrelated test

hiero-ledger · Oct 27, 2020 · 2ca564b · 2ca564b
1 parent 9d21ceb
commit 2ca564b
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/token_transfer_transaction_test.go b/token_transfer_transaction_test.go
@@ -74,16 +74,9 @@ func TestTokenTransferTransaction_Execute(t *testing.T) {
 		Execute(client)
 	assert.NoError(t, err)
 
-	balance, err := NewTokenBalanceQuery().
-		SetAccountID(accountID).
-		Execute(client)
-	assert.NoError(t, err)
-
 	receipt, err = txID.GetReceipt(client)
 	assert.NoError(t, err)
 
-	println("balance",receipt.tokenID.String(), " ", balance[*receipt.tokenID])
-
 	txID, err = NewTokenBurnTransaction().
 		SetTokenID(tokenID).
 		SetAmount(10).
@@ -92,7 +85,7 @@ func TestTokenTransferTransaction_Execute(t *testing.T) {
 
 	txID, err = NewTokenAssociateTransaction().
 		SetAccountID(accountID).
-		SetTokenIDs(tokenID).
+		AddTokenID(tokenID).
 		Execute(client)
 	assert.NoError(t, err)
 
@@ -135,7 +128,7 @@ func TestTokenTransferTransaction_Execute(t *testing.T) {
 
 	txID, err = NewTokenDissociateTransaction().
 		SetAccountID(accountID).
-		SetTokenIDs(tokenID).
+		AddTokenID(tokenID).
 		Execute(client)
 	assert.NoError(t, err)