예제 #1
0
    def test_add_remove_change_rating(self):

        new_rating = Rating(rating=2.5, user_id=11111, drink_id=1)
        new_rating.id = 10000
        db.session.add(new_rating)

        rating = Rating.query.get(10000)

        self.assertEqual(rating.id, 10000)
        self.assertEqual(rating.rating, 2.5)
        self.assertEqual(rating.user_id, 11111)
        self.assertEqual(rating.drink_id, 1)

        rating.rating = 1.0
        db.session.commit()

        self.assertEqual(rating.rating, 1.0)

        db.session.delete(rating)
        db.session.commit()
        check_rating = Rating.query.get(10000)

        self.assertEqual(check_rating, None)