150ca963f2
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Co-authored-by: Marko <marko@baricevic.me>
93 lines
3.0 KiB
Go
93 lines
3.0 KiB
Go
package mint_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"cosmossdk.io/log"
|
|
"cosmossdk.io/math"
|
|
storetypes "cosmossdk.io/store/types"
|
|
authtypes "cosmossdk.io/x/auth/types"
|
|
"cosmossdk.io/x/mint"
|
|
"cosmossdk.io/x/mint/keeper"
|
|
minttestutil "cosmossdk.io/x/mint/testutil"
|
|
"cosmossdk.io/x/mint/types"
|
|
|
|
codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
moduletestutil "github.com/cosmos/cosmos-sdk/types/module/testutil"
|
|
)
|
|
|
|
const govModuleNameStr = "cosmos10d07y265gmmuvt4z0w9aw880jnsr700j6zn9kn"
|
|
|
|
type ModuleTestSuite struct {
|
|
suite.Suite
|
|
|
|
mintKeeper keeper.Keeper
|
|
ctx sdk.Context
|
|
msgServer types.MsgServer
|
|
stakingKeeper *minttestutil.MockStakingKeeper
|
|
bankKeeper *minttestutil.MockBankKeeper
|
|
|
|
appmodule mint.AppModule
|
|
}
|
|
|
|
func TestKeeperTestSuite(t *testing.T) {
|
|
suite.Run(t, new(ModuleTestSuite))
|
|
}
|
|
|
|
func (s *ModuleTestSuite) SetupTest() {
|
|
encCfg := moduletestutil.MakeTestEncodingConfig(codectestutil.CodecOptions{}, mint.AppModule{})
|
|
key := storetypes.NewKVStoreKey(types.StoreKey)
|
|
storeService := runtime.NewKVStoreService(key)
|
|
env := runtime.NewEnvironment(storeService, log.NewNopLogger())
|
|
testCtx := testutil.DefaultContextWithDB(s.T(), key, storetypes.NewTransientStoreKey("transient_test"))
|
|
s.ctx = testCtx.Ctx
|
|
|
|
// gomock initializations
|
|
ctrl := gomock.NewController(s.T())
|
|
accountKeeper := minttestutil.NewMockAccountKeeper(ctrl)
|
|
bankKeeper := minttestutil.NewMockBankKeeper(ctrl)
|
|
stakingKeeper := minttestutil.NewMockStakingKeeper(ctrl)
|
|
|
|
accountKeeper.EXPECT().GetModuleAddress(types.ModuleName).Return(sdk.AccAddress{})
|
|
|
|
s.mintKeeper = keeper.NewKeeper(
|
|
encCfg.Codec,
|
|
env,
|
|
stakingKeeper,
|
|
accountKeeper,
|
|
bankKeeper,
|
|
authtypes.FeeCollectorName,
|
|
govModuleNameStr,
|
|
)
|
|
s.stakingKeeper = stakingKeeper
|
|
s.bankKeeper = bankKeeper
|
|
|
|
err := s.mintKeeper.Params.Set(s.ctx, types.DefaultParams())
|
|
s.NoError(err)
|
|
|
|
s.NoError(s.mintKeeper.Minter.Set(s.ctx, types.DefaultInitialMinter()))
|
|
s.msgServer = keeper.NewMsgServerImpl(s.mintKeeper)
|
|
|
|
s.appmodule = mint.NewAppModule(encCfg.Codec, s.mintKeeper, accountKeeper, s.mintKeeper.DefaultMintFn(types.DefaultInflationCalculationFn))
|
|
}
|
|
|
|
func (s *ModuleTestSuite) TestEpochHooks() {
|
|
s.stakingKeeper.EXPECT().StakingTokenSupply(s.ctx).Return(math.NewIntFromUint64(100000000000), nil).AnyTimes()
|
|
bondedRatio := math.LegacyNewDecWithPrec(15, 2)
|
|
s.stakingKeeper.EXPECT().BondedRatio(s.ctx).Return(bondedRatio, nil).AnyTimes()
|
|
s.bankKeeper.EXPECT().MintCoins(s.ctx, types.ModuleName, sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(792)))).Return(nil)
|
|
s.bankKeeper.EXPECT().SendCoinsFromModuleToModule(s.ctx, types.ModuleName, authtypes.FeeCollectorName, gomock.Any()).Return(nil)
|
|
|
|
err := s.appmodule.BeforeEpochStart(s.ctx, "block", -1)
|
|
s.NoError(err)
|
|
|
|
err = s.appmodule.AfterEpochEnd(s.ctx, "epochIdentifier", 1) // just to get coverage up
|
|
s.NoError(err)
|
|
}
|