From f81a70b3150606c2d24ed028f2605990500eb208 Mon Sep 17 00:00:00 2001
From: Sunny Aggarwal <sunnya97@gmail.com>
Date: Fri, 25 May 2018 20:48:27 -0700
Subject: [PATCH] added keeper tests

---
 x/auth/feekeeper_test.go | 75 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 x/auth/feekeeper_test.go

diff --git a/x/auth/feekeeper_test.go b/x/auth/feekeeper_test.go
new file mode 100644
index 000000000..2f1ffc59b
--- /dev/null
+++ b/x/auth/feekeeper_test.go
@@ -0,0 +1,75 @@
+package auth
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	abci "github.com/tendermint/abci/types"
+	"github.com/tendermint/tmlibs/log"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	wire "github.com/cosmos/cosmos-sdk/wire"
+)
+
+var (
+	emptyCoins = sdk.Coins{}
+	oneCoin    = sdk.Coins{{"foocoin", 1}}
+	twoCoins   = sdk.Coins{{"foocoin", 2}}
+)
+
+func TestFeeCollectionKeeperGetSet(t *testing.T) {
+	ms, _, capKey2 := setupMultiStore()
+	cdc := wire.NewCodec()
+
+	// make context and keeper
+	ctx := sdk.NewContext(ms, abci.Header{}, false, nil, log.NewNopLogger())
+	fck := NewFeeCollectionKeeper(cdc, capKey2)
+
+	// no coins initially
+	currFees := fck.GetCollectedFees(ctx)
+	assert.True(t, currFees.IsEqual(emptyCoins))
+
+	// set feeCollection to oneCoin
+	fck.setCollectedFees(ctx, oneCoin)
+
+	// check that it is equal to oneCoin
+	assert.True(t, fck.GetCollectedFees(ctx).IsEqual(oneCoin))
+}
+
+func TestFeeCollectionKeeperAdd(t *testing.T) {
+	ms, _, capKey2 := setupMultiStore()
+	cdc := wire.NewCodec()
+
+	// make context and keeper
+	ctx := sdk.NewContext(ms, abci.Header{}, false, nil, log.NewNopLogger())
+	fck := NewFeeCollectionKeeper(cdc, capKey2)
+
+	// no coins initially
+	assert.True(t, fck.GetCollectedFees(ctx).IsEqual(emptyCoins))
+
+	// add oneCoin and check that pool is now oneCoin
+	fck.addCollectedFees(ctx, oneCoin)
+	assert.True(t, fck.GetCollectedFees(ctx).IsEqual(oneCoin))
+
+	// add oneCoin again and check that pool is now twoCoins
+	fck.addCollectedFees(ctx, oneCoin)
+	assert.True(t, fck.GetCollectedFees(ctx).IsEqual(twoCoins))
+}
+
+func TestFeeCollectionKeeperClear(t *testing.T) {
+	ms, _, capKey2 := setupMultiStore()
+	cdc := wire.NewCodec()
+
+	// make context and keeper
+	ctx := sdk.NewContext(ms, abci.Header{}, false, nil, log.NewNopLogger())
+	fck := NewFeeCollectionKeeper(cdc, capKey2)
+
+	// set coins initially
+	fck.setCollectedFees(ctx, twoCoins)
+	assert.True(t, fck.GetCollectedFees(ctx).IsEqual(twoCoins))
+
+	// clear fees and see that pool is now empty
+	fck.ClearCollectedFees(ctx)
+	assert.True(t, fck.GetCollectedFees(ctx).IsEqual(emptyCoins))
+}