Source code for evennia.contrib.tutorials.evadventure.tests.test_equipment

"""
Test the EvAdventure equipment handler.

"""

from unittest.mock import MagicMock, patch

from parameterized import parameterized

from evennia.utils.test_resources import BaseEvenniaTest

from ..enums import Ability, WieldLocation
from ..equipment import EquipmentError
from .mixins import EvAdventureMixin


[docs]class TestEquipment(EvAdventureMixin, BaseEvenniaTest):
[docs] def test_count_slots(self): self.assertEqual(self.character.equipment.count_slots(), 0)
[docs] def test_max_slots(self): self.assertEqual(self.character.equipment.max_slots, 11) setattr(self.character, Ability.CON.value, 3) self.assertEqual(self.character.equipment.max_slots, 13)
[docs] def test_add__remove(self): self.character.equipment.add(self.helmet) self.assertEqual(self.character.equipment.slots[WieldLocation.BACKPACK], [self.helmet]) self.character.equipment.remove(self.helmet) self.assertEqual(self.character.equipment.slots[WieldLocation.BACKPACK], [])
[docs] def test_move__get_current_slot(self): self.character.equipment.add(self.helmet) self.assertEqual( self.character.equipment.get_current_slot(self.helmet), WieldLocation.BACKPACK ) self.character.equipment.move(self.helmet) self.assertEqual(self.character.equipment.get_current_slot(self.helmet), WieldLocation.HEAD)
[docs] def test_get_wearable_or_wieldable_objects_from_backpack(self): self.character.equipment.add(self.helmet) self.character.equipment.add(self.weapon) self.assertEqual( self.character.equipment.get_wieldable_objects_from_backpack(), [self.weapon] ) self.assertEqual( self.character.equipment.get_wearable_objects_from_backpack(), [self.helmet] ) self.assertEqual( self.character.equipment.all(), [ (None, WieldLocation.WEAPON_HAND), (None, WieldLocation.SHIELD_HAND), (None, WieldLocation.TWO_HANDS), (None, WieldLocation.BODY), (None, WieldLocation.HEAD), (self.helmet, WieldLocation.BACKPACK), (self.weapon, WieldLocation.BACKPACK), ], )
def _get_empty_slots(self): return { WieldLocation.BACKPACK: [], WieldLocation.WEAPON_HAND: None, WieldLocation.SHIELD_HAND: None, WieldLocation.TWO_HANDS: None, WieldLocation.BODY: None, WieldLocation.HEAD: None, }
[docs] def test_equipmenthandler_max_slots(self): self.assertEqual(self.character.equipment.max_slots, 11)
@parameterized.expand( [ # size, pass_validation? (1, True), (2, True), (11, True), (12, False), (20, False), (25, False), ] ) def test_validate_slot_usage(self, size, is_ok): obj = MagicMock() obj.size = size with patch("evennia.contrib.tutorials.evadventure.equipment.inherits_from") as mock_inherit: mock_inherit.return_value = True if is_ok: self.assertTrue(self.character.equipment.validate_slot_usage(obj)) else: with self.assertRaises(EquipmentError): self.character.equipment.validate_slot_usage(obj) @parameterized.expand( [ # item, where ("helmet", WieldLocation.HEAD), ("shield", WieldLocation.SHIELD_HAND), ("armor", WieldLocation.BODY), ("weapon", WieldLocation.WEAPON_HAND), ("big_weapon", WieldLocation.TWO_HANDS), ("item", WieldLocation.BACKPACK), ] ) def test_move(self, itemname, where): self.assertEqual(self.character.equipment.slots, self._get_empty_slots()) obj = getattr(self, itemname) self.character.equipment.move(obj) # check that item ended up in the right place if where is WieldLocation.BACKPACK: self.assertTrue(obj in self.character.equipment.slots[where]) else: self.assertEqual(self.character.equipment.slots[where], obj)
[docs] def test_add(self): self.character.equipment.add(self.weapon) self.assertEqual(self.character.equipment.slots[WieldLocation.WEAPON_HAND], None) self.assertTrue(self.weapon in self.character.equipment.slots[WieldLocation.BACKPACK])
[docs] def test_two_handed_exclusive(self): """Two-handed weapons can't be used together with weapon+shield""" self.character.equipment.move(self.big_weapon) self.assertEqual(self.character.equipment.slots[WieldLocation.TWO_HANDS], self.big_weapon) # equipping sword or shield removes two-hander self.character.equipment.move(self.shield) self.assertEqual(self.character.equipment.slots[WieldLocation.SHIELD_HAND], self.shield) self.assertEqual(self.character.equipment.slots[WieldLocation.TWO_HANDS], None) self.character.equipment.move(self.weapon) self.assertEqual(self.character.equipment.slots[WieldLocation.WEAPON_HAND], self.weapon) # the two-hander removes the two weapons self.character.equipment.move(self.big_weapon) self.assertEqual(self.character.equipment.slots[WieldLocation.TWO_HANDS], self.big_weapon) self.assertEqual(self.character.equipment.slots[WieldLocation.SHIELD_HAND], None) self.assertEqual(self.character.equipment.slots[WieldLocation.WEAPON_HAND], None)
[docs] def test_remove__with_obj(self): self.character.equipment.move(self.shield) self.character.equipment.move(self.item) self.character.equipment.add(self.weapon) self.assertEqual(self.character.equipment.slots[WieldLocation.SHIELD_HAND], self.shield) self.assertEqual( self.character.equipment.slots[WieldLocation.BACKPACK], [self.item, self.weapon] ) self.assertEqual(self.character.equipment.remove(self.shield), [self.shield]) self.assertEqual(self.character.equipment.remove(self.item), [self.item]) self.assertEqual(self.character.equipment.slots[WieldLocation.SHIELD_HAND], None) self.assertEqual(self.character.equipment.slots[WieldLocation.BACKPACK], [self.weapon])
[docs] def test_remove__with_slot(self): self.character.equipment.move(self.shield) self.character.equipment.move(self.item) self.character.equipment.add(self.helmet) self.assertEqual(self.character.equipment.slots[WieldLocation.SHIELD_HAND], self.shield) self.assertEqual( self.character.equipment.slots[WieldLocation.BACKPACK], [self.item, self.helmet] ) self.assertEqual(self.character.equipment.remove(WieldLocation.SHIELD_HAND), [self.shield]) self.assertEqual( self.character.equipment.remove(WieldLocation.BACKPACK), [self.item, self.helmet] ) self.assertEqual(self.character.equipment.slots[WieldLocation.SHIELD_HAND], None) self.assertEqual(self.character.equipment.slots[WieldLocation.BACKPACK], [])
[docs] def test_properties(self): self.character.equipment.move(self.armor) self.assertEqual(self.character.equipment.armor, 1) self.character.equipment.move(self.shield) self.assertEqual(self.character.equipment.armor, 2) self.character.equipment.move(self.helmet) self.assertEqual(self.character.equipment.armor, 3) self.character.equipment.move(self.weapon) self.assertEqual(self.character.equipment.weapon, self.weapon) self.character.equipment.move(self.big_weapon) self.assertEqual(self.character.equipment.weapon, self.big_weapon)