def test_patch_contextmanager_with_public_fqdn():
with hosts({"example.org": "1.2.3.4"}):
assert socket.gethostbyname_ex("example.org") == (
"example.org",
[],
["1.2.3.4"],
)
def test_patch_contextmanager_with_unknown_hostname():
with hosts({"UNKNOWN_HOST_AZ": "127.0.0.1"}):
assert socket.gethostbyname_ex("UNKNOWN_HOST_AZ") == (
"UNKNOWN_HOST_AZ",
[],
["127.0.0.1"],
)
def test_patch_contextmanager_with_another_hostname_ipv6():
with hosts({"localhost2": "1.2.3.4"}):
r = socket.getaddrinfo("ip6-localhost",
80,
family=socket.AF_INET6,
proto=socket.IPPROTO_TCP)
for (_, _, _, _, sockaddr) in r:
assert sockaddr == ("::1", 80, 0, 0)
def test_patch_contextmanager_with_ip_ipv6():
with hosts({"::1": "1::23"}):
r = socket.getaddrinfo("::1",
80,
family=socket.AF_INET6,
proto=socket.IPPROTO_TCP)
for (_, _, _, _, sockaddr) in r:
assert sockaddr == ("1::23", 80, 0, 0)
def test_patch_contextmanager_with_public_fqdn_ipv6():
with hosts({"example.org": "::1"}):
r = socket.getaddrinfo("example.org",
80,
family=socket.AF_INET6,
proto=socket.IPPROTO_TCP)
for (_, _, _, _, sockaddr) in r:
assert sockaddr == ("::1", 80, 0, 0)
def test_patch_contextmanager_with_ip_ipv4():
with hosts({"127.0.0.1": "1.2.3.4"}):
r = socket.getaddrinfo("127.0.0.1",
80,
family=socket.AF_INET,
proto=socket.IPPROTO_TCP)
for (_, _, _, _, sockaddr) in r:
assert sockaddr == ("1.2.3.4", 80)
def test_patch_contextmanager_only_gethostbyname_ex():
with hosts({"localhost": "1.2.3.4"}, only=["gethostbyname_ex"]):
assert socket.gethostbyname("localhost") == "127.0.0.1"
assert socket.gethostbyname_ex("localhost") == ("localhost", [],
["1.2.3.4"])
r = socket.getaddrinfo("localhost",
80,
family=socket.AF_INET,
proto=socket.IPPROTO_TCP)
for (_, _, _, _, sockaddr) in r:
assert sockaddr == ("127.0.0.1", 80)
def test_patch_contextmanager_with_unknown_hostname(httpd, httpdport):
r = requests.Session()
# don't work if hostname is in upper case
with hosts({"unknown_host_az": "127.0.0.1"}):
assert (
"from unknown_host_az"
in r.get("http://unknown_host_az:%d/" % httpdport).text
)
assert (
"from unknown_host_az"
in r.get("http://unknown_host_az:%d/" % httpdport).text
)
def test_patch_contextmanager_only_getaddrinfo():
with hosts({"localhost": "1.2.3.4"}, only=["getaddrinfo"]):
assert socket.gethostbyname("localhost") == "127.0.0.1"
(hostname, _, ipaddrlist) = socket.gethostbyname_ex("localhost")
assert hostname == "localhost"
assert ipaddrlist == ["127.0.0.1"]
r = socket.getaddrinfo("localhost",
80,
family=socket.AF_INET,
proto=socket.IPPROTO_TCP)
for (_, _, _, _, sockaddr) in r:
assert sockaddr == ("1.2.3.4", 80)
def test_patch_contextmanager_with_public_fqdn_and_a_name_for_addr():
with hosts({"example.org": "localhost"}):
(hostname, _, ipaddrlist) = socket.gethostbyname_ex("example.org")
assert hostname == "example.org"
assert ipaddrlist == ["127.0.0.1"]
def test_patch_contextmanager_with_ipv4(httpd, httpdport):
r = requests.Session()
with hosts({"example.org": "127.0.0.1"}):
assert "from example.org" in r.get("http://example.org:%d/" % httpdport).text
assert "from example.org" in r.get("http://example.org:%d/" % httpdport).text
def test_patch_contextmanager_with_another_hostname(httpd, httpdport):
r = requests.Session()
with hosts({"anotherhostname": "1.2.3.4"}):
assert "from localhost" in r.get("http://localhost:%d/" % httpdport).text
assert "from localhost" in r.get("http://localhost:%d/" % httpdport).text
def test_patch_contextmanager_with_another_hostname():
with hosts({"localhost2": "1.2.3.4"}):
assert socket.gethostbyname("localhost") == "127.0.0.1"
def test_patch_contextmanager_with_public_fqdn_and_a_name_for_addr():
with hosts({"example.org": "localhost"}):
assert socket.gethostbyname("example.org") == "127.0.0.1"
def test_patch_contextmanager_with_ip():
with hosts({"127.0.0.1": "1.2.3.4"}):
assert socket.gethostbyname("127.0.0.1") == "1.2.3.4"
def test_patch_contextmanager_with_name():
with hosts({"localhost": "1.2.3.4"}):
assert socket.gethostbyname("localhost") == "1.2.3.4"
def test_patch_contextmanager_with_public_fqdn_and_a_name_for_addr():
with hosts({"example.org": "localhost"}):
assert socket.getaddrinfo("example.org",
80) == socket.getaddrinfo("localhost", 80)
def test_patch_contextmanager_with_name(httpd, httpdport):
with hosts({"example.org": "localhost"}):
assert ("from example.org"
in requests.get("http://example.org:%d/" % httpdport).text)
def test_patch_contextmanager_with_public_fqdn_and_a_name_for_addr_ipv6():
with hosts({"example.org": "ip6-localhost"}):
family = socket.AF_INET6
assert socket.getaddrinfo("example.org", 80,
family=family) == socket.getaddrinfo(
"ip6-localhost", 80, family=family)
def test_patch_contextmanager_with_unknown_hostname():
with hosts({"UNKNOWN_HOST_AZ": "127.0.0.1"}):
assert socket.getaddrinfo("UNKNOWN_HOST_AZ",
80) == socket.getaddrinfo("127.0.0.1", 80)
def test_patch_contextmanager_with_name():
with hosts({"localhost": "1.2.3.4"}):
(hostname, _, ipaddrlist) = socket.gethostbyname_ex("localhost")
assert hostname == "localhost"
assert ipaddrlist == ["1.2.3.4"]
