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)
