146 lines
4.4 KiB
Go
146 lines
4.4 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
|
|
"github.com/golang/mock/gomock"
|
|
)
|
|
|
|
type MockAccountKeeper struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockAccountKeeperMockRecorder
|
|
}
|
|
|
|
type MockAccountKeeperMockRecorder struct {
|
|
mock *MockAccountKeeper
|
|
}
|
|
|
|
func NewMockAccountKeeper(ctrl *gomock.Controller) *MockAccountKeeper {
|
|
mock := &MockAccountKeeper{ctrl: ctrl}
|
|
mock.recorder = &MockAccountKeeperMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
func (m *MockAccountKeeper) EXPECT() *MockAccountKeeperMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
func (m *MockAccountKeeper) GetModuleAddress(name string) sdk.AccAddress {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetModuleAddress", name)
|
|
ret0, _ := ret[0].(sdk.AccAddress)
|
|
return ret0
|
|
}
|
|
|
|
func (mr *MockAccountKeeperMockRecorder) GetModuleAddress(name interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModuleAddress", reflect.TypeOf((*MockAccountKeeper)(nil).GetModuleAddress), name)
|
|
}
|
|
|
|
type MockBankKeeper struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockBankKeeperMockRecorder
|
|
}
|
|
|
|
type MockBankKeeperMockRecorder struct {
|
|
mock *MockBankKeeper
|
|
}
|
|
|
|
func NewMockBankKeeper(ctrl *gomock.Controller) *MockBankKeeper {
|
|
mock := &MockBankKeeper{ctrl: ctrl}
|
|
mock.recorder = &MockBankKeeperMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
func (m *MockBankKeeper) EXPECT() *MockBankKeeperMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
func (m *MockBankKeeper) GetBalance(ctx context.Context, addr sdk.AccAddress, denom string) sdk.Coin {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetBalance", ctx, addr, denom)
|
|
ret0 := ret[0].(sdk.Coin)
|
|
return ret0
|
|
}
|
|
|
|
func (mr *MockBankKeeperMockRecorder) GetBalance(ctx, addr, denom interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBalance", reflect.TypeOf((*MockBankKeeper)(nil).GetBalance), ctx, addr, denom)
|
|
}
|
|
|
|
func (m *MockBankKeeper) SendCoins(ctx context.Context, fromAddr sdk.AccAddress, toAddr sdk.AccAddress, amt sdk.Coins) error {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "SendCoins", ctx, fromAddr, toAddr, amt)
|
|
return nil
|
|
}
|
|
|
|
func (mr *MockBankKeeperMockRecorder) SendCoins(ctx, fromAddr, toAddr, amt interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendCoins", reflect.TypeOf((*MockBankKeeper)(nil).SendCoins), ctx, fromAddr, toAddr, amt)
|
|
}
|
|
|
|
type MockDistributionKeeperRecorder struct {
|
|
mock *MockDistributionKeeper
|
|
}
|
|
|
|
type MockDistributionKeeper struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockDistributionKeeperRecorder
|
|
}
|
|
|
|
func NewMockDistributionKeeper(ctrl *gomock.Controller) *MockDistributionKeeper {
|
|
mock := &MockDistributionKeeper{ctrl: ctrl}
|
|
mock.recorder = &MockDistributionKeeperRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
func (m *MockDistributionKeeper) EXPECT() *MockDistributionKeeperRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
func (m *MockDistributionKeeper) GetPreviousProposerConsAddr(ctx context.Context) (sdk.ConsAddress, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetPreviousProposerConsAddr", ctx)
|
|
ret0 := ret[0].(sdk.ConsAddress)
|
|
return ret0, nil
|
|
}
|
|
|
|
func (mr *MockDistributionKeeperRecorder) GetPreviousProposerConsAddr(ctx any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPreviousProposerConsAddr", reflect.TypeOf((*MockDistributionKeeper)(nil).GetPreviousProposerConsAddr), ctx)
|
|
}
|
|
|
|
type MockStakingKeeperRecorder struct {
|
|
mock *MockStakingKeeper
|
|
}
|
|
|
|
type MockStakingKeeper struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockStakingKeeperRecorder
|
|
}
|
|
|
|
func NewMockStakingKeeper(ctrl *gomock.Controller) *MockStakingKeeper {
|
|
mock := &MockStakingKeeper{ctrl: ctrl}
|
|
mock.recorder = &MockStakingKeeperRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
func (m *MockStakingKeeper) EXPECT() *MockStakingKeeperRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
func (m *MockStakingKeeper) GetValidatorByConsAddr(ctx context.Context, consAddr sdk.ConsAddress) (stakingtypes.Validator, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetValidatorByConsAddr", ctx, consAddr)
|
|
ret0 := ret[0].(stakingtypes.Validator)
|
|
return ret0, nil
|
|
}
|
|
|
|
func (mr *MockStakingKeeperRecorder) GetValidatorByConsAddr(ctx, consAddr any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetValidatorByConsAddr", reflect.TypeOf((*MockStakingKeeper)(nil).GetValidatorByConsAddr), ctx, consAddr)
|
|
}
|