def test_some_existing(self): postfix = model_factory.rpm_models(1)[0] postfix.name = 'postfix' vim = model_factory.rpm_models(1)[0] vim.name = 'vim-common' existing = [ Unit(postfix.TYPE, postfix.unit_key, postfix.metadata, ''), Unit(vim.TYPE, vim.unit_key, vim.metadata, ''), ] conduit = ImportUnitConduit('', '', '', '') conduit.get_destination_units = mock.MagicMock( spec_set=conduit.get_destination_units, return_value=existing) ret = associate.get_rpms_to_copy_by_name(self.RPM_NAMES, conduit) self.assertEqual(set(ret), set(['python-mock'])) self.assertEqual(conduit.get_destination_units.call_count, 1) self.assertTrue( isinstance(conduit.get_destination_units.call_args[0][0], UnitAssociationCriteria)) self.assertEqual( conduit.get_destination_units.call_args[0][0].type_ids, [models.RPM.TYPE]) self.assertEqual( conduit.get_destination_units.call_args[0][0].unit_fields, models.RPM.UNIT_KEY_NAMES)
def test_none_existing(self, mock_get_existing): mock_conduit = mock.MagicMock(spec_set=ImportUnitConduit('', '', '', '', '', '')) ret = associate.get_rpms_to_copy_by_name(self.RPM_NAMES, mock_conduit) self.assertTrue(isinstance(ret, set)) self.assertEqual(ret, set(self.RPM_NAMES)) self.assertEqual(mock_get_existing.call_count, 1) self.assertEqual(list(mock_get_existing.call_args[0][0]), list({'name': name} for name in self.RPM_NAMES)) self.assertEqual(mock_get_existing.call_args[0][1], models.RPM.UNIT_KEY_NAMES) self.assertEqual(mock_get_existing.call_args[0][2], models.RPM.TYPE) self.assertEqual(mock_get_existing.call_args[0][3], mock_conduit.get_destination_units)
def test_none_existing(self, mock_get_existing): mock_conduit = mock.MagicMock( spec_set=ImportUnitConduit('', '', '', '')) ret = associate.get_rpms_to_copy_by_name(self.RPM_NAMES, mock_conduit) self.assertTrue(isinstance(ret, set)) self.assertEqual(ret, set(self.RPM_NAMES)) self.assertEqual(mock_get_existing.call_count, 1) self.assertEqual(list(mock_get_existing.call_args[0][0]), list({'name': name} for name in self.RPM_NAMES)) self.assertEqual(mock_get_existing.call_args[0][1], models.RPM.UNIT_KEY_NAMES) self.assertEqual(mock_get_existing.call_args[0][2], models.RPM.TYPE) self.assertEqual(mock_get_existing.call_args[0][3], mock_conduit.get_destination_units)
def test_some_existing(self): postfix = model_factory.rpm_models(1)[0] postfix.name = 'postfix' vim = model_factory.rpm_models(1)[0] vim.name = 'vim-common' existing = [ Unit(postfix.TYPE, postfix.unit_key, postfix.metadata, ''), Unit(vim.TYPE, vim.unit_key, vim.metadata, ''), ] conduit = ImportUnitConduit('', '','', '', '', '') conduit.get_destination_units = mock.MagicMock(spec_set=conduit.get_destination_units, return_value=existing) ret = associate.get_rpms_to_copy_by_name(self.RPM_NAMES, conduit) self.assertEqual(set(ret), set(['python-mock'])) self.assertEqual(conduit.get_destination_units.call_count, 1) self.assertTrue(isinstance(conduit.get_destination_units.call_args[0][0], UnitAssociationCriteria)) self.assertEqual(conduit.get_destination_units.call_args[0][0].type_ids, [models.RPM.TYPE]) self.assertEqual(conduit.get_destination_units.call_args[0][0].unit_fields, models.RPM.UNIT_KEY_NAMES)