From 0390254b29641460bd431152dd0049b6e5966480 Mon Sep 17 00:00:00 2001
From: Matt Kocubinski <mkocubinski@gmail.com>
Date: Fri, 17 Jan 2025 08:38:53 -0600
Subject: [PATCH] test: add BeginBlock extension interface to module mocks
 (#23442)

---
 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 5f3e1fe19c..b936ab42e1 100644
--- a/testutil/mock/types_mock_appmodule.go
+++ b/testutil/mock/types_mock_appmodule.go
@@ -197,6 +197,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 interface{}) *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
 }