From ea092fd77b0139b5d0253677f28ba96a987e0848 Mon Sep 17 00:00:00 2001
From: Janet Kuo <chiachenk@google.com>
Date: Wed, 8 Jul 2015 17:14:28 -0700
Subject: [PATCH] E2E tests for kubectl label command

---
 test/e2e/kubectl.go | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/test/e2e/kubectl.go b/test/e2e/kubectl.go
index 600d81ca5c..6a4f206b15 100644
--- a/test/e2e/kubectl.go
+++ b/test/e2e/kubectl.go
@@ -202,6 +202,42 @@ var _ = Describe("Kubectl client", func() {
 		})
 	})
 
+	Describe("Kubectl label", func() {
+		var podPath string
+		var nsFlag string
+		BeforeEach(func() {
+			podPath = filepath.Join(testContext.RepoRoot, "examples/pod.yaml")
+			By("creating the pod")
+			nsFlag = fmt.Sprintf("--namespace=%v", ns)
+			runKubectl("create", "-f", podPath, nsFlag)
+			checkPodsRunningReady(c, ns, []string{simplePodName}, podStartTimeout)
+		})
+		AfterEach(func() {
+			cleanup(podPath, ns, simplePodSelector)
+		})
+
+		It("should update the label on a resource", func() {
+			labelName := "testing-label"
+			labelValue := "testing-label-value"
+
+			By("adding the label " + labelName + " with value " + labelValue + " to a pod")
+			runKubectl("label", "pods", simplePodName, labelName+"="+labelValue, nsFlag)
+			By("verifying the pod has the label " + labelName + " with the value " + labelValue)
+			output := runKubectl("get", "pod", simplePodName, "-L", labelName, nsFlag)
+			if !strings.Contains(output, labelValue) {
+				Failf("Failed updating label " + labelName + " to the pod " + simplePodName)
+			}
+
+			By("removing the label " + labelName + " of a pod")
+			runKubectl("label", "pods", simplePodName, labelName+"-", nsFlag)
+			By("verifying the pod doesn't have the label " + labelName)
+			output = runKubectl("get", "pod", simplePodName, "-L", labelName, nsFlag)
+			if strings.Contains(output, labelValue) {
+				Failf("Failed removing label " + labelName + " of the pod " + simplePodName)
+			}
+		})
+	})
+
 })
 
 func curl(addr string) (string, error) {