def test_training_is_deleted_from_db(self): """Test if the training has been successfully deleted from the database""" training = self._create_sample_training(self.user) training.delete_from_db() found_training = TrainingModel.find_by_name("test training") self.assertIsNone(found_training)
def test_find_by_name(self): """Test if the training is found""" training = self._create_sample_training( user=self.user, name="test training" ) found_training = TrainingModel.find_by_name("test training") self.assertEqual(found_training, training)
def test_training_is_saved_in_db(self): """Test if the training has been successfully saved in the database""" training = TrainingModel( name="test training", user_id=self.user.id, distance=10, time_in_seconds=3600, ) training.save_to_db() found_training = TrainingModel.find_by_name("test training") self.assertIsNotNone(found_training)
def test_post_training_data_in_db(self): """Test if the correct data is saved in the database""" data = { "name": "test3", "distance": 10, "time_in_seconds": 3600, } self.client.post( path="trainings/", data=json.dumps(data), headers={ "Content-Type": "application/json", "Authorization": f"Bearer {self.access_token}", }, ) training = TrainingModel.find_by_name(data["name"]) self.assertEqual(training.name, data["name"]) self.assertEqual(training.distance, data["distance"]) self.assertEqual(training.time_in_seconds, data["time_in_seconds"])
def test_find_by_name_no_training(self): """Test if None is returned if the training with the given name doesn't exist""" found_training = TrainingModel.find_by_name("test training") self.assertIsNone(found_training)