From 7539072e723dc246e97bd971bb3aa4b51d34132c Mon Sep 17 00:00:00 2001 From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com> Date: Fri, 17 Jan 2025 20:25:06 +0100 Subject: [PATCH] test: add BeginBlock extension interface to module mocks (port #23442) (#23443) Co-authored-by: Matt Kocubinski Co-authored-by: Julien Robert Co-authored-by: Alex | Interchain Labs --- testutil/mock/types_mock_appmodule.go | 14 ++++++++++++++ types/module/mock_appmodule_test.go | 1 + 2 files changed, 15 insertions(+) diff --git a/testutil/mock/types_mock_appmodule.go b/testutil/mock/types_mock_appmodule.go index 342baad21d..1d6ee346ed 100644 --- a/testutil/mock/types_mock_appmodule.go +++ b/testutil/mock/types_mock_appmodule.go @@ -204,6 +204,20 @@ func (m *MockAppModuleWithAllExtensionsABCI) EXPECT() *MockAppModuleWithAllExten return m.recorder } +// BeginBlock mocks base method. +func (m *MockAppModuleWithAllExtensionsABCI) BeginBlock(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "BeginBlock", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// BeginBlock indicates an expected call of BeginBlock. +func (mr *MockAppModuleWithAllExtensionsABCIMockRecorder) BeginBlock(arg0 any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BeginBlock", reflect.TypeOf((*MockAppModuleWithAllExtensionsABCI)(nil).BeginBlock), arg0) +} + // ConsensusVersion mocks base method. func (m *MockAppModuleWithAllExtensionsABCI) ConsensusVersion() uint64 { m.ctrl.T.Helper() diff --git a/types/module/mock_appmodule_test.go b/types/module/mock_appmodule_test.go index ed4311a891..4e550a174a 100644 --- a/types/module/mock_appmodule_test.go +++ b/types/module/mock_appmodule_test.go @@ -25,6 +25,7 @@ type AppModuleWithAllExtensionsABCI interface { module.HasServices appmodulev2.HasABCIGenesis appmodulev2.HasConsensusVersion + appmodulev2.HasBeginBlocker module.HasABCIEndBlock }