diff options
-rw-r--r-- | tests/unit/utils/test_urlutils.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/unit/utils/test_urlutils.py b/tests/unit/utils/test_urlutils.py index bbd55e36c..d2eab5928 100644 --- a/tests/unit/utils/test_urlutils.py +++ b/tests/unit/utils/test_urlutils.py @@ -10,7 +10,7 @@ import dataclasses import urllib.parse from qutebrowser.qt.core import QUrl -from qutebrowser.qt.network import QNetworkProxy +from qutebrowser.qt.network import QNetworkProxy, QHostInfo import pytest import hypothesis import hypothesis.strategies @@ -38,7 +38,7 @@ class FakeDNS: @dataclasses.dataclass class FakeDNSAnswer: - error: bool + error: QHostInfo.HostInfoError def __init__(self): self.used = False @@ -53,7 +53,9 @@ class FakeDNS: self.answer = None def _get_error(self): - return not self.answer + if self.answer: + return QHostInfo.HostInfoError.NoError + return QHostInfo.HostInfoError.HostNotFound def fromname_mock(self, _host): """Simple mock for QHostInfo::fromName returning a FakeDNSAnswer.""" |