import unittest
from unittest.mock import MagicMock, patch

from parley.ibus_engine import IBusIntegration, ParleyIBusEngine


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()