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)