add unittest for common/ssh_client package

common/ssh_client/kubernetes_client.py

@@ -42,11 +42,14 @@ def __init__(self, context=None, node=None):
     def exec_cmd(self, cmd):
         exec_command = ['/bin/sh', '-c', cmd]
         self.stdio.verbose("KubernetesClient exec_cmd: {0}".format(cmd))
-        resp = stream(self.client.connect_get_namespaced_pod_exec, self.pod_name, self.namespace, command=exec_command, stderr=True, stdin=False, stdout=True, tty=False, container=self.container_name)
-        self.stdio.verbose("KubernetesClient exec_cmd.resp: {0}".format(resp))
-        if "init system (PID 1). Can't operate." in resp:
-            return "KubernetesClient can't get the resp by {0}".format(cmd)
-        return resp
+        try:
+            resp = stream(self.client.connect_get_namespaced_pod_exec, self.pod_name, self.namespace, command=exec_command, stderr=True, stdin=False, stdout=True, tty=False, container=self.container_name)
+            self.stdio.verbose("KubernetesClient exec_cmd.resp: {0}".format(resp))
+            if "init system (PID 1). Can't operate." in resp:
+                return "KubernetesClient can't get the resp by {0}".format(cmd)
+            return resp
+        except Exception as e:
+            return f"KubernetesClient can't get the resp by {cmd}: {str(e)}"
 
     def download(self, remote_path, local_path):
         return self.__download_file_from_pod(self.namespace, self.pod_name, self.container_name, remote_path, local_path)

test/common/ssh_client/test_kubernetes_cilent.yaml

@@ -0,0 +1,18 @@
+apiVersion: v1
+kind: Config
+clusters:
+- cluster:
+    certificate-authority-data: DATA+OMITTED
+    server: https://127.0.0.1:8443
+  name: dev-cluster
+users:
+- user:
+    client-certificate-data: DATA+OMITTED
+    client-key-data: DATA+OMITTED
+  name: dev-user
+contexts:
+- context:
+    cluster: dev-cluster
+    user: dev-user
+  name: dev-context
+current-context: dev-context