David Heinemeier HanssonandClaude Opus 5.5
7b260bed17
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
123 lines
6.8 KiB
Python
123 lines
6.8 KiB
Python
import importlib.util
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
ROOT = Path(os.environ["OMARCHY_PATH"])
|
|
sys.path.insert(0, str(ROOT / "default/input-methods"))
|
|
spec = importlib.util.spec_from_file_location("typing_setup", ROOT / "default/input-methods/typing_setup.py")
|
|
typing = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(typing)
|
|
|
|
|
|
class TypingSetupTest(unittest.TestCase):
|
|
def test_variants_and_installer_binding_safety(self):
|
|
self.assertEqual(typing.keyboard_values(["us:intl", "fr"]), ("us,fr", "intl,"))
|
|
self.assertEqual(typing.keyboard_values(["ru:phonetic"]), ("us,ru", ",phonetic"))
|
|
self.assertEqual(typing.keyboard_values(["ru:phonetic", "us"]), ("us,ru", ",phonetic"))
|
|
with self.assertRaises(ValueError):
|
|
typing.keyboard_values([])
|
|
|
|
def test_keyboard_catalog_uses_installer_choices(self):
|
|
catalog = typing.keyboard_catalog()
|
|
self.assertEqual(catalog["us"], "English (US)")
|
|
self.assertEqual(catalog["gb"], "English (UK)")
|
|
self.assertEqual(catalog["us:colemak"], "English (US, Colemak)")
|
|
self.assertEqual(catalog["ch"], "German (Switzerland)")
|
|
self.assertEqual(catalog["la"], "Lao")
|
|
# Converted as the installer does, without renaming French.
|
|
self.assertEqual(catalog["fr"], "French")
|
|
self.assertEqual(catalog["bg:phonetic"], "Bulgarian")
|
|
self.assertEqual(catalog["cz:qwerty"], "Czech")
|
|
self.assertNotIn("az", catalog)
|
|
self.assertEqual(catalog["latam"], "Spanish (Latin American)")
|
|
self.assertEqual(catalog["latam"], "Spanish (Latin American)")
|
|
self.assertNotIn("us:intl", catalog)
|
|
self.assertLess(len(catalog), 60)
|
|
|
|
def test_readding_input_keeps_its_original_override(self):
|
|
self.assertEqual(typing.input_items([["keyboard-us", ""]], ["mozc"], {"mozc": "jp"}),
|
|
[["keyboard-us", ""], ["mozc", "jp"]])
|
|
self.assertEqual(typing.input_items([["keyboard-us", ""], ["mozc", "us"]], ["mozc"], {"mozc": "jp"}),
|
|
[["keyboard-us", ""], ["mozc", "us"]])
|
|
|
|
def test_input_deselection_keeps_keyboard_and_selected_overrides(self):
|
|
items = [["keyboard-us", ""], ["mozc", "jp"], ["hangul", ""], ["custom", "de"]]
|
|
self.assertEqual(typing.input_items(items, ["custom", "mozc"]), [["keyboard-us", ""], ["custom", "de"], ["mozc", "jp"]])
|
|
self.assertEqual(typing.input_items(items, []), [["keyboard-us", ""]])
|
|
|
|
def test_cancellation_does_not_change_input_settings(self):
|
|
with patch.object(typing.setup, "live_group", return_value=("Default", "us", [["keyboard-us", ""]])), patch.object(typing, "input_catalog", return_value=({"mozc": "Japanese"}, {"mozc": "Japanese"})), patch.object(typing, "choose", return_value=None), patch.object(typing.setup, "live_set") as setter:
|
|
self.assertFalse(typing.configure_inputs())
|
|
setter.assert_not_called()
|
|
|
|
def test_failed_input_save_restores_the_original_group(self):
|
|
before = [["keyboard-us", ""], ["mozc", "jp"]]
|
|
after = [["keyboard-us", ""]]
|
|
with patch.object(typing.setup, "run"), patch.object(typing.setup, "live_set") as setter, patch.object(typing.setup, "live_group", return_value=("Default", "us", before)):
|
|
with self.assertRaises(RuntimeError):
|
|
typing.set_inputs("Default", "us", before, after)
|
|
self.assertEqual(setter.call_args_list[-1].args, ("Default", "us", before))
|
|
|
|
def test_picker_preselects_current_entries_and_returns_empty_selection(self):
|
|
result = subprocess.CompletedProcess([], 0, stdout="[]")
|
|
with patch.object(typing.subprocess, "run", return_value=result) as process:
|
|
self.assertIsNone(typing.choose("Inputs", {"mozc": "Japanese", "hangul": "Korean"}, ["hangul"], "input"))
|
|
args = process.call_args.args[0]
|
|
self.assertEqual(args[2], "\tKorean")
|
|
self.assertIn("--multiple", args)
|
|
self.assertEqual(args[args.index("--change-key") + 1], "typing:input")
|
|
self.assertEqual(args[-2:], ["--selected", "Korean"])
|
|
|
|
def test_apply_rejects_unknown_values(self):
|
|
with self.assertRaises(ValueError):
|
|
typing.apply_selection("input", {"catalog": {"mozc": "Japanese"}}, ["unknown"])
|
|
|
|
def test_immediate_selection_removes_an_input_and_accepts_keyboard_only(self):
|
|
with patch.object(typing, "save_inputs") as save, patch("builtins.print"):
|
|
typing.apply_selection("input", {"catalog": {"mozc": "Japanese", "hangul": "Korean"}}, ["Japanese"])
|
|
save.assert_called_once_with(["mozc"], None)
|
|
typing.apply_selection("input", {"catalog": {"mozc": "Japanese"}}, [])
|
|
self.assertEqual(save.call_args.args, ([], None))
|
|
|
|
def test_keyboard_checks_reflect_normalized_selection(self):
|
|
with patch.object(typing, "save_keyboard"), patch.object(typing, "keyboard_selection", return_value=["us", "ru"]), patch("builtins.print") as output:
|
|
typing.apply_selection("keyboard", {"catalog": {"us": "English", "ru": "Russian"}}, ["Russian"])
|
|
self.assertEqual(json.loads(output.call_args.args[0]), ["English", "Russian"])
|
|
|
|
def test_keyboard_override_failure_preserves_symlink_and_restores_settings(self):
|
|
with tempfile.TemporaryDirectory() as temporary:
|
|
config = Path(temporary)
|
|
target = config / "actual-layouts"
|
|
target.write_text("XKBLAYOUT=de\n")
|
|
target.chmod(0o600)
|
|
path = config / "omarchy/keyboard-layouts"
|
|
path.parent.mkdir()
|
|
path.symlink_to(target)
|
|
before = ("Default", "us", [["keyboard-us", ""], ["mozc", "jp"]])
|
|
with patch.dict(os.environ, {"XDG_CONFIG_HOME": str(config)}), patch.object(typing.setup, "live_group", return_value=before), patch.object(typing.setup, "run", return_value=""), patch.object(typing, "keyboard_selection", return_value=["de"]), patch.object(typing.setup, "live_set") as setter:
|
|
with self.assertRaisesRegex(RuntimeError, "override"):
|
|
typing.save_keyboard(["us", "fr"])
|
|
self.assertTrue(path.is_symlink())
|
|
self.assertEqual(target.read_text(), "XKBLAYOUT=de\n")
|
|
self.assertEqual(target.stat().st_mode & 0o777, 0o600)
|
|
setter.assert_not_called()
|
|
|
|
def test_keyboard_save_aligns_fcitx_and_preserves_language_overrides(self):
|
|
with tempfile.TemporaryDirectory() as temporary:
|
|
before = ("Default", "us", [["keyboard-us", ""], ["mozc", "jp"]])
|
|
after = ("Default", "fr", [["keyboard-fr", ""], ["mozc", "jp"]])
|
|
with patch.dict(os.environ, {"XDG_CONFIG_HOME": temporary}), patch.object(typing.setup, "live_group", side_effect=[before, after]), patch.object(typing.setup, "run", return_value=""), patch.object(typing, "keyboard_selection", return_value=["fr", "us:intl"]), patch.object(typing.setup, "live_set") as setter:
|
|
typing.save_keyboard(["fr", "us:intl"])
|
|
self.assertEqual(setter.call_args.args, after)
|
|
self.assertEqual((Path(temporary) / "omarchy/keyboard-layouts").read_text(), "XKBLAYOUT=fr,us\nXKBVARIANT=,intl\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|