From 6f55de1bdf5e0a9fe40d81419927db80c2d69bf3 Mon Sep 17 00:00:00 2001
From: Manjunath A Kumatagi <mkumatag@in.ibm.com>
Date: Wed, 26 Jul 2017 12:16:57 +0530
Subject: [PATCH] Test image utils for multi arch test images

---
 test/utils/image/manifest.go | 104 +++++++++++++++++++++++++++++++++++
 1 file changed, 104 insertions(+)
 create mode 100644 test/utils/image/manifest.go

diff --git a/test/utils/image/manifest.go b/test/utils/image/manifest.go
new file mode 100644
index 00000000000..31349c4c417
--- /dev/null
+++ b/test/utils/image/manifest.go
@@ -0,0 +1,104 @@
+/*
+Copyright 2017 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package image
+
+import (
+	"fmt"
+	"runtime"
+)
+
+const (
+	e2eRegistry     = "gcr.io/kubernetes-e2e-test-images"
+	gcRegistry      = "gcr.io/google-containers"
+	PrivateRegistry = "gcr.io/k8s-authenticated-test"
+)
+
+type ImageConfig struct {
+	registry string
+	name     string
+	version  string
+}
+
+func (i *ImageConfig) SetRegistry(registry string) {
+	i.registry = registry
+}
+
+func (i *ImageConfig) SetName(name string) {
+	i.name = name
+}
+
+func (i *ImageConfig) SetVersion(version string) {
+	i.version = version
+}
+
+var (
+	ClusterTester      = ImageConfig{e2eRegistry, "clusterapi-tester", "1.0"}
+	CudaVectorAdd      = ImageConfig{e2eRegistry, "cuda-vector-add", "1.0"}
+	Dnsutils           = ImageConfig{e2eRegistry, "dnsutils", "1.0"}
+	EntrypointTester   = ImageConfig{e2eRegistry, "entrypoint-tester", "1.0"}
+	Fakegitserver      = ImageConfig{e2eRegistry, "fakegitserver", "1.0"}
+	Frontend           = ImageConfig{e2eRegistry, "frontend", "1.0"}
+	Goproxy            = ImageConfig{e2eRegistry, "goproxy", "1.0"}
+	Hostexec           = ImageConfig{e2eRegistry, "hostexec", "1.0"}
+	Iperf              = ImageConfig{e2eRegistry, "iperf", "1.0"}
+	JessieDnsutils     = ImageConfig{e2eRegistry, "jessie-dnsutils", "1.0"}
+	Kitten             = ImageConfig{e2eRegistry, "kitten", "1.0"}
+	Liveness           = ImageConfig{e2eRegistry, "liveness", "1.0"}
+	LogsGenerator      = ImageConfig{e2eRegistry, "logs-generator", "1.0"}
+	Mounttest          = ImageConfig{e2eRegistry, "mounttest", "1.0"}
+	MounttestUser      = ImageConfig{e2eRegistry, "mounttest-user", "1.0"}
+	Nautilus           = ImageConfig{e2eRegistry, "nautilus", "1.0"}
+	Net                = ImageConfig{e2eRegistry, "net", "1.0"}
+	Netexec            = ImageConfig{e2eRegistry, "netexec", "1.0"}
+	Nettest            = ImageConfig{e2eRegistry, "nettest", "1.0"}
+	NginxSlim          = ImageConfig{gcRegistry, "nginx-slim", "0.20"}
+	NginxSlimNew       = ImageConfig{gcRegistry, "nginx-slim", "0.21"}
+	NoSnatTest         = ImageConfig{e2eRegistry, "no-snat-test", "1.0"}
+	NoSnatTestProxy    = ImageConfig{e2eRegistry, "no-snat-test-proxy", "1.0"}
+	NWayHTTP           = ImageConfig{e2eRegistry, "n-way-http", "1.0"}
+	Pause              = ImageConfig{gcRegistry, "pause", "3.0"}
+	Porter             = ImageConfig{e2eRegistry, "porter", "1.0"}
+	PortForwardTester  = ImageConfig{e2eRegistry, "port-forward-tester", "1.0"}
+	Redis              = ImageConfig{e2eRegistry, "redis", "1.0"}
+	Redisslave         = ImageConfig{e2eRegistry, "redis-slave", "1.0"}
+	ResourceConsumer   = ImageConfig{e2eRegistry, "resource-consumer", "1.1"}
+	ResourceController = ImageConfig{e2eRegistry, "resource-consumer/controller", "1.0"}
+	ServeHostname      = ImageConfig{e2eRegistry, "serve-hostname", "1.0"}
+	TestWebserver      = ImageConfig{e2eRegistry, "test-webserver", "1.0"}
+)
+
+func GetE2EImage(image ImageConfig) string {
+	return fmt.Sprintf("%s/%s-%s:%s", image.registry, image.name, runtime.GOARCH, image.version)
+}
+
+// GetBusyboxImage returns multi-arch busybox docker image
+func GetBusyBoxImage() string {
+	switch arch := runtime.GOARCH; arch {
+	case "amd64":
+		return "busybox"
+	case "arm":
+		return "arm32v6/busybox"
+	case "arm64":
+		return "arm64v8/busybox"
+	case "ppc64le":
+		return "ppc64le/busybox"
+	case "s390x":
+		return "s390x/busybox"
+	default:
+		return ""
+	}
+}