def test_handle_uploaded_unit(self): # Setup initialized_unit = mock.MagicMock() initialized_unit.storage_path = self.dest_dir self.conduit.init_unit.return_value = initialized_unit # Test upload.handle_uploaded_unit(self.repo, constants.TYPE_PUPPET_MODULE, self.unit_key, self.unit_metadata, self.source_file, self.conduit) # Verify self.assertTrue(os.path.exists(self.dest_file)) self.assertEqual(1, self.conduit.init_unit.call_count) self.assertEqual(1, self.conduit.save_unit.call_count)
def upload_unit(self, repo, type_id, unit_key, metadata, file_path, conduit, config): try: report = upload.handle_uploaded_unit(repo, type_id, unit_key, metadata, file_path, conduit) except Exception, e: _logger.exception(e) report = {'success_flag': False, 'summary': e.message, 'details': {}}
def upload_unit(self, repo, type_id, unit_key, metadata, file_path, conduit, config): try: report = upload.handle_uploaded_unit(repo, type_id, unit_key, metadata, file_path, conduit) except Exception, e: _logger.exception(e) report = { 'success_flag': False, 'summary': e.message, 'details': {} }
def test_handle_uploaded_unit_with_no_data(self, mock_repo_controller, mock_module): # Setup initialized_unit = mock.MagicMock() initialized_unit.storage_path = self.dest_dir self.conduit.init_unit.return_value = initialized_unit # Test report = upload.handle_uploaded_unit(self.repo, constants.TYPE_PUPPET_MODULE, {}, {}, self.source_file, self.conduit) mock_module.from_metadata.return_value.save_and_import_content.assert_called_once() self.assertTrue(report['success_flag'])
def test_handle_uploaded_unit_with_no_data(self): # Setup initialized_unit = mock.MagicMock() initialized_unit.storage_path = self.dest_dir self.conduit.init_unit.return_value = initialized_unit # Test report = upload.handle_uploaded_unit(self.repo, constants.TYPE_PUPPET_MODULE, {}, {}, self.source_file, self.conduit) # Verify self.assertTrue(os.path.exists(self.dest_file)) self.assertEqual(1, self.conduit.init_unit.call_count) self.assertEqual(1, self.conduit.save_unit.call_count) self.assertTrue(report['success_flag'])
def test_handle_uploaded_unit(self, mock_repo_controller, mock_module): # Setup initialized_unit = mock.MagicMock() initialized_unit.storage_path = self.dest_dir self.conduit.init_unit.return_value = initialized_unit # Test report = upload.handle_uploaded_unit(self.repo, constants.TYPE_PUPPET_MODULE, self.unit_key, self.unit_metadata, self.source_file, self.conduit) # Verify mock_module.from_metadata.return_value.save_and_import_content.assert_called_once() self.assertTrue(isinstance(report, dict)) self.assertTrue('success_flag' in report) self.assertTrue(report['success_flag']) self.assertTrue('summary' in report) self.assertTrue('details' in report)
def test_handle_uploaded_unit(self): # Setup initialized_unit = mock.MagicMock() initialized_unit.storage_path = self.dest_dir self.conduit.init_unit.return_value = initialized_unit # Test report = upload.handle_uploaded_unit(self.repo, constants.TYPE_PUPPET_MODULE, self.unit_key, self.unit_metadata, self.source_file, self.conduit) # Verify self.assertTrue(os.path.exists(self.dest_file)) self.assertEqual(1, self.conduit.init_unit.call_count) self.assertEqual(1, self.conduit.save_unit.call_count) self.assertTrue(isinstance(report, dict)) self.assertTrue('success_flag' in report) self.assertTrue(report['success_flag']) self.assertTrue('summary' in report) self.assertTrue('details' in report)
def upload_unit(self, repo, type_id, unit_key, metadata, file_path, conduit, config): report = upload.handle_uploaded_unit(repo, type_id, unit_key, metadata, file_path, conduit) return report
def upload_unit(self, repo, type_id, unit_key, metadata, file_path, conduit, config): upload.handle_uploaded_unit(repo, type_id, unit_key, metadata, file_path, conduit)