class LocalDownloaderTests(base_downloader.BaseDownloaderTests): def setUp(self): super(LocalDownloaderTests, self).setUp() self.downloader = LocalDownloader(self.repo, None, self.config, self.mock_cancelled_callback) def test_download_resource(self): # Setup indexes = self.dist.get_indexes() # Test self.downloader.download_resources(indexes, self.mock_progress_report) # Verify self.assertEqual(3, len(indexes)) self.assertEqual(3, self.mock_progress_report.query_total_count) self.assertEqual(3, self.mock_progress_report.query_finished_count) self.assertEqual(2, self.mock_progress_report.update_progress.call_count) def test_download_resource_not_found(self): # Setup indexes = self.dist.get_indexes() indexes[1]['url'] = indexes[1]['url'] + '_' # Test & Verify try: self.downloader.download_resources(indexes, self.mock_progress_report) self.fail() except FileNotFoundException, e: self.assertEqual(e.location in [r['url'] for r in indexes], True)
def setUp(self): super(LocalDownloaderTests, self).setUp() self.downloader = LocalDownloader(self.repo, None, self.config, self.mock_cancelled_callback)