import unittest

from MechFabrik import MechFabrik
from MechFabrik.bauteile import AreaNames, ZoneNames

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 = {
        ZoneNames.HEAD:               6,
        ZoneNames.MIDDLE_TORSO_FRONT: 9,
        ZoneNames.MIDDLE_TORSO_REAR:  4,
        ZoneNames.LEFT_TORSO_FRONT:   8,
        ZoneNames.LEFT_TORSO_REAR:    3,
        ZoneNames.RIGHT_TORSO_FRONT:  8,
        ZoneNames.RIGHT_TORSO_REAR:   3,
        ZoneNames.LEFT_ARM:           5,
        ZoneNames.RIGHT_ARM:          5,
        ZoneNames.LEFT_LEG:           7,
        ZoneNames.RIGHT_LEG:          7,
        }

NEXT_OUTER_ZONE = {
        ZoneNames.HEAD:               AreaNames.HEAD,
        ZoneNames.MIDDLE_TORSO_FRONT: AreaNames.MIDDLE_TORSO,
        ZoneNames.MIDDLE_TORSO_REAR:  AreaNames.MIDDLE_TORSO,
        ZoneNames.LEFT_TORSO_FRONT:   AreaNames.LEFT_TORSO,
        ZoneNames.LEFT_TORSO_REAR:    AreaNames.LEFT_TORSO,
        ZoneNames.RIGHT_TORSO_FRONT:  AreaNames.RIGHT_TORSO,
        ZoneNames.RIGHT_TORSO_REAR:   AreaNames.RIGHT_TORSO,
        ZoneNames.LEFT_ARM:           AreaNames.LEFT_ARM,
        ZoneNames.RIGHT_ARM:          AreaNames.RIGHT_ARM,
        ZoneNames.LEFT_LEG:           AreaNames.LEFT_LEG,
        ZoneNames.RIGHT_LEG:          AreaNames.RIGHT_LEG,
}


class AbstractTestMechAmor(unittest.TestCase):

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

        fabrik.set_head_armor(OUTER_ARMOR[ZoneNames.HEAD])
        fabrik.set_middle_torso_armor(OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_FRONT],OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_REAR])
        fabrik.set_side_torso_armor(OUTER_ARMOR[ZoneNames.LEFT_TORSO_FRONT],OUTER_ARMOR[ZoneNames.LEFT_TORSO_REAR])
        fabrik.set_arm_armor(OUTER_ARMOR[ZoneNames.LEFT_ARM])
        fabrik.set_leg_armor(OUTER_ARMOR[ZoneNames.LEFT_LEG])

        self.mech = fabrik.auslieferung()