import unittest

from test_mech import AbstractTestMechAmor, INNER_ARMOR, OUTER_ARMOR
from MechFabrik.bauteile import AreaNames


class TestMechAmor(AbstractTestMechAmor):

    def test_process(self):
        for area_name in AreaNames:
            self.mech.strike(area_name, OUTER_ARMOR[area_name] - 3)
            self.assertEqual(3, self.mech.inner_structure.areas.get(area_name).armor)
            self.mech.strike(area_name, 4)
            self.assertEqual(0, self.mech.inner_structure.areas.get(area_name).armor)
            self.assertEqual(INNER_ARMOR[area_name] - 1, self.mech.inner_structure.areas.get(area_name).inner_armor)