logind: Add SetWallMessage

`systemctl suspend` and friends call this. Add a test case.

Author: martinpitt

dbusmock/templates/logind.py

@@ -58,6 +58,7 @@ def load(mock, parameters):
             ("GetUser", "u", "o", 'ret = "/org/freedesktop/login1/user/" + args[0]'),
             ("KillUser", "us", "", ""),
             ("TerminateUser", "u", "", ""),
+            ("SetWallMessage", "sb", "", ""),
         ],
     )
 

tests/test_logind.py

@@ -6,6 +6,8 @@
 (c) 2017 - 2025 Martin Pitt <martin@piware.de>
 """
 
+import fcntl
+import os
 import re
 import shutil
 import subprocess
@@ -42,6 +44,9 @@ def setUp(self):
         self.addCleanup(self.p_mock.terminate)
         self.addCleanup(self.p_mock.stdout.close)
 
+        flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL)
+        fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK)
+
     def test_empty(self):
         cmd = ["loginctl"]
         if self.version >= "209":
@@ -123,6 +128,17 @@ def test_inhibit(self):
         out = subprocess.check_output(["systemd-inhibit"], text=True)
         self.assertRegex(out, "No inhibitors|0 inhibitors listed")
 
+    def test_suspend(self):
+        (p_mock_polkit, _obj_polkitd) = self.spawn_server_template("polkitd", {}, stdout=subprocess.DEVNULL)
+        self.addCleanup(p_mock_polkit.wait)
+        self.addCleanup(p_mock_polkit.terminate)
+
+        subprocess.check_call(["systemctl", "suspend"])
+
+        log = self.p_mock.stdout.read().decode()
+        self.assertIn('SetWallMessage "" True', log)
+        self.assertIn("Suspend True", log)
+
 
 if __name__ == "__main__":
     # avoid writing to stderr