Differential D2400 Diff 8495 swh/loader/git/tests/test_tasks.py

Changeset View

Standalone View

swh/loader/git/tests/test_tasks.py

	# Copyright (C) 2018 The Software Heritage developers			# Copyright (C) 2018-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 datetime
	from unittest.mock import patch

				anlambertUnsubmitted Not Done Inline Actions You should use the `mocker` fixture from `pytest-mock` instead anlambert: You should use the `mocker` fixture from `pytest-mock` instead
				def test_git_loader(mocker, swh_app, celery_session_worker):
	@patch('swh.loader.git.loader.GitLoader.load')			mock_loader = mocker.patch('swh.loader.git.loader.GitLoader.load')
	def test_git_loader(mock_loader, swh_app, celery_session_worker):
	mock_loader.return_value = {'status': 'eventful'}			mock_loader.return_value = {'status': 'eventful'}

	res = swh_app.send_task(			res = swh_app.send_task(
	'swh.loader.git.tasks.UpdateGitRepository',			'swh.loader.git.tasks.UpdateGitRepository',
	('origin_url',))			('origin_url',))
	assert res			assert res
	res.wait()			res.wait()
	assert res.successful()			assert res.successful()

	assert res.result == {'status': 'eventful'}			assert res.result == {'status': 'eventful'}
	mock_loader.assert_called_once_with('origin_url', base_url=None)			mock_loader.assert_called_once_with()


	@patch('swh.loader.git.from_disk.GitLoaderFromDisk.load')			def test_git_loader_from_disk(mocker, swh_app, celery_session_worker):
	def test_git_loader_from_disk(mock_loader, swh_app, celery_session_worker):			mock_loader = mocker.patch(
				'swh.loader.git.from_disk.GitLoaderFromDisk.load')
	mock_loader.return_value = {'status': 'uneventful'}			mock_loader.return_value = {'status': 'uneventful'}

	res = swh_app.send_task(			res = swh_app.send_task(
	'swh.loader.git.tasks.LoadDiskGitRepository',			'swh.loader.git.tasks.LoadDiskGitRepository',
	('origin_url2', '/some/repo', '2018-12-10 00:00'))			('origin_url2', '/some/repo', '2018-12-10 00:00'))
	assert res			assert res
	res.wait()			res.wait()
	assert res.successful()			assert res.successful()

	assert res.result == {'status': 'uneventful'}			assert res.result == {'status': 'uneventful'}
	mock_loader.assert_called_once_with(			mock_loader.assert_called_once_with()
	'origin_url2', '/some/repo', datetime.datetime(2018, 12, 10, 0, 0))

				def test_git_loader_from_archive(mocker, swh_app, celery_session_worker):
				mock_loader = mocker.patch(
				'swh.loader.git.from_disk.GitLoaderFromArchive.load')

	@patch('swh.loader.git.from_disk.GitLoaderFromArchive.load')
	def test_git_loader_from_archive(mock_loader, swh_app, celery_session_worker):
	mock_loader.return_value = {'status': 'failed'}			mock_loader.return_value = {'status': 'failed'}

	res = swh_app.send_task(			res = swh_app.send_task(
	'swh.loader.git.tasks.UncompressAndLoadDiskGitRepository',			'swh.loader.git.tasks.UncompressAndLoadDiskGitRepository',
	('origin_url3', '/some/repo', '2017-01-10 00:00'))			('origin_url3', '/some/repo', '2017-01-10 00:00'))
	assert res			assert res
	res.wait()			res.wait()
	assert res.successful()			assert res.successful()

	assert res.result == {'status': 'failed'}			assert res.result == {'status': 'failed'}
	mock_loader.assert_called_once_with(			mock_loader.assert_called_once_with()
	'origin_url3', '/some/repo', datetime.datetime(2017, 1, 10, 0, 0))