diff options
Diffstat (limited to 'tests/unit/browser/webengine/test_webengineinterceptor.py')
| -rw-r--r-- | tests/unit/browser/webengine/test_webengineinterceptor.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tests/unit/browser/webengine/test_webengineinterceptor.py b/tests/unit/browser/webengine/test_webengineinterceptor.py index eb71e0c6e..f48d4a715 100644 --- a/tests/unit/browser/webengine/test_webengineinterceptor.py +++ b/tests/unit/browser/webengine/test_webengineinterceptor.py @@ -22,22 +22,26 @@ import pytest -pytest.importorskip('PyQt5.QtWebEngineWidgets') +pytest.importorskip('qutebrowser.qt.webenginecore') -from PyQt5.QtWebEngineCore import QWebEngineUrlRequestInfo +from qutebrowser.qt.webenginecore import QWebEngineUrlRequestInfo from qutebrowser.browser.webengine import interceptor +from qutebrowser.utils import qtutils +from helpers import testutils def test_no_missing_resource_types(): request_interceptor = interceptor.RequestInterceptor() - qb_keys = request_interceptor._resource_types.keys() - qt_keys = {i for i in vars(QWebEngineUrlRequestInfo).values() - if isinstance(i, QWebEngineUrlRequestInfo.ResourceType)} + qb_keys = set(request_interceptor._resource_types.keys()) + qt_keys = set(testutils.enum_members( + QWebEngineUrlRequestInfo, + QWebEngineUrlRequestInfo.ResourceType, + ).values()) assert qt_keys == qb_keys def test_resource_type_values(): request_interceptor = interceptor.RequestInterceptor() for qt_value, qb_item in request_interceptor._resource_types.items(): - assert qt_value == qb_item.value + assert qtutils.extract_enum_val(qt_value) == qb_item.value |