Make IBus activation resilientHEADmain

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/test_ibus_engine.py b/tests/test_ibus_engine.py
index 18be819..e3bb825 100644
--- a/tests/test_ibus_engine.py
+++ b/tests/test_ibus_engine.py
@@ -4,6 +4,35 @@ from unittest.mock import MagicMock, patch
 from parley.ibus_engine import IBusIntegration, ParleyIBusEngine
 
 
+class IBusSessionTests(unittest.TestCase):
+    def setUp(self) -> None:
+        self.integration = object.__new__(IBusIntegration)
+        self.integration._session_active = False
+        self.integration._restore_engine = None
+
+    def test_session_remembers_and_replaces_the_current_engine(self) -> None:
+        self.integration._current_engine_name = MagicMock(
+            return_value="xkb:il::heb"
+        )
+        self.integration._schedule_activation = MagicMock()
+
+        self.integration.begin_session()
+
+        self.assertTrue(self.integration._session_active)
+        self.assertEqual(self.integration._restore_engine, "xkb:il::heb")
+        self.integration._schedule_activation.assert_called_once_with(0)
+
+    def test_gnome_engine_change_is_reacquired_and_restored_later(self) -> None:
+        self.integration._session_active = True
+        self.integration._restore_engine = "xkb:us::eng"
+        self.integration._schedule_activation = MagicMock()
+
+        self.integration._on_global_engine_changed(MagicMock(), "xkb:il::heb")
+
+        self.assertEqual(self.integration._restore_engine, "xkb:il::heb")
+        self.integration._schedule_activation.assert_called_once_with()
+
+
 class IBusSpinnerTests(unittest.TestCase):
     def setUp(self) -> None:
         self.integration = object.__new__(IBusIntegration)