Restore linked input settings and attribute test notifications to Fcitx

test/shell.d/input-acceptance-checks.py

@@ -0,0 +1,144 @@
+"""Regression fixtures for acceptance cleanup and notification attribution."""
+
+from pathlib import Path
+import runpy
+import subprocess
+import tempfile
+import unittest
+from unittest.mock import patch
+
+
+ROOT = Path(__file__).resolve().parents[2]
+state = runpy.run_path(str(ROOT / "test/acceptance.d/input-method-state.py"))
+notifications = runpy.run_path(str(ROOT / "test/acceptance.d/input-method-notifications.py"))
+
+
+class AcceptanceHelpers(unittest.TestCase):
+  def setUp(self):
+    self.temp = tempfile.TemporaryDirectory()
+    self.addCleanup(self.temp.cleanup)
+    self.root = Path(self.temp.name)
+
+  def test_nested_linked_settings_restore_targets_links_and_modes(self):
+    config = self.root / "fcitx5"
+    config.mkdir()
+    target = self.root / "dotfiles-config"
+    target.write_text("custom switching keys")
+    target.chmod(0o600)
+    link = config / "config"
+    link.symlink_to("../dotfiles-config")
+    profile = config / "profile"
+    profile.write_text("custom input methods")
+    snapshot = self.root / "snapshot"
+    state["backup"](snapshot, [config])
+    target.write_text("test keys")
+    target.chmod(0o644)
+    profile.write_text("test methods")
+    (config / "new-engine.conf").write_text("created by test")
+    state["restore"](snapshot)
+    self.assertTrue(link.is_symlink())
+    self.assertEqual(link.readlink(), Path("../dotfiles-config"))
+    self.assertEqual(target.read_text(), "custom switching keys")
+    self.assertEqual(target.stat().st_mode & 0o777, 0o600)
+    self.assertEqual(profile.read_text(), "custom input methods")
+    self.assertFalse((config / "new-engine.conf").exists())
+
+  def test_directory_links_and_external_font_preferences_are_restored(self):
+    target = self.root / "dotfiles"
+    target.mkdir()
+    (target / "profile").write_text("original")
+    config = self.root / "fcitx5"
+    config.symlink_to(target)
+    font_target = self.root / "font-preference"
+    font_target.write_text("original font")
+    font = self.root / "font.conf"
+    font.symlink_to(font_target)
+    snapshot = self.root / "snapshot"
+    state["backup"](snapshot, [config, font])
+    (target / "profile").write_text("changed")
+    font_target.write_text("test font")
+    state["restore"](snapshot)
+    self.assertTrue(config.is_symlink())
+    self.assertTrue(font.is_symlink())
+    self.assertEqual((target / "profile").read_text(), "original")
+    self.assertEqual(font_target.read_text(), "original font")
+
+  def test_missing_targets_and_paths_remain_missing(self):
+    absent = self.root / "absent"
+    target = self.root / "missing-target"
+    link = self.root / "dangling"
+    link.symlink_to(target)
+    snapshot = self.root / "snapshot"
+    state["backup"](snapshot, [absent, link])
+    absent.mkdir()
+    (absent / "test").write_text("new")
+    target.write_text("created through link")
+    state["restore"](snapshot)
+    self.assertFalse(absent.exists())
+    self.assertTrue(link.is_symlink())
+    self.assertFalse(target.exists())
+
+  def test_failed_restore_still_restarts_the_service(self):
+    script = (ROOT / "test/acceptance.d/input-methods-test.sh").read_text()
+    cleanup = "cleanup() {" + script.split("cleanup() {", 1)[1].split("\n}\ntrap cleanup", 1)[0] + "\n}"
+    commands = """
+notification_monitor=""
+work=$1
+close_windows() { :; }
+systemctl() { printf '%s\n' "$*" >> "$work/service.log"; }
+python() { return 1; }
+busctl() { :; }
+"""
+    result = subprocess.run(["bash", "-ec", commands + cleanup + "\ncleanup", "test", str(self.root)], capture_output=True, text=True)
+    self.assertEqual(result.returncode, 1)
+    self.assertIn("start omarchy-fcitx5.service", (self.root / "service.log").read_text())
+    self.assertIn("snapshot retained", result.stderr)
+
+  def test_unchanged_link_targets_are_not_rewritten(self):
+    target = self.root / "read-only-config"
+    target.write_text("unchanged")
+    target.chmod(0o444)
+    link = self.root / "config"
+    link.symlink_to(target)
+    snapshot = self.root / "snapshot"
+    state["backup"](snapshot, [link])
+    with patch.object(Path, "unlink", side_effect=PermissionError("read-only")):
+      state["restore"](snapshot)
+    self.assertTrue(link.is_symlink())
+    self.assertEqual(target.read_text(), "unchanged")
+
+  def test_new_parent_directories_are_removed(self):
+    service = self.root / "dbus-1/services/org.fcitx.Fcitx5.service"
+    snapshot = self.root / "snapshot"
+    state["backup"](snapshot, [service])
+    service.parent.mkdir(parents=True)
+    service.write_text("created by setup")
+    state["restore"](snapshot)
+    self.assertFalse((self.root / "dbus-1").exists())
+
+  def test_notifications_follow_process_ownership_across_restarts(self):
+    def call(sender, app, body):
+      return f'''method call time=1 sender={sender} -> destination=org.freedesktop.Notifications serial=1 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+   string "{app}"
+   uint32 0
+   string "icon"
+   string "summary"
+   string "{body}"
+'''
+    reminder = call(":1.99", "Fcitx", "unrelated reminder")
+    first = call(":1.42", "Input Method", "input setup prompt")
+    second = call(":1.43", "拼音", "cloud input prompt")
+    changed = '''signal time=2 sender=org.freedesktop.DBus -> destination=(null destination) serial=2 path=/org/freedesktop/DBus; interface=org.freedesktop.DBus; member=NameOwnerChanged
+   string "org.fcitx.Fcitx5"
+   string ":1.42"
+   string ":1.43"
+'''
+    selected = list(notifications["input_notifications"](reminder + first + changed + second, ":1.42"))
+    self.assertEqual(selected, [first, second])
+    self.assertEqual(list(notifications["input_notifications"](reminder, ":1.42")), [])
+    self.assertEqual(list(notifications["input_notifications"]("", ":1.42")), [])
+
+
+
+if __name__ == "__main__":
+  unittest.main()

test/shell.d/input-acceptance-test.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+set -euo pipefail
+source "$(dirname "${BASH_SOURCE[0]}")/base-test.sh"
+python "$ROOT/test/shell.d/input-acceptance-checks.py"