import unittest
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)
        self.integration._processing = False
        self.integration._spinner_frame = 0
        self.integration._spinner_source = 0
        self.engine = MagicMock()
        ParleyIBusEngine.focused = self.engine

    def tearDown(self) -> None:
        ParleyIBusEngine.focused = None

    @patch("parley.ibus_engine.GLib.timeout_add", return_value=42)
    def test_processing_uses_temporary_preedit_text(
        self, timeout_add: MagicMock
    ) -> None:
        self.integration.set_processing(True)

        arguments = self.engine.update_preedit_text_with_mode.call_args.args
        self.assertEqual(arguments[0].get_text(), "⠋ Transcribing…")
        self.assertTrue(arguments[2])
        timeout_add.assert_called_once_with(120, self.integration._advance_spinner)

    @patch("parley.ibus_engine.GLib.Source.remove")
    def test_stopping_processing_hides_preedit(self, remove: MagicMock) -> None:
        self.integration._processing = True
        self.integration._spinner_source = 42

        self.integration.set_processing(False)

        remove.assert_called_once_with(42)
        self.engine.hide_preedit_text.assert_called_once_with()


if __name__ == "__main__":
    unittest.main()