Support to init k8s cluster

Author: LinuxSuRen

Dockerfile

@@ -33,6 +33,7 @@ RUN apk add --no-cache \
 COPY entrypoint.sh /entrypoint.sh
 COPY --from=builder /workspace/atest /usr/bin/atest
 COPY --from=hd /usr/local/bin/hd /usr/local/bin/hd
-RUN hd i kubernetes-sigs/kubectl
+RUN hd i kubernetes-sigs/kubectl && \
+    hd i k3d
 
 ENTRYPOINT ["/entrypoint.sh"]

action.yml

@@ -5,8 +5,20 @@ inputs:
     description: 'The pattern of the items'
     required: true
     default: 'testcase-*.yaml'
+  kustomization:
+    description: 'The kustomization file path'
+    required: true
+  waitNamespace:
+    description: 'waitNamespace'
+    required: true
+  waitResource:
+    description: 'waitResource'
+    required: true
 runs:
   using: 'docker'
   image: 'Dockerfile'
   args:
     - ${{ inputs.pattern }}
+    - ${{ inputs.kustomization }}
+    - ${{ inputs.waitNamespace }}
+    - ${{ inputs.waitResource }}

cmd/init.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"github.com/linuxsuren/api-testing/pkg/exec"
+	"github.com/spf13/cobra"
+)
+
+type initOption struct {
+	kustomization string
+	waitNamespace string
+	waitResource  string
+}
+
+func createInitCommand() (cmd *cobra.Command) {
+	opt := &initOption{}
+	cmd = &cobra.Command{
+		Use:  "init",
+		Long: "Support to init Kubernetes cluster with kustomization, and wait it with command: kubectl wait",
+		RunE: opt.runE,
+	}
+
+	flags := cmd.Flags()
+	flags.StringVarP(&opt.kustomization, "kustomization", "k", "", "The kustomization file path")
+	flags.StringVarP(&opt.waitNamespace, "wait-namespace", "", "", "")
+	flags.StringVarP(&opt.waitResource, "wait-resource", "", "", "")
+	return
+}
+
+func (o *initOption) runE(cmd *cobra.Command, args []string) (err error) {
+	if o.kustomization != "" {
+		if err = exec.RunCommand("kubectl", "apply", "-k", o.kustomization, "--wait=true"); err != nil {
+			return
+		}
+	}
+
+	if o.waitNamespace != "" && o.waitResource != "" {
+		if err = exec.RunCommand("kubectl", "wait", "-n", o.waitNamespace, o.waitResource, "--for", "condition=Available=True", "--timeout=900s"); err != nil {
+			return
+		}
+	}
+	return
+}

cmd/root.go

@@ -1,62 +1,18 @@
 package main
 
 import (
-	"github.com/linuxsuren/api-testing/pkg/runner"
-	"github.com/linuxsuren/api-testing/pkg/testing"
 	"github.com/spf13/cobra"
 	"os"
-	"path"
-	"path/filepath"
 )
 
-type option struct {
-	pattern string
-}
-
 func main() {
-	opt := &option{}
-
 	cmd := &cobra.Command{
-		Use:  "atest",
-		RunE: opt.runE,
+		Use: "atest",
 	}
-
-	// set flags
-	flags := cmd.Flags()
-	flags.StringVarP(&opt.pattern, "pattern", "p", "testcase-*.yaml",
-		"The file pattern which try to execute the test cases")
+	cmd.AddCommand(createInitCommand(), createRunCommand())
 
 	// run command
 	if err := cmd.Execute(); err != nil {
 		os.Exit(1)
 	}
 }
-
-func (o *option) runE(cmd *cobra.Command, args []string) (err error) {
-	var files []string
-	if files, err = filepath.Glob(o.pattern); err == nil {
-		for i := range files {
-			item := files[i]
-
-			var testcase *testing.TestCase
-			if testcase, err = testing.Parse(item); err != nil {
-				return
-			}
-
-			setRelativeDir(item, testcase)
-
-			if err = runner.RunTestCase(testcase); err != nil {
-				return
-			}
-		}
-	}
-	return
-}
-
-func setRelativeDir(configFile string, testcase *testing.TestCase) {
-	dir := filepath.Dir(configFile)
-
-	for i := range testcase.Prepare.Kubernetes {
-		testcase.Prepare.Kubernetes[i] = path.Join(dir, testcase.Prepare.Kubernetes[i])
-	}
-}

cmd/run.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"github.com/linuxsuren/api-testing/pkg/runner"
+	"github.com/linuxsuren/api-testing/pkg/testing"
+	"github.com/spf13/cobra"
+	"path"
+	"path/filepath"
+)
+
+type runOption struct {
+	pattern string
+}
+
+func createRunCommand() (cmd *cobra.Command) {
+	opt := &runOption{}
+	cmd = &cobra.Command{
+		Use:  "run",
+		RunE: opt.runE,
+	}
+
+	// set flags
+	flags := cmd.Flags()
+	flags.StringVarP(&opt.pattern, "pattern", "p", "testcase-*.yaml",
+		"The file pattern which try to execute the test cases")
+	return
+}
+
+func (o *runOption) runE(cmd *cobra.Command, args []string) (err error) {
+	var files []string
+	if files, err = filepath.Glob(o.pattern); err == nil {
+		for i := range files {
+			item := files[i]
+
+			var testcase *testing.TestCase
+			if testcase, err = testing.Parse(item); err != nil {
+				return
+			}
+
+			setRelativeDir(item, testcase)
+
+			if err = runner.RunTestCase(testcase); err != nil {
+				return
+			}
+		}
+	}
+	return
+}
+
+func setRelativeDir(configFile string, testcase *testing.TestCase) {
+	dir := filepath.Dir(configFile)
+
+	for i := range testcase.Prepare.Kubernetes {
+		testcase.Prepare.Kubernetes[i] = path.Join(dir, testcase.Prepare.Kubernetes[i])
+	}
+}

entrypoint.sh

@@ -1,3 +1,7 @@
 #!/bin/sh
 
-atest -p "$1"
+k3d cluster create
+k3d cluster list
+
+atest init -k $2 --wait-namespace $3 --wait-resource $4
+atest run -p "$1"