Allow kubernete context to be supplied

Closes #554

Co-authored-by: Philippe Pepiot <[email protected]>

Author: NickLarsenNZ

doc/source/backends.rst

@@ -127,6 +127,8 @@ namespace::
     $ py.test --hosts='kubectl://mypod-a1b2c3'
     # specify container name and namespace
     $ py.test --hosts='kubectl://somepod-2536ab?container=nginx&namespace=web'
+    # specify the kubeconfig context to use
+    $ py.test --hosts='kubectl://somepod-2536ab?context=k8s-cluster-a&container=nginx'
     # you can specify kubeconfig either from KUBECONFIG environment variable
     # or when working with multiple configuration with the "kubeconfig" option
     $ py.test --hosts='kubectl://somepod-123?kubeconfig=/path/kubeconfig,kubectl://otherpod-123?kubeconfig=/other/kubeconfig'

test/test_backends.py

@@ -441,18 +441,27 @@ def test_parse_hostspec(hostspec, expected):
     assert BaseBackend.parse_hostspec(hostspec) == expected
 
 
-@pytest.mark.parametrize('hostspec,pod,container,namespace,kubeconfig', [
-    ('kubectl://pod', 'pod', None, None, None),
-    ('kubectl://pod?namespace=n', 'pod', None, 'n', None),
-    ('kubectl://pod?container=c&namespace=n', 'pod', 'c', 'n', None),
-    ('kubectl://pod?namespace=n&kubeconfig=k', 'pod', None, 'n', 'k')
-])
-def test_kubectl_hostspec(hostspec, pod, container, namespace, kubeconfig):
+@pytest.mark.parametrize(
+    'hostspec,pod,container,namespace,kubeconfig,context', [
+        ('kubectl://pod', 'pod', None, None, None, None),
+        ('kubectl://pod?namespace=n', 'pod', None, 'n', None, None),
+        ('kubectl://pod?container=c&namespace=n',
+         'pod', 'c', 'n', None, None),
+        ('kubectl://pod?namespace=n&kubeconfig=k',
+         'pod', None, 'n', 'k', None),
+        ('kubectl://pod?context=ctx&container=c',
+         'pod', 'c', None, None, 'ctx')
+    ]
+)
+def test_kubectl_hostspec(
+    hostspec, pod, container, namespace, kubeconfig, context,
+):
     backend = testinfra.get_host(hostspec).backend
     assert backend.name == pod
     assert backend.container == container
     assert backend.namespace == namespace
     assert backend.kubeconfig == kubeconfig
+    assert backend.context == context
 
 
 @pytest.mark.parametrize('hostspec,pod,container,namespace,kubeconfig', [

testinfra/backend/__init__.py

@@ -52,7 +52,7 @@ def parse_hostspec(hostspec):
                 kw[key] = True
         for key in ("sudo_user", 'namespace', 'container', 'read_timeout_sec',
                     'operation_timeout_sec', 'timeout', 'controlpersist',
-                    'kubeconfig'):
+                    'kubeconfig', 'context'):
             if key in query:
                 kw[key] = query.get(key)[0]
         for key in (

testinfra/backend/kubectl.py

@@ -21,6 +21,7 @@ def __init__(self, name, *args, **kwargs):
         self.container = kwargs.get('container')
         self.namespace = kwargs.get('namespace')
         self.kubeconfig = kwargs.get('kubeconfig')
+        self.context = kwargs.get('context')
         super().__init__(self.name, *args, **kwargs)
 
     def run(self, command, *args, **kwargs):
@@ -32,6 +33,9 @@ def run(self, command, *args, **kwargs):
         if self.kubeconfig is not None:
             kcmd += '--kubeconfig="%s" '
             kcmd_args.append(self.kubeconfig)
+        if self.context is not None:
+            kcmd += '--context="%s" '
+            kcmd_args.append(self.context)
         if self.namespace is not None:
             kcmd += '-n %s '
             kcmd_args.append(self.namespace)