93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
|
package signing_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/tendermint/tendermint/crypto/secp256k1"
|
||
|
|
|
||
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||
|
|
txtypes "github.com/cosmos/cosmos-sdk/types/tx"
|
||
|
|
"github.com/cosmos/cosmos-sdk/x/auth/signing"
|
||
|
|
"github.com/cosmos/cosmos-sdk/x/auth/signing/amino"
|
||
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
||
|
|
"github.com/cosmos/cosmos-sdk/x/bank"
|
||
|
|
)
|
||
|
|
|
||
|
|
func MakeTestHandlerMap() signing.SignModeHandler {
|
||
|
|
return signing.NewSignModeHandlerMap(
|
||
|
|
txtypes.SignMode_SIGN_MODE_LEGACY_AMINO_JSON,
|
||
|
|
[]signing.SignModeHandler{
|
||
|
|
amino.LegacyAminoJSONHandler{},
|
||
|
|
},
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandlerMap_GetSignBytes(t *testing.T) {
|
||
|
|
priv1 := secp256k1.GenPrivKey()
|
||
|
|
addr1 := sdk.AccAddress(priv1.PubKey().Address())
|
||
|
|
priv2 := secp256k1.GenPrivKey()
|
||
|
|
addr2 := sdk.AccAddress(priv2.PubKey().Address())
|
||
|
|
|
||
|
|
coins := sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}
|
||
|
|
|
||
|
|
fee := authtypes.StdFee{
|
||
|
|
Amount: coins,
|
||
|
|
Gas: 10000,
|
||
|
|
}
|
||
|
|
memo := "foo"
|
||
|
|
msgs := []sdk.Msg{
|
||
|
|
&bank.MsgSend{
|
||
|
|
FromAddress: addr1,
|
||
|
|
ToAddress: addr2,
|
||
|
|
Amount: coins,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
tx := authtypes.StdTx{
|
||
|
|
Msgs: msgs,
|
||
|
|
Fee: fee,
|
||
|
|
Signatures: nil,
|
||
|
|
Memo: memo,
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
chainId = "test-chain"
|
||
|
|
accNum uint64 = 7
|
||
|
|
seqNum uint64 = 7
|
||
|
|
)
|
||
|
|
|
||
|
|
handler := MakeTestHandlerMap()
|
||
|
|
aminoJSONHandler := amino.LegacyAminoJSONHandler{}
|
||
|
|
|
||
|
|
signingData := signing.SignerData{
|
||
|
|
ChainID: chainId,
|
||
|
|
AccountNumber: accNum,
|
||
|
|
AccountSequence: seqNum,
|
||
|
|
}
|
||
|
|
signBz, err := handler.GetSignBytes(txtypes.SignMode_SIGN_MODE_LEGACY_AMINO_JSON, signingData, tx)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
expectedSignBz, err := aminoJSONHandler.GetSignBytes(txtypes.SignMode_SIGN_MODE_LEGACY_AMINO_JSON, signingData, tx)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.Equal(t, expectedSignBz, signBz)
|
||
|
|
|
||
|
|
// expect error with wrong sign mode
|
||
|
|
_, err = aminoJSONHandler.GetSignBytes(txtypes.SignMode_SIGN_MODE_DIRECT, signingData, tx)
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandlerMap_DefaultMode(t *testing.T) {
|
||
|
|
handler := MakeTestHandlerMap()
|
||
|
|
require.Equal(t, txtypes.SignMode_SIGN_MODE_LEGACY_AMINO_JSON, handler.DefaultMode())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandlerMap_Modes(t *testing.T) {
|
||
|
|
handler := MakeTestHandlerMap()
|
||
|
|
modes := handler.Modes()
|
||
|
|
require.Contains(t, modes, txtypes.SignMode_SIGN_MODE_LEGACY_AMINO_JSON)
|
||
|
|
require.Len(t, modes, 1)
|
||
|
|
}
|