Ejemplo n.º 1
0
    def test_get_dead_unit_by_name(self):
        dungeon_model = self._get_dungeon_model()
        with LogCapture():
            logger = logging.getLogger()
            dungeon = Dungeon(dungeon=dungeon_model,
                              announcer={},
                              log=logger)

        unit_name = "Oliver Queen"
        unit_generator = UnitGenerator()
        unit = unit_generator.generate(unit_name=unit_name)

        dungeon.add_unit(unit)

        unit.kill()

        # test typical use case
        actual = dungeon.get_dead_unit_by_name(unit.get_name())
        self.assertIsInstance(actual, Unit)

        unit.hp = 1

        actual_alive = dungeon.get_living_unit_by_name(unit.get_name())
        self.assertIsInstance(actual_alive, Unit)