def test_download_movie(self): with self.instance.test_request_context() as request: u = UserFactory(login='******', email='*****@*****.**') m = MovieFactory() w = WatchlistFactory(user_id=u.id, movie_id=m.id) result = w.download() self.assertEqual(result, w.id) self.assertTrue(self.redis.exists(Watchlist.downloading_key)) self.assertTrue(self.redis.llen(Watchlist.downloading_key), 1) self.assertEqual(self.redis.lpop(Watchlist.downloading_key), w.id)
def test_download_movie_with_wrong_status(self): with self.instance.test_request_context() as request: u = UserFactory(login='******', email='*****@*****.**') m = MovieFactory() w = WatchlistFactory(user_id=u.id, movie_id=m.id, status=Watchlist.LOADING_MOVIE) result = w.download() self.assertEqual(result, None) w.status = Watchlist.LOADED_MOVIE result = w.download() self.assertEqual(result, None) w.status = Watchlist.ENCODING_MOVIE result = w.download() self.assertEqual(result, None) w.status = Watchlist.ENCODED_MOVIE result = w.download() self.assertEqual(result, None)