def test_import_with_model_instance(self):
file_mock = mock.MagicMock(spec=FileHistory, name='FileHistoryMock')
file_mock.file_upload = '/media/test.csv'
importer = GenericImporter(source=file_mock)
self.assertEqual(importer.get_source_file_extension(), 'csv')
def test_unsuported_raise_error_message(self):
with self.assertRaisesMessage(UnsuportedFile, 'Unsuported File'):
GenericImporter(source=self.unsuported_file)
def test_import_with_file_instance(self):
file_instance = open(self.csv_file)
importer = GenericImporter(source=file_instance)
self.assertEqual(importer.get_source_file_extension(), 'csv')
def test_xml_reader_set(self):
importer = GenericImporter(source=self.xml_file)
self.assertEqual(importer.get_reader_class(), XMLReader)
def test_getting_source_file_extension(self):
importer = GenericImporter(source=self.csv_file)
self.assertEqual(importer.get_source_file_extension(), 'csv')
def test_import_with_model_instance(self):
file_mock = mock.MagicMock(spec=FileHistory, name='FileHistoryMock')
file_mock.file_upload = '/media/test.csv'
importer = GenericImporter(source=file_mock)
self.assertEquals(importer.get_source_file_extension(), 'csv')
def test_csv_reader_set(self):
importer = GenericImporter(source=self.csv_file)
self.assertEqual(importer.get_reader_class(), CSVReader)
def test_import_with_file_instance(self):
file_instance = open(self.csv_file)
importer = GenericImporter(source=file_instance)
self.assertEquals(importer.get_source_file_extension(), 'csv')
def test_getting_source_file_extension(self):
importer = GenericImporter(source=self.csv_file)
self.assertEquals(importer.get_source_file_extension(), 'csv')
def test_xml_reader_set(self):
importer = GenericImporter(source=self.xml_file)
self.assertEquals(importer.get_reader_class(), XMLReader)
def test_csv_reader_set(self):
importer = GenericImporter(source=self.csv_file)
self.assertEquals(importer.get_reader_class(), CSVReader)
def test_xlsx_reader_set(self):
importer = GenericImporter(source=self.xlsx_file)
self.assertEquals(importer.get_reader_class(), XLSXReader)
