252 lines
11 KiB
Python
252 lines
11 KiB
Python
import unittest
|
|
from unittest import mock
|
|
|
|
import gi
|
|
gi.require_version("Atspi", "2.0")
|
|
from gi.repository import Atspi
|
|
|
|
from cthulhu import ax_object
|
|
from cthulhu import cthulhu_state
|
|
from cthulhu import focus_manager
|
|
from cthulhu.scripts import default
|
|
from cthulhu.scripts.toolkits.Chromium import script as chromium_script
|
|
from cthulhu.scripts.toolkits.Chromium import script_utilities as chromium_utilities
|
|
|
|
|
|
class ChromiumOmniboxRegressionTests(unittest.TestCase):
|
|
def tearDown(self):
|
|
cthulhu_state.activeWindow = None
|
|
cthulhu_state.locusOfFocus = None
|
|
|
|
def test_detects_chromium_popup_item_with_broken_ancestry(self):
|
|
popupItem = object()
|
|
|
|
with (
|
|
mock.patch.object(ax_object.AXObject, "is_dead", return_value=False),
|
|
mock.patch.object(ax_object.AXObject, "get_toolkit_name", return_value="chromium"),
|
|
mock.patch.object(ax_object.AXObject, "get_role", return_value=Atspi.Role.LIST_ITEM),
|
|
mock.patch.object(ax_object.AXObject, "get_parent", return_value=None),
|
|
mock.patch.object(ax_object.debug, "print_tokens"),
|
|
):
|
|
self.assertTrue(ax_object.AXObject.has_broken_popup_ancestry(popupItem))
|
|
|
|
def test_ignores_chromium_popup_item_with_valid_ancestry(self):
|
|
popupItem = object()
|
|
parent = object()
|
|
|
|
def get_role(obj):
|
|
if obj is popupItem:
|
|
return Atspi.Role.LIST_ITEM
|
|
return Atspi.Role.APPLICATION
|
|
|
|
with (
|
|
mock.patch.object(ax_object.AXObject, "is_dead", return_value=False),
|
|
mock.patch.object(ax_object.AXObject, "get_toolkit_name", return_value="chromium"),
|
|
mock.patch.object(ax_object.AXObject, "get_role", side_effect=get_role),
|
|
mock.patch.object(ax_object.AXObject, "get_parent", side_effect=[parent, None]),
|
|
):
|
|
self.assertFalse(ax_object.AXObject.has_broken_popup_ancestry(popupItem))
|
|
|
|
def test_preserves_active_window_when_chromium_popup_ancestry_is_broken(self):
|
|
activeWindow = object()
|
|
popupItem = object()
|
|
cthulhu_state.activeWindow = activeWindow
|
|
cthulhu_state.locusOfFocus = popupItem
|
|
|
|
controller = mock.Mock()
|
|
app = mock.Mock()
|
|
manager = None
|
|
with (
|
|
mock.patch.object(focus_manager.dbus_service, "get_remote_controller", return_value=controller),
|
|
mock.patch.object(focus_manager.AXObject, "has_broken_popup_ancestry", return_value=True),
|
|
):
|
|
manager = focus_manager.FocusManager(app)
|
|
manager.set_active_window(None)
|
|
|
|
self.assertIs(manager.get_active_window(), activeWindow)
|
|
self.assertIs(cthulhu_state.activeWindow, activeWindow)
|
|
|
|
def test_detects_nameless_frame_with_listbox_as_autocomplete_popup(self):
|
|
utilities = chromium_utilities.Utilities.__new__(chromium_utilities.Utilities)
|
|
frame = object()
|
|
listbox = object()
|
|
|
|
with (
|
|
mock.patch.object(chromium_utilities.AXUtilities, "is_frame", return_value=True),
|
|
mock.patch.object(chromium_utilities.AXObject, "get_name", return_value=""),
|
|
mock.patch.object(chromium_utilities.AXObject, "get_child_count", return_value=1),
|
|
mock.patch.object(chromium_utilities.AXObject, "get_child", return_value=listbox),
|
|
mock.patch.object(chromium_utilities.AXUtilities, "is_list_box", return_value=True),
|
|
):
|
|
result = utilities.autocompletePopupForFrame(frame)
|
|
|
|
self.assertIs(result, listbox)
|
|
|
|
def test_does_not_treat_nested_listbox_as_autocomplete_popup(self):
|
|
utilities = chromium_utilities.Utilities.__new__(chromium_utilities.Utilities)
|
|
frame = object()
|
|
|
|
with (
|
|
mock.patch.object(chromium_utilities.AXUtilities, "is_frame", return_value=True),
|
|
mock.patch.object(chromium_utilities.AXObject, "get_name", return_value=""),
|
|
mock.patch.object(chromium_utilities.AXObject, "get_child_count", return_value=2),
|
|
):
|
|
result = utilities.autocompletePopupForFrame(frame)
|
|
|
|
self.assertIsNone(result)
|
|
|
|
@staticmethod
|
|
def _make_chromium_script():
|
|
testScript = chromium_script.Script.__new__(chromium_script.Script)
|
|
testScript.utilities = mock.Mock()
|
|
testScript.utilities.canBeActiveWindow.return_value = True
|
|
testScript.utilities.autocompletePopupForFrame.return_value = None
|
|
testScript.utilities.popupMenuForFrame.return_value = None
|
|
testScript._lastAutocompletePopupItem = None
|
|
testScript.presentObject = mock.Mock()
|
|
return testScript
|
|
|
|
def test_autocomplete_popup_activation_presents_item_without_frame_fallback(self):
|
|
testScript = self._make_chromium_script()
|
|
frame = object()
|
|
listbox = object()
|
|
item = object()
|
|
event = mock.Mock(source=frame)
|
|
testScript.utilities.autocompletePopupForFrame.return_value = listbox
|
|
testScript.utilities.selectedChildren.return_value = [item]
|
|
|
|
with (
|
|
mock.patch.object(chromium_script.AXUtilities, "get_focused_object", return_value=item),
|
|
mock.patch.object(chromium_script.cthulhu, "setActiveWindow") as setActiveWindow,
|
|
mock.patch.object(chromium_script.cthulhu, "setLocusOfFocus") as setLocusOfFocus,
|
|
mock.patch.object(default.Script, "onWindowActivated") as defaultHandler,
|
|
):
|
|
chromium_script.Script.onWindowActivated(testScript, event)
|
|
|
|
setActiveWindow.assert_not_called()
|
|
setLocusOfFocus.assert_not_called()
|
|
testScript.presentObject.assert_called_once_with(item, interrupt=True)
|
|
self.assertIs(testScript._lastAutocompletePopupItem, item)
|
|
defaultHandler.assert_not_called()
|
|
|
|
def test_autocomplete_popup_activation_uses_focused_item_when_none_selected(self):
|
|
testScript = self._make_chromium_script()
|
|
frame = object()
|
|
listbox = object()
|
|
item = object()
|
|
event = mock.Mock(source=frame)
|
|
testScript.utilities.autocompletePopupForFrame.return_value = listbox
|
|
testScript.utilities.selectedChildren.return_value = []
|
|
|
|
with mock.patch.object(
|
|
chromium_script.AXUtilities,
|
|
"get_focused_object",
|
|
return_value=item,
|
|
):
|
|
chromium_script.Script.onWindowActivated(testScript, event)
|
|
|
|
testScript.presentObject.assert_called_once_with(item, interrupt=True)
|
|
self.assertIs(testScript._lastAutocompletePopupItem, item)
|
|
|
|
def test_autocomplete_popup_selection_is_presented_without_changing_document_focus(self):
|
|
testScript = self._make_chromium_script()
|
|
comboBox = object()
|
|
listbox = object()
|
|
frame = object()
|
|
item = object()
|
|
event = mock.Mock(source=item, detail1=True)
|
|
cthulhu_state.locusOfFocus = comboBox
|
|
testScript.utilities.inDocumentContent.return_value = False
|
|
|
|
with (
|
|
mock.patch.object(chromium_script.web.Script, "onSelectedChanged", return_value=False),
|
|
mock.patch.object(
|
|
chromium_script.AXObject,
|
|
"find_ancestor_inclusive",
|
|
return_value=listbox,
|
|
),
|
|
mock.patch.object(chromium_script.AXObject, "get_parent", return_value=frame),
|
|
mock.patch.object(chromium_script.AXObject, "get_name", return_value=""),
|
|
mock.patch.object(chromium_script.AXUtilities, "is_frame", return_value=True),
|
|
):
|
|
chromium_script.Script.onSelectedChanged(testScript, event)
|
|
chromium_script.Script.onSelectedChanged(testScript, event)
|
|
|
|
self.assertIs(cthulhu_state.locusOfFocus, comboBox)
|
|
testScript.presentObject.assert_called_once_with(item, interrupt=True)
|
|
|
|
def test_main_window_deactivation_for_autocomplete_popup_preserves_state(self):
|
|
testScript = self._make_chromium_script()
|
|
mainFrame = object()
|
|
comboBox = object()
|
|
event = mock.Mock(source=mainFrame)
|
|
cthulhu_state.locusOfFocus = comboBox
|
|
testScript.utilities.lastKeyAndModifiers.return_value = ("Down", 8)
|
|
|
|
with (
|
|
mock.patch.object(chromium_script.AXUtilities, "is_combo_box", return_value=True),
|
|
mock.patch.object(
|
|
chromium_script.AXUtilities,
|
|
"supports_autocompletion",
|
|
return_value=True,
|
|
),
|
|
mock.patch.object(default.Script, "onWindowDeactivated") as defaultHandler,
|
|
):
|
|
chromium_script.Script.onWindowDeactivated(testScript, event)
|
|
|
|
defaultHandler.assert_not_called()
|
|
|
|
def test_unrelated_autocomplete_window_deactivation_reaches_default(self):
|
|
testScript = self._make_chromium_script()
|
|
mainFrame = object()
|
|
comboBox = object()
|
|
app = object()
|
|
event = mock.Mock(source=mainFrame)
|
|
cthulhu_state.locusOfFocus = comboBox
|
|
testScript.utilities.lastKeyAndModifiers.return_value = ("w", 4)
|
|
|
|
with (
|
|
mock.patch.object(chromium_script.AXUtilities, "is_combo_box", return_value=True),
|
|
mock.patch.object(
|
|
chromium_script.AXUtilities,
|
|
"supports_autocompletion",
|
|
return_value=True,
|
|
),
|
|
mock.patch.object(chromium_script.AXObject, "get_application", return_value=app),
|
|
mock.patch.object(chromium_script.AXObject, "iter_children", return_value=iter(())),
|
|
mock.patch.object(chromium_script.web.Script, "onWindowDeactivated", return_value=False),
|
|
mock.patch.object(default.Script, "onWindowDeactivated") as defaultHandler,
|
|
):
|
|
chromium_script.Script.onWindowDeactivated(testScript, event)
|
|
|
|
defaultHandler.assert_called_once_with(testScript, event)
|
|
|
|
def test_unmodified_arrow_deactivation_reaches_default(self):
|
|
testScript = self._make_chromium_script()
|
|
mainFrame = object()
|
|
comboBox = object()
|
|
app = object()
|
|
event = mock.Mock(source=mainFrame)
|
|
cthulhu_state.locusOfFocus = comboBox
|
|
testScript.utilities.lastKeyAndModifiers.return_value = ("Down", 0)
|
|
|
|
with (
|
|
mock.patch.object(chromium_script.AXUtilities, "is_combo_box", return_value=True),
|
|
mock.patch.object(
|
|
chromium_script.AXUtilities,
|
|
"supports_autocompletion",
|
|
return_value=True,
|
|
),
|
|
mock.patch.object(chromium_script.AXObject, "get_application", return_value=app),
|
|
mock.patch.object(chromium_script.AXObject, "iter_children", return_value=iter(())),
|
|
mock.patch.object(chromium_script.web.Script, "onWindowDeactivated", return_value=False),
|
|
mock.patch.object(default.Script, "onWindowDeactivated") as defaultHandler,
|
|
):
|
|
chromium_script.Script.onWindowDeactivated(testScript, event)
|
|
|
|
defaultHandler.assert_called_once_with(testScript, event)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|