Differential D2492 Diff 8898 swh/web/tests/api/views/test_vault.py

Changeset View

Standalone View

swh/web/tests/api/views/test_vault.py

	# Copyright (C) 2017-2019 The Software Heritage developers			# Copyright (C) 2017-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 Affero General Public License version 3, or any later version			# License: GNU Affero 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 swh.model import hashutil			from swh.model import hashutil
				from swh.web.common.utils import reverse

	TEST_OBJ_ID = 'd4905454cc154b492bd6afed48694ae3c579345e'			TEST_OBJ_ID = 'd4905454cc154b492bd6afed48694ae3c579345e'

	OBJECT_TYPES = {'directory': ('directory', None),			OBJECT_TYPES = ('directory', 'revision_gitfast')
	'revision_gitfast': ('revision', 'gitfast')}


	def test_api_vault_cook(api_client, mocker):			def test_api_vault_cook(api_client, mocker):
	mock_service = mocker.patch('swh.web.api.views.vault.service')			mock_service = mocker.patch('swh.web.api.views.vault.service')

				for obj_type in OBJECT_TYPES:

				fetch_url = reverse(f'api-1-vault-fetch-{obj_type}',
				url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID})
	stub_cook = {			stub_cook = {
	'fetch_url': ('http://127.0.0.1:5004/api/1/vault/directory/{}/raw/'			'fetch_url': fetch_url,
	.format(TEST_OBJ_ID)),
	'obj_id': TEST_OBJ_ID,			'obj_id': TEST_OBJ_ID,
	'obj_type': 'test_type',			'obj_type': obj_type,
	'progress_message': None,			'progress_message': None,
	'status': 'done',			'status': 'done',
	'task_uuid': 'de75c902-5ee5-4739-996e-448376a93eff',			'task_uuid': 'de75c902-5ee5-4739-996e-448376a93eff',
	}			}
	stub_fetch = b'content'			stub_fetch = b'content'

	mock_service.vault_cook.return_value = stub_cook			mock_service.vault_cook.return_value = stub_cook
	mock_service.vault_fetch.return_value = stub_fetch			mock_service.vault_fetch.return_value = stub_fetch

	for obj_type, (obj_type_name, obj_type_format) in OBJECT_TYPES.items():			url = reverse(f'api-1-vault-cook-{obj_type}',
	url = '/api/1/vault/{}/{}/'.format(obj_type_name, TEST_OBJ_ID)			url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID})
	if obj_type_format:
	url += '{}/'.format(obj_type_format)
	rv = api_client.post(url, {'email': 'test@test.mail'})			rv = api_client.post(url, {'email': 'test@test.mail'})

	assert rv.status_code == 200, rv.data			assert rv.status_code == 200, rv.data
	assert rv['Content-Type'] == 'application/json'			assert rv['Content-Type'] == 'application/json'

				stub_cook['fetch_url'] = rv.wsgi_request.build_absolute_uri(
				stub_cook['fetch_url'])

	assert rv.data == stub_cook			assert rv.data == stub_cook
	mock_service.vault_cook.assert_called_with(			mock_service.vault_cook.assert_called_with(
	obj_type,			obj_type,
	hashutil.hash_to_bytes(TEST_OBJ_ID),			hashutil.hash_to_bytes(TEST_OBJ_ID),
	'test@test.mail')			'test@test.mail')

	rv = api_client.get(url + 'raw/')			rv = api_client.get(fetch_url)

	assert rv.status_code == 200			assert rv.status_code == 200
	assert rv['Content-Type'] == 'application/gzip'			assert rv['Content-Type'] == 'application/gzip'
	assert rv.content == stub_fetch			assert rv.content == stub_fetch
	mock_service.vault_fetch.assert_called_with(			mock_service.vault_fetch.assert_called_with(
	obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID))			obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID))


	def test_api_vault_cook_uppercase_hash(api_client, mocker):			def test_api_vault_cook_uppercase_hash(api_client):
	mock_service = mocker.patch('swh.web.api.views.vault.service')
	stub_cook = {
	'fetch_url': ('http://127.0.0.1:5004/api/1/vault/directory/{}/raw/'
	.format(TEST_OBJ_ID.upper())),
	'obj_id': TEST_OBJ_ID.upper(),
	'obj_type': 'test_type',
	'progress_message': None,
	'status': 'done',
	'task_uuid': 'de75c902-5ee5-4739-996e-448376a93eff',
	}
	stub_fetch = b'content'

	mock_service.vault_cook.return_value = stub_cook			for obj_type in OBJECT_TYPES:
	mock_service.vault_fetch.return_value = stub_fetch

	for obj_type, (obj_type_name, obj_type_format) in OBJECT_TYPES.items():			url = reverse(f'api-1-vault-cook-{obj_type}-uppercase-checksum',
	url = '/api/1/vault/{}/{}/'.format(obj_type_name, TEST_OBJ_ID)			url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID.upper()})
	if obj_type_format:
	url += '{}/'.format(obj_type_format)
	rv = api_client.post(url, {'email': 'test@test.mail'})			rv = api_client.post(url, {'email': 'test@test.mail'})

	assert rv.status_code == 200, rv.data			assert rv.status_code == 302
	assert rv['Content-Type'] == 'application/json'

	assert rv.data == stub_cook			redirect_url = reverse(f'api-1-vault-cook-{obj_type}',
	mock_service.vault_cook.assert_called_with(			url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID})
	obj_type,
	hashutil.hash_to_bytes(TEST_OBJ_ID),
	'test@test.mail')

	rv = api_client.get(url + 'raw/')			assert rv['location'] == redirect_url

	assert rv.status_code == 200			fetch_url = reverse(
	assert rv['Content-Type'] == 'application/gzip'			f'api-1-vault-fetch-{obj_type}-uppercase-checksum',
	assert rv.content == stub_fetch			url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID.upper()})
	mock_service.vault_fetch.assert_called_with(
	obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID))			rv = api_client.get(fetch_url)

				assert rv.status_code == 302

				redirect_url = reverse(f'api-1-vault-fetch-{obj_type}',
				url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID})

				assert rv['location'] == redirect_url


	def test_api_vault_cook_notfound(api_client, mocker):			def test_api_vault_cook_notfound(api_client, mocker):
	mock_service = mocker.patch('swh.web.api.views.vault.service')			mock_service = mocker.patch('swh.web.api.views.vault.service')
	mock_service.vault_cook.return_value = None			mock_service.vault_cook.return_value = None
	mock_service.vault_fetch.return_value = None			mock_service.vault_fetch.return_value = None

	for obj_type, (obj_type_name, obj_type_format) in OBJECT_TYPES.items():			for obj_type in OBJECT_TYPES:
	url = '/api/1/vault/{}/{}/'.format(obj_type_name, TEST_OBJ_ID)			url = reverse(f'api-1-vault-cook-{obj_type}',
	if obj_type_format:			url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID})
	url += '{}/'.format(obj_type_format)
	rv = api_client.post(url)			rv = api_client.post(url)

	assert rv.status_code == 404, rv.data			assert rv.status_code == 404, rv.data
	assert rv['Content-Type'] == 'application/json'			assert rv['Content-Type'] == 'application/json'

	assert rv.data['exception'] == 'NotFoundExc'			assert rv.data['exception'] == 'NotFoundExc'
	mock_service.vault_cook.assert_called_with(			mock_service.vault_cook.assert_called_with(
	obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID), None)			obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID), None)

	rv = api_client.get(url + 'raw/')			fetch_url = reverse(f'api-1-vault-fetch-{obj_type}',
				url_args={f'{obj_type[:3]}_id': TEST_OBJ_ID})

				rv = api_client.get(fetch_url)

	assert rv.status_code == 404, rv.data			assert rv.status_code == 404, rv.data
	assert rv['Content-Type'] == 'application/json'			assert rv['Content-Type'] == 'application/json'
	assert rv.data['exception'] == 'NotFoundExc'			assert rv.data['exception'] == 'NotFoundExc'
	mock_service.vault_fetch.assert_called_with(			mock_service.vault_fetch.assert_called_with(
	obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID))			obj_type, hashutil.hash_to_bytes(TEST_OBJ_ID))