75 lines
3.8 KiB
Python
75 lines
3.8 KiB
Python
"""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',
|
|
'--user try-restart xdg-desktop-portal-ferese.service', '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)
|
|
self.assertNotIn('--user try-restart xdg-desktop-portal-ferese.service', 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()
|