81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
)
|
|
|
|
func TestAccountMapperGetSet(t *testing.T) {
|
|
app, ctx := createTestApp(true)
|
|
addr := sdk.AccAddress([]byte("some-address"))
|
|
|
|
// no account before its created
|
|
acc := app.AccountKeeper.GetAccount(ctx, addr)
|
|
require.Nil(t, acc)
|
|
|
|
// create account and check default values
|
|
acc = app.AccountKeeper.NewAccountWithAddress(ctx, addr)
|
|
require.NotNil(t, acc)
|
|
require.Equal(t, addr, acc.GetAddress())
|
|
require.EqualValues(t, nil, acc.GetPubKey())
|
|
require.EqualValues(t, 0, acc.GetSequence())
|
|
|
|
// NewAccount doesn't call Set, so it's still nil
|
|
require.Nil(t, app.AccountKeeper.GetAccount(ctx, addr))
|
|
|
|
// set some values on the account and save it
|
|
newSequence := uint64(20)
|
|
acc.SetSequence(newSequence)
|
|
app.AccountKeeper.SetAccount(ctx, acc)
|
|
|
|
// check the new values
|
|
acc = app.AccountKeeper.GetAccount(ctx, addr)
|
|
require.NotNil(t, acc)
|
|
require.Equal(t, newSequence, acc.GetSequence())
|
|
}
|
|
|
|
func TestAccountMapperRemoveAccount(t *testing.T) {
|
|
app, ctx := createTestApp(true)
|
|
addr1 := sdk.AccAddress([]byte("addr1"))
|
|
addr2 := sdk.AccAddress([]byte("addr2"))
|
|
|
|
// create accounts
|
|
acc1 := app.AccountKeeper.NewAccountWithAddress(ctx, addr1)
|
|
acc2 := app.AccountKeeper.NewAccountWithAddress(ctx, addr2)
|
|
|
|
accSeq1 := uint64(20)
|
|
accSeq2 := uint64(40)
|
|
|
|
acc1.SetSequence(accSeq1)
|
|
acc2.SetSequence(accSeq2)
|
|
app.AccountKeeper.SetAccount(ctx, acc1)
|
|
app.AccountKeeper.SetAccount(ctx, acc2)
|
|
|
|
acc1 = app.AccountKeeper.GetAccount(ctx, addr1)
|
|
require.NotNil(t, acc1)
|
|
require.Equal(t, accSeq1, acc1.GetSequence())
|
|
|
|
// remove one account
|
|
app.AccountKeeper.RemoveAccount(ctx, acc1)
|
|
acc1 = app.AccountKeeper.GetAccount(ctx, addr1)
|
|
require.Nil(t, acc1)
|
|
|
|
acc2 = app.AccountKeeper.GetAccount(ctx, addr2)
|
|
require.NotNil(t, acc2)
|
|
require.Equal(t, accSeq2, acc2.GetSequence())
|
|
}
|
|
|
|
func TestGetSetParams(t *testing.T) {
|
|
app, ctx := createTestApp(true)
|
|
params := types.DefaultParams()
|
|
|
|
app.AccountKeeper.SetParams(ctx, params)
|
|
|
|
actualParams := app.AccountKeeper.GetParams(ctx)
|
|
require.Equal(t, params, actualParams)
|
|
}
|