Generate mocks for automation types

smartcontractkit · Dec 20, 2023 · 39e5668 · 39e5668
1 parent 13f49fd
commit 39e5668
Show file tree

Hide file tree

Showing 20 changed files with 1,207 additions and 43 deletions.
diff --git a/pkg/monitoring/exporter_factory_mock.go b/pkg/monitoring/exporter_factory_mock.go
diff --git a/pkg/monitoring/exporter_mock.go b/pkg/monitoring/exporter_mock.go
diff --git a/pkg/monitoring/metrics_mock.go b/pkg/monitoring/metrics_mock.go
diff --git a/pkg/monitoring/source_factory_mock.go b/pkg/monitoring/source_factory_mock.go
diff --git a/pkg/monitoring/source_mock.go b/pkg/monitoring/source_mock.go
diff --git a/pkg/types/automation/interfaces.go b/pkg/types/automation/interfaces.go
@@ -48,47 +48,47 @@ type UpkeepStateReader interface {
 	SelectByWorkIDs(ctx context.Context, workIDs ...string) ([]UpkeepState, error)
 }
 
-//go:generate mockery --name Encoder --structname MockEncoder --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename encoder.generated.go
+//go:generate mockery --name Encoder --structname MockEncoder --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename encoder.generated.go
 type Encoder interface {
 	Encode(...CheckResult) ([]byte, error)
 	Extract([]byte) ([]ReportedUpkeep, error)
 }
 
-//go:generate mockery --name LogEventProvider --structname MockLogEventProvider --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename logeventprovider.generated.go
+//go:generate mockery --name LogEventProvider --structname MockLogEventProvider --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename logeventprovider.generated.go
 type LogEventProvider interface {
 	GetLatestPayloads(context.Context) ([]UpkeepPayload, error)
 	Start(context.Context) error
 	Close() error
 }
 
-//go:generate mockery --name RecoverableProvider --structname MockRecoverableProvider --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename recoverableprovider.generated.go
+//go:generate mockery --name RecoverableProvider --structname MockRecoverableProvider --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename recoverableprovider.generated.go
 type RecoverableProvider interface {
 	GetRecoveryProposals(context.Context) ([]UpkeepPayload, error)
 }
 
-//go:generate mockery --name TransmitEventProvider --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename transmit_event_provider.generated.go
+//go:generate mockery --name TransmitEventProvider --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename transmit_event_provider.generated.go
 type TransmitEventProvider interface {
 	GetLatestEvents(context.Context) ([]TransmitEvent, error)
 }
 
-//go:generate mockery --name ConditionalUpkeepProvider --structname MockConditionalUpkeepProvider --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename conditionalupkeepprovider.generated.go
+//go:generate mockery --name ConditionalUpkeepProvider --structname MockConditionalUpkeepProvider --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename conditionalupkeepprovider.generated.go
 type ConditionalUpkeepProvider interface {
 	GetActiveUpkeeps(context.Context) ([]UpkeepPayload, error)
 }
 
-//go:generate mockery --name PayloadBuilder --structname MockPayloadBuilder --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename payloadbuilder.generated.go
+//go:generate mockery --name PayloadBuilder --structname MockPayloadBuilder --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename payloadbuilder.generated.go
 type PayloadBuilder interface {
 	// Can get payloads for a subset of proposals along with an error
 	BuildPayloads(context.Context, ...CoordinatedBlockProposal) ([]UpkeepPayload, error)
 }
 
-//go:generate mockery --name Runnable --structname MockRunnable --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename runnable.generated.go
+//go:generate mockery --name Runnable --structname MockRunnable --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename runnable.generated.go
 type Runnable interface {
 	// Can get results for a subset of payloads along with an error
 	CheckUpkeeps(context.Context, ...UpkeepPayload) ([]CheckResult, error)
 }
 
-//go:generate mockery --name BlockSubscriber --structname MockBlockSubscriber --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename block_subscriber.generated.go
+//go:generate mockery --name BlockSubscriber --structname MockBlockSubscriber --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename block_subscriber.generated.go
 type BlockSubscriber interface {
 	// Subscribe provides an identifier integer, a new channel, and potentially an error
 	Subscribe() (int, chan BlockHistory, error)
@@ -98,7 +98,7 @@ type BlockSubscriber interface {
 	Close() error
 }
 
-//go:generate mockery --name UpkeepStateUpdater --structname MockUpkeepStateUpdater --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename upkeep_state_updater.generated.go
+//go:generate mockery --name UpkeepStateUpdater --structname MockUpkeepStateUpdater --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename upkeep_state_updater.generated.go
 type UpkeepStateUpdater interface {
 	SetUpkeepState(context.Context, CheckResult, UpkeepState) error
 }
@@ -117,14 +117,14 @@ type ProposalQueue interface {
 	Dequeue(t UpkeepType, n int) ([]CoordinatedBlockProposal, error)
 }
 
-//go:generate mockery --name ResultStore --structname MockResultStore --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename result_store.generated.go
+//go:generate mockery --name ResultStore --structname MockResultStore --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename result_store.generated.go
 type ResultStore interface {
 	Add(...CheckResult)
 	Remove(...string)
 	View() ([]CheckResult, error)
 }
 
-//go:generate mockery --name Coordinator --structname MockCoordinator --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename coordinator.generated.go
+//go:generate mockery --name Coordinator --structname MockCoordinator --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename coordinator.generated.go
 type Coordinator interface {
 	PreProcess(_ context.Context, payloads []UpkeepPayload) ([]UpkeepPayload, error)
 
@@ -134,7 +134,7 @@ type Coordinator interface {
 	FilterProposals([]CoordinatedBlockProposal) ([]CoordinatedBlockProposal, error)
 }
 
-//go:generate mockery --name MetadataStore --structname MockMetadataStore --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename metadatastore.generated.go
+//go:generate mockery --name MetadataStore --structname MockMetadataStore --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename metadatastore.generated.go
 type MetadataStore interface {
 	SetBlockHistory(BlockHistory)
 	GetBlockHistory() BlockHistory
@@ -147,7 +147,7 @@ type MetadataStore interface {
 	Close() error
 }
 
-//go:generate mockery --name Ratio --structname MockRatio --srcpkg "github.com/smartcontractkit/chainlink-automation/pkg/v3/types" --case underscore --filename ratio.generated.go
+//go:generate mockery --name Ratio --structname MockRatio --srcpkg "github.com/smartcontractkit/chainlink-common/pkg/types/automation" --case underscore --filename ratio.generated.go
 type Ratio interface {
 	// OfInt should return n out of x such that n/x ~ r (ratio)
 	OfInt(int) int

diff --git a/pkg/types/automation/mocks/block_subscriber.generated.go b/pkg/types/automation/mocks/block_subscriber.generated.go