feat: Add ExecWithContext (#10)

* exec parameters

* exec test

* exec with context

README.md

@@ -74,3 +74,9 @@ You can find the docs at [go docs](https://pkg.go.dev/github.com/shipengqi/kube)
 ```bash
 go test -v . -kubeconfig <kubeconfig file>
 ```
+
+## Test Client.Exec
+
+```bash
+go test -v -coverprofile=coverage.out -kubeconfig <kubeconfig file> -container <container name> -pod <pod name> -namespace <namespace> .
+```

client_test.go

@@ -23,6 +23,9 @@ type testcm struct {
 
 func TestMain(m *testing.M) {
 	flag.StringVar(&kubeconfig, "kubeconfig", "", "kubeconfig file")
+	flag.StringVar(&podname, "pod", "", "pod name for exec")
+	flag.StringVar(&containername, "container", "", "container name for exec")
+	flag.StringVar(&namespace, "namespace", "", "namespace for exec")
 	flag.Parse()
 
 	flags := genericclioptions.NewConfigFlags(false)

exec.go

@@ -13,7 +13,16 @@ import (
 
 // Exec is like kubectl exec.
 func (c *Client) Exec(pod, container, namespace string, command ...string) (string, string, error) {
-	p, err := c.GetPod(context.TODO(), namespace, pod)
+	return c.exec(context.TODO(), pod, container, namespace, command...)
+}
+
+// ExecWithContext executes exec with context.
+func (c *Client) ExecWithContext(ctx context.Context, pod, container, namespace string, command ...string) (string, string, error) {
+	return c.exec(ctx, pod, container, namespace, command...)
+}
+
+func (c *Client) exec(ctx context.Context, pod, container, namespace string, command ...string) (string, string, error) {
+	p, err := c.GetPod(ctx, namespace, pod)
 	if err != nil {
 		return "", "", err
 	}
@@ -43,7 +52,7 @@ func (c *Client) Exec(pod, container, namespace string, command ...string) (stri
 	if err != nil {
 		return "", "", err
 	}
-	err = exec.Stream(remotecommand.StreamOptions{
+	err = exec.StreamWithContext(ctx, remotecommand.StreamOptions{
 		Stdin:  nil,
 		Stdout: &stdout,
 		Stderr: &stderr,

exec_test.go

@@ -14,26 +14,37 @@ const (
 )
 
 func TestClientExec(t *testing.T) {
-	list, err := mockcli.GetPods(context.TODO(), _defaultTestNamespace)
-	require.NoError(t, err)
-	require.NotEqual(t, len(list.Items), 0)
-
-	var (
-		podName       string
-		containerName string
-	)
-	// exec coredns
-	for _, v := range list.Items {
-		if strings.Contains(strings.ToLower(v.Name), _defaultTestPod) {
-			podName = v.Name
-			containerName = v.Spec.Containers[0].Name
-			break
+
+	var containerName string
+
+	podNamespace := _defaultTestNamespace
+	podName := _defaultTestPod
+
+	if namespace != "" {
+		podNamespace = namespace
+	}
+
+	if podname != "" && containername != "" {
+		podName = podname
+		containerName = containername
+	} else {
+		list, err := mockcli.GetPods(context.TODO(), podNamespace)
+		require.NoError(t, err)
+		require.NotEqual(t, len(list.Items), 0)
+
+		// exec coredns
+		for _, v := range list.Items {
+			if strings.Contains(strings.ToLower(v.Name), podName) {
+				podName = v.Name
+				containerName = v.Spec.Containers[0].Name
+				break
+			}
 		}
 	}
 
 	// Todo, create a pod, and test Exec(), currently just skip validating the error
 	t.Log("exec:", podName, containerName)
-	stdout, _, err := mockcli.Exec(podName, containerName, _defaultTestNamespace, "/bin/sh", "-c", "/bin/ls /")
+	stdout, _, err := mockcli.Exec(podName, containerName, podNamespace, "/bin/sh", "-c", "/bin/ls /")
 	require.Error(t, err)
 	t.Log(stdout)
 }