2019-06-05 16:26:17 -07:00
|
|
|
package types
|
2018-01-06 20:54:04 -08:00
|
|
|
|
|
|
|
import (
|
2019-11-06 12:06:26 -08:00
|
|
|
"encoding/json"
|
2019-09-12 12:22:25 -07:00
|
|
|
"errors"
|
2018-01-06 20:54:04 -08:00
|
|
|
"testing"
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
"github.com/stretchr/testify/require"
|
2019-09-12 12:22:25 -07:00
|
|
|
"github.com/tendermint/tendermint/crypto/secp256k1"
|
2019-01-14 08:11:24 -08:00
|
|
|
|
2018-12-20 11:09:43 -08:00
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
2018-03-02 01:24:07 -08:00
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
2019-09-12 12:22:25 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/x/auth/exported"
|
2018-01-06 20:54:04 -08:00
|
|
|
)
|
|
|
|
|
2018-05-26 14:21:29 -07:00
|
|
|
func TestBaseAddressPubKey(t *testing.T) {
|
2019-06-05 16:26:17 -07:00
|
|
|
_, pub1, addr1 := KeyTestPubAddr()
|
|
|
|
_, pub2, addr2 := KeyTestPubAddr()
|
2018-03-17 13:20:24 -07:00
|
|
|
acc := NewBaseAccountWithAddress(addr1)
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-03-17 13:20:24 -07:00
|
|
|
// check the address (set) and pubkey (not set)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.EqualValues(t, addr1, acc.GetAddress())
|
|
|
|
require.EqualValues(t, nil, acc.GetPubKey())
|
2018-01-17 16:42:05 -08:00
|
|
|
|
2018-03-17 13:53:27 -07:00
|
|
|
// can't override address
|
2018-03-17 13:20:24 -07:00
|
|
|
err := acc.SetAddress(addr2)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, err)
|
|
|
|
require.EqualValues(t, addr1, acc.GetAddress())
|
2018-03-17 13:20:24 -07:00
|
|
|
|
|
|
|
// set the pubkey
|
|
|
|
err = acc.SetPubKey(pub1)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, pub1, acc.GetPubKey())
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-05-06 22:13:32 -07:00
|
|
|
// can override pubkey
|
2018-03-17 13:20:24 -07:00
|
|
|
err = acc.SetPubKey(pub2)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, pub2, acc.GetPubKey())
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-03-17 13:20:24 -07:00
|
|
|
//------------------------------------
|
|
|
|
|
|
|
|
// can set address on empty account
|
|
|
|
acc2 := BaseAccount{}
|
|
|
|
err = acc2.SetAddress(addr2)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.EqualValues(t, addr2, acc2.GetAddress())
|
2018-03-17 13:20:24 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestBaseAccountCoins(t *testing.T) {
|
2019-06-05 16:26:17 -07:00
|
|
|
_, _, addr := KeyTestPubAddr()
|
2018-03-17 13:20:24 -07:00
|
|
|
acc := NewBaseAccountWithAddress(addr)
|
|
|
|
|
2018-07-30 17:09:50 -07:00
|
|
|
someCoins := sdk.Coins{sdk.NewInt64Coin("atom", 123), sdk.NewInt64Coin("eth", 246)}
|
2018-03-17 13:20:24 -07:00
|
|
|
|
|
|
|
err := acc.SetCoins(someCoins)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, someCoins, acc.GetCoins())
|
2018-03-17 13:20:24 -07:00
|
|
|
}
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-03-17 13:20:24 -07:00
|
|
|
func TestBaseAccountSequence(t *testing.T) {
|
2019-06-05 16:26:17 -07:00
|
|
|
_, _, addr := KeyTestPubAddr()
|
2018-03-17 13:20:24 -07:00
|
|
|
acc := NewBaseAccountWithAddress(addr)
|
|
|
|
|
2018-11-26 03:29:21 -08:00
|
|
|
seq := uint64(7)
|
2018-03-17 13:20:24 -07:00
|
|
|
|
|
|
|
err := acc.SetSequence(seq)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, seq, acc.GetSequence())
|
2018-03-17 13:20:24 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestBaseAccountMarshal(t *testing.T) {
|
2019-06-05 16:26:17 -07:00
|
|
|
_, pub, addr := KeyTestPubAddr()
|
2018-03-17 13:20:24 -07:00
|
|
|
acc := NewBaseAccountWithAddress(addr)
|
|
|
|
|
2018-07-30 17:09:50 -07:00
|
|
|
someCoins := sdk.Coins{sdk.NewInt64Coin("atom", 123), sdk.NewInt64Coin("eth", 246)}
|
2018-11-26 03:29:21 -08:00
|
|
|
seq := uint64(7)
|
2018-03-17 13:20:24 -07:00
|
|
|
|
|
|
|
// set everything on the account
|
|
|
|
err := acc.SetPubKey(pub)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
2018-03-17 13:20:24 -07:00
|
|
|
err = acc.SetSequence(seq)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
2018-03-17 13:20:24 -07:00
|
|
|
err = acc.SetCoins(someCoins)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
2018-03-17 13:20:24 -07:00
|
|
|
|
|
|
|
// need a codec for marshaling
|
2018-09-13 11:17:32 -07:00
|
|
|
cdc := codec.New()
|
|
|
|
codec.RegisterCrypto(cdc)
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-11-04 18:28:38 -08:00
|
|
|
b, err := cdc.MarshalBinaryLengthPrefixed(acc)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-03-17 13:20:24 -07:00
|
|
|
acc2 := BaseAccount{}
|
2018-11-04 18:28:38 -08:00
|
|
|
err = cdc.UnmarshalBinaryLengthPrefixed(b, &acc2)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, acc, acc2)
|
2018-01-06 20:54:04 -08:00
|
|
|
|
2018-03-17 13:20:24 -07:00
|
|
|
// error on bad bytes
|
2018-01-15 17:19:39 -08:00
|
|
|
acc2 = BaseAccount{}
|
2018-11-04 18:28:38 -08:00
|
|
|
err = cdc.UnmarshalBinaryLengthPrefixed(b[:len(b)/2], &acc2)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, err)
|
2018-03-17 13:20:24 -07:00
|
|
|
}
|
2019-01-14 08:11:24 -08:00
|
|
|
|
2019-09-12 12:22:25 -07:00
|
|
|
func TestGenesisAccountValidate(t *testing.T) {
|
|
|
|
pubkey := secp256k1.GenPrivKey().PubKey()
|
|
|
|
addr := sdk.AccAddress(pubkey.Address())
|
|
|
|
baseAcc := NewBaseAccount(addr, nil, pubkey, 0, 0)
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
acc exported.GenesisAccount
|
|
|
|
expErr error
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"valid base account",
|
|
|
|
baseAcc,
|
|
|
|
nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"invalid base valid account",
|
|
|
|
NewBaseAccount(addr, sdk.NewCoins(), secp256k1.GenPrivKey().PubKey(), 0, 0),
|
|
|
|
errors.New("pubkey and address pair is invalid"),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
2019-10-17 06:47:35 -07:00
|
|
|
tt := tt
|
2019-09-12 12:22:25 -07:00
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
err := tt.acc.Validate()
|
|
|
|
require.Equal(t, tt.expErr, err)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2019-11-06 12:06:26 -08:00
|
|
|
|
|
|
|
func TestBaseAccountJSON(t *testing.T) {
|
|
|
|
pubkey := secp256k1.GenPrivKey().PubKey()
|
|
|
|
addr := sdk.AccAddress(pubkey.Address())
|
|
|
|
coins := sdk.NewCoins(sdk.NewInt64Coin("test", 5))
|
|
|
|
baseAcc := NewBaseAccount(addr, coins, pubkey, 10, 50)
|
|
|
|
|
|
|
|
bz, err := json.Marshal(baseAcc)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
bz1, err := baseAcc.MarshalJSON()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, string(bz1), string(bz))
|
|
|
|
|
|
|
|
var a BaseAccount
|
|
|
|
require.NoError(t, json.Unmarshal(bz, &a))
|
|
|
|
require.Equal(t, baseAcc.String(), a.String())
|
|
|
|
}
|