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