1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
# SPDX-FileCopyrightText: Florian Bruhin (The Compiler) <mail@qutebrowser.org>
#
# SPDX-License-Identifier: GPL-3.0-or-later
import re
import dataclasses
import mimetypes
import pytest
from unittest import mock
webview = pytest.importorskip('qutebrowser.browser.webengine.webview')
from qutebrowser.qt.webenginecore import QWebEnginePage
from qutebrowser.utils import qtutils
from helpers import testutils
@dataclasses.dataclass
class Naming:
prefix: str = ""
suffix: str = ""
def camel_to_snake(naming, name):
if naming.prefix:
assert name.startswith(naming.prefix)
name = name[len(naming.prefix):]
if naming.suffix:
assert name.endswith(naming.suffix)
name = name[:-len(naming.suffix)]
# https://stackoverflow.com/a/1176023
return re.sub(r'(?<!^)(?=[A-Z])', '_', name).lower()
@pytest.mark.parametrize("naming, name, expected", [
(Naming(prefix="NavigationType"), "NavigationTypeLinkClicked", "link_clicked"),
(Naming(prefix="NavigationType"), "NavigationTypeTyped", "typed"),
(Naming(prefix="NavigationType"), "NavigationTypeBackForward", "back_forward"),
(Naming(suffix="MessageLevel"), "InfoMessageLevel", "info"),
])
def test_camel_to_snake(naming, name, expected):
assert camel_to_snake(naming, name) == expected
@pytest.mark.parametrize("enum_type, naming, mapping", [
(
QWebEnginePage.JavaScriptConsoleMessageLevel,
Naming(suffix="MessageLevel"),
webview.WebEnginePage._JS_LOG_LEVEL_MAPPING,
),
(
QWebEnginePage.NavigationType,
Naming(prefix="NavigationType"),
webview.WebEnginePage._NAVIGATION_TYPE_MAPPING,
)
])
def test_enum_mappings(enum_type, naming, mapping):
members = testutils.enum_members(QWebEnginePage, enum_type).items()
for name, val in members:
mapped = mapping[val]
assert camel_to_snake(naming, name) == mapped.name
@pytest.fixture
def suffix_mocks(monkeypatch):
def guess(mime):
mimetypes_map = {
"image/jpeg": [".jpg", ".jpe"],
"video/mp4": [".m4v", ".mpg4"],
}
return mimetypes_map.get(mime, [])
monkeypatch.setattr(mimetypes, "guess_all_extensions", guess)
def version(string):
if string == "6.2.3":
return True
if string == "6.7.0":
return False
raise AssertionError(f"unexpected version {string}")
monkeypatch.setattr(qtutils, "version_check", version)
EXTRA_SUFFIXES_PARAMS = [
(["image/jpeg"], {".jpg", ".jpe"}),
(["image/jpeg", ".jpeg"], {".jpg", ".jpe"}),
(["image/jpeg", ".jpg", ".jpe"], set()),
(
[
".jpg",
],
set(),
), # not sure why black reformats this one and not the others
(["image/jpeg", "video/mp4"], {".jpg", ".jpe", ".m4v", ".mpg4"}),
]
@pytest.mark.parametrize("before, extra", EXTRA_SUFFIXES_PARAMS)
def test_suffixes_workaround_extras_returned(suffix_mocks, before, extra):
assert extra == webview.WebEnginePage.extra_suffixes_workaround(before)
@pytest.mark.parametrize("before, extra", EXTRA_SUFFIXES_PARAMS)
@mock.patch("qutebrowser.browser.webengine.webview.super") # mock super() calls!
def test_suffixes_workaround_choosefiles_args(
mocked_super,
suffix_mocks,
config_stub,
before,
extra,
):
# We can pass the class as "self" because we are only calling a static
# method of it. That saves us having to initilize the class and mock all
# the stuff required for __init__()
webview.WebEnginePage.chooseFiles(
webview.WebEnginePage,
QWebEnginePage.FileSelectionMode.FileSelectOpen,
[],
before,
)
expected = set(before).union(extra)
assert len(mocked_super().chooseFiles.call_args_list) == 1
called_with = mocked_super().chooseFiles.call_args_list[0][0][2]
assert sorted(called_with) == sorted(expected)
|
