コード例 #1
0
    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)
コード例 #2
0
    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)