mirror of https://github.com/pypa/pip
141 lines
4.9 KiB
Python
141 lines
4.9 KiB
Python
import pytest
|
|
|
|
from pip._internal.models.link import Link
|
|
from pip._internal.utils.hashes import Hashes
|
|
|
|
|
|
class TestLink:
|
|
|
|
@pytest.mark.parametrize('url, expected', [
|
|
(
|
|
'https://user:password@example.com/path/page.html',
|
|
'<Link https://user:****@example.com/path/page.html>',
|
|
),
|
|
])
|
|
def test_repr(self, url, expected):
|
|
link = Link(url)
|
|
assert repr(link) == expected
|
|
|
|
@pytest.mark.parametrize('url, expected', [
|
|
('http://yo/wheel.whl', 'wheel.whl'),
|
|
('http://yo/wheel', 'wheel'),
|
|
('https://example.com/path/page.html', 'page.html'),
|
|
# Test a quoted character.
|
|
('https://example.com/path/page%231.html', 'page#1.html'),
|
|
(
|
|
'http://yo/myproject-1.0%2Bfoobar.0-py2.py3-none-any.whl',
|
|
'myproject-1.0+foobar.0-py2.py3-none-any.whl',
|
|
),
|
|
# Test a path that ends in a slash.
|
|
('https://example.com/path/', 'path'),
|
|
('https://example.com/path//', 'path'),
|
|
# Test a url with no filename.
|
|
('https://example.com/', 'example.com'),
|
|
# Test a url with no filename and with auth information.
|
|
(
|
|
'https://user:password@example.com/',
|
|
'example.com',
|
|
),
|
|
])
|
|
def test_filename(self, url, expected):
|
|
link = Link(url)
|
|
assert link.filename == expected
|
|
|
|
def test_splitext(self):
|
|
assert ('wheel', '.whl') == Link('http://yo/wheel.whl').splitext()
|
|
|
|
def test_no_ext(self):
|
|
assert '' == Link('http://yo/wheel').ext
|
|
|
|
def test_ext(self):
|
|
assert '.whl' == Link('http://yo/wheel.whl').ext
|
|
|
|
def test_ext_fragment(self):
|
|
assert '.whl' == Link('http://yo/wheel.whl#frag').ext
|
|
|
|
def test_ext_query(self):
|
|
assert '.whl' == Link('http://yo/wheel.whl?a=b').ext
|
|
|
|
def test_is_wheel(self):
|
|
assert Link('http://yo/wheel.whl').is_wheel
|
|
|
|
def test_is_wheel_false(self):
|
|
assert not Link('http://yo/not_a_wheel').is_wheel
|
|
|
|
def test_fragments(self):
|
|
url = 'git+https://example.com/package#egg=eggname'
|
|
assert 'eggname' == Link(url).egg_fragment
|
|
assert None is Link(url).subdirectory_fragment
|
|
url = 'git+https://example.com/package#egg=eggname&subdirectory=subdir'
|
|
assert 'eggname' == Link(url).egg_fragment
|
|
assert 'subdir' == Link(url).subdirectory_fragment
|
|
url = 'git+https://example.com/package#subdirectory=subdir&egg=eggname'
|
|
assert 'eggname' == Link(url).egg_fragment
|
|
assert 'subdir' == Link(url).subdirectory_fragment
|
|
|
|
@pytest.mark.parametrize('yanked_reason, expected', [
|
|
(None, False),
|
|
('', True),
|
|
('there was a mistake', True),
|
|
])
|
|
def test_is_yanked(self, yanked_reason, expected):
|
|
link = Link(
|
|
'https://example.com/wheel.whl',
|
|
yanked_reason=yanked_reason,
|
|
)
|
|
assert link.is_yanked == expected
|
|
|
|
@pytest.mark.parametrize('hash_name, hex_digest, expected', [
|
|
# Test a value that matches but with the wrong hash_name.
|
|
('sha384', 128 * 'a', False),
|
|
# Test matching values, including values other than the first.
|
|
('sha512', 128 * 'a', True),
|
|
('sha512', 128 * 'b', True),
|
|
# Test a matching hash_name with a value that doesn't match.
|
|
('sha512', 128 * 'c', False),
|
|
# Test a link without a hash value.
|
|
('sha512', '', False),
|
|
])
|
|
def test_is_hash_allowed(self, hash_name, hex_digest, expected):
|
|
url = (
|
|
'https://example.com/wheel.whl#{hash_name}={hex_digest}'.format(
|
|
hash_name=hash_name,
|
|
hex_digest=hex_digest,
|
|
)
|
|
)
|
|
link = Link(url)
|
|
hashes_data = {
|
|
'sha512': [128 * 'a', 128 * 'b'],
|
|
}
|
|
hashes = Hashes(hashes_data)
|
|
assert link.is_hash_allowed(hashes) == expected
|
|
|
|
def test_is_hash_allowed__no_hash(self):
|
|
link = Link('https://example.com/wheel.whl')
|
|
hashes_data = {
|
|
'sha512': [128 * 'a'],
|
|
}
|
|
hashes = Hashes(hashes_data)
|
|
assert not link.is_hash_allowed(hashes)
|
|
|
|
@pytest.mark.parametrize('hashes, expected', [
|
|
(None, False),
|
|
# Also test a success case to show the test is correct.
|
|
(Hashes({'sha512': [128 * 'a']}), True),
|
|
])
|
|
def test_is_hash_allowed__none_hashes(self, hashes, expected):
|
|
url = 'https://example.com/wheel.whl#sha512={}'.format(128 * 'a')
|
|
link = Link(url)
|
|
assert link.is_hash_allowed(hashes) == expected
|
|
|
|
@pytest.mark.parametrize('url, expected', [
|
|
('git+https://github.com/org/repo', True),
|
|
('bzr+http://bzr.myproject.org/MyProject/trunk/#egg=MyProject', True),
|
|
('hg+file://hg.company.com/repo', True),
|
|
('https://example.com/some.whl', False),
|
|
('file://home/foo/some.whl', False),
|
|
])
|
|
def test_is_vcs(self, url, expected):
|
|
link = Link(url)
|
|
assert link.is_vcs is expected
|