@@ -7,7 +7,6 @@ SPDX-License-Identifier: Apache-2.0
77package nfttx_test
88
99import (
10- "context"
1110 "encoding/base64"
1211 "encoding/json"
1312 "errors"
@@ -42,11 +41,11 @@ func TestQueryExecutor_QueryByKey_FilterErrors(t *testing.T) {
4241 qe := nfttx .NewTestQueryExecutor (fakeSelector , fakeVault , 64 )
4342
4443 fakeSelector .FilterReturns (nil , nfttx .ErrNoResults )
45- err := qe .QueryByKey (context . TODO (), & House {}, "LinearID" , "123" )
44+ err := qe .QueryByKey (t . Context (), & House {}, "LinearID" , "123" )
4645 require .ErrorIs (t , err , nfttx .ErrNoResults )
4746
4847 fakeSelector .FilterReturns (nil , errors .New ("some filter error" ))
49- err = qe .QueryByKey (context . TODO (), & House {}, "LinearID" , "123" )
48+ err = qe .QueryByKey (t . Context (), & House {}, "LinearID" , "123" )
5049 require .Error (t , err )
5150 assert .Contains (t , err .Error (), "some filter error" )
5251}
@@ -59,7 +58,7 @@ func TestQueryExecutor_QueryByKey_VaultErrors(t *testing.T) {
5958
6059 fakeSelector .FilterReturns ([]* token2.ID {{TxId : "tx1" , Index : 0 }}, nil )
6160 fakeVault .GetTokensReturns (nil , errors .New ("vault error" ))
62- err := qe .QueryByKey (context . TODO (), & House {}, "LinearID" , "123" )
61+ err := qe .QueryByKey (t . Context (), & House {}, "LinearID" , "123" )
6362 require .Error (t , err )
6463 assert .Contains (t , err .Error (), "vault error" )
6564}
@@ -88,19 +87,19 @@ func TestQueryExecutor_QueryByKey_Success(t *testing.T) {
8887 fakeVault .GetTokensReturns (tokens , nil )
8988
9089 var house House
91- err = qe .QueryByKey (context . TODO (), & house , "LinearID" , "123" )
90+ err = qe .QueryByKey (t . Context (), & house , "LinearID" , "123" )
9291 require .NoError (t , err )
9392 assert .Equal (t , "123" , house .LinearID )
9493
9594 // fail decoding type
9695 tokens [0 ].Type = "not-base64"
97- err = qe .QueryByKey (context . TODO (), & house , "LinearID" , "123" )
96+ err = qe .QueryByKey (t . Context (), & house , "LinearID" , "123" )
9897 require .Error (t , err )
9998 assert .Contains (t , err .Error (), "failed to decode type" )
10099
101100 // test no matching token, empty tokens list
102101 fakeVault .GetTokensReturns ([]* token2.Token {}, nil )
103- err = qe .QueryByKey (context . TODO (), & house , "LinearID" , "123" )
102+ err = qe .QueryByKey (t . Context (), & house , "LinearID" , "123" )
104103 require .ErrorIs (t , err , nfttx .ErrNoResults )
105104}
106105
@@ -116,7 +115,7 @@ func TestQueryExecutor_QueryByKey_BadQuantity(t *testing.T) {
116115 fakeSelector .FilterReturns ([]* token2.ID {{TxId : "tx1" , Index : 0 }}, nil )
117116 fakeVault .GetTokensReturns (tokens , nil )
118117
119- err := qe .QueryByKey (context . TODO (), & House {}, "LinearID" , "123" )
118+ err := qe .QueryByKey (t . Context (), & House {}, "LinearID" , "123" )
120119 require .Error (t , err )
121120 assert .Contains (t , err .Error (), "failed to convert quantity" )
122121}
0 commit comments