diff options
Diffstat (limited to 'tests/test_hashes.py')
-rw-r--r-- | tests/test_hashes.py | 194 |
1 files changed, 194 insertions, 0 deletions
diff --git a/tests/test_hashes.py b/tests/test_hashes.py new file mode 100644 index 000000000..79c4f1649 --- /dev/null +++ b/tests/test_hashes.py @@ -0,0 +1,194 @@ +import os + +from nose.tools import assert_raises + +from pip.download import _get_hash_from_file, _check_hash +from pip.exceptions import InstallationError +from pip.index import Link + + +def test_get_hash_from_file_md5(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#md5=d41d8cd98f00b204e9800998ecf8427e") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash.digest_size == 16 + assert download_hash.hexdigest() == "d41d8cd98f00b204e9800998ecf8427e" + + +def test_get_hash_from_file_sha1(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha1=da39a3ee5e6b4b0d3255bfef95601890afd80709") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash.digest_size == 20 + assert download_hash.hexdigest() == "da39a3ee5e6b4b0d3255bfef95601890afd80709" + + +def test_get_hash_from_file_sha224(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha224=d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash.digest_size == 28 + assert download_hash.hexdigest() == "d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f" + + +def test_get_hash_from_file_sha384(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha384=38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash.digest_size == 48 + assert download_hash.hexdigest() == "38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b" + + +def test_get_hash_from_file_sha256(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash.digest_size == 32 + assert download_hash.hexdigest() == "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855" + + +def test_get_hash_from_file_sha512(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha512=cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash.digest_size == 64 + assert download_hash.hexdigest() == "cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e" + + +def test_get_hash_from_file_unknown(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#unknown_hash=d41d8cd98f00b204e9800998ecf8427e") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert download_hash is None + + +def test_check_hash_md5_valid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#md5=d41d8cd98f00b204e9800998ecf8427e") + + download_hash = _get_hash_from_file(file_path, file_link) + + _check_hash(download_hash, file_link) + + +def test_check_hash_md5_invalid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#md5=deadbeef") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, file_link) + + +def test_check_hash_sha1_valid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha1=da39a3ee5e6b4b0d3255bfef95601890afd80709") + + download_hash = _get_hash_from_file(file_path, file_link) + + _check_hash(download_hash, file_link) + + +def test_check_hash_sha1_invalid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha1=deadbeef") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, file_link) + + +def test_check_hash_sha224_valid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha224=d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f'") + + download_hash = _get_hash_from_file(file_path, file_link) + + _check_hash(download_hash, file_link) + + +def test_check_hash_sha224_invalid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha224=deadbeef") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, file_link) + + +def test_check_hash_sha384_valid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha384=38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b") + + download_hash = _get_hash_from_file(file_path, file_link) + + _check_hash(download_hash, file_link) + + +def test_check_hash_sha384_invalid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha384=deadbeef") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, file_link) + + +def test_check_hash_sha256_valid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855") + + download_hash = _get_hash_from_file(file_path, file_link) + + _check_hash(download_hash, file_link) + + +def test_check_hash_sha256_invalid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha256=deadbeef") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, file_link) + + +def test_check_hash_sha512_valid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha512=cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e") + + download_hash = _get_hash_from_file(file_path, file_link) + + _check_hash(download_hash, file_link) + + +def test_check_hash_sha512_invalid(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#sha512=deadbeef") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, file_link) + + +def test_check_hasher_mismsatch(): + file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "packages", "gmpy-1.15.tar.gz") + file_link = Link("http://testserver/gmpy-1.15.tar.gz#md5=d41d8cd98f00b204e9800998ecf8427e") + other_link = Link("http://testserver/gmpy-1.15.tar.gz#sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855") + + download_hash = _get_hash_from_file(file_path, file_link) + + assert_raises(InstallationError, _check_hash, download_hash, other_link) |