blob: 18be8199fe74c8a0eb047c7ccbf6c4e6ebc24eaf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
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()
|
