89 lines
2.9 KiB
Go
89 lines
2.9 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
gocontext "context"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"cosmossdk.io/log"
|
|
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"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
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"
|
|
)
|
|
|
|
type MintTestSuite struct {
|
|
suite.Suite
|
|
|
|
ctx sdk.Context
|
|
queryClient types.QueryClient
|
|
mintKeeper keeper.Keeper
|
|
}
|
|
|
|
func (suite *MintTestSuite) SetupTest() {
|
|
encCfg := moduletestutil.MakeTestEncodingConfig(codectestutil.CodecOptions{}, mint.AppModule{})
|
|
key := storetypes.NewKVStoreKey(types.StoreKey)
|
|
storeService := runtime.NewEnvironment(runtime.NewKVStoreService(key), log.NewNopLogger())
|
|
testCtx := testutil.DefaultContextWithDB(suite.T(), key, storetypes.NewTransientStoreKey("transient_test"))
|
|
suite.ctx = testCtx.Ctx
|
|
|
|
// gomock initializations
|
|
ctrl := gomock.NewController(suite.T())
|
|
accountKeeper := minttestutil.NewMockAccountKeeper(ctrl)
|
|
bankKeeper := minttestutil.NewMockBankKeeper(ctrl)
|
|
stakingKeeper := minttestutil.NewMockStakingKeeper(ctrl)
|
|
|
|
accountKeeper.EXPECT().GetModuleAddress("mint").Return(sdk.AccAddress{})
|
|
|
|
suite.mintKeeper = keeper.NewKeeper(
|
|
encCfg.Codec,
|
|
storeService,
|
|
stakingKeeper,
|
|
accountKeeper,
|
|
bankKeeper,
|
|
authtypes.FeeCollectorName,
|
|
govModuleNameStr,
|
|
)
|
|
|
|
err := suite.mintKeeper.Params.Set(suite.ctx, types.DefaultParams())
|
|
suite.Require().NoError(err)
|
|
suite.Require().NoError(suite.mintKeeper.Minter.Set(suite.ctx, types.DefaultInitialMinter()))
|
|
queryHelper := baseapp.NewQueryServerTestHelper(testCtx.Ctx, encCfg.InterfaceRegistry)
|
|
types.RegisterQueryServer(queryHelper, keeper.NewQueryServerImpl(suite.mintKeeper))
|
|
|
|
suite.queryClient = types.NewQueryClient(queryHelper)
|
|
}
|
|
|
|
func (suite *MintTestSuite) TestGRPCParams() {
|
|
params, err := suite.queryClient.Params(gocontext.Background(), &types.QueryParamsRequest{})
|
|
suite.Require().NoError(err)
|
|
kparams, err := suite.mintKeeper.Params.Get(suite.ctx)
|
|
suite.Require().NoError(err)
|
|
suite.Require().Equal(params.Params, kparams)
|
|
|
|
inflation, err := suite.queryClient.Inflation(gocontext.Background(), &types.QueryInflationRequest{})
|
|
suite.Require().NoError(err)
|
|
minter, err := suite.mintKeeper.Minter.Get(suite.ctx)
|
|
suite.Require().NoError(err)
|
|
suite.Require().Equal(inflation.Inflation, minter.Inflation)
|
|
|
|
annualProvisions, err := suite.queryClient.AnnualProvisions(gocontext.Background(), &types.QueryAnnualProvisionsRequest{})
|
|
suite.Require().NoError(err)
|
|
suite.Require().Equal(annualProvisions.AnnualProvisions, minter.AnnualProvisions)
|
|
}
|
|
|
|
func TestMintTestSuite(t *testing.T) {
|
|
suite.Run(t, new(MintTestSuite))
|
|
}
|