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

"""
Helpers for testing evadventure modules.

"""

from evennia.utils import create

from .. import enums
from ..characters import EvAdventureCharacter
from ..objects import (
    EvAdventureArmor,
    EvAdventureHelmet,
    EvAdventureObject,
    EvAdventureShield,
    EvAdventureWeapon,
)
from ..rooms import EvAdventureRoom


class EvAdventureMixin:
    """
    Provides a set of pre-made characters.

    """

    def setUp(self):
        super().setUp()
        self.location = create.create_object(EvAdventureRoom, key="testroom")
        self.character = create.create_object(
            EvAdventureCharacter, key="testchar", location=self.location
        )
        self.helmet = create.create_object(
            EvAdventureHelmet,
            key="helmet",
        )
        self.shield = create.create_object(
            EvAdventureShield,
            key="shield",
        )
        self.armor = create.create_object(
            EvAdventureArmor,
            key="armor",
        )
        self.weapon = create.create_object(
            EvAdventureWeapon,
            key="weapon",
        )
        self.big_weapon = create.create_object(
            EvAdventureWeapon,
            key="big_weapon",
            attributes=[("inventory_use_slot", enums.WieldLocation.TWO_HANDS)],
        )
        self.item = create.create_object(EvAdventureObject, key="backpack item")