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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
from urllib.parse import quote
import pytest
from streamlink.utils.url import absolute_url, prepend_www, update_qsd, update_scheme, url_concat, url_equal
@pytest.mark.parametrize(
("baseurl", "url", "expected"),
[
("http://test.se", "/test", "http://test.se/test"),
("http://test.se", "http/test.se/test", "http://test.se/http/test.se/test"),
("http://test.se", "http://test2.se/test", "http://test2.se/test"),
],
)
def test_absolute_url(baseurl, url, expected):
assert expected == absolute_url(baseurl, url)
@pytest.mark.parametrize(
("url", "expected"),
[
("http://test.se/test", "http://www.test.se/test"),
("http://www.test.se", "http://www.test.se"),
],
)
def test_prepend_www(url, expected):
assert expected == prepend_www(url)
@pytest.mark.parametrize(
("args", "expected"),
[
pytest.param(
("https://other.com/bar", "http://example.com/foo"),
"https://example.com/foo",
id="override-target-http-scheme",
),
pytest.param(
("http://other.com/bar", "https://example.com/foo"),
"http://example.com/foo",
id="override-target-https-scheme",
),
pytest.param(
("http://other.com/bar", "https://example.com/foo", False),
"https://example.com/foo",
id="keep-target-https-scheme-no-force",
),
pytest.param(
("https://other.com/bar", "http://example.com/foo", False),
"http://example.com/foo",
id="keep-target-http-scheme-no-force",
),
pytest.param(
("https://other.com/bar", "//example.com/foo"),
"https://example.com/foo",
id="add-current-scheme-to-schemeless-target",
),
pytest.param(
("https://other.com/bar", "//example.com/foo", False),
"https://example.com/foo",
id="add-current-scheme-to-schemeless-target-no-force",
),
pytest.param(
("https://other.com/bar", "example.com/foo"),
"https://example.com/foo",
id="add-current-scheme-to-target-string",
),
pytest.param(
("https://other.com/bar", "example.com/foo", False),
"https://example.com/foo",
id="add-current-scheme-to-target-string-no-force",
),
pytest.param(
("http://", "127.0.0.1:1234"),
"http://127.0.0.1:1234",
id="implicit-scheme-with-ipv4-port",
),
pytest.param(
("http://", "foo.bar:1234"),
"http://foo.bar:1234",
id="implicit-scheme-with-hostname-port",
),
pytest.param(
("foo.1+2-bar://baz", "FOO.1+2-BAR://qux"),
"foo.1+2-bar://qux",
id="weird-schemes",
),
pytest.param(
("https://", "file://./foo", False),
"file://./foo",
id="keep-target",
),
pytest.param(
("https://", "file:///foo", False),
"file:///foo",
id="keep-target-no-netloc",
),
],
)
def test_update_scheme(args, expected):
assert update_scheme(*args) == expected
def test_url_equal():
assert url_equal("http://test.com/test", "http://test.com/test")
assert not url_equal("http://test.com/test", "http://test.com/test2")
assert url_equal("http://test.com/test", "http://test.com/test2", ignore_path=True)
assert url_equal("http://test.com/test", "https://test.com/test", ignore_scheme=True)
assert not url_equal("http://test.com/test", "https://test.com/test")
assert url_equal("http://test.com/test", "http://test.com/test#hello", ignore_fragment=True)
assert url_equal("http://test.com/test", "http://test2.com/test", ignore_netloc=True)
assert not url_equal("http://test.com/test", "http://test2.com/test1", ignore_netloc=True)
def test_url_concat():
assert url_concat("http://test.se", "one", "two", "three") == "http://test.se/one/two/three"
assert url_concat("http://test.se", "/one", "/two", "/three") == "http://test.se/one/two/three"
assert url_concat("http://test.se/one", "../two", "three") == "http://test.se/two/three"
assert url_concat("http://test.se/one", "../two", "../three") == "http://test.se/three"
def test_update_qsd():
assert update_qsd("http://test.se?one=1&two=3", {"two": 2}) == "http://test.se?one=1&two=2"
assert update_qsd("http://test.se?one=1&two=3", remove=["two"]) == "http://test.se?one=1"
assert update_qsd("http://test.se?one=1&two=3", {"one": None}, remove="*") == "http://test.se?one=1"
assert update_qsd("http://test.se", {"one": "", "two": ""}) == "http://test.se?one=&two=", "should add empty params"
assert update_qsd("http://test.se?one=", {"one": None}) == "http://test.se?one=", "should leave empty params unchanged"
assert update_qsd("http://test.se?one=", keep_blank_values=False) == "http://test.se", "should strip blank params"
assert update_qsd("http://test.se?one=&two=", {"one": None}, keep_blank_values=False) == "http://test.se?one=", \
"should leave one" # fmt: skip
assert update_qsd("http://test.se?&two=", {"one": ""}, keep_blank_values=False) == "http://test.se?one=", \
"should set one blank" # fmt: skip
assert update_qsd("http://test.se?one=", {"two": 2}) == "http://test.se?one=&two=2"
assert update_qsd("http://test.se?foo=%3F", {"bar": "!"}) == "http://test.se?foo=%3F&bar=%21", \
"urlencode - encoded URL" # fmt: skip
assert update_qsd("http://test.se?foo=?", {"bar": "!"}) == "http://test.se?foo=%3F&bar=%21", \
"urlencode - fix URL" # fmt: skip
assert update_qsd("http://test.se?foo=?", {"bar": "!"}, quote_via=lambda s, *_: s) == "http://test.se?foo=?&bar=!", \
"urlencode - dummy quote method" # fmt: skip
assert update_qsd("http://test.se", {"foo": "/ "}) == "http://test.se?foo=%2F+", \
"urlencode - default quote_plus" # fmt: skip
assert update_qsd("http://test.se", {"foo": "/ "}, safe="/", quote_via=quote) == "http://test.se?foo=/%20", \
"urlencode - regular quote with reserved slash" # fmt: skip
assert update_qsd("http://test.se", {"foo": "/ "}, safe="", quote_via=quote) == "http://test.se?foo=%2F%20", \
"urlencode - regular quote without reserved slash" # fmt: skip
|
