qutebrowser/tests/unit/browser/test_shared.py

146 lines
4.3 KiB
Python

# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:
# Copyright 2016-2021 Florian Bruhin (The Compiler) <mail@qutebrowser.org>
#
# This file is part of qutebrowser.
#
# qutebrowser is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# qutebrowser is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with qutebrowser. If not, see <https://www.gnu.org/licenses/>.
import logging
import pytest
from qutebrowser.browser import shared
from qutebrowser.utils import usertypes
@pytest.mark.parametrize('dnt, accept_language, custom_headers, expected', [
# DNT
(True, None, {}, {b'DNT': b'1'}),
(False, None, {}, {b'DNT': b'0'}),
(None, None, {}, {}),
# Accept-Language
(False, 'de, en', {}, {b'DNT': b'0', b'Accept-Language': b'de, en'}),
# Custom headers
(False, None, {'X-Qute': 'yes'}, {b'DNT': b'0', b'X-Qute': b'yes'}),
# Mixed
(False, 'de, en', {'X-Qute': 'yes'}, {b'DNT': b'0',
b'Accept-Language': b'de, en',
b'X-Qute': b'yes'}),
])
def test_custom_headers(config_stub, dnt, accept_language, custom_headers,
expected):
headers = config_stub.val.content.headers
headers.do_not_track = dnt
headers.accept_language = accept_language
headers.custom = custom_headers
expected_items = sorted(expected.items())
assert shared.custom_headers(url=None) == expected_items
@pytest.mark.parametrize(
(
"levels_setting, excludes_setting, level, source, msg, expected_ret, "
"expected_level"
), [
# Empty settings
(
{},
{},
usertypes.JsLogLevel.error,
"qute:test",
"msg",
False,
None,
),
# Simple error message
(
{"qute:*": ["error"]},
{},
usertypes.JsLogLevel.error,
"qute:bla",
"msg",
True,
usertypes.MessageLevel.error,
),
# Unfiltered error message
(
{"qute:*": ["error"]},
{"qute:*": ["filter*"]},
usertypes.JsLogLevel.error,
"qute:bla",
"notfiltered",
True,
usertypes.MessageLevel.error,
),
# Filtered error message
(
{"qute:*": ["error"]},
{"qute:*": ["filter*"]},
usertypes.JsLogLevel.error,
"qute:bla",
"filtered",
False,
None,
),
# Filter with different domain
(
{"qute:*": ["error"]},
{"qutie:*": ["*"]},
usertypes.JsLogLevel.error,
"qute:bla",
"msg",
True,
usertypes.MessageLevel.error,
),
# Info message, not logged
(
{"qute:*": ["error"]},
{},
usertypes.JsLogLevel.info,
"qute:bla",
"msg",
False,
None,
),
# Info message, logged
(
{"qute:*": ["error", "info"]},
{},
usertypes.JsLogLevel.info,
"qute:bla",
"msg",
True,
usertypes.MessageLevel.info,
),
]
)
def test_js_log_to_ui(
config_stub, message_mock, caplog,
levels_setting, excludes_setting, level, source, msg, expected_ret, expected_level,
):
config_stub.val.content.javascript.log_message.levels = levels_setting
config_stub.val.content.javascript.log_message.excludes = excludes_setting
with caplog.at_level(logging.ERROR):
ret = shared._js_log_to_ui(level=level, source=source, line=0, msg=msg)
assert ret == expected_ret
if expected_level is not None:
assert message_mock.getmsg(expected_level).text == f"JS: [{source}:0] {msg}"
else:
assert not message_mock.messages