FrameBridge 0.3.1 community source release

source/tests/test_console.py

@@ -0,0 +1,104 @@
+import os
+from pathlib import Path
+import tempfile
+from types import SimpleNamespace
+import unittest
+from unittest import mock
+
+from framebridge_expanded import console
+
+class ConsoleTests(unittest.TestCase):
+    def test_remote_alias_cannot_create_a_local_console(self):
+        with tempfile.TemporaryDirectory() as path, mock.patch.object(Path, "mkdir") as mkdir:
+            with self.assertRaisesRegex(RuntimeError, "REMOTE_CONSOLE_SSH_CONFIGURATION_REQUIRED"):
+                console.context(Path(path) / "profile", "a-different-node", create=True)
+            mkdir.assert_not_called()
+
+    def test_profile_socket_and_job_roots_are_isolated(self):
+        with tempfile.TemporaryDirectory() as path:
+            one = console.context(Path(path) / "one")
+            two = console.context(Path(path) / "two")
+        self.assertNotEqual(one["socket"], two["socket"])
+        self.assertNotEqual(one["session"], two["session"])
+        self.assertEqual(one["jobs_root"], Path(path) / "one/home/.local/share/framebridge/fbterm")
+        self.assertLess(len(str(one["socket"]).encode()), 108)
+
+    def desktop(self, *, display=":0", vt="7", user=None):
+        return SimpleNamespace(returncode=0, stdout=(
+            "Type=x11\nUser=" + str(os.getuid() if user is None else user)
+            + "\nActive=yes\nRemote=no\nDisplay=" + display + "\nVTNr=" + vt + "\nTTY=\nLeader=3668\n"))
+
+    def test_lightdm_empty_tty_does_not_hide_active_user_desktop(self):
+        with mock.patch.object(console.subprocess, "run", return_value=self.desktop()), \
+             mock.patch.object(Path, "read_text", return_value="tty7\n"):
+            self.assertTrue(console._active_graphical_display({"DISPLAY": ":0", "XDG_SESSION_ID": "19"}))
+
+    def test_private_virtual_display_is_not_chosen_as_user_desktop(self):
+        with mock.patch.object(console.subprocess, "run", return_value=self.desktop()), \
+             mock.patch.object(Path, "read_text", return_value="tty7\n"):
+            self.assertFalse(console._active_graphical_display({"DISPLAY": ":99", "XDG_SESSION_ID": "19"}))
+
+    def test_changed_foreground_vt_prevents_gui_auto_open(self):
+        with mock.patch.object(console.subprocess, "run", return_value=self.desktop()), \
+             mock.patch.object(Path, "read_text", return_value="tty1\n"):
+            self.assertFalse(console._active_graphical_display({"DISPLAY": ":0", "XDG_SESSION_ID": "19"}))
+
+    def test_foreign_user_session_is_not_selected(self):
+        with mock.patch.object(console.subprocess, "run", return_value=self.desktop(user=os.getuid() + 1)), \
+             mock.patch.object(Path, "read_text", return_value="tty7\n"):
+            self.assertFalse(console._active_graphical_display({"DISPLAY": ":0", "XDG_SESSION_ID": "19"}))
+
+    def test_noninteractive_terminal_fails_before_starting_a_console(self):
+        with tempfile.TemporaryDirectory() as path:
+            ctx = console.context(Path(path) / "one")
+            with mock.patch.object(console, "context", return_value=ctx), \
+                 mock.patch.object(console, "_active_graphical_display", return_value=False), \
+                 mock.patch.object(console.os, "isatty", return_value=False), \
+                 mock.patch.object(console.subprocess, "run") as run:
+                with self.assertRaisesRegex(RuntimeError, "CONSOLE_INTERACTIVE_TERMINAL_REQUIRED"):
+                    console.run(ctx["state"])
+                run.assert_not_called()
+
+    def test_terminal_attach_keeps_shared_identity_and_profile_jobs(self):
+        with tempfile.TemporaryDirectory() as path:
+            ctx = console.context(Path(path) / "one")
+            with mock.patch.object(console, "context", return_value=ctx), \
+                 mock.patch.object(console, "_active_graphical_display", return_value=False), \
+                 mock.patch.object(console.os, "isatty", return_value=True), \
+                 mock.patch.object(console.subprocess, "run", return_value=SimpleNamespace(returncode=0)), \
+                 mock.patch.object(console.os, "execve") as execute:
+                self.assertEqual(console.run(ctx["state"], inside=True), 0)
+                program, args, env = execute.call_args.args
+        self.assertEqual(program, "/usr/bin/tmux")
+        self.assertIn(str(ctx["socket"]), args)
+        self.assertIn("=" + ctx["session"], args)
+        self.assertEqual(env["FRAMEBRIDGE_FBTERM_ROOT"], str(ctx["jobs_root"]))
+        self.assertNotIn("sudo", args)
+
+    def test_layout_failure_does_not_attach_or_replace_a_shell(self):
+        with tempfile.TemporaryDirectory() as path:
+            ctx = console.context(Path(path) / "one")
+            with mock.patch.object(console, "context", return_value=ctx), \
+                 mock.patch.object(console, "_active_graphical_display", return_value=False), \
+                 mock.patch.object(console.os, "isatty", return_value=True), \
+                 mock.patch.object(console.subprocess, "run", return_value=SimpleNamespace(returncode=1)), \
+                 mock.patch.object(console.os, "execve") as execute:
+                with self.assertRaisesRegex(RuntimeError, "CONSOLE_LAYOUT_FAILED"):
+                    console.run(ctx["state"], inside=True)
+                execute.assert_not_called()
+
+    def test_dangling_socket_symlink_is_refused(self):
+        with tempfile.TemporaryDirectory() as path:
+            sock = Path(path) / "console.sock"
+            sock.symlink_to(Path(path) / "missing")
+            with self.assertRaisesRegex(RuntimeError, "CONSOLE_SOCKET_UNTRUSTED"):
+                console._socket_exists(sock)
+
+    def test_public_runtime_directory_is_refused(self):
+        with tempfile.TemporaryDirectory() as path:
+            os.chmod(path, 0o755)
+            with self.assertRaisesRegex(RuntimeError, "CONSOLE_RUNTIME_DIRECTORY_UNTRUSTED"):
+                console._private_runtime(Path(path))
+
+if __name__ == "__main__":
+    unittest.main()