def test_tmdb_status(self):
"""
Test tmdb_status
"""
self.initialize()
# take all not imported movies without posters from db
films = Film.objects.filter(image="", tmdb_import_status=Film.NOT_IMPORTED)
# there should be only 1 movie
self.assertEqual(len(films), 1)
#try to fetch posters
for film in films:
# check if there is no poster
self.assertEqual(film.image is not None, True)
# check tmdb_import_status, should be NOT_IMPORTED
self.assertEqual(film.tmdb_import_status, Film.NOT_IMPORTED)
# fetch poster for film
tmdb_movie = fetch_film_by_id(film)
film.tmdb_import_status = Film.IMPORTED
film.save()
self.assertEqual(tmdb_movie is not None, True)
result = save_tmdb_poster(film, tmdb_movie)
self.assertEqual(result, True)
film = Film.objects.get(title="The Terminator")
# check tmdb_status should be IMPORTED
self.assertEqual(film.tmdb_import_status, Film.IMPORTED)
# there are no other movies without posters in db
films = Film.objects.filter(tmdb_import_status=Film.NOT_IMPORTED)
self.assertEqual(len(films), 0)
def test_fetch_by_id(self):
"""
Test fetching by imdb code
"""
self.initialize()
films = Film.objects.filter(image="", tmdb_import_status=Film.NOT_IMPORTED)
for film in films:
self.assertEqual(film.image is not None, True)
tmdb_movie = fetch_film_by_id(film)
self.assertEqual(tmdb_movie is not None, True)
result = save_tmdb_poster(film, tmdb_movie)
self.assertEqual(result, True)
film = Film.objects.get(title="The Terminator")
self.assertEqual(film.image is not None, True)
