76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
package auth
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
codec "github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
var (
|
|
emptyCoins = sdk.Coins{}
|
|
oneCoin = sdk.Coins{sdk.NewInt64Coin("foocoin", 1)}
|
|
twoCoins = sdk.Coins{sdk.NewInt64Coin("foocoin", 2)}
|
|
)
|
|
|
|
func TestFeeCollectionKeeperGetSet(t *testing.T) {
|
|
ms, _, capKey2 := setupMultiStore()
|
|
cdc := codec.New()
|
|
|
|
// make context and keeper
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
|
|
fck := NewFeeCollectionKeeper(cdc, capKey2)
|
|
|
|
// no coins initially
|
|
currFees := fck.GetCollectedFees(ctx)
|
|
require.True(t, currFees.IsEqual(emptyCoins))
|
|
|
|
// set feeCollection to oneCoin
|
|
fck.setCollectedFees(ctx, oneCoin)
|
|
|
|
// check that it is equal to oneCoin
|
|
require.True(t, fck.GetCollectedFees(ctx).IsEqual(oneCoin))
|
|
}
|
|
|
|
func TestFeeCollectionKeeperAdd(t *testing.T) {
|
|
ms, _, capKey2 := setupMultiStore()
|
|
cdc := codec.New()
|
|
|
|
// make context and keeper
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
|
|
fck := NewFeeCollectionKeeper(cdc, capKey2)
|
|
|
|
// no coins initially
|
|
require.True(t, fck.GetCollectedFees(ctx).IsEqual(emptyCoins))
|
|
|
|
// add oneCoin and check that pool is now oneCoin
|
|
fck.addCollectedFees(ctx, oneCoin)
|
|
require.True(t, fck.GetCollectedFees(ctx).IsEqual(oneCoin))
|
|
|
|
// add oneCoin again and check that pool is now twoCoins
|
|
fck.addCollectedFees(ctx, oneCoin)
|
|
require.True(t, fck.GetCollectedFees(ctx).IsEqual(twoCoins))
|
|
}
|
|
|
|
func TestFeeCollectionKeeperClear(t *testing.T) {
|
|
ms, _, capKey2 := setupMultiStore()
|
|
cdc := codec.New()
|
|
|
|
// make context and keeper
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
|
|
fck := NewFeeCollectionKeeper(cdc, capKey2)
|
|
|
|
// set coins initially
|
|
fck.setCollectedFees(ctx, twoCoins)
|
|
require.True(t, fck.GetCollectedFees(ctx).IsEqual(twoCoins))
|
|
|
|
// clear fees and see that pool is now empty
|
|
fck.ClearCollectedFees(ctx)
|
|
require.True(t, fck.GetCollectedFees(ctx).IsEqual(emptyCoins))
|
|
}
|