You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.6 KiB
71 lines
1.6 KiB
// +build integration
|
|
|
|
package integration
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
var kubebenchImg = flag.String("kubebenchImg", "aquasec/kube-bench:latest", "kube-bench image used as part of this test")
|
|
|
|
func TestRunWithKind(t *testing.T) {
|
|
flag.Parse()
|
|
fmt.Printf("kube-bench Container Image: %s\n", *kubebenchImg)
|
|
timeout := time.Duration(10 * time.Minute)
|
|
ticker := time.Duration(2 * time.Second)
|
|
|
|
mustMatch := func(expFname, data string) {
|
|
d, err := ioutil.ReadFile(expFname)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
expectedData := strings.TrimSpace(string(d))
|
|
data = strings.TrimSpace(data)
|
|
if expectedData != data {
|
|
t.Errorf("expected: %q\n\n Got %q\n\n", expectedData, data)
|
|
}
|
|
}
|
|
|
|
cases := []struct {
|
|
TestName string
|
|
KindCfg string
|
|
KubebenchYAML string
|
|
ExpectedFile string
|
|
ExpectError bool
|
|
}{
|
|
{
|
|
TestName: "job",
|
|
KindCfg: "./testdata/add-tls-kind.yaml",
|
|
KubebenchYAML: "../job.yaml",
|
|
ExpectedFile: "./testdata/job.data",
|
|
},
|
|
{
|
|
TestName: "job-node",
|
|
KindCfg: "./testdata/add-tls-kind.yaml",
|
|
KubebenchYAML: "../job-node.yaml",
|
|
ExpectedFile: "./testdata/job-node.data",
|
|
},
|
|
{
|
|
TestName: "job-master",
|
|
KindCfg: "./testdata/add-tls-kind.yaml",
|
|
KubebenchYAML: "../job-master.yaml",
|
|
ExpectedFile: "./testdata/job-master.data",
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.TestName, func(t *testing.T) {
|
|
data, err := runWithKind(c.TestName, c.KindCfg, c.KubebenchYAML, *kubebenchImg, timeout, ticker)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
return
|
|
}
|
|
mustMatch(c.ExpectedFile, data)
|
|
})
|
|
}
|
|
}
|