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 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 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'])
