07edcce697
Validate update_completion sets the completions widget's model correctly based on the command text.
137 lines
4.7 KiB
Python
137 lines
4.7 KiB
Python
# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:
|
|
|
|
# Copyright 2016 Ryan Roden-Corrent (rcorre) <ryan@rcorre.net>
|
|
#
|
|
# This file is part of qutebrowser.
|
|
#
|
|
# qutebrowser is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# qutebrowser is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with qutebrowser. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""Tests for the Completer Object."""
|
|
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
from PyQt5.QtGui import QStandardItemModel
|
|
|
|
from qutebrowser.completion.completer import Completer
|
|
from qutebrowser.utils.usertypes import Completion
|
|
|
|
|
|
class FakeCompletionModel(QStandardItemModel):
|
|
|
|
"""Stub for a completion model."""
|
|
|
|
DUMB_SORT = Mock()
|
|
|
|
def __init__(self, kind, parent=None):
|
|
super().__init__(parent)
|
|
self.kind = kind
|
|
|
|
|
|
@pytest.fixture
|
|
def cmd(stubs):
|
|
"""Create the statusbar command prompt the completer uses."""
|
|
return stubs.FakeStatusbarCommand()
|
|
|
|
|
|
@pytest.fixture
|
|
def completer(qtbot, cmd, config_stub):
|
|
"""Create the completer used for testing."""
|
|
config_stub.data = {'completion': {'auto-open': False}}
|
|
return Completer(cmd, 0)
|
|
|
|
|
|
@pytest.fixture
|
|
def instances(monkeypatch):
|
|
"""Mock the instances module so get returns a fake completion model."""
|
|
# populate a model for each completion type, with a nested structure for
|
|
# option and value completion
|
|
instances = {kind: FakeCompletionModel(kind) for kind in Completion}
|
|
instances[Completion.option] = {
|
|
'general': FakeCompletionModel(Completion.option),
|
|
}
|
|
instances[Completion.value] = {
|
|
'general': {
|
|
'ignore-case': FakeCompletionModel(Completion.value),
|
|
}
|
|
}
|
|
monkeypatch.setattr('qutebrowser.completion.completer.instances',
|
|
instances)
|
|
return instances
|
|
|
|
|
|
@pytest.fixture
|
|
def cmdutils_patch(monkeypatch, stubs):
|
|
"""Patch the cmdutils module to provide fake commands."""
|
|
cmds = {
|
|
'set': [Completion.section, Completion.option, Completion.value],
|
|
'help': [Completion.helptopic],
|
|
'quickmark-load': [Completion.quickmark_by_name],
|
|
'bookmark-load': [Completion.bookmark_by_url],
|
|
'open': [Completion.url],
|
|
'buffer': [Completion.tab],
|
|
'session-load': [Completion.sessions],
|
|
'bind': [Completion.empty, Completion.command],
|
|
}
|
|
cmd_utils = stubs.FakeCmdUtils({
|
|
name: stubs.FakeCommand(completion=compl)
|
|
for name, compl in cmds.items()
|
|
})
|
|
monkeypatch.setattr('qutebrowser.completion.completer.cmdutils',
|
|
cmd_utils)
|
|
return cmd_utils
|
|
|
|
|
|
@pytest.mark.parametrize('txt, expected', [
|
|
(':nope|', Completion.command),
|
|
(':nope |', None),
|
|
(':set |', Completion.section),
|
|
(':set gen|', Completion.section),
|
|
(':set general |', Completion.option),
|
|
(':set what |', None),
|
|
(':set general ignore-case |', Completion.value),
|
|
(':set general huh |', None),
|
|
(':help |', Completion.helptopic),
|
|
(':quickmark-load |', Completion.quickmark_by_name),
|
|
(':bookmark-load |', Completion.bookmark_by_url),
|
|
(':open |', Completion.url),
|
|
(':buffer |', Completion.tab),
|
|
(':session-load |', Completion.sessions),
|
|
(':bind |', Completion.empty),
|
|
(':bind <c-x> |', Completion.command),
|
|
(':bind <c-x> foo|', Completion.command),
|
|
(':bind <c-x>| foo', Completion.empty),
|
|
(':set| general ', Completion.command),
|
|
(':|set general ', Completion.command),
|
|
(':set gene|ral ignore-case', Completion.section),
|
|
(':|', Completion.command),
|
|
(': |', Completion.command),
|
|
(':bookmark-load |', Completion.bookmark_by_url),
|
|
])
|
|
def test_update_completion(txt, expected, cmd, completer, instances,
|
|
cmdutils_patch, completion_widget_stub):
|
|
"""Test setting the completion widget's model based on command text."""
|
|
# this test uses | as a placeholder for the current cursor position
|
|
cursor_pos = txt.index('|')
|
|
cmd.setText(txt.replace('|', ''))
|
|
cmd.setCursorPosition(cursor_pos)
|
|
completer.update_completion()
|
|
if expected is None:
|
|
assert not completion_widget_stub.set_model.called
|
|
else:
|
|
assert completion_widget_stub.set_model.call_count == 1
|
|
arg = completion_widget_stub.set_model.call_args[0][0]
|
|
# the outer model is just for sorting; srcmodel is the completion model
|
|
assert arg.srcmodel.kind == expected
|