From 1955e3436fd9d0c6bedcc2c1196b6d40920df5fa Mon Sep 17 00:00:00 2001
From: Ethan Buchman <ethan@coinculture.info>
Date: Mon, 12 Mar 2018 23:25:55 +0100
Subject: [PATCH] x/auth: mapper_test.go

---
 x/auth/mapper.go      |  4 +++
 x/auth/mapper_test.go | 81 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 85 insertions(+)
 create mode 100644 x/auth/mapper_test.go

diff --git a/x/auth/mapper.go b/x/auth/mapper.go
index b50e4405fd..1176db3cbe 100644
--- a/x/auth/mapper.go
+++ b/x/auth/mapper.go
@@ -11,6 +11,9 @@ import (
 	wire "github.com/cosmos/cosmos-sdk/wire"
 )
 
+var _ sdk.AccountMapper = (*accountMapper)(nil)
+var _ sdk.AccountMapper = (*sealedAccountMapper)(nil)
+
 // Implements sdk.AccountMapper.
 // This AccountMapper encodes/decodes accounts using the
 // go-wire (binary) encoding/decoding library.
@@ -108,6 +111,7 @@ func (sam sealedAccountMapper) WireCodec() *wire.Codec {
 //----------------------------------------
 // misc.
 
+// NOTE: currently unused
 func (am accountMapper) clonePrototypePtr() interface{} {
 	protoRt := reflect.TypeOf(am.proto)
 	if protoRt.Kind() == reflect.Ptr {
diff --git a/x/auth/mapper_test.go b/x/auth/mapper_test.go
new file mode 100644
index 0000000000..4ac96c3810
--- /dev/null
+++ b/x/auth/mapper_test.go
@@ -0,0 +1,81 @@
+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() })
+}