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)