Add salt backend

testinfra/backend/__init__.py

@@ -17,6 +17,7 @@
 
 from testinfra.backend import local
 from testinfra.backend import paramiko
+from testinfra.backend import salt
 from testinfra.backend import ssh
 
 
@@ -27,6 +28,7 @@ def get_backend(backend_type, *args, **kwargs):
             "ssh": ssh.SshBackend,
             "safe_ssh": ssh.SafeSshBackend,
             "paramiko": paramiko.ParamikoBakend,
+            "salt": salt.SaltBackend,
         }[backend_type]
     except KeyError:
         raise RuntimeError("Unknown backend '%s'" % (backend_type,))

testinfra/backend/paramiko.py

@@ -23,9 +23,10 @@
 
 try:
     import paramiko
-    HAS_PARAMIKO = True
 except ImportError:
     HAS_PARAMIKO = False
+else:
+    HAS_PARAMIKO = True
 
 logger = logging.getLogger("testinfra.backend")
 

testinfra/backend/salt.py

@@ -0,0 +1,51 @@
+# -*- coding: utf8 -*-
+# Copyright © 2015 Philippe Pepiot
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import unicode_literals
+from __future__ import absolute_import
+
+from testinfra.backend import base
+
+try:
+    import salt.client
+except ImportError:
+    HAS_SALT = False
+else:
+    HAS_SALT = True
+
+
+class SaltBackend(base.BaseBackend):
+
+    def __init__(self, host, *args, **kwargs):
+        self.host = host
+        self._client = None
+        super(SaltBackend, self).__init__(*args, **kwargs)
+
+    @property
+    def client(self):
+        if self._client is None:
+            if not HAS_SALT:
+                raise RuntimeError(
+                    "You must install salt package to use the salt backend")
+            self._client = salt.client.LocalClient()
+        return self._client
+
+    def run(self, command, *args):
+        command = self.quote(command, *args)
+        out = self.client.cmd(
+            self.host, 'cmd.run_all', [command],
+        )[self.host]
+        return base.CommandResult(
+            out['retcode'], out['stdout'], out['stderr'], command)

testinfra/plugin.py

@@ -34,11 +34,13 @@
 def testinfra_backend(pytestconfig, _testinfra_host):
     if _testinfra_host is not None:
         backend_type = pytestconfig.option.connection or "paramiko"
+        kwargs = {}
+        if pytestconfig.option.ssh_config is not None:
+            kwargs["ssh_config"] = pytestconfig.option.ssh_config
         testinfra.set_backend(
             backend_type,
             _testinfra_host,
-            ssh_config=pytestconfig.option.ssh_config,
-        )
+            **kwargs)
     else:
         testinfra.set_backend("local")
 
@@ -49,7 +51,7 @@ def pytest_addoption(parser):
         "--connection",
         action="store",
         dest="connection",
-        help="Remote connection backend ssh|paramiko|safe_ssh",
+        help="Remote connection backend paramiko|ssh|safe_ssh|salt",
     )
     group._addoption(
         "--hosts",