def test_unwanted_modules(self, mock_criteria): associated = [ {'name': 'pulp1', 'author': 'john', 'version': '1.0'}, {'name': 'pulp2', 'author': 'josh', 'version': '2.0'}, {'name': 'pulp3', 'author': 'josh', 'version': '3.0'} ] conduit = Mock() conduit.get_units = Mock(return_value=[]) criteria = Mock() mock_criteria.return_value = criteria # test and validation inventory = Inventory(conduit) inventory.associated = set() inventory.associated.add(tuple(associated[0].items())) inventory.associated.add(tuple(associated[1].items())) inventory.associated.add(tuple(associated[2].items())) wanted = associated[:1] unwanted = inventory.unwanted_modules(wanted) self.assertEqual(len(unwanted), 2) self.assertTrue(associated[1] in unwanted) self.assertTrue(associated[2] in unwanted)
def test_already_associated(self, mock_criteria): associated = [ {'name': 'pulp1', 'author': 'john', 'version': '1.0'}, {'name': 'pulp2', 'author': 'josh', 'version': '2.0'} ] conduit = Mock() conduit.get_units = Mock(return_value=[]) criteria = Mock() mock_criteria.return_value = criteria # test and validation module = Mock() inventory = Inventory(conduit) inventory.associated = set() inventory.associated.add(tuple(associated[0].items())) inventory.associated.add(tuple(associated[1].items())) # should be associated module.unit_key = Mock(return_value=associated[0]) self.assertTrue(inventory.already_associated(module)) # should be associated module.unit_key = Mock(return_value=associated[1]) self.assertTrue(inventory.already_associated(module)) # should not be associated module.unit_key = Mock(return_value={}) self.assertFalse(inventory.already_associated(module))