Service: systemd support

testinfra/modules/service.py

@@ -40,9 +40,15 @@ def is_enabled(self):
     @classmethod
     def as_fixture(cls):
         @pytest.fixture(scope="module")
-        def f(SystemInfo):
+        def f(SystemInfo, Command, File):
             if SystemInfo.type == "linux":
-                return LinuxService
+                if (
+                    Command.run_test("which systemctl").rc == 0
+                    and "systemd" in File("/sbin/init").linked_to
+                ):
+                    return SystemdService
+                else:
+                    return LinuxService
             elif SystemInfo.type == "freebsd":
                 return FreeBSDService
             elif SystemInfo.type == "openbsd":
@@ -84,6 +90,18 @@ def is_enabled_with_level(self, level):
             return False
 
 
+class SystemdService(Service):
+
+    @property
+    def is_running(self):
+        return self.run_expect(
+            [0, 3], "systemctl is-active %s", self.name).rc == 0
+
+    @property
+    def is_enabled(self):
+        return self.run_test("systemctl is-enabled %s", self.name).rc == 0
+
+
 class FreeBSDService(Service):
 
     @property

testinfra/test/test_service.py

@@ -21,8 +21,14 @@ def test_ssh_service(_testinfra_host, Service, SystemInfo):
     if _testinfra_host in (
         "ubuntu_trusty", "debian_wheezy", "debian_jessie",
     ):
-        ssh = Service("ssh")
-        assert ssh.is_running
-        assert ssh.is_enabled
+        name = "ssh"
+    elif _testinfra_host in (
+        "centos_7", "fedora_21",
+    ):
+        name = "sshd"
     else:
         pytest.skip()
+
+    ssh = Service(name)
+    assert ssh.is_running
+    assert ssh.is_enabled