operations/openrc: support runlevel for enabling services

Author: sengo4hd

pyinfra/operations/openrc.py

@@ -50,14 +50,14 @@ def service(
 
         if enabled is True:
             if not is_enabled:
-                yield "rc-update add {0}".format(service)
+                yield "rc-update add {0} {1}".format(service, runlevel)
                 openrc_enabled[service] = True
             else:
                 host.noop("service {0} is enabled".format(service))
 
         if enabled is False:
             if is_enabled:
-                yield "rc-update del {0}".format(service)
+                yield "rc-update del {0} {1}".format(service, runlevel)
                 openrc_enabled[service] = False
             else:
                 host.noop("service {0} is disabled".format(service))

tests/operations/openrc.service/disable.json

@@ -16,6 +16,6 @@
         }
     },
     "commands": [
-        "rc-update del nginx"
+        "rc-update del nginx default"
     ]
 }

tests/operations/openrc.service/disable_runlevel.json

@@ -0,0 +1,22 @@
+{
+    "args": ["nftables"],
+    "kwargs": {
+        "runlevel": "boot",
+        "enabled": false
+    },
+    "facts": {
+        "openrc.OpenrcStatus": {
+            "runlevel=boot": {
+                "nftables": true
+            }
+        },
+        "openrc.OpenrcEnabled": {
+            "runlevel=boot": {
+                "nftables": true
+            }
+        }
+    },
+    "commands": [
+        "rc-update del nftables boot"
+    ]
+}

tests/operations/openrc.service/enable.json

@@ -16,6 +16,6 @@
         }
     },
     "commands": [
-        "rc-update add nginx"
+        "rc-update add nginx default"
     ]
 }

tests/operations/openrc.service/enable_runlevel.json

@@ -0,0 +1,22 @@
+{
+    "args": ["nftables"],
+    "kwargs": {
+        "runlevel": "boot",
+        "enabled": true
+    },
+    "facts": {
+        "openrc.OpenrcStatus": {
+            "runlevel=boot": {
+                "nftables": true
+            }
+        },
+        "openrc.OpenrcEnabled": {
+            "runlevel=boot": {
+                "nftables": false
+            }
+        }
+    },
+    "commands": [
+        "rc-update add nftables boot"
+    ]
+}