def test_create_download_tmp_dir(self): # Test created = web._create_download_tmp_dir(self.working_dir) # Verify self.assertTrue(os.path.exists(created)) self.assertEqual(created, os.path.join(self.working_dir, web.DOWNLOAD_TMP_DIR))
def test_retrieve_module_missing_module(self, mock_curl_constructor): # Setup mock_curl = mock.MagicMock() mock_curl.getinfo.return_value = 404 # simulate a not found mock_curl_constructor.return_value = mock_curl # Test try: self.downloader.retrieve_module(self.mock_progress_report, self.module) self.fail() except exceptions.FileNotFoundException, e: self.assertTrue(self.module.filename() in e.location) expected_filename = web._create_download_tmp_dir(self.working_dir) expected_filename = os.path.join(expected_filename, self.module.filename()) self.assertTrue(not os.path.exists(os.path.join(expected_filename)))
def test_retrieve_module_missing_module(self, mock_curl_constructor): # Setup mock_curl = mock.MagicMock() mock_curl.getinfo.return_value = 404 # simulate a not found mock_curl_constructor.return_value = mock_curl # Test try: self.downloader.retrieve_module(self.mock_progress_report, self.module) self.fail() except exceptions.FileNotFoundException, e: self.assertTrue(self.module.filename() in e.location) expected_filename = web._create_download_tmp_dir(self.working_dir) expected_filename = os.path.join(expected_filename, self.module.filename()) self.assertTrue( not os.path.exists(os.path.join(expected_filename)))
def test_retrieve_module_missing_module(self, mock_downloader_download, mock_listener_constructor): # Setup mock_listener = mock.MagicMock() report = DownloadReport(None, None) report.error_msg = 'oops' mock_listener.failed_reports = [report] mock_listener_constructor.return_value = mock_listener # Test try: self.downloader.retrieve_module(self.mock_progress_report, self.module) self.fail() except exceptions.FileRetrievalException: expected_filename = web._create_download_tmp_dir(self.working_dir) expected_filename = os.path.join(expected_filename, self.module.filename()) self.assertFalse(os.path.exists(os.path.join(expected_filename)))
def test_retrieve_module_missing_module(self, mock_get_working_dir, mock_downloader_download, mock_listener_constructor): # Setup self.module.author = 'asdf' self.module.puppet_standard_filename.return_value = 'puppet-filename.tar.gz' mock_listener = mock.MagicMock() report = DownloadReport(None, None) report.error_msg = 'oops' mock_listener.failed_reports = [report] mock_listener_constructor.return_value = mock_listener # Test try: self.downloader.retrieve_module(self.mock_progress_report, self.module) self.fail() except exceptions.FileRetrievalException: expected_filename = web._create_download_tmp_dir(self.working_dir) expected_filename = os.path.join(expected_filename, self.module.filename())