def get_game_variables(constants):
player = Entity(0, 0, '@', libtcod.white, 'Player', blocks=True, render_order=RenderOrder.ACTOR)
Fighter(hp=100, defense=11, power=2, hit=11).add_to_entity(player)
Level().add_to_entity(player)
inventory_component = Inventory(26)
inventory_component.add_to_entity(player)
equipment_component = Equipment()
equipment_component.add_to_entity(player)
dagger = Entity(0, 0, '-', libtcod.sky, 'Dagger')
equippable_component = Equippable(EquipmentSlots.MAIN_HAND, power_bonus=2)
equippable_component.add_to_entity(dagger)
inventory_component.add_item(dagger)
equipment_component.toggle_equip(dagger)
StatusEffects().add_to_entity(player)
world = World(player, constants['map_width'], constants['map_height'])
message_log = MessageLog(constants['message_x'], constants['message_width'], constants['message_height'])
game_state = GameStates.PLAYERS_TURN
return player, world, message_log, game_state
def test_toggle_equip__non_equippable__raisesImpossible():
e = Equipment()
vial = factory.make("healing vial")
with pytest.raises(exceptions.Impossible):
e.toggle_equip(item=vial)
def test_toggle_equip__armor2armor__msg(leather_armor):
e = Equipment(leather_armor)
bp_vest = factory.make("bulletproof vest")
result = e.toggle_equip(item=bp_vest)
assert result == "You remove the leather vest. You equip the bulletproof vest. "
def test_toggle_equip__armor2armor(leather_armor):
e = Equipment(leather_armor)
bp_vest = factory.make("bulletproof vest")
e.toggle_equip(item=bp_vest)
assert e.slots['ARMOR'] == bp_vest
def test_toggle_equip__weapon2weapon__msg(dagger):
e = Equipment(dagger)
baton = factory.make("riot baton")
result = e.toggle_equip(item=baton)
assert result == "You remove the dagger. You equip the riot baton. "
def test_toggle_equip__weapon2weapon(dagger):
e = Equipment(dagger)
baton = factory.make("riot baton")
e.toggle_equip(item=baton)
assert e.slots['WEAPON'] == baton
def test_toggle_equip__armor2none__msg(leather_armor):
e = Equipment(leather_armor)
result = e.toggle_equip(item=leather_armor)
assert result == "You remove the leather vest. "
def test_toggle_equip__armor2none(leather_armor):
e = Equipment(leather_armor)
e.toggle_equip(item=leather_armor)
assert e.slots['ARMOR'] is None
def test_toggle_equip__weapon2none__msg(dagger):
e = Equipment(dagger)
result = e.toggle_equip(item=dagger)
assert result == "You remove the dagger. "
def test_toggle_equip__weapon2none(dagger):
e = Equipment(dagger)
e.toggle_equip(item=dagger)
assert e.slots['WEAPON'] is None
def test_toggle_equip__none2armor__msg(leather_armor):
e = Equipment()
result = e.toggle_equip(item=leather_armor)
assert result == "You equip the leather vest. "
def test_toggle_equip__none2armor(leather_armor):
e = Equipment()
e.toggle_equip(item=leather_armor)
assert e.slots['ARMOR'] == leather_armor
def test_toggle_equip__none2weapon__msg(dagger):
e = Equipment()
result = e.toggle_equip(item=dagger)
assert result == "You equip the dagger. "
def test_toggle_equip__none2weapon(dagger):
e = Equipment()
e.toggle_equip(item=dagger)
assert e.slots['WEAPON'] == dagger
