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))
