def test_no_instant_heal_in_resort(self):
self.hero.health = self.hero.max_health
self.hero.position.place.modifier = Resort(self.hero.position.place)
old_messages_len = len(self.hero.messages.messages)
prototypes.ActionInPlacePrototype.create(hero=self.hero)
self.assertEqual(self.hero.health, self.hero.max_health)
self.storage._test_save()
def test_companion_heal_in_resort__no_companion(self):
self.assertEqual(self.hero.companion, None)
self.hero.position.place.modifier = Resort(self.hero.position.place)
prototypes.ActionInPlacePrototype.create(hero=self.hero)
self.storage._test_save()
def test_companion_heal_in_resort__healed_companion(self):
companion_record = companions_storage.companions.enabled_companions(
).next()
self.hero.set_companion(
companions_logic.create_companion(companion_record))
self.assertEqual(self.hero.companion.health,
self.hero.companion.max_health)
self.hero.position.place.modifier = Resort(self.hero.position.place)
prototypes.ActionInPlacePrototype.create(hero=self.hero)
self.assertFalse(self.hero.messages.messages[-1].key.
is_ACTION_INPLACE_COMPANION_HEAL)
self.storage._test_save()
def test_companion_heal_in_resort__damaged_companion(self):
companion_record = companions_storage.companions.enabled_companions(
).next()
self.hero.set_companion(
companions_logic.create_companion(companion_record))
self.hero.companion.health = 1
self.hero.position.place.modifier = Resort(self.hero.position.place)
with self.check_increased(lambda: self.hero.companion.health):
prototypes.ActionInPlacePrototype.create(hero=self.hero)
self.assertTrue(self.hero.messages.messages[-1].key.
is_ACTION_INPLACE_COMPANION_HEAL)
self.storage._test_save()
def test_resort(self):
self.assertFalse(CraftCenter(self.place_1).full_regen_allowed())
self.assertTrue(Resort(self.place_1).full_regen_allowed())