diff options
Diffstat (limited to 'tests/helpers/stubs.py')
-rw-r--r-- | tests/helpers/stubs.py | 88 |
1 files changed, 42 insertions, 46 deletions
diff --git a/tests/helpers/stubs.py b/tests/helpers/stubs.py index 493bfd55e..f8770a34e 100644 --- a/tests/helpers/stubs.py +++ b/tests/helpers/stubs.py @@ -29,19 +29,15 @@ import dataclasses import builtins import importlib import types - -from qutebrowser.qt.core import pyqtSignal, QPoint, QProcess, QObject, QUrl, QByteArray -from qutebrowser.qt.gui import QIcon -from qutebrowser.qt.network import (QNetworkRequest, QAbstractNetworkCache, - QNetworkCacheMetaData) -from qutebrowser.qt.widgets import QCommonStyle, QLineEdit, QWidget, QTabBar +from qutebrowser.qt import widgets, network, gui from qutebrowser.browser import browsertab, downloads from qutebrowser.utils import usertypes from qutebrowser.commands import runners +from qutebrowser.qt import core -class FakeNetworkCache(QAbstractNetworkCache): +class FakeNetworkCache(network.QAbstractNetworkCache): """Fake cache with no data.""" @@ -55,7 +51,7 @@ class FakeNetworkCache(QAbstractNetworkCache): pass def metaData(self, _url): - return QNetworkCacheMetaData() + return network.QNetworkCacheMetaData() def prepare(self, _metadata): return None @@ -94,7 +90,7 @@ class FakeWebFrame: parent: The parent frame. """ if scroll is None: - scroll = QPoint(0, 0) + scroll = core.QPoint(0, 0) self.geometry = mock.Mock(return_value=geometry) self.scrollPosition = mock.Mock(return_value=scroll) self.parentFrame = mock.Mock(return_value=parent) @@ -121,7 +117,7 @@ class FakeQApplication: def __init__(self, *, style=None, all_widgets=None, active_window=None, arguments=None, platform_name=None): - self.style = mock.Mock(spec=QCommonStyle) + self.style = mock.Mock(spec=widgets.QCommonStyle) self.style().metaObject().className.return_value = style self.allWidgets = lambda: all_widgets @@ -135,12 +131,12 @@ class FakeNetworkReply: """QNetworkReply stub which provides a Content-Disposition header.""" KNOWN_HEADERS = { - QNetworkRequest.KnownHeaders.ContentTypeHeader: 'Content-Type', + network.QNetworkRequest.KnownHeaders.ContentTypeHeader: 'Content-Type', } def __init__(self, headers=None, url=None): if url is None: - url = QUrl() + url = core.QUrl() if headers is None: self.headers = {} else: @@ -193,18 +189,18 @@ class FakeNetworkReply: self.headers[key] = value -class FakeProcess(QProcess): +class FakeProcess(core.QProcess): - def __init__(self, parent: QObject = None) -> None: + def __init__(self, parent: core.QObject = None) -> None: super().__init__(parent) - self.start = mock.Mock(spec=QProcess.start) - self.startDetached = mock.Mock(spec=QProcess.startDetached) + self.start = mock.Mock(spec=core.QProcess.start) + self.startDetached = mock.Mock(spec=core.QProcess.startDetached) self.readAllStandardOutput = mock.Mock( - spec=QProcess.readAllStandardOutput, return_value=QByteArray(b'')) + spec=core.QProcess.readAllStandardOutput, return_value=core.QByteArray(b'')) self.readAllStandardError = mock.Mock( - spec=QProcess.readAllStandardError, return_value=QByteArray(b'')) - self.terminate = mock.Mock(spec=QProcess.terminate) - self.kill = mock.Mock(spec=QProcess.kill) + spec=core.QProcess.readAllStandardError, return_value=core.QByteArray(b'')) + self.terminate = mock.Mock(spec=core.QProcess.terminate) + self.kill = mock.Mock(spec=core.QProcess.kill) class FakeWebTabScroller(browsertab.AbstractScroller): @@ -256,7 +252,7 @@ class FakeWebTab(browsertab.AbstractTab): """Fake AbstractTab to use in tests.""" - def __init__(self, url=QUrl(), title='', tab_id=0, *, + def __init__(self, url=core.QUrl(), title='', tab_id=0, *, scroll_pos_perc=(0, 0), load_status=usertypes.LoadStatus.success, progress=0, can_go_back=None, can_go_forward=None): @@ -270,7 +266,7 @@ class FakeWebTab(browsertab.AbstractTab): self.scroller = FakeWebTabScroller(self, scroll_pos_perc) self.audio = FakeWebTabAudio(self) self.private_api = FakeWebTabPrivate(tab=self, mode_manager=None) - wrapped = QWidget() + wrapped = widgets.QWidget() self._layout.wrap(self, wrapped) def url(self, *, requested=False): @@ -287,7 +283,7 @@ class FakeWebTab(browsertab.AbstractTab): return self._load_status def icon(self): - return QIcon() + return gui.QIcon() def renderer_process_pid(self): return None @@ -345,11 +341,11 @@ class FakeCommand: modes: Tuple[usertypes.KeyMode] = (usertypes.KeyMode.normal, ) -class FakeTimer(QObject): +class FakeTimer(core.QObject): """Stub for a usertypes.Timer.""" - timeout_signal = pyqtSignal() + timeout_signal = core.pyqtSignal() def __init__(self, parent=None, name=None): super().__init__(parent) @@ -395,14 +391,14 @@ class FakeTimer(QObject): return self._started -class InstaTimer(QObject): +class InstaTimer(core.QObject): """Stub for a QTimer that fires instantly on start(). Useful to test a time-based event without inserting an artificial delay. """ - timeout = pyqtSignal() + timeout = core.pyqtSignal() def start(self, interval=None): self.timeout.emit() @@ -418,27 +414,27 @@ class InstaTimer(QObject): fun() -class StatusBarCommandStub(QLineEdit): +class StatusBarCommandStub(widgets.QLineEdit): """Stub for the statusbar command prompt.""" - got_cmd = pyqtSignal(str) - clear_completion_selection = pyqtSignal() - hide_completion = pyqtSignal() - update_completion = pyqtSignal() - show_cmd = pyqtSignal() - hide_cmd = pyqtSignal() + got_cmd = core.pyqtSignal(str) + clear_completion_selection = core.pyqtSignal() + hide_completion = core.pyqtSignal() + update_completion = core.pyqtSignal() + show_cmd = core.pyqtSignal() + hide_cmd = core.pyqtSignal() def prefix(self): return self.text()[0] -class UrlMarkManagerStub(QObject): +class UrlMarkManagerStub(core.QObject): """Stub for the quickmark-manager or bookmark-manager object.""" - added = pyqtSignal(str, str) - removed = pyqtSignal(str) + added = core.pyqtSignal(str, str) + removed = core.pyqtSignal(str) def __init__(self, parent=None): super().__init__(parent) @@ -476,7 +472,7 @@ class SessionManagerStub: pass -class TabbedBrowserStub(QObject): +class TabbedBrowserStub(core.QObject): """Stub for the tabbed-browser object.""" @@ -506,16 +502,16 @@ class TabbedBrowserStub(QObject): return self.cur_url -class TabWidgetStub(QObject): +class TabWidgetStub(core.QObject): """Stub for the tab-widget object.""" - new_tab = pyqtSignal(browsertab.AbstractTab, int) + new_tab = core.pyqtSignal(browsertab.AbstractTab, int) def __init__(self, parent=None): super().__init__(parent) self.tabs = [] - self._qtabbar = QTabBar() + self._qtabbar = widgets.QTabBar() self.index_of = None self.current_index = None @@ -551,7 +547,7 @@ class TabWidgetStub(QObject): return self.tabs[idx - 1] -class HTTPPostStub(QObject): +class HTTPPostStub(core.QObject): """A stub class for HTTPClient. @@ -560,8 +556,8 @@ class HTTPPostStub(QObject): data: the last data send by post() """ - success = pyqtSignal(str) - error = pyqtSignal(str) + success = core.pyqtSignal(str) + error = core.pyqtSignal(str) def __init__(self, parent=None): super().__init__(parent) @@ -573,11 +569,11 @@ class HTTPPostStub(QObject): self.data = data -class FakeDownloadItem(QObject): +class FakeDownloadItem(core.QObject): """Mock browser.downloads.DownloadItem.""" - finished = pyqtSignal() + finished = core.pyqtSignal() def __init__(self, fileobj, name, parent=None): super().__init__(parent) |