def test_copy_units_only_specified(self):
# Setup
conduit = mock.MagicMock()
specified_units = ['a', 'b']
# Test
copier.copy_units(conduit, specified_units)
# Verify
self.assertEqual(0, conduit.get_source_units.call_count)
self.assertEqual(len(specified_units), conduit.associate_unit.call_count)
self._assert_associated_units(conduit, specified_units)
def test_copy_units_all_units(self):
# Setup
conduit = mock.MagicMock()
all_source_units = ['a', 'b', 'c'] # content is irrelevant
conduit.get_source_units.return_value = all_source_units
# Test
copier.copy_units(conduit, None)
# Verify
self.assertEqual(1, conduit.get_source_units.call_count)
call_args = conduit.get_source_units.call_args
self.assertTrue('criteria' in call_args[1])
self.assertTrue(isinstance(call_args[1]['criteria'], UnitAssociationCriteria))
self.assertEqual(call_args[1]['criteria'].type_ids, [constants.TYPE_PUPPET_MODULE])
self.assertEqual(len(all_source_units), conduit.associate_unit.call_count)
self._assert_associated_units(conduit, all_source_units)
def import_units(self, source_repo, dest_repo, import_conduit, config,
units=None):
copier.copy_units(import_conduit, units)