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
|
import pytest
from mitmproxy.net import server_spec
def test_parse():
assert server_spec.parse("example.com") == ("https", ("example.com", 443))
assert server_spec.parse("example.com") == ("https", ("example.com", 443))
assert server_spec.parse("http://example.com") == ("http", ("example.com", 80))
assert server_spec.parse("http://127.0.0.1") == ("http", ("127.0.0.1", 80))
assert server_spec.parse("http://[::1]") == ("http", ("::1", 80))
assert server_spec.parse("http://[::1]/") == ("http", ("::1", 80))
assert server_spec.parse("https://[::1]/") == ("https", ("::1", 443))
assert server_spec.parse("http://[::1]:8080") == ("http", ("::1", 8080))
with pytest.raises(ValueError, match="Invalid server specification"):
server_spec.parse(":")
with pytest.raises(ValueError, match="Invalid server scheme"):
server_spec.parse("ftp://example.com")
with pytest.raises(ValueError, match="Invalid hostname"):
server_spec.parse("$$$")
with pytest.raises(ValueError, match="Invalid port"):
server_spec.parse("example.com:999999")
def test_parse_with_mode():
assert server_spec.parse_with_mode("m:example.com") == ("m", ("https", ("example.com", 443)))
with pytest.raises(ValueError):
server_spec.parse_with_mode("moo")
|