2018-03-12 15:25:55 -07:00
|
|
|
package auth
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
"github.com/stretchr/testify/require"
|
2018-12-10 06:27:25 -08:00
|
|
|
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
2018-03-12 15:25:55 -07:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestAccountMapperGetSet(t *testing.T) {
|
2018-12-20 11:09:43 -08:00
|
|
|
input := setupTestInput()
|
2018-07-06 00:06:53 -07:00
|
|
|
addr := sdk.AccAddress([]byte("some-address"))
|
2018-03-12 15:25:55 -07:00
|
|
|
|
|
|
|
// no account before its created
|
2018-12-20 11:09:43 -08:00
|
|
|
acc := input.ak.GetAccount(input.ctx, addr)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, acc)
|
2018-03-12 15:25:55 -07:00
|
|
|
|
|
|
|
// create account and check default values
|
2018-12-20 11:09:43 -08:00
|
|
|
acc = input.ak.NewAccountWithAddress(input.ctx, addr)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, acc)
|
|
|
|
require.Equal(t, addr, acc.GetAddress())
|
|
|
|
require.EqualValues(t, nil, acc.GetPubKey())
|
|
|
|
require.EqualValues(t, 0, acc.GetSequence())
|
2018-03-12 15:25:55 -07:00
|
|
|
|
|
|
|
// NewAccount doesn't call Set, so it's still nil
|
2018-12-20 11:09:43 -08:00
|
|
|
require.Nil(t, input.ak.GetAccount(input.ctx, addr))
|
2018-03-12 15:25:55 -07:00
|
|
|
|
|
|
|
// set some values on the account and save it
|
2018-11-26 03:29:21 -08:00
|
|
|
newSequence := uint64(20)
|
2018-03-12 15:25:55 -07:00
|
|
|
acc.SetSequence(newSequence)
|
2018-12-20 11:09:43 -08:00
|
|
|
input.ak.SetAccount(input.ctx, acc)
|
2018-03-12 15:25:55 -07:00
|
|
|
|
|
|
|
// check the new values
|
2018-12-20 11:09:43 -08:00
|
|
|
acc = input.ak.GetAccount(input.ctx, addr)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, acc)
|
|
|
|
require.Equal(t, newSequence, acc.GetSequence())
|
2018-03-12 15:25:55 -07:00
|
|
|
}
|
2018-09-18 20:00:49 -07:00
|
|
|
|
2018-10-04 07:20:31 -07:00
|
|
|
func TestAccountMapperRemoveAccount(t *testing.T) {
|
2018-12-20 11:09:43 -08:00
|
|
|
input := setupTestInput()
|
2018-10-04 07:20:31 -07:00
|
|
|
addr1 := sdk.AccAddress([]byte("addr1"))
|
|
|
|
addr2 := sdk.AccAddress([]byte("addr2"))
|
|
|
|
|
|
|
|
// create accounts
|
2018-12-20 11:09:43 -08:00
|
|
|
acc1 := input.ak.NewAccountWithAddress(input.ctx, addr1)
|
|
|
|
acc2 := input.ak.NewAccountWithAddress(input.ctx, addr2)
|
2018-10-04 07:20:31 -07:00
|
|
|
|
2018-11-26 03:29:21 -08:00
|
|
|
accSeq1 := uint64(20)
|
|
|
|
accSeq2 := uint64(40)
|
2018-10-04 07:20:31 -07:00
|
|
|
|
|
|
|
acc1.SetSequence(accSeq1)
|
|
|
|
acc2.SetSequence(accSeq2)
|
2018-12-20 11:09:43 -08:00
|
|
|
input.ak.SetAccount(input.ctx, acc1)
|
|
|
|
input.ak.SetAccount(input.ctx, acc2)
|
2018-10-04 07:20:31 -07:00
|
|
|
|
2018-12-20 11:09:43 -08:00
|
|
|
acc1 = input.ak.GetAccount(input.ctx, addr1)
|
2018-10-04 07:20:31 -07:00
|
|
|
require.NotNil(t, acc1)
|
|
|
|
require.Equal(t, accSeq1, acc1.GetSequence())
|
|
|
|
|
|
|
|
// remove one account
|
2018-12-20 11:09:43 -08:00
|
|
|
input.ak.RemoveAccount(input.ctx, acc1)
|
|
|
|
acc1 = input.ak.GetAccount(input.ctx, addr1)
|
2018-10-04 07:20:31 -07:00
|
|
|
require.Nil(t, acc1)
|
|
|
|
|
2018-12-20 11:09:43 -08:00
|
|
|
acc2 = input.ak.GetAccount(input.ctx, addr2)
|
2018-10-04 07:20:31 -07:00
|
|
|
require.NotNil(t, acc2)
|
|
|
|
require.Equal(t, accSeq2, acc2.GetSequence())
|
|
|
|
}
|
2018-12-20 11:09:43 -08:00
|
|
|
|
|
|
|
func TestSetParams(t *testing.T) {
|
|
|
|
input := setupTestInput()
|
|
|
|
params := DefaultParams()
|
|
|
|
|
|
|
|
input.ak.SetParams(input.ctx, params)
|
|
|
|
|
|
|
|
newParams := Params{}
|
|
|
|
input.ak.paramSubspace.Get(input.ctx, KeyTxSigLimit, &newParams.TxSigLimit)
|
|
|
|
require.Equal(t, newParams.TxSigLimit, DefaultTxSigLimit)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestGetParams(t *testing.T) {
|
|
|
|
input := setupTestInput()
|
|
|
|
params := DefaultParams()
|
|
|
|
|
|
|
|
input.ak.SetParams(input.ctx, params)
|
|
|
|
|
|
|
|
newParams := input.ak.GetParams(input.ctx)
|
|
|
|
require.Equal(t, params, newParams)
|
|
|
|
}
|