def test_csv_rename(self, site_mock, base_init_mock, browser_mock,
user_id_mock): # pylint: disable=too-many-arguments
parser = IMDBRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'IMDB'
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(
os.path.join(TESTDATA_PATH, 'exports'))
parser.csv_filename = '1234567890_imdb.csv'
parser.downloaded_file_name = os.path.join(os.pardir, 'imdb',
'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_csv_rename(self, site_mock, base_init_mock, browser_mock,
user_id_mock): # pylint: disable=too-many-arguments
parser = IMDBRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = "IMDB"
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(
os.path.join(TESTDATA_PATH, "exports"))
parser.csv_filename = "1234567890_imdb.csv"
parser.downloaded_file_name = os.path.join(os.pardir, "imdb",
"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
parser = IMDBRatingsParser(None)
parser.movies = []
parser.site = site_mock
parser.site.site_name = 'IMDB'
parser.site.browser = browser_mock
parser.exports_folder = os.path.abspath(os.path.join(TESTDATA_PATH, 'exports'))
parser.csv_filename = '1234567890_imdb.csv'
parser.downloaded_file_name = os.path.join(os.pardir, 'imdb', 'my_ratings.csv')
parser.parse()
self.assertEqual(1, rename_csv_mock.call_count)
self.assertEqual(1, parse_csv_mock.call_count)
def test_init(self, browser_mock, base_init_mock, user_id_mock):
IMDBRatingsParser(None)
self.assertTrue(base_init_mock.called)