fix go generate command in interface files

Signed-off-by: Alex Wu <[email protected]>
flyteorg · Jan 28, 2025 · 0e69dd2 · 0e69dd2
1 parent 3f86dca
commit 0e69dd2
Show file tree

Hide file tree

Showing 11 changed files with 34 additions and 58 deletions.
diff --git a/flyteadmin/pkg/manager/interfaces/execution.go b/flyteadmin/pkg/manager/interfaces/execution.go
@@ -7,6 +7,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=ExecutionInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing Flyte Workflow Executions
 type ExecutionInterface interface {
 	CreateExecution(ctx context.Context, request *admin.ExecutionCreateRequest, requestedAt time.Time) (

diff --git a/flyteadmin/pkg/manager/interfaces/launch_plan.go b/flyteadmin/pkg/manager/interfaces/launch_plan.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=LaunchPlanInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing Flyte Launch Plans
 type LaunchPlanInterface interface {
 	// Interface to create Launch Plans based on the request.

diff --git a/flyteadmin/pkg/manager/interfaces/named_entity.go b/flyteadmin/pkg/manager/interfaces/named_entity.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=NamedEntityInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing metadata associated with NamedEntityIdentifiers
 type NamedEntityInterface interface {
 	GetNamedEntity(ctx context.Context, request *admin.NamedEntityGetRequest) (*admin.NamedEntity, error)

diff --git a/flyteadmin/pkg/manager/interfaces/node_execution.go b/flyteadmin/pkg/manager/interfaces/node_execution.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=NodeExecutionInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing Flyte Workflow NodeExecutions
 type NodeExecutionInterface interface {
 	CreateNodeEvent(ctx context.Context, request *admin.NodeExecutionEventRequest) (

diff --git a/flyteadmin/pkg/manager/interfaces/project.go b/flyteadmin/pkg/manager/interfaces/project.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=ProjectInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing projects (and domains).
 type ProjectInterface interface {
 	CreateProject(ctx context.Context, request *admin.ProjectRegisterRequest) (*admin.ProjectRegisterResponse, error)

diff --git a/flyteadmin/pkg/manager/interfaces/resource.go b/flyteadmin/pkg/manager/interfaces/resource.go
@@ -6,7 +6,7 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
-//go:generate mockery -name ResourceInterface -output=../mocks -case=underscore
+//go:generate mockery-v2 --name=ResourceInterface --output=../mocks --case=underscore --with-expecter
 
 // ResourceInterface manages project, domain and workflow -specific attributes.
 type ResourceInterface interface {

diff --git a/flyteadmin/pkg/manager/interfaces/task.go b/flyteadmin/pkg/manager/interfaces/task.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=TaskInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing Flyte Tasks
 type TaskInterface interface {
 	CreateTask(ctx context.Context, request *admin.TaskCreateRequest) (*admin.TaskCreateResponse, error)

diff --git a/flyteadmin/pkg/manager/interfaces/task_execution.go b/flyteadmin/pkg/manager/interfaces/task_execution.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=TaskExecutionInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing Flyte Workflow TaskExecutions
 type TaskExecutionInterface interface {
 	CreateTaskExecutionEvent(ctx context.Context, request *admin.TaskExecutionEventRequest) (

diff --git a/flyteadmin/pkg/manager/interfaces/version.go b/flyteadmin/pkg/manager/interfaces/version.go
@@ -6,7 +6,7 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
-//go:generate mockery -name VersionInterface -output=../mocks -case=underscore
+//go:generate mockery-v2 --name=VersionInterface --output=../mocks --case=underscore --with-expecter
 
 // Interface for managing Flyte admin version
 type VersionInterface interface {

diff --git a/flyteadmin/pkg/manager/interfaces/workflow.go b/flyteadmin/pkg/manager/interfaces/workflow.go
@@ -6,6 +6,8 @@ import (
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin"
 )
 
+//go:generate mockery-v2 --name=WorkflowInterface --output=../mocks --case=underscore --with-expecter
+
 // Interface for managing Flyte Workflows
 type WorkflowInterface interface {
 	CreateWorkflow(ctx context.Context, request *admin.WorkflowCreateRequest) (*admin.WorkflowCreateResponse, error)

diff --git a/flyteadmin/pkg/manager/mocks/metrics_interface.go b/flyteadmin/pkg/manager/mocks/metrics_interface.go