Merge branch 'master' into unnecessary-warning

pull/68/head
Abubakr-Sadik Nii Nai Davis 7 years ago
commit 53eb720952

@ -539,7 +539,7 @@ groups:
scored: true scored: true
- id: 1.1.33 - id: 1.1.33
text: "1.1.34 Ensure that the --experimental-encryption-provider-config argument is text: "Ensure that the --experimental-encryption-provider-config argument is
set as appropriate (Scored)" set as appropriate (Scored)"
audit: "ps -ef | grep $apiserverbin | grep -v grep" audit: "ps -ef | grep $apiserverbin | grep -v grep"
tests: tests:

@ -25,16 +25,17 @@ import (
) )
var ( var (
envVarsPrefix = "KUBE_BENCH" envVarsPrefix = "KUBE_BENCH"
cfgDir = "./cfg" cfgDir = "./cfg"
cfgFile string defaultKubeVersion = "1.6"
jsonFmt bool cfgFile string
pgSql bool jsonFmt bool
checkList string pgSql bool
groupList string checkList string
masterFile string groupList string
nodeFile string masterFile string
federatedFile string nodeFile string
federatedFile string
) )
// RootCmd represents the base command when called without any subcommands // RootCmd represents the base command when called without any subcommands

@ -213,39 +213,30 @@ func multiWordReplace(s string, subname string, sub string) string {
return strings.Replace(s, subname, sub, -1) return strings.Replace(s, subname, sub, -1)
} }
type version struct { func getKubeVersion() string {
Server string
Client string
}
func getKubeVersion() *version {
ver := new(version)
// These executables might not be on the user's path. // These executables might not be on the user's path.
_, err := exec.LookPath("kubectl") _, err := exec.LookPath("kubectl")
if err != nil { if err != nil {
s := fmt.Sprintf("Kubernetes version check skipped with error %v", err) exitWithError(fmt.Errorf("kubernetes version check failed: %v", err))
continueWithError(err, sprintlnWarn(s))
return nil
} }
cmd := exec.Command("kubectl", "version") cmd := exec.Command("kubectl", "version", "--short")
out, err := cmd.Output() out, err := cmd.CombinedOutput()
if err != nil { if err != nil {
s := fmt.Sprintf("Kubernetes version check skipped, with error getting kubectl version") continueWithError(fmt.Errorf("%s", out), "")
continueWithError(err, sprintlnWarn(s))
return nil
} }
clientVerRe := regexp.MustCompile(`Client.*Major:"(\d+)".*Minor:"(\d+)"`) return getVersionFromKubectlOutput(string(out))
svrVerRe := regexp.MustCompile(`Server.*Major:"(\d+)".*Minor:"(\d+)"`) }
sub := clientVerRe.FindStringSubmatch(string(out))
ver.Client = sub[1] + "." + sub[2]
sub = svrVerRe.FindStringSubmatch(string(out))
ver.Server = sub[1] + "." + sub[2]
return ver 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))
return defaultKubeVersion
}
return subs[1]
} }
func makeSubstitutions(s string, ext string, m map[string]string) string { func makeSubstitutions(s string, ext string, m map[string]string) string {

@ -17,7 +17,6 @@ package cmd
import ( import (
"os" "os"
"reflect" "reflect"
"regexp"
"strconv" "strconv"
"testing" "testing"
@ -182,19 +181,17 @@ func TestMultiWordReplace(t *testing.T) {
} }
} }
func TestGetKubeVersion(t *testing.T) { func TestKubeVersionRegex(t *testing.T) {
ver := getKubeVersion() ver := getVersionFromKubectlOutput(`Client Version: v1.8.0
if ver == nil { Server Version: v1.8.12
t.Log("Expected non nil version info.") `)
} else { if ver != "1.8" {
if ok, err := regexp.MatchString(`\d+.\d+`, ver.Client); !ok && err != nil { t.Fatalf("Expected 1.8 got %s", ver)
t.Logf("Expected:%v got %v\n", "n.m", ver.Client) }
}
if ok, err := regexp.MatchString(`\d+.\d+`, ver.Server); !ok && err != nil {
t.Logf("Expected:%v got %v\n", "n.m", ver.Server)
}
ver = getVersionFromKubectlOutput("Something completely different")
if ver != "1.6" {
t.Fatalf("Expected 1.6 got %s", ver)
} }
} }

Loading…
Cancel
Save