	# 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


	from typing import Any, Dict, Mapping			from typing import Any, Dict, Mapping

				import pkg_resources


	try:			try:
	from swh.loader.core._version import __version__ # type: ignore			__version__ = pkg_resources.get_distribution('swh.loader.core').version
	except ImportError:			except ImportError:
	__version__ = 'devel'			__version__ = 'devel'


	DEFAULT_PARAMS: Dict[str, Any] = {			DEFAULT_PARAMS: Dict[str, Any] = {
	'headers': {			'headers': {
	'User-Agent': 'Software Heritage Loader (%s)' % (			'User-Agent': 'Software Heritage Loader (%s)' % (
	__version__			__version__
	Show All 9 Lines

View Options

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

Use pkg_resources to get the package version instead of vcversionerClosedPublicActions

Details

Diff Detail

Event Timeline

Revision ContentsChangeset List

Diff 8044

swh/loader/package/__init__.py

swh/loader/package/tests/test_utils.py

Use pkg_resources to get the package version instead of vcversioner
ClosedPublic
Actions

Revision Contents
Changeset List

swh/loader/package/init.py