def test_add_favourite(self):
user = User(email='*****@*****.**', password='******')
db.session.add(user)
db.session.commit()
user.add_favourite(tmdb_id=69740)
favourites = user.favourites
favourite = favourites.pop()
self.assertEqual(favourite.user_id, 1)
self.assertEqual(favourite.tmdb_id, 69740)
def test_get_show(self):
user = User(email='*****@*****.**', password='******')
db.session.add(user)
db.session.commit()
user.add_favourite(tmdb_id=69740)
favourites = user.favourites
favourite = favourites.pop()
show = favourite.get_show()
self.assertEqual(show.name, 'Ozark')
def test_remove_favourite(self):
user = User(email='*****@*****.**', password='******')
db.session.add(user)
db.session.commit()
user.add_favourite(tmdb_id=69740)
user.remove_favourite(tmdb_id=69740)
favourites = user.favourites
self.assertTrue(
len([
favourite for favourite in favourites
if favourite.tmdb_id == 69740
]) == 0)
