Use go run for mockgen

buildkite · Apr 17, 2023 · cb8cea2 · cb8cea2
1 parent c4df24f
commit cb8cea2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/internal/scheduler/limiter_test.go b/internal/scheduler/limiter_test.go
@@ -18,7 +18,7 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
-//go:generate mockgen -destination=mock_handler_test.go -source=../monitor/monitor.go -package scheduler_test
+//go:generate go run github.com/golang/mock/mockgen -destination=mock_handler_test.go -source=../monitor/monitor.go -package scheduler_test
 
 func TestLimiter(t *testing.T) {
 	t.Parallel()

diff --git a/internal/scheduler/scheduler_test.go b/internal/scheduler/scheduler_test.go
@@ -14,8 +14,6 @@ import (
 	corev1 "k8s.io/api/core/v1"
 )
 
-//go:generate mockgen -destination=mock_handler_test.go -source=scheduler.go -package scheduler_test
-
 func TestJobPluginConversion(t *testing.T) {
 	t.Parallel()
 	pluginConfig := scheduler.KubernetesPlugin{

diff --git a/tools.go b/tools.go
@@ -4,5 +4,6 @@ package main
 
 import (
 	_ "github.com/Khan/genqlient"
+	_ "github.com/golang/mock/mockgen"
 	_ "gotest.tools/gotestsum"
 )