def test_get_mirrors(mock_gethostbyname_ex):
# Test when the expected result comes back
# from socket.gethostbyname_ex
mock_gethostbyname_ex.return_value = ('g.pypi.python.org', [DEFAULT_MIRROR_HOSTNAME], ['129.21.171.98'])
mirrors = get_mirrors()
# Expect [a-g].pypi.python.org, since last mirror
# is returned as g.pypi.python.org
assert len(mirrors) == 7
for c in "abcdefg":
assert c + ".pypi.python.org" in mirrors
def test_get_mirrors_no_cname(mock_gethostbyname_ex):
# Test when the UNexpected result comes back
# from socket.gethostbyname_ex
# (seeing this in Japan and was resulting in 216k
# invalid mirrors and a hot CPU)
mock_gethostbyname_ex.return_value = (DEFAULT_MIRROR_HOSTNAME, [DEFAULT_MIRROR_HOSTNAME], ['129.21.171.98'])
mirrors = get_mirrors()
# Falls back to [a-z].pypi.python.org
assert len(mirrors) == 26
for c in ascii_lowercase:
assert c + ".pypi.python.org" in mirrors
def test_get_mirrors(mock_gethostbyname_ex):
# Test when the expected result comes back
# from socket.gethostbyname_ex
mock_gethostbyname_ex.return_value = ('g.pypi.python.org',
[DEFAULT_MIRROR_HOSTNAME
], ['129.21.171.98'])
mirrors = get_mirrors()
# Expect [a-g].pypi.python.org, since last mirror
# is returned as g.pypi.python.org
assert len(mirrors) == 7
for c in "abcdefg":
assert c + ".pypi.python.org" in mirrors
def test_get_mirrors_no_cname(mock_gethostbyname_ex):
# Test when the UNexpected result comes back
# from socket.gethostbyname_ex
# (seeing this in Japan and was resulting in 216k
# invalid mirrors and a hot CPU)
mock_gethostbyname_ex.return_value = (DEFAULT_MIRROR_HOSTNAME,
[DEFAULT_MIRROR_HOSTNAME
], ['129.21.171.98'])
mirrors = get_mirrors()
# Falls back to [a-z].pypi.python.org
assert len(mirrors) == 26
for c in ascii_lowercase:
assert c + ".pypi.python.org" in mirrors
