Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/package/tests/test_utils.py
| # Copyright (C) 2019 The Software Heritage developers | # Copyright (C) 2019 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| import os | import os | ||||
| import pytest | import pytest | ||||
| import swh.loader.package | |||||
| from swh.loader.package.utils import download, api_info, release_name | from swh.loader.package.utils import download, api_info, release_name | ||||
| def test_version_generation(): | |||||
| assert swh.loader.package.__version__ != 'devel' | |||||
| @pytest.mark.fs | @pytest.mark.fs | ||||
| def test_download_fail_to_download(tmp_path, requests_mock): | def test_download_fail_to_download(tmp_path, requests_mock): | ||||
| url = 'https://pypi.org/pypi/arrow/json' | url = 'https://pypi.org/pypi/arrow/json' | ||||
| status_code = 404 | status_code = 404 | ||||
| requests_mock.get(url, status_code=status_code) | requests_mock.get(url, status_code=status_code) | ||||
| with pytest.raises(ValueError) as e: | with pytest.raises(ValueError) as e: | ||||
| download(url, tmp_path) | download(url, tmp_path) | ||||
| Show All 39 Lines | def test_download_ok(tmp_path, requests_mock): | ||||
| assert actual_filename == filename | assert actual_filename == filename | ||||
| assert actual_hashes['length'] == len(data) | assert actual_hashes['length'] == len(data) | ||||
| assert actual_hashes['checksums']['sha1'] == 'fdd1ce606a904b08c816ba84f3125f2af44d92b2' # noqa | assert actual_hashes['checksums']['sha1'] == 'fdd1ce606a904b08c816ba84f3125f2af44d92b2' # noqa | ||||
| assert (actual_hashes['checksums']['sha256'] == | assert (actual_hashes['checksums']['sha256'] == | ||||
| '1d9224378d77925d612c9f926eb9fb92850e6551def8328011b6a972323298d5') | '1d9224378d77925d612c9f926eb9fb92850e6551def8328011b6a972323298d5') | ||||
| @pytest.mark.fs | @pytest.mark.fs | ||||
| def test_download_headers(tmp_path, requests_mock): | |||||
| """Check that we send proper headers when downloading files""" | |||||
| filename = 'requests-0.0.1.tar.gz' | |||||
| url = 'https://pypi.org/pypi/requests/%s' % filename | |||||
| data = 'this is something' | |||||
| requests_mock.get(url, text=data, headers={ | |||||
| 'content-length': str(len(data)) | |||||
| }) | |||||
| actual_filepath, actual_hashes = download(url, dest=str(tmp_path)) | |||||
| assert len(requests_mock.request_history) == 1 | |||||
| req = requests_mock.request_history[0] | |||||
| assert 'User-Agent' in req.headers | |||||
| user_agent = req.headers['User-Agent'] | |||||
| assert 'Software Heritage Loader' in user_agent | |||||
| assert swh.loader.package.__version__ in user_agent | |||||
| @pytest.mark.fs | |||||
| def test_download_ok_with_hashes(tmp_path, requests_mock): | def test_download_ok_with_hashes(tmp_path, requests_mock): | ||||
| """Download without issue should provide filename and hashes""" | """Download without issue should provide filename and hashes""" | ||||
| filename = 'requests-0.0.1.tar.gz' | filename = 'requests-0.0.1.tar.gz' | ||||
| url = 'https://pypi.org/pypi/requests/%s' % filename | url = 'https://pypi.org/pypi/requests/%s' % filename | ||||
| data = 'this is something' | data = 'this is something' | ||||
| requests_mock.get(url, text=data, headers={ | requests_mock.get(url, text=data, headers={ | ||||
| 'content-length': str(len(data)) | 'content-length': str(len(data)) | ||||
| }) | }) | ||||
| ▲ Show 20 Lines • Show All 78 Lines • Show Last 20 Lines | |||||