import unittest

from mech_factory import MechFabrik
from mech_factory.component_storage 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,
        }
## entspricht 64 Punkte → 4 t Panzerung
OUTER_ARMOR = {
        ZoneNames.HEAD:               6,
        ZoneNames.MIDDLE_TORSO_FRONT: 8,
        ZoneNames.MIDDLE_TORSO_REAR:  4,
        ZoneNames.LEFT_TORSO_FRONT:   8,
        ZoneNames.LEFT_TORSO_REAR:    2,
        ZoneNames.RIGHT_TORSO_FRONT:  8,
        ZoneNames.RIGHT_TORSO_REAR:   2,
        ZoneNames.LEFT_ARM:           5,
        ZoneNames.RIGHT_ARM:          5,
        ZoneNames.LEFT_LEG:           8,
        ZoneNames.RIGHT_LEG:          8,
        }

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 AbstractTestMech(unittest.TestCase):

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


class AbstractTestMechAmor(AbstractTestMech):

    def setUp(self):
        super(AbstractTestMechAmor, self).setUp()
        self.factory.set_head_armor(OUTER_ARMOR[ZoneNames.HEAD])
        self.factory.set_middle_torso_armor(OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_FRONT],OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_REAR])
        self.factory.set_side_torso_armor(OUTER_ARMOR[ZoneNames.LEFT_TORSO_FRONT],OUTER_ARMOR[ZoneNames.LEFT_TORSO_REAR])
        self.factory.set_arm_armor(OUTER_ARMOR[ZoneNames.LEFT_ARM])
        self.factory.set_leg_armor(OUTER_ARMOR[ZoneNames.LEFT_LEG])