def test_get_item_by_item_name(self): item = self._get_item() collection = ItemCollection() collection.add(item) expected = item # Typical use case actual = collection.get_item_by_item_name(item.name) self.assertEqual(len(collection.items), 1) self.assertEqual(expected, actual) # Test that we don't get the wrong thing expected = None actual = collection.get_item_by_item_name("hello world") self.assertEqual(expected, actual)
def test_add_item(self): item = self._get_item() collection = ItemCollection() collection.add(item) expected = item actual = collection.get_item_by_item_name(item.name) self.assertEqual(len(collection.items), 1) self.assertEqual(expected, actual) # Test that we can't add duplicate items collection.add(item) self.assertEqual(len(collection.items), 1)