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)