Differential D1729 Diff 5975 swh/loader/gnu/tests/test_download.py

Changeset View

Standalone View

				# Copyright (C) 2019 The Software Heritage developers
				# See the AUTHORS file at the top-level directory of this distribution
				# License: GNU General Public License version 3, or any later version
				# See top-level LICENSE file for more information

				import unittest
				from swh.loader.gnu.loader import GNULoader
				from swh.loader.base.tests.download_test_base import BaseTestcompareField
				from swh.loader.gnu.tests.common import tarballs

				_LOADER_TESTS_CONFIG = {
				'content_packet_size': 10000,
				'content_packet_size_bytes': 104857600,
				'content_size_limit': 104857600,
				'debug': False,
				'directory_packet_size': 25000,
				'occurrence_packet_size': 100000,
				'release_packet_size': 100000,
				'revision_packet_size': 100000,
				'send_contents': True,
				'send_directories': True,
				'send_releases': True,
				'send_revisions': True,
				'send_snapshot': True,
				'storage': {'args': {}, 'cls': 'memory'},
				'temp_directory': '/tmp/swh.loader.gnu/'
				}


				class GNULoaderTest(GNULoader):
				def parse_config_file(self, args, *kwargs):
				return _LOADER_TESTS_CONFIG


				class TestGNULoader(unittest.TestCase, BaseTestcompareField):

				tarballs = tarballs

				def get_fl(self):
				"""Creates and returns a fake loader instance

				Returns:
				A fake loader instance
				"""
				loader = GNULoaderTest()
				return loader