def test_find_movie_success(self, browser_mock, base_init_mock, site_mock):
site_mock.browser = browser_mock
browser_mock.page_source = self.search_result
inserter = IMDBRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = "IMDB"
inserter.failed_movies = []
result = inserter._find_movie(self.movie) # pylint: disable=protected-access
self.assertTrue(result)
def test_is_requested_movie_success(self, browser_mock, base_init_mock, site_mock):
site_mock.browser = browser_mock
inserter = IMDBRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = 'IMDB'
inserter.failed_movies = []
search_result_page = BeautifulSoup(self.search_result, 'html.parser')
search_result = search_result_page.find(class_='findList').find_all(class_='findResult')[0]
result = inserter._is_requested_movie(self.movie, search_result) # pylint: disable=protected-access
self.assertTrue(result)
def test_find_movie_fail(self, browser_mock, base_init_mock, site_mock):
site_mock.browser = browser_mock
browser_mock.page_source = self.search_result
inserter = IMDBRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = "IMDB"
inserter.failed_movies = []
movie2 = dict()
movie2["title"] = "The Matrix"
movie2["year"] = 1995
result = inserter._find_movie(movie2) # pylint: disable=protected-access
self.assertFalse(result)
def test_is_requested_movie_no_movie_with_that_year(self, browser_mock, base_init_mock, site_mock):
site_mock.browser = browser_mock
inserter = IMDBRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = 'IMDB'
inserter.failed_movies = []
search_result_page = BeautifulSoup(self.search_result, 'html.parser')
search_result = search_result_page.find(class_='findList').find_all(class_='findResult')[0]
movie2 = dict()
movie2['title'] = 'SomeMovie'
movie2['year'] = 1995
result = inserter._is_requested_movie(movie2, search_result) # pylint: disable=protected-access
self.assertFalse(result)
def test_is_requested_movie_fail(self, browser_mock, base_init_mock,
site_mock):
site_mock.browser = browser_mock
inserter = IMDBRatingsInserter(None)
inserter.site = site_mock
inserter.site.site_name = "IMDB"
inserter.failed_movies = []
search_result_page = BeautifulSoup(self.search_result, "html.parser")
search_result = search_result_page.find(class_="findList").find_all(
class_="findResult")[0]
movie2 = dict()
movie2["title"] = "Arrival"
movie2["year"] = 2006
result = inserter._is_requested_movie(movie2, search_result) # pylint: disable=protected-access
self.assertFalse(result)
def test_insert(self, browser_mock, base_init_mock, site_mock, progress_print_mock, action_mock): # pylint: disable=too-many-arguments
site_mock.browser = browser_mock
inserter = IMDBRatingsInserter(None)
inserter.args = False
inserter.site = site_mock
inserter.site.site_name = 'IMDB'
inserter.failed_movies = []
inserter.insert([self.movie], 'Trakt')
self.assertTrue(base_init_mock.called)
self.assertTrue(progress_print_mock.called)
def test_init(self, browser_mock, base_init_mock):
IMDBRatingsInserter(None)
self.assertTrue(base_init_mock.called)
