def setUp(self):
self.user_details = {
'username': '******',
'email': '*****@*****.**',
'password': '******'
}
self.user = User.objects.create_user(**self.user_details)
self.login_url = reverse('landing:login')
self.org = Organization.objects.create()
OrganizationUser.objects.create(user=self.user, organization=self.org)
self.import_record = ImportRecord.objects.create(owner=self.user)
self.import_record.super_organization = self.org
self.import_record.save()
self.import_file = ImportFile.objects.create(
import_record=self.import_record)
self.import_file.source_type = 'ASSESSED_RAW'
self.import_file.file = File(
open(
path.join(path.dirname(__file__),
'../data/covered-buildings-sample-with-errors.csv')))
self.import_file.save()
self.import_file_mapping = path.join(
path.dirname(__file__),
"../data/covered-buildings-sample-with-errors-mappings.csv")
tasks.save_raw_data(self.import_file.id)
Column.create_mappings_from_file(self.import_file_mapping, self.org,
self.user, self.import_file.id)
tasks.map_data(self.import_file.id)
def test_save_column_mapping_by_file(self):
self.mapping_import_file = os.path.abspath("./seed/tests/data/test_mapping.csv")
Column.create_mappings_from_file(self.mapping_import_file, self.fake_org, self.fake_user)
expected = {
u'City': (u'PropertyState', u'city'),
u'Custom ID': (u'PropertyState', u'custom_id_1'),
u'Zip': (u'PropertyState', u'postal_code'),
u'GBA': (u'PropertyState', u'gross_floor_area'),
u'PM Property ID': (u'PropertyState', u'pm_property_id'),
u'BLDGS': (u'PropertyState', u'building_count'),
u'AYB_YearBuilt': (u'PropertyState', u'year_build'),
u'State': (u'PropertyState', u'state'),
u'Address': (u'PropertyState', u'address_line_1'),
u'Owner': (u'PropertyState', u'owner'),
u'Raw Column': (u'Table Name', u'Field Name'),
u'Property Type': (u'PropertyState', u'property_type'),
u'UBI': (u'TaxLotState', u'jurisdiction_tax_lot_id')
}
test_mapping, _ = ColumnMapping.get_column_mappings(self.fake_org)
self.assertItemsEqual(expected, test_mapping)
def test_save_column_mapping_by_file_exception(self):
self.mapping_import_file = os.path.abspath("./no-file.csv")
with self.assertRaisesRegexp(Exception, "Mapping file does not exist: .*/seed/no-file.csv"):
Column.create_mappings_from_file(self.mapping_import_file, self.fake_org, self.fake_user)
