def TestUploadUnitFails(self, mock_handle_upload): module_importer = PuppetModuleImporter() mock_handle_upload.side_effect = Exception('bad') report = module_importer.upload_unit(Mock(), Mock(), Mock(), Mock(), Mock(), Mock(), Mock()) self.assertFalse(report['success_flag']) self.assertEquals(report['summary'], 'bad')
def testUploadUnit(self, mock_handle_upload): module_importer = PuppetModuleImporter() mock_handle_upload.return_value = { 'success_flag': True, 'summary': '', 'details': {} } report = module_importer.upload_unit(Mock(), Mock(), Mock(), Mock(), Mock(), Mock(), Mock()) self.assertTrue(report['success_flag'])
def test_directory_synchronization(self, forge_call, mock_call): conduit = Mock() repository = Mock() config = {constants.CONFIG_FEED: 'http://host/tmp/%s' % constants.MANIFEST_FILENAME} progress_report = SyncProgressReport(conduit) progress_report.metadata_state = constants.STATE_SUCCESS progress_report.modules_state = constants.STATE_SUCCESS mock_call.return_value = progress_report # test plugin = PuppetModuleImporter() report = plugin.sync_repo(repository, conduit, config) # validation mock_call.assert_called_with(repository) self.assertEquals(report, conduit.build_success_report.return_value) self.assertFalse(forge_call.called)
def test_directory_synchronization(self, forge_call, mock_call): conduit = Mock() repository = Mock() config = {constants.CONFIG_FEED: 'http://host/tmp/%s' % constants.MANIFEST_FILENAME} progress_report = SyncProgressReport(conduit) progress_report.metadata_state = constants.STATE_SUCCESS progress_report.modules_state = constants.STATE_SUCCESS mock_call.return_value = progress_report # test plugin = PuppetModuleImporter() report = plugin.sync_repo(repository, conduit, config) # validation mock_call.assert_called_with() self.assertEquals(report, conduit.build_success_report.return_value) self.assertFalse(forge_call.called)
def test_forge_synchronization(self, failed_call, mock_call): conduit = Mock() repository = Mock() config = {constants.CONFIG_FEED: 'http://host/tmp/forge'} # directory synchronization failure needed so the importer # will retry using the forge synchronization. failed_report = SyncProgressReport(conduit) failed_report.metadata_state = constants.STATE_FAILED failed_call.return_value = failed_report progress_report = SyncProgressReport(conduit) progress_report.metadata_state = constants.STATE_FAILED mock_call.return_value = progress_report # test plugin = PuppetModuleImporter() report = plugin.sync_repo(repository, conduit, config) # validation mock_call.assert_called_with() self.assertEquals(report, conduit.build_failure_report.return_value)
def testUploadUnit(self, mock_handle_upload): module_importer = PuppetModuleImporter() mock_handle_upload.return_value = {'success_flag': True, 'summary': '', 'details': {}} report = module_importer.upload_unit(Mock(), Mock(), Mock(), Mock(), Mock(), Mock(), Mock()) self.assertTrue(report['success_flag'])