Пример #1
0
    def test_target_movie_detail_page_has_no_year(
        self, browser_mock, base_init_mock, site_mock
    ):  # pylint: disable=too-many-arguments
        with open(
            os.path.join(
                TESTDATA_PATH, "listal", "movie_details_page_without_release_year.html"
            ),
            encoding="UTF-8",
        ) as movie_details_page_without_release_year:
            movie_details_page_without_release_year = (
                movie_details_page_without_release_year.read()
            )

        site_mock.browser = browser_mock
        browser_mock.page_source = movie_details_page_without_release_year
        inserter = ListalRatingsInserter(None)
        inserter.site = site_mock
        inserter.site.site_name = "Listal"
        inserter.failed_movies = []
        inserter.args = None

        movie2 = dict()
        movie2["title"] = "Fight Club"
        movie2["year"] = 1999

        search_result = BeautifulSoup(self.search_result_tile_list[0], "html.parser")
        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, overview_page_mock,  # pylint: disable=too-many-arguments
                    eq_check_mock, progress_print_mock, post_rating_mock):
        overview_page_mock.return_value = self.search_result_tile_list
        eq_check_mock.return_value = True
        site_mock.browser = browser_mock
        inserter = ListalRatingsInserter(None)
        inserter.args = False
        inserter.site = site_mock
        inserter.site.site_name = 'Listal'
        inserter.failed_movies = []

        inserter.insert([self.movie], 'IMDB')

        self.assertTrue(base_init_mock.called)
        self.assertTrue(progress_print_mock.called)