e2e: Support use customize kubectl

Signed-off-by: zhanglei <[email protected]>

test/framework/exec/kubectl.go

@@ -20,16 +20,21 @@ import (
 	"bytes"
 	"context"
 	"fmt"
+	"os"
 )
 
 // KubectlApply shells out to kubectl apply.
 //
 // TODO: Remove this usage of kubectl and replace with a function from apply.go using the controller-runtime client.
 func KubectlApply(ctx context.Context, kubeconfigPath string, resources []byte, args ...string) error {
+	kubectlPath, ok := os.LookupEnv("CAPI_KUBECTL_PATH")
+	if !ok {
+		kubectlPath = "kubectl"
+	}
 	aargs := append([]string{"apply", "--kubeconfig", kubeconfigPath, "-f", "-"}, args...)
 	rbytes := bytes.NewReader(resources)
 	applyCmd := NewCommand(
-		WithCommand("kubectl"),
+		WithCommand(kubectlPath),
 		WithArgs(aargs...),
 		WithStdin(rbytes),
 	)
@@ -44,9 +49,13 @@ func KubectlApply(ctx context.Context, kubeconfigPath string, resources []byte,
 
 // KubectlWait shells out to kubectl wait.
 func KubectlWait(ctx context.Context, kubeconfigPath string, args ...string) error {
+	kubectlPath, ok := os.LookupEnv("CAPI_KUBECTL_PATH")
+	if !ok {
+		kubectlPath = "kubectl"
+	}
 	wargs := append([]string{"wait", "--kubeconfig", kubeconfigPath}, args...)
 	wait := NewCommand(
-		WithCommand("kubectl"),
+		WithCommand(kubectlPath),
 		WithArgs(wargs...),
 	)
 	_, stderr, err := wait.Run(ctx)