From 1fb12b84ddbccc851701ddbdd6937dfc081c5a84 Mon Sep 17 00:00:00 2001
From: Yang Guo <ygg@google.com>
Date: Mon, 14 Aug 2017 17:11:05 -0700
Subject: [PATCH] Allow passing image description from e2e node test config

---
 .../e2e_node/jenkins/benchmark/benchmark-config.yaml |  3 +++
 test/e2e_node/remote/node_e2e.go                     |  2 +-
 test/e2e_node/runner/remote/run_remote.go            | 12 ++++++++++--
 3 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/test/e2e_node/jenkins/benchmark/benchmark-config.yaml b/test/e2e_node/jenkins/benchmark/benchmark-config.yaml
index 58494307d3..d92e92c941 100644
--- a/test/e2e_node/jenkins/benchmark/benchmark-config.yaml
+++ b/test/e2e_node/jenkins/benchmark/benchmark-config.yaml
@@ -113,6 +113,7 @@ images:
       - 'resource tracking for 105 pods per node \[Benchmark\]'
   cos-docker112-resource1:
     image: cos-stable-60-9592-76-0
+    image_description: cos-stable-60-9592-76-0 with docker 1.12.6
     project: cos-cloud
     machine: n1-standard-1
     metadata: "user-data<test/e2e_node/jenkins/cos-init-docker.yaml,gci-update-strategy=update_disabled,gci-docker-version=1.12.6"
@@ -120,6 +121,7 @@ images:
       - 'resource tracking for 0 pods per node \[Benchmark\]'
   cos-docker112-resource2:
     image: cos-stable-60-9592-76-0
+    image_description: cos-stable-60-9592-76-0 with docker 1.12.6
     project: cos-cloud
     machine: n1-standard-1
     metadata: "user-data<test/e2e_node/jenkins/cos-init-docker.yaml,gci-update-strategy=update_disabled,gci-docker-version=1.12.6"
@@ -127,6 +129,7 @@ images:
       - 'resource tracking for 35 pods per node \[Benchmark\]'
   cos-docker112-resource3:
     image: cos-stable-60-9592-76-0
+    image_description: cos-stable-60-9592-76-0 with docker 1.12.6
     project: cos-cloud
     machine: n1-standard-1
     metadata: "user-data<test/e2e_node/jenkins/cos-init-docker.yaml,gci-update-strategy=update_disabled,gci-docker-version=1.12.6"
diff --git a/test/e2e_node/remote/node_e2e.go b/test/e2e_node/remote/node_e2e.go
index 66a94f411c..4d46e3f245 100644
--- a/test/e2e_node/remote/node_e2e.go
+++ b/test/e2e_node/remote/node_e2e.go
@@ -211,7 +211,7 @@ func (n *NodeE2ERemote) RunTest(host, workspace, results, imageDesc, junitFilePr
 	glog.V(2).Infof("Starting tests on %q", host)
 	cmd := getSSHCommand(" && ",
 		fmt.Sprintf("cd %s", workspace),
-		fmt.Sprintf("timeout -k 30s %fs ./ginkgo %s ./e2e_node.test -- --system-spec-name=%s --system-spec-file=%s --logtostderr --v 4 --node-name=%s --report-dir=%s --report-prefix=%s --image-description=%s %s",
+		fmt.Sprintf("timeout -k 30s %fs ./ginkgo %s ./e2e_node.test -- --system-spec-name=%s --system-spec-file=%s --logtostderr --v 4 --node-name=%s --report-dir=%s --report-prefix=%s --image-description=\"%s\" %s",
 			timeout.Seconds(), ginkgoArgs, systemSpecName, systemSpecFile, host, results, junitFilePrefix, imageDesc, testArgs),
 	)
 	return SSH(host, "sh", "-c", cmd)
diff --git a/test/e2e_node/runner/remote/run_remote.go b/test/e2e_node/runner/remote/run_remote.go
index 84db8c6225..1446021906 100644
--- a/test/e2e_node/runner/remote/run_remote.go
+++ b/test/e2e_node/runner/remote/run_remote.go
@@ -113,6 +113,7 @@ type Resources struct {
 
 type GCEImage struct {
 	Image      string `json:"image, omitempty"`
+	ImageDesc  string `json:"image_description, omitempty"`
 	Project    string `json:"project"`
 	Metadata   string `json:"metadata"`
 	ImageRegex string `json:"image_regex, omitempty"`
@@ -131,7 +132,10 @@ type internalImageConfig struct {
 }
 
 type internalGCEImage struct {
-	image     string
+	image string
+	// imageDesc is the description of the image. If empty, the value in the
+	// 'image' will be used.
+	imageDesc string
 	project   string
 	resources Resources
 	metadata  *compute.Metadata
@@ -206,12 +210,16 @@ func main() {
 			for _, image := range images {
 				gceImage := internalGCEImage{
 					image:     image,
+					imageDesc: imageConfig.ImageDesc,
 					project:   imageConfig.Project,
 					metadata:  getImageMetadata(imageConfig.Metadata),
 					machine:   imageConfig.Machine,
 					tests:     imageConfig.Tests,
 					resources: imageConfig.Resources,
 				}
+				if gceImage.imageDesc == "" {
+					gceImage.imageDesc = gceImage.image
+				}
 				if isRegex && len(images) > 1 {
 					// Use image name when shortName is not unique.
 					name = image
@@ -485,7 +493,7 @@ func testImage(imageConfig *internalGCEImage, junitFilePrefix string) *TestResul
 	// If we are going to delete the instance, don't bother with cleaning up the files
 	deleteFiles := !*deleteInstances && *cleanup
 
-	result := testHost(host, deleteFiles, imageConfig.image, junitFilePrefix, ginkgoFlagsStr)
+	result := testHost(host, deleteFiles, imageConfig.imageDesc, junitFilePrefix, ginkgoFlagsStr)
 	// This is a temporary solution to collect serial node serial log. Only port 1 contains useful information.
 	// TODO(random-liu): Extract out and unify log collection logic with cluste e2e.
 	serialPortOutput, err := computeService.Instances.GetSerialPortOutput(*project, *zone, host).Port(1).Do()