From 9506444dfec5d567b2c699160a8ebfdd6595d7d1 Mon Sep 17 00:00:00 2001
From: Kirill <paltsev.kir@gmail.com>
Date: Thu, 16 Jan 2025 09:39:08 +0400
Subject: [PATCH] Trigger mock generations through go generate (#120)

---
 Makefile                                | 6 ++----
 internal/clients/bbnclient/interface.go | 1 +
 internal/clients/btcclient/interface.go | 1 +
 internal/db/interface.go                | 1 +
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/Makefile b/Makefile
index 9a9a3e0..dbf32e9 100644
--- a/Makefile
+++ b/Makefile
@@ -42,10 +42,8 @@ run-local:
 	./bin/local-startup.sh;
 	go run cmd/babylon-staking-indexer/main.go --config config/config-local.yml
 
-generate-mock-interface:
-	cd internal/db && mockery --name=DbInterface --output=../../tests/mocks --outpkg=mocks --filename=mock_db_client.go
-	cd internal/clients/btcclient && mockery --name=BtcInterface --output=../../../tests/mocks --outpkg=mocks --filename=mock_btc_client.go
-	cd internal/clients/bbnclient && mockery --name=BbnInterface --output=../../../tests/mocks --outpkg=mocks --filename=mock_bbn_client.go
+generate:
+	go generate ./...
 
 test:
 	./bin/local-startup.sh;
diff --git a/internal/clients/bbnclient/interface.go b/internal/clients/bbnclient/interface.go
index e660789..c035dc1 100644
--- a/internal/clients/bbnclient/interface.go
+++ b/internal/clients/bbnclient/interface.go
@@ -6,6 +6,7 @@ import (
 	ctypes "github.com/cometbft/cometbft/rpc/core/types"
 )
 
+//go:generate mockery --name=BbnInterface --output=../../../tests/mocks --outpkg=mocks --filename=mock_bbn_client.go
 type BbnInterface interface {
 	GetCheckpointParams(ctx context.Context) (*CheckpointParams, error)
 	GetAllStakingParams(ctx context.Context) (map[uint32]*StakingParams, error)
diff --git a/internal/clients/btcclient/interface.go b/internal/clients/btcclient/interface.go
index a6f89c6..4936b2c 100644
--- a/internal/clients/btcclient/interface.go
+++ b/internal/clients/btcclient/interface.go
@@ -1,5 +1,6 @@
 package btcclient
 
+//go:generate mockery --name=BtcInterface --output=../../../tests/mocks --outpkg=mocks --filename=mock_btc_client.go
 type BtcInterface interface {
 	GetTipHeight() (uint64, error)
 }
diff --git a/internal/db/interface.go b/internal/db/interface.go
index 7de7851..dba3432 100644
--- a/internal/db/interface.go
+++ b/internal/db/interface.go
@@ -8,6 +8,7 @@ import (
 	"github.com/babylonlabs-io/babylon-staking-indexer/internal/types"
 )
 
+//go:generate mockery --name=DbInterface --output=../../tests/mocks --outpkg=mocks --filename=mock_db_client.go
 type DbInterface interface {
 	/**
 	 * Ping checks the database connection.