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 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 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_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):
# 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 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)
