blob: c537172aae057f92e4a2aed06f7ccba0131a5850 (
plain)
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
|
from pip.index import package_to_requirement, HTMLPage, get_mirrors, DEFAULT_MIRROR_HOSTNAME
from string import ascii_lowercase
from mock import patch
def test_package_name_should_be_converted_to_requirement():
"""
Test that it translates a name like Foo-1.2 to Foo==1.3
"""
assert package_to_requirement('Foo-1.2') == 'Foo==1.2'
assert package_to_requirement('Foo-dev') == 'Foo==dev'
assert package_to_requirement('Foo') == 'Foo'
def test_html_page_should_be_able_to_scrap_rel_links():
"""
Test scraping page looking for url in href
"""
page = HTMLPage("""
<!-- The <th> elements below are a terrible terrible hack for setuptools -->
<li>
<strong>Home Page:</strong>
<!-- <th>Home Page -->
<a href="http://supervisord.org/">http://supervisord.org/</a>
</li>""", "supervisor")
links = list(page.scraped_rel_links())
assert len(links) == 1
assert links[0].url == 'http://supervisord.org/'
@patch('socket.gethostbyname_ex')
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
@patch('socket.gethostbyname_ex')
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
|
