def test_bad_metadata(self):
# Set up a distribution unit is no metadata
distribution_unit = [AssociatedUnit(ids.TYPE_ID_DISTRO, None, {}, None, None, None, None, None)]
# Without the metadata, the unit should not be exported
summary, details = export_utils.export_distribution('/working/dir', distribution_unit)
self.assertEqual(0, summary['num_distribution_units_exported'])
def test_failed_copy(self, mock_copy):
# Set up a distribution unit
metadata = {'files': [{'relativepath': 'unit'}]}
distribution_unit = [AssociatedUnit(ids.TYPE_ID_DISTRO, None, metadata, '/fake/path', None,
None, None, None)]
# Test that when create_copy fails, units are not exported
summary, details = export_utils.export_distribution('/working/dir', distribution_unit)
mock_copy.assert_called_once_with('/fake/path/unit', '/working/dir/unit')
self.assertEqual(0, summary['num_distribution_units_exported'])
def test_successful_export(self, mock_copy):
# Setup a distribution unit
metadata = {'files': [{'relativepath': 'unit'}]}
distribution_units = [AssociatedUnit(ids.TYPE_ID_DISTRO, None, metadata, '/fake/path', None,
None, None, None)]
# Test that the unit is successfully exported
summary, details = export_utils.export_distribution('/working/dir', distribution_units)
self.assertEqual(('/fake/path/unit', '/working/dir/unit'), mock_copy.call_args[0])
self.assertEqual(1, summary['num_distribution_units_exported'])