5
5
6
6
"github.com/stretchr/testify/require"
7
7
8
+ apisigning "cosmossdk.io/api/cosmos/tx/signing/v1beta1"
9
+
8
10
"github.com/cosmos/cosmos-sdk/client/tx"
9
11
"github.com/cosmos/cosmos-sdk/codec"
10
12
"github.com/cosmos/cosmos-sdk/codec/testutil"
@@ -16,7 +18,6 @@ import (
16
18
sdk "github.com/cosmos/cosmos-sdk/types"
17
19
moduletestutil "github.com/cosmos/cosmos-sdk/types/module/testutil"
18
20
typestx "github.com/cosmos/cosmos-sdk/types/tx"
19
- "github.com/cosmos/cosmos-sdk/types/tx/signing"
20
21
)
21
22
22
23
const (
@@ -53,7 +54,7 @@ func TestAuxTxBuilder(t *testing.T) {
53
54
{
54
55
"cannot set SIGN_MODE_DIRECT" ,
55
56
func () error {
56
- return b .SetSignMode (signing .SignMode_SIGN_MODE_DIRECT )
57
+ return b .SetSignMode (apisigning .SignMode_SIGN_MODE_DIRECT )
57
58
},
58
59
true , "AuxTxBuilder can only sign with SIGN_MODE_DIRECT_AUX or SIGN_MODE_LEGACY_AMINO_JSON" ,
59
60
},
@@ -105,7 +106,7 @@ func TestAuxTxBuilder(t *testing.T) {
105
106
func () error {
106
107
require .NoError (t , b .SetMsgs (msg1 ))
107
108
require .NoError (t , b .SetPubKey (pub1 ))
108
- require .NoError (t , b .SetSignMode (signing .SignMode_SIGN_MODE_DIRECT_AUX ))
109
+ require .NoError (t , b .SetSignMode (apisigning .SignMode_SIGN_MODE_DIRECT_AUX ))
109
110
110
111
_ , err := b .GetSignBytes ()
111
112
return err
@@ -117,7 +118,7 @@ func TestAuxTxBuilder(t *testing.T) {
117
118
func () error {
118
119
require .NoError (t , b .SetMsgs (msg1 ))
119
120
require .NoError (t , b .SetPubKey (pub1 ))
120
- require .NoError (t , b .SetSignMode (signing .SignMode_SIGN_MODE_DIRECT_AUX ))
121
+ require .NoError (t , b .SetSignMode (apisigning .SignMode_SIGN_MODE_DIRECT_AUX ))
121
122
122
123
_ , err := b .GetSignBytes ()
123
124
require .NoError (t , err )
@@ -133,7 +134,7 @@ func TestAuxTxBuilder(t *testing.T) {
133
134
require .NoError (t , b .SetMsgs (msg1 ))
134
135
require .NoError (t , b .SetPubKey (pub1 ))
135
136
b .SetAddress (addr1Str )
136
- require .NoError (t , b .SetSignMode (signing .SignMode_SIGN_MODE_DIRECT_AUX ))
137
+ require .NoError (t , b .SetSignMode (apisigning .SignMode_SIGN_MODE_DIRECT_AUX ))
137
138
138
139
_ , err := b .GetSignBytes ()
139
140
require .NoError (t , err )
@@ -154,7 +155,7 @@ func TestAuxTxBuilder(t *testing.T) {
154
155
require .NoError (t , b .SetMsgs (msg1 ))
155
156
require .NoError (t , b .SetPubKey (pub1 ))
156
157
b .SetAddress (addr1Str )
157
- err := b .SetSignMode (signing .SignMode_SIGN_MODE_DIRECT_AUX )
158
+ err := b .SetSignMode (apisigning .SignMode_SIGN_MODE_DIRECT_AUX )
158
159
require .NoError (t , err )
159
160
160
161
_ , err = b .GetSignBytes ()
@@ -164,7 +165,7 @@ func TestAuxTxBuilder(t *testing.T) {
164
165
auxSignerData , err := b .GetAuxSignerData ()
165
166
166
167
// Make sure auxSignerData is correctly populated
167
- checkCorrectData (t , cdc , auxSignerData , signing .SignMode_SIGN_MODE_DIRECT_AUX )
168
+ checkCorrectData (t , cdc , auxSignerData , apisigning .SignMode_SIGN_MODE_DIRECT_AUX )
168
169
169
170
return err
170
171
},
@@ -176,7 +177,7 @@ func TestAuxTxBuilder(t *testing.T) {
176
177
require .NoError (t , b .SetMsgs (msg1 ))
177
178
require .NoError (t , b .SetPubKey (pub1 ))
178
179
b .SetAddress (addr1Str )
179
- err := b .SetSignMode (signing .SignMode_SIGN_MODE_LEGACY_AMINO_JSON )
180
+ err := b .SetSignMode (apisigning .SignMode_SIGN_MODE_LEGACY_AMINO_JSON )
180
181
require .NoError (t , err )
181
182
182
183
_ , err = b .GetSignBytes ()
@@ -195,7 +196,7 @@ func TestAuxTxBuilder(t *testing.T) {
195
196
require .NoError (t , b .SetMsgs (msg1 ))
196
197
require .NoError (t , b .SetPubKey (pub1 ))
197
198
b .SetAddress (addr1Str )
198
- err := b .SetSignMode (signing .SignMode_SIGN_MODE_LEGACY_AMINO_JSON )
199
+ err := b .SetSignMode (apisigning .SignMode_SIGN_MODE_LEGACY_AMINO_JSON )
199
200
require .NoError (t , err )
200
201
201
202
_ , err = b .GetSignBytes ()
@@ -205,7 +206,7 @@ func TestAuxTxBuilder(t *testing.T) {
205
206
auxSignerData , err := b .GetAuxSignerData ()
206
207
207
208
// Make sure auxSignerData is correctly populated
208
- checkCorrectData (t , cdc , auxSignerData , signing .SignMode_SIGN_MODE_LEGACY_AMINO_JSON )
209
+ checkCorrectData (t , cdc , auxSignerData , apisigning .SignMode_SIGN_MODE_LEGACY_AMINO_JSON )
209
210
210
211
return err
211
212
},
@@ -229,7 +230,7 @@ func TestAuxTxBuilder(t *testing.T) {
229
230
}
230
231
231
232
// checkCorrectData that the auxSignerData's content matches the inputs we gave.
232
- func checkCorrectData (t * testing.T , cdc codec.Codec , auxSignerData typestx.AuxSignerData , signMode signing .SignMode ) {
233
+ func checkCorrectData (t * testing.T , cdc codec.Codec , auxSignerData typestx.AuxSignerData , signMode apisigning .SignMode ) {
233
234
t .Helper ()
234
235
pkAny , err := codectypes .NewAnyWithValue (pub1 )
235
236
require .NoError (t , err )
@@ -247,6 +248,6 @@ func checkCorrectData(t *testing.T, cdc codec.Codec, auxSignerData typestx.AuxSi
247
248
require .Equal (t , chainID , auxSignerData .SignDoc .ChainId )
248
249
require .Equal (t , msgAny , body .GetMessages ()[0 ])
249
250
require .Equal (t , pkAny , auxSignerData .SignDoc .PublicKey )
250
- require .Equal (t , signMode , auxSignerData .Mode )
251
+ require .Equal (t , signMode , apisigning . SignMode ( auxSignerData .Mode ) )
251
252
require .Equal (t , rawSig , auxSignerData .Sig )
252
253
}
0 commit comments