add more unit tests

Author: LinuxSuRen

pkg/exec/command.go

@@ -31,6 +31,7 @@ func NewDefaultReader(server, token string) Reader {
 	}
 }
 
+// GetResource returns the resource
 func (r *defualtReader) GetResource(group, kind, version, namespace, name string) (map[string]interface{}, error) {
 	api := fmt.Sprintf("%s/api/%s/%s/namespaces/%s/%s/%s", r.server, group, version, namespace, kind, name)
 	api = strings.ReplaceAll(api, "api//", "api/")
@@ -72,6 +73,7 @@ func GetClient() *http.Client {
 	return client
 }
 
+// ResourceValidator represents a generic resource validator
 type ResourceValidator interface {
 	Exist() bool
 	ExpectField(value interface{}, fields ...string) bool

pkg/runner/kubernetes/client.go

@@ -7,6 +7,7 @@ func PodValidatorFunc() expr.Option {
 	return expr.Function("pod", podValidator, new(func(...string) ResourceValidator))
 }
 
+// KubernetesValidatorFunc returns a expr for checking the generic Kubernetes resources
 func KubernetesValidatorFunc() expr.Option {
 	return expr.Function("k8s", resourceValidator, new(func(interface{}, ...string) ResourceValidator))
 }

pkg/runner/kubernetes/verify.go

@@ -0,0 +1,135 @@
+package kubernetes_test
+
+import (
+	"net/http"
+	"os"
+	"testing"
+
+	"github.com/antonmedv/expr"
+	"github.com/h2non/gock"
+	"github.com/linuxsuren/api-testing/pkg/runner/kubernetes"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestKubernetesValidatorFunc(t *testing.T) {
+	os.Setenv("KUBERNETES_SERVER", "http://foo")
+	os.Setenv("KUBERNETES_TOKEN", "token")
+	gock.InterceptClient(kubernetes.GetClient())
+	defer gock.RestoreClient(http.DefaultClient)
+	defer gock.Off()
+
+	tests := []struct {
+		name       string
+		prepare    func()
+		expression string
+		expectBool bool
+		expectErr  bool
+	}{{
+		name:       "pod exist expr",
+		prepare:    preparePod,
+		expression: `pod('ns', 'foo').Exist()`,
+		expectBool: true,
+	}, {
+		name:       "pod expectField expr",
+		prepare:    preparePod,
+		expression: `pod('ns', 'foo').ExpectField('pod', 'kind')`,
+		expectBool: true,
+	}, {
+		name:       "pod expectField expr, not match",
+		prepare:    preparePod,
+		expression: `pod('ns', 'foo').ExpectField('pods', 'kind')`,
+		expectBool: false,
+	}, {
+		name:       "pod expectField expr, not find field",
+		prepare:    preparePod,
+		expression: `pod('ns', 'foo').ExpectField('pods', 'kinds')`,
+		expectBool: false,
+	}, {
+		name:       "no enough params",
+		expression: `k8s('crd')`,
+		prepare:    emptyPrepare,
+		expectBool: false,
+		expectErr:  true,
+	}, {
+		name:       "crd",
+		expression: `k8s({"kind":"vms","group":"bar","version":"v2"}, "ns", "foo").Exist()`,
+		prepare:    prepareCRDVM,
+		expectBool: true,
+	}, {
+		name:       "deploy",
+		expression: `k8s("deployments", "ns", "foo").Exist()`,
+		prepare:    prepareDeploy,
+		expectBool: true,
+	}, {
+		name:       "statefulset",
+		expression: `k8s("statefulsets", "ns", "foo").Exist()`,
+		prepare:    prepareStatefulset,
+		expectBool: true,
+	}, {
+		name:       "daemonset",
+		expression: `k8s("daemonsets", "ns", "foo").Exist()`,
+		prepare:    prepareDaemonset,
+		expectBool: true,
+	}, {
+		name:       "no kind",
+		expression: `k8s({"foo": "bar"}, "ns", "foo").Exist()`,
+		prepare:    emptyPrepare,
+		expectErr:  true,
+	}}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			tt.prepare()
+			vm, err := expr.Compile(tt.expression, kubernetes.KubernetesValidatorFunc(),
+				kubernetes.PodValidatorFunc())
+			assert.Nil(t, err)
+
+			result, err := expr.Run(vm, expr.Env(tt))
+			assert.Equal(t, tt.expectErr, err != nil)
+			if err == nil {
+				assert.Equal(t, tt.expectBool, result)
+			}
+		})
+	}
+}
+
+func emptyPrepare() {}
+
+func preparePod() {
+	gock.New("http://foo").
+		Get("/api/v1/namespaces/ns/pods/foo").
+		MatchHeader("Authorization", "Bearer token").
+		Reply(http.StatusOK).
+		JSON(`{"kind":"pod"}`)
+}
+
+func prepareDeploy() {
+	gock.New("http://foo").
+		Get("/apis/apps/v1/namespaces/ns/deployments/foo").
+		MatchHeader("Authorization", "Bearer token").
+		Reply(http.StatusOK).
+		JSON(`{"kind":"deploy"}`)
+}
+
+func prepareStatefulset() {
+	gock.New("http://foo").
+		Get("/apis/apps/v1/namespaces/ns/statefulsets/foo").
+		MatchHeader("Authorization", "Bearer token").
+		Reply(http.StatusOK).
+		JSON(`{"kind":"statefulset"}`)
+}
+
+func prepareDaemonset() {
+	gock.New("http://foo").
+		Get("/apis/apps/v1/namespaces/ns/daemonsets/foo").
+		MatchHeader("Authorization", "Bearer token").
+		Reply(http.StatusOK).
+		JSON(`{"kind":"daemonset"}`)
+}
+
+func prepareCRDVM() {
+	gock.New("http://foo").
+		Get("/apis/bar/v2/namespaces/ns/vms/foo").
+		MatchHeader("Authorization", "Bearer token").
+		Reply(http.StatusOK).
+		JSON(`{"kind":"vm"}`)
+}