Fix kubeVersion regex tests

cmd/common.go

@@ -62,6 +62,8 @@ func runChecks(t check.NodeType) {
 	}
 
 	ver := getKubeVersion()
+	glog.V(1).Info(fmt.Sprintf("Running tests for Kubernetes version: %s", ver))
+
 	path := fmt.Sprintf("%s/%s/%s", cfgDir, ver, file)
 	in, err := ioutil.ReadFile(path)
 	if err != nil {

cmd/util.go

@@ -214,14 +214,10 @@ func multiWordReplace(s string, subname string, sub string) string {
 }
 
 func getKubeVersion() string {
-	var ver string
-	var matched bool
-
-	failmsg := "kubernetes version check failed"
 	// These executables might not be on the user's path.
 	_, err := exec.LookPath("kubectl")
 	if err != nil {
-		continueWithError(err, failmsg)
+		exitWithError(fmt.Errorf("kubernetes version check failed: %v", err))
 	}
 
 	cmd := exec.Command("kubectl", "version", "--short")
@@ -230,22 +226,17 @@ func getKubeVersion() string {
 		continueWithError(fmt.Errorf("%s", out), "")
 	}
 
-	serverVersionRe := regexp.MustCompile(`Server Version: v(\d+.\d+)`)
-	subs := serverVersionRe.FindStringSubmatch(string(out))
-	if len(subs) == 2 {
-		ver = string(subs[1])
-		validVersionPttn := `\d.\d.\d`
-		if matched, _ = regexp.MatchString(validVersionPttn, ver); !matched {
-			continueWithError(fmt.Errorf("%s: invalid server version ", ver), failmsg)
-		}
-	}
+	return getVersionFromKubectlOutput(string(out))
+}
 
-	if ver == "" || !matched {
+func getVersionFromKubectlOutput(s string) string {
+	serverVersionRe := regexp.MustCompile(`Server Version: v(\d+.\d+)`)
+	subs := serverVersionRe.FindStringSubmatch(s)
+	if len(subs) < 2 {
 		printlnWarn(fmt.Sprintf("Unable to get kubectl version, using default version: %s", defaultKubeVersion))
-		ver = defaultKubeVersion
+		return defaultKubeVersion
 	}
-
-	return ver
+	return subs[1]
 }
 
 func makeSubstitutions(s string, ext string, m map[string]string) string {

cmd/util_test.go

@@ -191,6 +191,20 @@ func TestGetKubeVersion(t *testing.T) {
 
 }
 
+func TestKubeVersionRegex(t *testing.T) {
+	ver := getVersionFromKubectlOutput(`Client Version: v1.8.0
+		Server Version: v1.8.12
+		`)
+	if ver != "1.8" {
+		t.Fatalf("Expected 1.8 got %s", ver)
+	}
+
+	ver = getVersionFromKubectlOutput("Something completely different")
+	if ver != "1.6" {
+		t.Fatalf("Expected 1.6 got %s", ver)
+	}
+}
+
 func TestFindConfigFile(t *testing.T) {
 	cases := []struct {
 		input       []string