From 411540cde2698a74f71abe75cd78b97a5d7ebb06 Mon Sep 17 00:00:00 2001
From: jamesbeedy <jamesbeedy@gmail.com>
Date: Wed, 4 Dec 2024 20:58:17 +0000
Subject: [PATCH] Support stop/start systemd calls

These changes add the stop/start systemd calls to the service
manager and additionally remove the --now flag from enable and
disable commands.
---
 lib/charms/hpc_libs/v0/slurm_ops.py | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/lib/charms/hpc_libs/v0/slurm_ops.py b/lib/charms/hpc_libs/v0/slurm_ops.py
index 83cc47d..e78e949 100644
--- a/lib/charms/hpc_libs/v0/slurm_ops.py
+++ b/lib/charms/hpc_libs/v0/slurm_ops.py
@@ -338,17 +338,37 @@ def type(self) -> _ServiceType:
 class _SystemctlServiceManager(_ServiceManager):
     """Control a Slurm service using systemctl services."""
 
+    def start(self) -> None:
+        """Start service.
+
+        Raises:
+            SlurmOpsError: Raised if `systemctl start ...` returns a non-zero returncode.
+        """
+        _systemctl("start", self._service.value)
+
+    def stop(self) -> None:
+        """Stop service.
+
+        Raises:
+            SlurmOpsError: Raised if `systemctl stop ...` returns a non-zero returncode.
+        """
+        _systemctl("stop", self._service.value)
+
     def enable(self) -> None:
         """Enable service.
 
         Raises:
             SlurmOpsError: Raised if `systemctl enable ...` returns a non-zero returncode.
         """
-        _systemctl("enable", "--now", self._service.value)
+        _systemctl("enable", self._service.value)
 
     def disable(self) -> None:
-        """Disable service."""
-        _systemctl("disable", "--now", self._service.value)
+        """Disable service.
+
+        Raises:
+            SlurmOpsError: Raised if `systemctl disable ...` returns a non-zero returncode.
+        """
+        _systemctl("disable", self._service.value)
 
     def restart(self) -> None:
         """Restart service."""