class LocalDownloaderTests(base_downloader.BaseDownloaderTests):
def setUp(self):
super(LocalDownloaderTests, self).setUp()
self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
self.downloader = LocalDownloader(self.repo, None, self.config, self.mock_cancelled_callback)
def test_retrieve_metadata(self):
# Test
docs = self.downloader.retrieve_metadata(self.mock_progress_report)
# Verify
self.assertEqual(1, len(docs))
metadata = model.RepositoryMetadata()
metadata.update_from_json(docs[0])
self.assertEqual(2, len(metadata.modules))
self.assertEqual(1, self.mock_progress_report.metadata_query_total_count)
self.assertEqual(1, self.mock_progress_report.metadata_query_finished_count)
expected_query = os.path.join(VALID_REPO_DIR, constants.REPO_METADATA_FILENAME)
self.assertEqual(expected_query, self.mock_progress_report.metadata_current_query)
self.assertEqual(2, self.mock_progress_report.update_progress.call_count)
def test_retrieve_metadata_no_metadata_found(self):
# Setup
self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + INVALID_REPO_DIR
# Test
try:
self.downloader.retrieve_metadata(self.mock_progress_report)
self.fail()
except FileNotFoundException, e:
expected = os.path.join(INVALID_REPO_DIR, constants.REPO_METADATA_FILENAME)
self.assertEqual(expected, e.location)
class LocalDownloaderTests(base_downloader.BaseDownloaderTests):
def setUp(self):
super(LocalDownloaderTests, self).setUp()
self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
self.downloader = LocalDownloader(self.repo, None, self.config)
@mock.patch('nectar.config.DownloaderConfig.finalize')
def test_retrieve_metadata(self, mock_finalize):
# Test
docs = self.downloader.retrieve_metadata(self.mock_progress_report)
# Verify
self.assertEqual(1, len(docs))
metadata = model.RepositoryMetadata()
metadata.update_from_json(docs[0])
self.assertEqual(2, len(metadata.modules))
self.assertEqual(1, self.mock_progress_report.metadata_query_total_count)
self.assertEqual(1, self.mock_progress_report.metadata_query_finished_count)
expected_query = os.path.join(VALID_REPO_DIR, constants.REPO_METADATA_FILENAME)
self.assertEqual(expected_query, self.mock_progress_report.metadata_current_query)
self.assertEqual(2, self.mock_progress_report.update_progress.call_count)
mock_finalize.assert_called_once()
def test_retrieve_metadata_no_metadata_found(self):
# Setup
self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + INVALID_REPO_DIR
# Test
try:
self.downloader.retrieve_metadata(self.mock_progress_report)
self.fail()
except FileRetrievalException:
pass
def test_retrieve_module(self):
# Test
mod_path = self.downloader.retrieve_module(self.mock_progress_report, self.module)
# Verify
expected = os.path.join(VALID_REPO_DIR, self.module.filename())
self.assertEqual(expected, mod_path)
def test_retrieve_module_no_file(self):
# Setup
self.module.author = 'foo'
# Test
try:
self.downloader.retrieve_module(self.mock_progress_report, self.module)
self.fail()
except FileRetrievalException:
pass
def test_cleanup_module(self):
# Test
self.downloader.cleanup_module(self.module)
def setUp(self):
super(LocalDownloaderTests, self).setUp()
self.config.repo_plugin_config[
constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
self.downloader = LocalDownloader(self.repo, None, self.config)
class LocalDownloaderTests(base_downloader.BaseDownloaderTests):
def setUp(self):
super(LocalDownloaderTests, self).setUp()
self.config.repo_plugin_config[
constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
self.downloader = LocalDownloader(self.repo, None, self.config)
@mock.patch('nectar.config.DownloaderConfig.finalize')
def test_retrieve_metadata(self, mock_finalize):
# Test
docs = self.downloader.retrieve_metadata(self.mock_progress_report)
# Verify
self.assertEqual(1, len(docs))
metadata = RepositoryMetadata()
metadata.update_from_json(docs[0])
self.assertEqual(2, len(metadata.modules))
self.assertEqual(1,
self.mock_progress_report.metadata_query_total_count)
self.assertEqual(
1, self.mock_progress_report.metadata_query_finished_count)
expected_query = os.path.join(VALID_REPO_DIR,
constants.REPO_METADATA_FILENAME)
self.assertEqual(expected_query,
self.mock_progress_report.metadata_current_query)
self.assertEqual(2,
self.mock_progress_report.update_progress.call_count)
mock_finalize.assert_called_once()
def test_retrieve_metadata_no_metadata_found(self):
# Setup
self.config.repo_plugin_config[
constants.CONFIG_FEED] = 'file://' + INVALID_REPO_DIR
# Test
try:
self.downloader.retrieve_metadata(self.mock_progress_report)
self.fail()
except FileRetrievalException:
pass
@mock.patch(MODULE_PATH + '.os.path.exists')
def test_retrieve_module(self, mock_exists):
mock_exists.return_value = True
# Test
mod_path = self.downloader.retrieve_module(self.mock_progress_report,
self.module)
# Verify
expected = os.path.join(VALID_REPO_DIR, self.module.filename())
self.assertEqual(expected, mod_path)
@mock.patch(MODULE_PATH + '.os.path.exists')
def test_retrieve_module_no_file(self, mock_exists):
mock_exists.return_value = False
# Setup
self.module.author = 'foo'
# Test
try:
self.downloader.retrieve_module(self.mock_progress_report,
self.module)
self.fail()
except FileRetrievalException:
pass
def test_cleanup_module(self):
# Test
self.downloader.cleanup_module(self.module)
def setUp(self):
super(LocalDownloaderTests, self).setUp()
self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
self.downloader = LocalDownloader(self.repo, None, self.config)
