!test: add msg server tests

testutil/address.go

@@ -2,9 +2,14 @@ package testutil
 
 import (
 	"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
+	cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
+func GenPubKey() cryptotypes.PubKey {
+	return secp256k1.GenPrivKey().PubKey()
+}
+
 func GenAddress() sdk.AccAddress {
 	priv := secp256k1.GenPrivKey()
 

x/multi-staking/keeper/keeper_test.go

@@ -9,24 +9,36 @@ import (
 	tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	minttypes "github.com/cosmos/cosmos-sdk/x/mint/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
 )
 
 type KeeperTestSuite struct {
 	suite.Suite
 
-	ctx      sdk.Context
-	msKeeper *multistakingkeeper.Keeper
+	app       *simapp.SimApp
+	ctx       sdk.Context
+	msKeeper  *multistakingkeeper.Keeper
+	msgServer stakingtypes.MsgServer
 }
 
 func (suite *KeeperTestSuite) SetupTest() {
 	app := simapp.Setup(false)
 	ctx := app.BaseApp.NewContext(false, tmproto.Header{})
+	multiStakingMsgServer := multistakingkeeper.NewMsgServerImpl(app.MultiStakingKeeper)
 
-	suite.ctx, suite.msKeeper = ctx, &app.MultiStakingKeeper
+	suite.app, suite.ctx, suite.msKeeper, suite.msgServer = app, ctx, &app.MultiStakingKeeper, multiStakingMsgServer
 }
 
 func TestKeeperTestSuite(t *testing.T) {
 	suite.Run(t, new(KeeperTestSuite))
 }
 
 // Todo: add CheckBalance; AddAccountWithCoin; FundAccount
+func (suite *KeeperTestSuite) FundAccount(addr sdk.AccAddress, amounts sdk.Coins) error {
+	err := suite.app.BankKeeper.MintCoins(suite.ctx, minttypes.ModuleName, amounts)
+	if err != nil {
+		return err
+	}
+	return suite.app.BankKeeper.SendCoinsFromModuleToAccount(suite.ctx, minttypes.ModuleName, addr, amounts)
+}