def test_count_number_of_trainings(self):
training = Training()
start = datetime.now(timezone.utc) - timedelta(seconds=405)
training.save()
training.start_training_date = start
training.save()
number, rest = count_number_of_trainings(training)
self.assertEqual(number, 20)
self.assertEqual(rest, 5)
def test_add_training_points_to_character(self):
training = Training()
training.character = self.hero
training.save()
training.start_training_date = datetime.now(
timezone.utc) - timedelta(seconds=885)
training.save()
self.hero.busy = True
set_is_ending_training(training)
add_training_points_to_character(self.hero)
self.assertEqual(self.hero.growth_points, 45)