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)