def test__detect_type(self):
fh = Factory.people_mailing_list_spreadsheet_file(self.a1, file_type=CSV_TYPE)
s = SpreadsheetAbstraction(self.a1, fh, "people", filename="test.foo")
self.assertEqual(s.type, CSV_TYPE)
self.assertEqual(s.is_valid,True)
fh = Factory.people_mailing_list_spreadsheet_file(self.a1, file_type=EXCEL_TYPE)
s = SpreadsheetAbstraction(self.a1, fh, "people", filename="test.bar")
self.assertEqual(s.type, EXCEL_TYPE)
self.assertEqual(s.is_valid,True)
def created_and_imported_excel_spreadsheet(self, **kwargs):
fh = Factory.people_mailing_list_spreadsheet_file(self.a1, file_type=EXCEL_TYPE, **kwargs)
# import it
fh.seek(0)
s = SpreadsheetAbstraction(self.a1, fh, "people", filename="test.xls")
self.assertEqual(s.is_valid,True)
return s
def _create_a_person_spreadsheet_and_clear_the_data(self, file_type=CSV_TYPE, extension=None,):
if not extension:
if file_type == CSV_TYPE:
extension = "csv"
elif file_type == EXCEL_TYPE:
extension = "xls"
# create a person spreadsheet
fh = Factory.people_mailing_list_spreadsheet_file(self.a1, file_type=file_type)
# delete all the people
Person.objects_by_account(self.a1).all().delete()
# assert that they're gone
self.assertEqual(Person.objects_by_account(self.a1).count(), 0)
# import the spreadsheet
s = SpreadsheetAbstraction(self.a1, fh, "people", filename="test.%s" % (extension,))
return s
