diff options
Diffstat (limited to 'tests/helpers/messagemock.py')
| -rw-r--r-- | tests/helpers/messagemock.py | 24 |
1 files changed, 7 insertions, 17 deletions
diff --git a/tests/helpers/messagemock.py b/tests/helpers/messagemock.py index 41f0bac64..6f020618b 100644 --- a/tests/helpers/messagemock.py +++ b/tests/helpers/messagemock.py @@ -20,7 +20,6 @@ """pytest helper to monkeypatch the message module.""" import logging -import dataclasses import pytest from PyQt5.QtCore import pyqtSlot, pyqtSignal, QObject @@ -28,15 +27,6 @@ from PyQt5.QtCore import pyqtSlot, pyqtSignal, QObject from qutebrowser.utils import usertypes, message -@dataclasses.dataclass -class Message: - - """Information about a shown message.""" - - level: usertypes.MessageLevel - text: str - - class MessageMock(QObject): """Helper object for message_mock. @@ -47,7 +37,7 @@ class MessageMock(QObject): _logger: The logger to use for messages/questions. """ - got_message = pyqtSignal(usertypes.MessageLevel, str) + got_message = pyqtSignal(message.MessageInfo) got_question = pyqtSignal(usertypes.Question) def __init__(self, parent=None): @@ -56,18 +46,18 @@ class MessageMock(QObject): self.questions = [] self._logger = logging.getLogger('messagemock') - @pyqtSlot(usertypes.MessageLevel, str) - def _record_message(self, level, text): - self.got_message.emit(level, text) + @pyqtSlot(message.MessageInfo) + def _record_message(self, info): + self.got_message.emit(info) log_levels = { usertypes.MessageLevel.error: logging.ERROR, usertypes.MessageLevel.info: logging.INFO, usertypes.MessageLevel.warning: logging.WARNING, } - log_level = log_levels[level] + log_level = log_levels[info.level] - self._logger.log(log_level, text) - self.messages.append(Message(level, text)) + self._logger.log(log_level, info.text) + self.messages.append(info) @pyqtSlot(usertypes.Question) def _record_question(self, question): |