From 7a4f491ba6df7d922e0d33aa73bc0d15f05324a5 Mon Sep 17 00:00:00 2001
From: Manav Darji <manavdarji.india@gmail.com>
Date: Fri, 22 Sep 2023 15:18:09 +0530
Subject: [PATCH] use ethdb package to generate mocks

---
 Makefile                 |  2 +-
 eth/filters/IDatabase.go | 14 ++++++++------
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/Makefile b/Makefile
index 6a20344a7f..8a482387c8 100644
--- a/Makefile
+++ b/Makefile
@@ -35,7 +35,7 @@ protoc:
 generate-mocks:
 	go generate mockgen -destination=./tests/bor/mocks/IHeimdallClient.go -package=mocks ./consensus/bor IHeimdallClient
 	go generate mockgen -destination=./eth/filters/IBackend.go -package=filters ./eth/filters Backend
-	go generate mockgen -destination=../eth/filters/IDatabase.go -package=filters . Database
+	go generate mockgen -destination=../eth/filters/IDatabase.go -package=filters ./ethdb Database
 
 geth:
 	$(GORUN) build/ci.go install ./cmd/geth
diff --git a/eth/filters/IDatabase.go b/eth/filters/IDatabase.go
index 88c5b41233..6f22bb755c 100644
--- a/eth/filters/IDatabase.go
+++ b/eth/filters/IDatabase.go
@@ -355,11 +355,12 @@ func (mr *MockDatabaseMockRecorder) Tail() *gomock.Call {
 }
 
 // TruncateHead mocks base method.
-func (m *MockDatabase) TruncateHead(arg0 uint64) error {
+func (m *MockDatabase) TruncateHead(arg0 uint64) (uint64, error) {
 	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "TruncateHead", arg0)
-	ret0, _ := ret[0].(error)
-	return ret0
+	ret0, _ := ret[0].(uint64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
 }
 
 // TruncateHead indicates an expected call of TruncateHead.
@@ -369,11 +370,12 @@ func (mr *MockDatabaseMockRecorder) TruncateHead(arg0 interface{}) *gomock.Call
 }
 
 // TruncateTail mocks base method.
-func (m *MockDatabase) TruncateTail(arg0 uint64) error {
+func (m *MockDatabase) TruncateTail(arg0 uint64) (uint64, error) {
 	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "TruncateTail", arg0)
-	ret0, _ := ret[0].(error)
-	return ret0
+	ret0, _ := ret[0].(uint64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
 }
 
 // TruncateTail indicates an expected call of TruncateTail.