def test_csv_rename(self, site_mock, base_init_mock, browser_mock): # pylint: disable=too-many-arguments
parser = LetterboxdRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'Letterboxd'
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(
os.path.join(TESTDATA_PATH, 'exports'))
parser.csv_filename = '1234567890_letterboxd.csv'
parser.downloaded_file_name = os.path.join(os.pardir, 'letterboxd',
'my_ratings.csv')
self.assertFalse(
os.path.isfile(
os.path.join(TESTDATA_PATH, 'exports', 'ratings.csv')))
with open(os.path.join(TESTDATA_PATH, 'exports', 'ratings.csv'), 'w+'):
self.assertTrue(
os.path.isfile(
os.path.join(TESTDATA_PATH, 'exports', 'ratings.csv')))
self.assertFalse(
os.path.isfile(
os.path.join(TESTDATA_PATH, 'exports', parser.csv_filename)))
parser._rename_csv_file('ratings.csv') # pylint: disable=protected-access
self.assertFalse(
os.path.isfile(
os.path.join(TESTDATA_PATH, 'exports', 'ratings.csv')))
self.assertTrue(
os.path.isfile(
os.path.join(TESTDATA_PATH, 'exports', parser.csv_filename)))
os.remove(os.path.join(TESTDATA_PATH, 'exports', parser.csv_filename))
def test_parser(
self,
site_mock,
base_init_mock,
browser_mock,
rename_csv_mock,
parse_csv_mock, # pylint: disable=too-many-arguments
parsed_filename_mock,
zip_extraction_mock):
parser = LetterboxdRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'Letterboxd'
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(
os.path.join(TESTDATA_PATH, 'exports'))
parser.csv_filename = '1234567890_letterboxd.csv'
parser.downloaded_file_name = os.path.join(os.pardir, 'letterboxd',
'my_ratings.csv')
parsed_filename_mock.return_value = ['letterboxd.zip']
parser.parse()
self.assertEqual(1, zip_extraction_mock.call_count)
self.assertEqual(1, rename_csv_mock.call_count)
self.assertEqual(1, parse_csv_mock.call_count)
def test_parse_movies_from_csv(self, site_mock, base_init_mock, browser_mock):
parser = LetterboxdRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'Letterboxd'
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(os.path.join(TESTDATA_PATH, 'exports'))
parser.csv_filename = '1234567890_letterboxd.csv'
parser.downloaded_file_name = os.path.join(os.pardir, 'letterboxd', 'my_ratings.csv')
movies = parser._parse_movies_from_csv(os.path.join(TESTDATA_PATH, 'letterboxd', 'my_ratings.csv')) # pylint: disable=protected-access
self.assertEqual(1056, len(movies))
self.assertEqual(dict, type(movies[0]))
self.assertEqual('Life', movies[0]['title'])
self.assertEqual(2017, movies[0]['year'])
self.assertEqual('https://letterboxd.com/film/life-2017/', movies[0]['letterboxd']['url'])
self.assertEqual(7, movies[0]['letterboxd']['my_rating'])
def test_parse_movies_from_csv(self, site_mock, base_init_mock,
browser_mock):
parser = LetterboxdRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = "Letterboxd"
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(
os.path.join(TESTDATA_PATH, "exports"))
parser.csv_filename = "1234567890_letterboxd.csv"
parser.downloaded_file_name = os.path.join(os.pardir, "letterboxd",
"my_ratings.csv")
movies = parser._parse_movies_from_csv(
os.path.join(TESTDATA_PATH, "letterboxd", "my_ratings.csv")) # pylint: disable=protected-access
self.assertEqual(1056, len(movies))
self.assertEqual(dict, type(movies[0]))
self.assertEqual("Life", movies[0]["title"])
self.assertEqual(2017, movies[0]["year"])
self.assertEqual("https://letterboxd.com/film/life-2017/",
movies[0]["letterboxd"]["url"])
self.assertEqual(7, movies[0]["letterboxd"]["my_rating"])
def test_init(self, browser_mock, base_init_mock):
LetterboxdRatingsParser(None)
self.assertTrue(base_init_mock.called)