66 lines
2.0 KiB
Go
66 lines
2.0 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
storetypes "cosmossdk.io/store/types"
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/golang/mock/gomock"
|
|
testutils "github.com/skip-mev/block-sdk/testutils"
|
|
"github.com/skip-mev/block-sdk/x/builder/keeper"
|
|
"github.com/skip-mev/block-sdk/x/builder/types"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type KeeperTestSuite struct {
|
|
suite.Suite
|
|
|
|
builderKeeper keeper.Keeper
|
|
bankKeeper *testutils.MockBankKeeper
|
|
accountKeeper *testutils.MockAccountKeeper
|
|
distrKeeper *testutils.MockDistributionKeeper
|
|
stakingKeeper *testutils.MockStakingKeeper
|
|
encCfg testutils.EncodingConfig
|
|
ctx sdk.Context
|
|
msgServer types.MsgServer
|
|
key *storetypes.KVStoreKey
|
|
authorityAccount sdk.AccAddress
|
|
}
|
|
|
|
func TestKeeperTestSuite(t *testing.T) {
|
|
suite.Run(t, new(KeeperTestSuite))
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) SetupTest() {
|
|
suite.encCfg = testutils.CreateTestEncodingConfig()
|
|
suite.key = storetypes.NewKVStoreKey(types.StoreKey)
|
|
testCtx := testutil.DefaultContextWithDB(suite.T(), suite.key, storetypes.NewTransientStoreKey("transient_test"))
|
|
suite.ctx = testCtx.Ctx
|
|
|
|
ctrl := gomock.NewController(suite.T())
|
|
|
|
suite.accountKeeper = testutils.NewMockAccountKeeper(ctrl)
|
|
suite.accountKeeper.EXPECT().GetModuleAddress(types.ModuleName).Return(sdk.AccAddress{}).AnyTimes()
|
|
|
|
suite.bankKeeper = testutils.NewMockBankKeeper(ctrl)
|
|
suite.distrKeeper = testutils.NewMockDistributionKeeper(ctrl)
|
|
suite.stakingKeeper = testutils.NewMockStakingKeeper(ctrl)
|
|
suite.authorityAccount = sdk.AccAddress([]byte("authority"))
|
|
suite.builderKeeper = keeper.NewKeeper(
|
|
suite.encCfg.Codec,
|
|
suite.key,
|
|
suite.accountKeeper,
|
|
suite.bankKeeper,
|
|
suite.distrKeeper,
|
|
suite.stakingKeeper,
|
|
suite.authorityAccount.String(),
|
|
)
|
|
|
|
err := suite.builderKeeper.SetParams(suite.ctx, types.DefaultParams())
|
|
suite.Require().NoError(err)
|
|
|
|
suite.msgServer = keeper.NewMsgServerImpl(suite.builderKeeper)
|
|
}
|