82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
|
package auth
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
abci "github.com/tendermint/abci/types"
|
||
|
|
crypto "github.com/tendermint/go-crypto"
|
||
|
|
oldwire "github.com/tendermint/go-wire"
|
||
|
|
dbm "github.com/tendermint/tmlibs/db"
|
||
|
|
|
||
|
|
"github.com/cosmos/cosmos-sdk/store"
|
||
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func setupMultiStore() (sdk.MultiStore, *sdk.KVStoreKey) {
|
||
|
|
db := dbm.NewMemDB()
|
||
|
|
capKey := sdk.NewKVStoreKey("capkey")
|
||
|
|
ms := store.NewCommitMultiStore(db)
|
||
|
|
ms.MountStoreWithDB(capKey, sdk.StoreTypeIAVL, db)
|
||
|
|
ms.LoadLatestVersion()
|
||
|
|
|
||
|
|
// wire registration while we're at it ... TODO
|
||
|
|
var _ = oldwire.RegisterInterface(
|
||
|
|
struct{ sdk.Account }{},
|
||
|
|
oldwire.ConcreteType{&BaseAccount{}, 0x1},
|
||
|
|
)
|
||
|
|
|
||
|
|
return ms, capKey
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAccountMapperGetSet(t *testing.T) {
|
||
|
|
ms, capKey := setupMultiStore()
|
||
|
|
|
||
|
|
// make context and mapper
|
||
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, nil)
|
||
|
|
mapper := NewAccountMapper(capKey, &BaseAccount{})
|
||
|
|
|
||
|
|
addr := sdk.Address([]byte("some-address"))
|
||
|
|
|
||
|
|
// no account before its created
|
||
|
|
acc := mapper.GetAccount(ctx, addr)
|
||
|
|
assert.Nil(t, acc)
|
||
|
|
|
||
|
|
// create account and check default values
|
||
|
|
acc = mapper.NewAccountWithAddress(ctx, addr)
|
||
|
|
assert.NotNil(t, acc)
|
||
|
|
assert.Equal(t, addr, acc.GetAddress())
|
||
|
|
assert.EqualValues(t, crypto.PubKey{}, acc.GetPubKey())
|
||
|
|
assert.EqualValues(t, 0, acc.GetSequence())
|
||
|
|
|
||
|
|
// NewAccount doesn't call Set, so it's still nil
|
||
|
|
assert.Nil(t, mapper.GetAccount(ctx, addr))
|
||
|
|
|
||
|
|
// set some values on the account and save it
|
||
|
|
newSequence := int64(20)
|
||
|
|
acc.SetSequence(newSequence)
|
||
|
|
mapper.SetAccount(ctx, acc)
|
||
|
|
|
||
|
|
// check the new values
|
||
|
|
acc = mapper.GetAccount(ctx, addr)
|
||
|
|
assert.NotNil(t, acc)
|
||
|
|
assert.Equal(t, newSequence, acc.GetSequence())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAccountMapperSealed(t *testing.T) {
|
||
|
|
_, capKey := setupMultiStore()
|
||
|
|
|
||
|
|
// normal mapper exposes the wire codec
|
||
|
|
mapper := NewAccountMapper(capKey, &BaseAccount{})
|
||
|
|
assert.NotNil(t, mapper.WireCodec())
|
||
|
|
|
||
|
|
// seal mapper, should panic when we try to get the codec
|
||
|
|
mapperSealed := mapper.Seal()
|
||
|
|
assert.Panics(t, func() { mapperSealed.WireCodec() })
|
||
|
|
|
||
|
|
// another way to get a sealed mapper
|
||
|
|
mapperSealed = NewAccountMapperSealed(capKey, &BaseAccount{})
|
||
|
|
assert.Panics(t, func() { mapperSealed.WireCodec() })
|
||
|
|
}
|