import unittest

from MechFabrik import MechFabrik
from MechFabrik.bauteile import AreaNames

INNER_ARMOR = {
        AreaNames.HEAD:         3,
        AreaNames.MIDDLE_TORSO: 6,
        AreaNames.LEFT_TORSO:   5,
        AreaNames.RIGHT_TORSO:  5,
        AreaNames.LEFT_ARM:     3,
        AreaNames.RIGHT_ARM:    3,
        AreaNames.LEFT_LEG:     4,
        AreaNames.RIGHT_LEG:    4,
        }

OUTER_ARMOR = {
        AreaNames.HEAD:         6,
        AreaNames.MIDDLE_TORSO: 9,
        AreaNames.LEFT_TORSO:   8,
        AreaNames.RIGHT_TORSO:  8,
        AreaNames.LEFT_ARM:     5,
        AreaNames.RIGHT_ARM:    5,
        AreaNames.LEFT_LEG:     7,
        AreaNames.RIGHT_LEG:    7,
        }


class AbstractTestMechAmor(unittest.TestCase):

    def setUp(self):
        fabrik = MechFabrik('Heuschrecke', 20.0, 8)

        fabrik.set_head_armor(OUTER_ARMOR[AreaNames.HEAD])
        fabrik.set_middle_torso_armor(OUTER_ARMOR[AreaNames.MIDDLE_TORSO])
        fabrik.set_side_torso_armor(OUTER_ARMOR[AreaNames.LEFT_TORSO])
        fabrik.set_arm_armor(OUTER_ARMOR[AreaNames.LEFT_ARM])
        fabrik.set_leg_armor(OUTER_ARMOR[AreaNames.LEFT_LEG])

        self.mech = fabrik.auslieferung()