From af739fd0bd6094d9b22e422be5055c45053f626a Mon Sep 17 00:00:00 2001
From: Davanum Srinivas <davanum@gmail.com>
Date: Thu, 22 Aug 2019 12:16:27 -0400
Subject: [PATCH] Add --include-test-files flag to verify-import-boss

---
 hack/verify-import-boss.sh                                | 2 +-
 staging/src/k8s.io/code-generator/cmd/import-boss/BUILD   | 1 +
 staging/src/k8s.io/code-generator/cmd/import-boss/main.go | 2 ++
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/hack/verify-import-boss.sh b/hack/verify-import-boss.sh
index 7b12e514d0f..cbd611c9800 100755
--- a/hack/verify-import-boss.sh
+++ b/hack/verify-import-boss.sh
@@ -25,4 +25,4 @@ kube::golang::setup_env
 
 make -C "${KUBE_ROOT}" WHAT=vendor/k8s.io/code-generator/cmd/import-boss
 
-$(kube::util::find-binary "import-boss") --verify-only
+$(kube::util::find-binary "import-boss") --verify-only "$@"
diff --git a/staging/src/k8s.io/code-generator/cmd/import-boss/BUILD b/staging/src/k8s.io/code-generator/cmd/import-boss/BUILD
index 6fec2c33388..76721fd3ed9 100644
--- a/staging/src/k8s.io/code-generator/cmd/import-boss/BUILD
+++ b/staging/src/k8s.io/code-generator/cmd/import-boss/BUILD
@@ -18,6 +18,7 @@ go_library(
     importpath = "k8s.io/code-generator/cmd/import-boss",
     deps = [
         "//staging/src/k8s.io/code-generator/pkg/util:go_default_library",
+        "//vendor/github.com/spf13/pflag:go_default_library",
         "//vendor/k8s.io/gengo/args:go_default_library",
         "//vendor/k8s.io/gengo/examples/import-boss/generators:go_default_library",
         "//vendor/k8s.io/klog:go_default_library",
diff --git a/staging/src/k8s.io/code-generator/cmd/import-boss/main.go b/staging/src/k8s.io/code-generator/cmd/import-boss/main.go
index da099fda718..0080f01eb04 100644
--- a/staging/src/k8s.io/code-generator/cmd/import-boss/main.go
+++ b/staging/src/k8s.io/code-generator/cmd/import-boss/main.go
@@ -63,6 +63,7 @@ import (
 	"os"
 	"path/filepath"
 
+	"github.com/spf13/pflag"
 	"k8s.io/code-generator/pkg/util"
 	"k8s.io/gengo/args"
 	"k8s.io/gengo/examples/import-boss/generators"
@@ -81,6 +82,7 @@ func main() {
 		"k8s.io/kubernetes/cmd/...",
 		"k8s.io/kubernetes/plugin/...",
 	}
+	pflag.CommandLine.BoolVar(&arguments.IncludeTestFiles, "include-test-files", false, "If true, include *_test.go files.")
 
 	if err := arguments.Execute(
 		generators.NameSystems(),