diff options
Diffstat (limited to 'tests/test_windows.py')
| -rw-r--r-- | tests/test_windows.py | 34 |
1 files changed, 26 insertions, 8 deletions
diff --git a/tests/test_windows.py b/tests/test_windows.py index e52e23d..e632406 100644 --- a/tests/test_windows.py +++ b/tests/test_windows.py @@ -31,10 +31,12 @@ class DesktopIcons(unittest.TestCase): from pathlib import Path base = Path(tempfile.mkdtemp()) (base / "com.obsproject.Studio.desktop").write_text( - "[Desktop Entry]\nIcon=com.obsproject.Studio\nStartupWMClass=obs\n") - mapping = windows._desktop_icons([base]) + "[Desktop Entry]\nName=OBS Studio\nIcon=com.obsproject.Studio\nStartupWMClass=obs\n" + "[Desktop Action x]\nName=Other\n") + mapping = windows._desktop_icons(windows._desktop_entries([base])) self.assertEqual(mapping["com.obsproject.studio"], ["com.obsproject.Studio", "obs"]) self.assertEqual(mapping["obs"], ["com.obsproject.Studio", "obs"]) + self.assertEqual(windows._desktop_entries([base])["com.obsproject.Studio"]["name"], "OBS Studio") class Move(unittest.TestCase): @@ -86,14 +88,30 @@ class Commands(unittest.TestCase): ("hyprctl", "activewindow", "-j"): "{}", ("hyprctl", "workspaces", "-j"): '[{"id": 3}]', } - m = windows.make({ - "windows": {"workspaces": [1, 2, 3, 4]}, - "launch": {"obs": {"icon": "obsstudio", "command": "obs"}}, - "run": lambda cmd: responses[tuple(cmd)], - }) + entries = {"com.obsproject.Studio": {"name": "OBS Studio", "icon": "obs", "wmclass": None}} + with mock.patch.object(windows, "_desktop_entries", lambda: entries): + m = windows.make({ + "windows": {"workspaces": [1, 2, 3, 4]}, + "launch": {"apps": ["com.obsproject.Studio", "not-installed"]}, + "run": lambda cmd: responses[tuple(cmd)], + }) st = m.state() self.assertEqual(st["workspaces"], [1, 2, 3, 4]) - self.assertEqual(st["launches"], [{"key": "obs", "icon": "obsstudio", "label": "obs"}]) + self.assertEqual(st["launches"], + [{"key": "com.obsproject.Studio", "icon": "obs", "label": "OBS Studio"}]) + + def test_launch_runs_only_listed_installed_apps(self): + entries = {"kitty": {"name": "kitty", "icon": "kitty", "wmclass": None}, + "other": {"name": "Other", "icon": None, "wmclass": None}} + calls = [] + with mock.patch.object(windows, "_desktop_entries", lambda: entries): + m = windows.make({"launch": {"apps": ["kitty", "gone"]}, "run": lambda cmd: "[]"}) + with mock.patch.object(windows, "_popen", calls.append): + m.actions["launch"]({"key": "kitty"}) + for bad in ("other", "gone", "kitty; rm"): + with self.assertRaises(windows.ModuleError): + m.actions["launch"]({"key": bad}) + self.assertEqual(calls, [["gtk-launch", "kitty"]]) if __name__ == "__main__": |