Activate the graphical session and refresh portal services at login

scripts/tests/test_session_startup.py

@@ -0,0 +1,72 @@
+"""Session startup regressions using fake services; never touches the host bus."""
+import os
+from pathlib import Path
+import shutil
+import signal
+import subprocess
+import tempfile
+import time
+import unittest
+
+SCRIPT = Path(__file__).resolve().parents[2] / 'packaging/ferese-session-shell'
+
+
+class SessionStartupTest(unittest.TestCase):
+    def run_session(self, direct=True, fail_target=False, shell_status=0, terminate=False):
+        with tempfile.TemporaryDirectory() as directory:
+            root = Path(directory)
+            helper = root / 'ferese-session-shell'
+            shutil.copyfile(SCRIPT, helper)
+            programs = {
+                'dbus-update-activation-environment': 'printf "import:%s\\n" "$WAYLAND_DISPLAY" >> "$TEST_ROOT/events"\n',
+                'dbus-send': 'echo reload-bus >> "$TEST_ROOT/events"\n',
+                'systemctl': 'echo "$*" >> "$TEST_ROOT/events"\nif [[ $* == "--user start ferese-session.target" && $TEST_FAIL == 1 ]]; then exit 1; fi\n',
+                'ferese-polkit-agent': 'exec sleep 30\n',
+                'ferese-shell': 'printf "shell:%s\\n" "$WAYLAND_SOCKET" >> "$TEST_ROOT/events"\nif [[ $TEST_WAIT == 1 ]]; then exec sleep 30; fi\nexit "$TEST_STATUS"\n',
+            }
+            for name, body in programs.items():
+                path = root / name
+                path.write_text('#!/bin/bash\n' + body)
+                path.chmod(0o755)
+            env = dict(os.environ, PATH=f'{root}:{os.environ["PATH"]}', TEST_ROOT=directory,
+                       FERESE_SESSION_IMPORT_ENV=str(int(direct)), WAYLAND_DISPLAY='private-display',
+                       FERESE_PUBLIC_WAYLAND_DISPLAY='public-display', WAYLAND_SOCKET='77',
+                       TEST_FAIL=str(int(fail_target)), TEST_STATUS=str(shell_status), TEST_WAIT=str(int(terminate)))
+            child = subprocess.Popen(['bash', str(helper)], env=env, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            if terminate:
+                deadline = time.monotonic() + 3
+                while not (root / 'events').exists() or 'shell:77' not in (root / 'events').read_text():
+                    if time.monotonic() > deadline:
+                        child.kill()
+                        self.fail('Session shell did not start')
+                    time.sleep(.01)
+                child.send_signal(signal.SIGTERM)
+            child.communicate(timeout=4)
+            return child.returncode, (root / 'events').read_text().splitlines()
+
+    def test_direct_session_imports_public_display_before_activation_and_cleans_up(self):
+        status, events = self.run_session(shell_status=7)
+        self.assertEqual(status, 7)
+        self.assertEqual(events, ['import:public-display', 'reload-bus', '--user daemon-reload',
+                                  '--user start ferese-session.target', 'shell:77',
+                                  '--user stop ferese-session.target'])
+
+    def test_nested_session_never_changes_host_activation(self):
+        status, events = self.run_session(direct=False)
+        self.assertEqual(status, 0)
+        self.assertEqual(events, ['shell:77'])
+
+    def test_failed_target_does_not_prevent_shell_or_stop_unowned_target(self):
+        status, events = self.run_session(fail_target=True)
+        self.assertEqual(status, 0)
+        self.assertIn('shell:77', events)
+        self.assertNotIn('--user stop ferese-session.target', events)
+
+    def test_termination_stops_the_session_target(self):
+        status, events = self.run_session(terminate=True)
+        self.assertEqual(status, 143)
+        self.assertEqual(events[-1], '--user stop ferese-session.target')
+
+
+if __name__ == '__main__':
+    unittest.main()