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
|
# This file is part of Hypothesis, which may be found at
# https://github.com/HypothesisWorks/hypothesis/
#
# Copyright the Hypothesis Authors.
# Individual contributors are listed in AUTHORS.rst and the git log.
#
# This Source Code Form is subject to the terms of the Mozilla Public License,
# v. 2.0. If a copy of the MPL was not distributed with this file, You can
# obtain one at https://mozilla.org/MPL/2.0/.
import re
import string
import pytest
from hypothesis import given, settings
from hypothesis.errors import InvalidArgument
from hypothesis.provisional import (
FRAGMENT_SAFE_CHARACTERS,
_url_fragments_strategy,
domains,
urls,
)
from tests.common.debug import check_can_generate_examples, find_any
@given(urls())
def test_is_URL(url):
allowed_chars = set(string.ascii_letters + string.digits + "$-_.+!*'(),~%/")
url_schemeless = url.split("://", 1)[1]
components = url_schemeless.split("#", 1)
domain_path = components[0]
path = domain_path.split("/", 1)[1] if "/" in domain_path else ""
assert all(c in allowed_chars for c in path)
assert all(
re.match("^[0-9A-Fa-f]{2}", after_perc) for after_perc in path.split("%")[1:]
)
fragment = components[1] if "#" in url_schemeless else ""
fragment_allowed_chars = allowed_chars | {"?"}
assert all(c in fragment_allowed_chars for c in fragment)
assert all(
re.match("^[0-9A-Fa-f]{2}", after_perc)
for after_perc in fragment.split("%")[1:]
)
@pytest.mark.parametrize("max_length", [-1, 0, 3, 4.0, 256])
@pytest.mark.parametrize("max_element_length", [-1, 0, 4.0, 64, 128])
def test_invalid_domain_arguments(max_length, max_element_length):
with pytest.raises(InvalidArgument):
check_can_generate_examples(
domains(max_length=max_length, max_element_length=max_element_length)
)
@pytest.mark.skipif(
settings._current_profile == "crosshair",
reason="takes ~300s each; the `sampled_from(get_top_level_domains())` decision is realized via iterative comparisons https://github.com/pschanely/CrossHair/issues/332",
)
@pytest.mark.parametrize("max_length", [None, 4, 8, 255])
@pytest.mark.parametrize("max_element_length", [None, 1, 2, 4, 8, 63])
def test_valid_domains_arguments(max_length, max_element_length):
check_can_generate_examples(
domains(max_length=max_length, max_element_length=max_element_length)
)
@pytest.mark.parametrize("strategy", [domains(), urls()])
def test_find_any_non_empty(strategy):
find_any(strategy, lambda s: len(s) > 0)
@given(_url_fragments_strategy)
# There's a lambda in the implementation that only gets run if we generate at
# least one percent-escape sequence, so we derandomize to ensure that coverage
# isn't flaky.
@settings(derandomize=True)
def test_url_fragments_contain_legal_chars(fragment):
assert fragment.startswith("#")
# Strip all legal escape sequences. Any remaining % characters were not
# part of a legal escape sequence.
without_escapes = re.sub(r"(?ai)%[0-9a-f][0-9a-f]", "", fragment[1:])
assert set(without_escapes).issubset(FRAGMENT_SAFE_CHARACTERS)
|
