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)