2017-08-11 16:59:57 +00:00
|
|
|
// Copyright © 2017 Aqua Security Software Ltd. <info@aquasec.com>
|
|
|
|
//
|
|
|
|
// 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 cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"regexp"
|
2017-08-15 15:44:40 +00:00
|
|
|
"strconv"
|
2017-08-11 16:59:57 +00:00
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestCheckVersion(t *testing.T) {
|
|
|
|
kubeoutput := `Client Version: version.Info{Major:"1", Minor:"7", GitVersion:"v1.7.0", GitCommit:"d3ada0119e776222f11ec7945e6d860061339aad", GitTreeState:"clean", BuildDate:"2017-06-30T09:51:01Z", GoVersion:"go1.8.3", Compiler:"gc", Platform:"darwin/amd64"}
|
|
|
|
Server Version: version.Info{Major:"1", Minor:"7", GitVersion:"v1.7.0", GitCommit:"d3ada0119e776222f11ec7945e6d860061339aad", GitTreeState:"clean", BuildDate:"2017-07-26T00:12:31Z", GoVersion:"go1.8.3", Compiler:"gc", Platform:"linux/amd64"}`
|
|
|
|
cases := []struct {
|
|
|
|
t string
|
|
|
|
s string
|
|
|
|
major string
|
|
|
|
minor string
|
|
|
|
exp string
|
|
|
|
}{
|
|
|
|
{t: "Client", s: kubeoutput, major: "1", minor: "7"},
|
|
|
|
{t: "Server", s: kubeoutput, major: "1", minor: "7"},
|
|
|
|
{t: "Client", s: kubeoutput, major: "1", minor: "6", exp: "Unexpected Client version 1.7"},
|
|
|
|
{t: "Client", s: kubeoutput, major: "2", minor: "0", exp: "Unexpected Client version 1.7"},
|
|
|
|
{t: "Server", s: "something unexpected", major: "2", minor: "0", exp: "Couldn't find Server version from kubectl output 'something unexpected'"},
|
|
|
|
}
|
|
|
|
|
2017-08-15 15:44:40 +00:00
|
|
|
for id, c := range cases {
|
|
|
|
t.Run(strconv.Itoa(id), func(t *testing.T) {
|
2017-08-11 16:59:57 +00:00
|
|
|
m := checkVersion(c.t, c.s, c.major, c.minor)
|
|
|
|
if m != c.exp {
|
|
|
|
t.Fatalf("Got: %s, expected: %s", m, c.exp)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestVersionMatch(t *testing.T) {
|
|
|
|
minor := regexVersionMinor
|
|
|
|
major := regexVersionMajor
|
|
|
|
client := `Client Version: version.Info{Major:"1", Minor:"7", GitVersion:"v1.7.0", GitCommit:"d3ada0119e776222f11ec7945e6d860061339aad", GitTreeState:"clean", BuildDate:"2017-06-30T09:51:01Z", GoVersion:"go1.8.3", Compiler:"gc", Platform:"darwin/amd64"}`
|
|
|
|
server := `Server Version: version.Info{Major:"1", Minor:"7", GitVersion:"v1.7.0", GitCommit:"d3ada0119e776222f11ec7945e6d860061339aad", GitTreeState:"clean", BuildDate:"2017-07-26T00:12:31Z", GoVersion:"go1.8.3", Compiler:"gc", Platform:"linux/amd64"}`
|
|
|
|
|
|
|
|
cases := []struct {
|
|
|
|
r *regexp.Regexp
|
|
|
|
s string
|
|
|
|
exp string
|
|
|
|
}{
|
|
|
|
{r: major, s: server, exp: "1"},
|
|
|
|
{r: minor, s: server, exp: "7"},
|
|
|
|
{r: major, s: client, exp: "1"},
|
|
|
|
{r: minor, s: client, exp: "7"},
|
|
|
|
{r: major, s: "Some unexpected string"},
|
|
|
|
{r: minor}, // Checking that we don't fall over if the string is empty
|
|
|
|
}
|
|
|
|
|
2017-08-15 15:44:40 +00:00
|
|
|
for id, c := range cases {
|
|
|
|
t.Run(strconv.Itoa(id), func(t *testing.T) {
|
2017-08-11 16:59:57 +00:00
|
|
|
m := versionMatch(c.r, c.s)
|
|
|
|
if m != c.exp {
|
|
|
|
t.Fatalf("Got %s expected %s", m, c.exp)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2017-08-15 15:44:40 +00:00
|
|
|
|
|
|
|
var g string
|
|
|
|
|
|
|
|
func fakeps(proc string) string {
|
|
|
|
return g
|
|
|
|
}
|
|
|
|
func TestVerifyBin(t *testing.T) {
|
|
|
|
cases := []struct {
|
|
|
|
proc string
|
|
|
|
psOut string
|
|
|
|
exp bool
|
|
|
|
}{
|
|
|
|
{proc: "single", psOut: "single", exp: true},
|
|
|
|
{proc: "single", psOut: "", exp: false},
|
|
|
|
{proc: "two words", psOut: "two words", exp: true},
|
|
|
|
{proc: "two words", psOut: "", exp: false},
|
|
|
|
{proc: "cmd", psOut: "cmd param1 param2", exp: true},
|
|
|
|
{proc: "cmd param", psOut: "cmd param1 param2", exp: true},
|
|
|
|
{proc: "cmd param", psOut: "cmd", exp: false},
|
|
|
|
}
|
|
|
|
|
|
|
|
for id, c := range cases {
|
|
|
|
t.Run(strconv.Itoa(id), func(t *testing.T) {
|
|
|
|
g = c.psOut
|
|
|
|
v := verifyBin(c.proc, fakeps)
|
|
|
|
if v != c.exp {
|
|
|
|
t.Fatalf("Expected %v got %v", c.exp, v)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|