Esempio n. 1
0
    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)
Esempio n. 2
0
    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)
Esempio n. 3
0
 def import_units(self, source_repo, dest_repo, import_conduit, config,
                  units=None):
     copier.copy_units(import_conduit, units)