示例#1
0
    def test_insert(self, browser_mock, base_init_mock, site_mock, impex_mock):  # pylint: disable=too-many-arguments
        site_mock.browser = browser_mock
        inserter = MovielensRatingsInserter(None)
        inserter.site = site_mock
        inserter.site.site_name = 'Movielens'
        inserter.failed_movies = []
        inserter.exports_folder = TESTDATA_PATH
        inserter.csv_filename = 'converted.csv'

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

        self.assertTrue(base_init_mock.called)
        self.assertTrue(impex_mock.called)
示例#2
0
    def test_init(self, browser_mock, base_init_mock):
        MovielensRatingsInserter(None)

        self.assertTrue(base_init_mock.called)