mockgen: use typed mocks for compile time check (#10103)

Use `mockgen -typed=true` to generate mocks with type-safe `Return`, `Do`, `DoAndReturn` function - https://github.com/uber-go/mock?tab=readme-ov-file#flags
erigontech · Apr 28, 2024 · 4bc3bc6 · 4bc3bc6
1 parent 63578a8
commit 4bc3bc6
Show file tree

Hide file tree

Showing 48 changed files with 4,983 additions and 464 deletions.
diff --git a/cl/aggregation/mock_services/aggregation_pool_mock.go b/cl/aggregation/mock_services/aggregation_pool_mock.go
diff --git a/cl/aggregation/pool.go b/cl/aggregation/pool.go
@@ -5,7 +5,7 @@ import (
 	"github.com/ledgerwatch/erigon/cl/cltypes/solid"
 )
 
-//go:generate mockgen -destination=./mock_services/aggregation_pool_mock.go -package=mock_services . AggregationPool
+//go:generate mockgen -typed=true -destination=./mock_services/aggregation_pool_mock.go -package=mock_services . AggregationPool
 type AggregationPool interface {
 	// AddAttestation adds a single attestation to the pool.
 	AddAttestation(att *solid.Attestation) error

diff --git a/cl/beacon/synced_data/interface.go b/cl/beacon/synced_data/interface.go
@@ -2,7 +2,7 @@ package synced_data
 
 import "github.com/ledgerwatch/erigon/cl/phase1/core/state"
 
-//go:generate mockgen -destination=./mock_services/synced_data_mock.go -package=mock_services . SyncedData
+//go:generate mockgen -typed=true -destination=./mock_services/synced_data_mock.go -package=mock_services . SyncedData
 type SyncedData interface {
 	OnHeadState(newState *state.CachingBeaconState) (err error)
 	HeadState() *state.CachingBeaconState

diff --git a/cl/beacon/synced_data/mock_services/synced_data_mock.go b/cl/beacon/synced_data/mock_services/synced_data_mock.go
diff --git a/cl/phase1/core/state/interface.go b/cl/phase1/core/state/interface.go
@@ -4,7 +4,7 @@ import libcommon "github.com/ledgerwatch/erigon-lib/common"
 
 // BeaconStateReader is an interface for reading the beacon state.
 //
-//go:generate mockgen -destination=./mock_services/beacon_state_reader_mock.go -package=mock_services . BeaconStateReader
+//go:generate mockgen -typed=true -destination=./mock_services/beacon_state_reader_mock.go -package=mock_services . BeaconStateReader
 type BeaconStateReader interface {
 	ValidatorPublicKey(index int) (libcommon.Bytes48, error)
 	GetDomain(domainType [4]byte, epoch uint64) ([]byte, error)