def test_find_movie_success(self, browser_mock, base_init_mock, site_mock):
site_mock.browser = browser_mock
browser_mock.page_source = self.search_results
inserter = AlloCineRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = 'AlloCine'
inserter.failed_movies = []
movie2 = dict()
movie2['title'] = 'Fight Club'
movie2['year'] = 1999
result = inserter._find_movie(movie2) # pylint: disable=protected-access
self.assertTrue(result)
def test_find_movie_fail(self, browser_mock, base_init_mock, site_mock,
tiles_mock, equality_mock): # pylint: disable=too-many-arguments
site_mock.browser = browser_mock
browser_mock.page_source = self.search_results
inserter = AlloCineRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = 'AlloCine'
inserter.failed_movies = []
tiles_mock.return_value = self.search_result_tile_list
equality_mock.return_value = False
movie2 = dict()
movie2['title'] = 'The Matrix'
movie2['year'] = 1995
result = inserter._find_movie(movie2) # pylint: disable=protected-access
self.assertFalse(result)
def test_insert(
self,
browser_mock,
base_init_mock,
site_mock,
overview_page_mock, # pylint: disable=too-many-arguments
eq_check_mock,
progress_print_mock):
overview_page_mock.return_value = self.search_result_tile_list
eq_check_mock.return_value = True
site_mock.browser = browser_mock
inserter = AlloCineRatingsInserter(None)
inserter.args = False
inserter.site = site_mock
inserter.site.site_name = 'AlloCine'
inserter.failed_movies = []
inserter.insert([self.movie], 'IMDB')
self.assertTrue(base_init_mock.called)
self.assertTrue(progress_print_mock.called)
def test_init(self, browser_mock, base_init_mock):
AlloCineRatingsInserter(None)
self.assertTrue(base_init_mock.called)