def test_parser(self, site_mock, base_init_mock, browser_mock,
parse_movie_mock, progress_print_mock): # pylint: disable=too-many-arguments
browser_mock.page_source = self.my_ratings
parser = TMDBRatingsParser(None)
parser.args = False
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'TMDB'
parser.site.browser = browser_mock
parser.args = None
parser.parse()
self.assertEqual(20, parse_movie_mock.call_count)
self.assertEqual(20, len(parser.movies))
self.assertEqual(dict, type(parser.movies[0]))
self.assertEqual('Logan', parser.movies[0]['title'])
self.assertEqual('263115', parser.movies[0]['tmdb']['id'])
self.assertEqual('https://www.themoviedb.org/movie/263115',
parser.movies[0]['tmdb']['url'])
def test_parser(self, site_mock, base_init_mock, browser_mock,
progress_print_mock): # pylint: disable=too-many-arguments
browser_mock.page_source = self.my_ratings
parser = TMDBRatingsParser(None)
parser.args = False
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'TMDB'
parser.site.browser = browser_mock
parser.args = None
parser.parse()
self.assertEqual(50, len(parser.movies))
self.assertEqual(dict, type(parser.movies[0]))
self.assertEqual('Ghost in the Shell: The New Movie',
parser.movies[0]['title'])
self.assertEqual(2017, parser.movies[0]['year'])
self.assertEqual('334376', parser.movies[0]['tmdb']['id'])
self.assertEqual('https://www.themoviedb.org/movie/334376',
parser.movies[0]['tmdb']['url'])
self.assertEqual(6, parser.movies[0]['tmdb']['my_rating'])
def test_parser_single_movie(self, site_mock, base_init_mock,
browser_mock):
browser_mock.page_source = self.my_ratings
parser = TMDBRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'TMDB'
parser.site.browser = browser_mock
browser_mock.page_source = self.detail_page
movie = dict()
parser.parse_movie_details_page(movie)
self.assertEqual(8, movie['tmdb']['my_rating'])
self.assertEqual(2017, movie['year'])
def test_parser(
self, site_mock, base_init_mock, browser_mock, progress_print_mock
): # pylint: disable=too-many-arguments
browser_mock.page_source = self.my_ratings
parser = TMDBRatingsParser(None)
parser.args = False
parser.movies = []
parser.site = site_mock
parser.site.site_name = "TMDB"
parser.site.browser = browser_mock
parser.args = None
parser.parse()
self.assertEqual(50, len(parser.movies))
self.assertEqual(dict, type(parser.movies[0]))
self.assertEqual("Ghost in the Shell: The New Movie", parser.movies[0]["title"])
self.assertEqual(2017, parser.movies[0]["year"])
self.assertEqual("334376", parser.movies[0]["tmdb"]["id"])
self.assertEqual(
"https://www.themoviedb.org/movie/334376", parser.movies[0]["tmdb"]["url"]
)
self.assertEqual(6, parser.movies[0]["tmdb"]["my_rating"])
def test_init(self, browser_mock, base_init_mock):
TMDBRatingsParser(None)
self.assertTrue(base_init_mock.called)