def test_two_failed_movies_to_be_printed(
self,
browser_mock,
search_success_mock,
click_rating_mock, # pylint: disable=too-many-arguments
progress_bar_mock,
save_movies_to_json_mock,
):
with patch("RatS.base.base_site.Site") as site_mock:
inserter = RatingsInserter(site_mock, None)
movie = dict()
movie["title"] = "Fight Club"
movie["year"] = 1999
movie["imdb"] = dict()
movie["imdb"]["id"] = "tt0137523"
movie["imdb"]["url"] = "https://www.imdb.com/title/tt0137523"
movie["imdb"]["my_rating"] = 9
movie2 = dict()
movie2["title"] = "unreadable movie"
movie2["year"] = 1111
movie2["imdb"] = dict()
movie2["imdb"]["id"] = "xxx"
movie2["imdb"]["url"] = "https://www.imdb.com/title/xxx"
movie2["imdb"]["my_rating"] = 4
movies = [movie, movie2, movie2]
search_success_mock.side_effect = [True, False, False]
inserter.insert(movies, "imdb")
self.assertEqual(2, len(inserter.failed_movies))
save_movies_to_json_mock.assert_called_with(
[movie2, movie2],
folder=inserter.exports_folder,
filename=inserter.failed_movies_filename,
)
def test_two_failed_movies_to_be_printed(
self,
browser_mock,
search_success_mock,
click_rating_mock, # pylint: disable=too-many-arguments
progress_bar_mock,
save_movies_to_json_mock):
with patch('RatS.base.base_site.Site') as site_mock:
inserter = RatingsInserter(site_mock, None)
movie = dict()
movie['title'] = 'Fight Club'
movie['year'] = 1999
movie['imdb'] = dict()
movie['imdb']['id'] = 'tt0137523'
movie['imdb']['url'] = 'https://www.imdb.com/title/tt0137523'
movie['imdb']['my_rating'] = 9
movie2 = dict()
movie2['title'] = 'unreadable movie'
movie2['year'] = 1111
movie2['imdb'] = dict()
movie2['imdb']['id'] = 'xxx'
movie2['imdb']['url'] = 'https://www.imdb.com/title/xxx'
movie2['imdb']['my_rating'] = 4
movies = [movie, movie2, movie2]
search_success_mock.side_effect = [True, False, False]
inserter.insert(movies, 'imdb')
self.assertEqual(2, len(inserter.failed_movies))
save_movies_to_json_mock.assert_called_with(
[movie2, movie2],
folder=inserter.exports_folder,
filename=inserter.failed_movies_filename)
def test_init(self, browser_mock):
with patch('RatS.base.base_site.Site') as site_mock:
inserter = RatingsInserter(site_mock, None)
self.assertEqual(inserter.site, site_mock)