コード例 #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)
コード例 #2
0
ファイル: importer.py プロジェクト: seandst/pulp_puppet
 def import_units(self,
                  source_repo,
                  dest_repo,
                  import_conduit,
                  config,
                  units=None):
     return copier.copy_units(import_conduit, units)
コード例 #3
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)
コード例 #4
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
        returned = copier.copy_units(conduit, None)

        # Verify
        self.assertEqual(returned, all_source_units)
        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)
コード例 #5
0
ファイル: importer.py プロジェクト: kimjohansen/pulp_puppet
 def import_units(self, source_repo, dest_repo, import_conduit, config,
                  units=None):
     return copier.copy_units(import_conduit, units)