def test_find_ammo(self):
player = Mock()
inventory = standard_items.Inventory(player)
import questgame.game_items.weapons as weapons
ammo_poor = MockHelper.get_mock_item(spec=weapons.WoodArrow)
ammo_poor.cost = 1
ammo_best = MockHelper.get_mock_item(spec=weapons.SteelArrow)
ammo_best.cost = 100
ammo_good = MockHelper.get_mock_item(spec=weapons.IronArrow)
ammo_good.cost = 10
#Add poor arrow
self.assertEqual(inventory.find_ammo(weapons.Arrow), False)
inventory.add(ammo_poor)
#Add best arrow
self.assertEqual(inventory.find_ammo(weapons.Arrow), ammo_poor)
inventory.add(ammo_best)
self.assertEqual(inventory.find_ammo(weapons.Arrow), ammo_best)
#Add good arrow - should still select best arrow
inventory.add(ammo_good)
self.assertEqual(inventory.find_ammo(weapons.Arrow), ammo_best)
def test_contains(self):
player = Mock()
inventory = standard_items.Inventory(player)
beer = MockHelper.get_mock_beer()
self.assertFalse(inventory.contains(beer))
self.assertTrue(inventory.add(beer))
self.assertTrue(inventory.contains(beer))
def test_add_money(self):
player = Mock()
inventory = standard_items.Inventory(player)
#Add some money
self.assertTrue(inventory.add_money(10))
self.assertTrue(inventory.items() != [])
self.assertEqual(inventory.money, 10)
def test_get_item(self):
player = Mock()
inventory = standard_items.Inventory(player)
beer = MockHelper.get_mock_item()
#Nothing in inventroy
self.assertFalse(inventory.get_item(beer))
#Beer is in inventory
self.assertTrue(inventory.add(beer))
self.assertEqual(inventory.get_item(beer), beer)
def test_remove_money(self):
player = Mock()
inventory = standard_items.Inventory(player)
#Remove money we don't have
self.assertFalse(inventory.remove_money(3))
self.assertEqual(inventory.money, 0)
self.assertTrue(inventory.add_money(3.5))
self.assertEqual(inventory.money, 3.5)
self.assertTrue(inventory.remove_money(3))
self.assertEqual(inventory.money, 0.5)
self.assertTrue(inventory.remove_money(1))
self.assertEqual(inventory.money, 0)
def test_add(self):
player = Mock()
inventory = standard_items.Inventory(player)
#Add None
self.assertFalse(inventory.add(None))
#Add non-item
self.assertFalse(inventory.add(Mock()))
#Add item
beer = MockHelper.get_mock_beer()
self.assertTrue(inventory.add(beer))
self.assertTrue(inventory.contains(beer))
def test_weight(self):
player = Mock()
inventory = standard_items.Inventory(player)
self.assertEqual(inventory.weight, 0)
beer = MockHelper.get_mock_beer()
scroll = MockHelper.get_mock_scroll()
inventory.add(beer)
self.assertEqual(inventory.weight, 1)
inventory.add(beer)
self.assertEqual(inventory.weight, 2)
inventory.add(scroll)
self.assertEqual(inventory.weight, 3)
inventory.remove(scroll)
self.assertEqual(inventory.weight, 2)
def test_money(self):
player = Mock()
inventory = standard_items.Inventory(player)
self.assertEqual(inventory.money, 0.00)
#Add some money
coin = MockHelper.get_mock_money()
coin.cost = 10
self.assertTrue(inventory.add(coin))
self.assertEqual(inventory.money, 10.00)
#Add some more money
coin = MockHelper.get_mock_money()
coin.cost = 0.05
self.assertTrue(inventory.add(coin))
self.assertEqual(inventory.money, 10.05)
def test_items(self):
player = Mock()
inventory = standard_items.Inventory(player)
#Nothing in inventory
self.assertEqual(inventory.items(), [])
beer = MockHelper.get_mock_beer()
self.assertTrue(inventory.add(beer))
self.assertTrue(inventory.add(MockHelper.get_mock_beer()))
self.assertEqual(inventory.items(), [beer])
scroll = MockHelper.get_mock_scroll()
self.assertTrue(inventory.add(scroll))
self.assertTrue(inventory.add(MockHelper.get_mock_scroll()))
self.assertEqual(inventory.items(), [beer, scroll])
def test_count_items(self):
player = Mock()
beer = MockHelper.get_mock_item()
inventory = standard_items.Inventory(player)
#Test None
self.assertEqual(inventory.count_items(None), 0)
self.assertEqual(inventory.count_items(beer), 0)
#Test 1 item
self.assertTrue(inventory.add(beer))
self.assertEqual(inventory.count_items(beer), 1)
#Test 2 items
self.assertTrue(inventory.add(beer))
self.assertEqual(inventory.count_items(beer), 2)
def test_count(self):
player = Mock()
inventory = standard_items.Inventory(player)
self.assertEqual(inventory.count, 0)
