def test_table_layout_when_no_location_type_or_location_yet(self): LocationType.objects.all().delete() LocationTypeDetails.objects.all().delete() Location.objects.all().delete() ea_csv_layout = UploadEACSVLayoutHelper() self.assertEqual( [["No Location/LocationType added yet. Please add those first."]], ea_csv_layout.table_layout_example())
class EAUploadCSVLayoutHelperTest(BaseTest): def setUp(self): country = LocationType.objects.create(name='Country', slug='country') uganda = Location.objects.create(name="Uganda", type=country) LocationTypeDetails.objects.create(country=uganda, location_type=country) region_type = LocationType.objects.create(name="Regiontype", slug="regiontype") district_type = LocationType.objects.create(name="Districttype", slug='districttype') county_type = LocationType.objects.create(name="Countytype", slug='countytype') parish_type = LocationType.objects.create(name="Parishtype", slug='parishtype') region = Location.objects.create(name="region1", type=region_type, tree_parent=uganda) district = Location.objects.create(name="district1", tree_parent=region, type=district_type) county_1 = Location.objects.create(name="county1", tree_parent=district, type=county_type) parish_1 = Location.objects.create(name="parish_1", tree_parent=county_1, type=parish_type) parish_1_b = Location.objects.create(name="parish_1b", tree_parent=county_1, type=parish_type) region = Location.objects.create(name="region2", tree_parent=uganda, type=region_type) district = Location.objects.create(name="district2", tree_parent=region, type=district_type) county_2 = Location.objects.create(name="county2", tree_parent=district, type=county_type) parish_2 = Location.objects.create(name="parish_2", tree_parent=county_2, type=parish_type) self.ea_csv_layout = UploadEACSVLayoutHelper() def test_headers_format(self): expected_headers = ['Regiontype', 'Districttype', 'Countytype', 'EA', 'Parishtype', 'EA'] [self.assertIn(header, expected_headers) for header in self.ea_csv_layout._header_format()] def test_table_layout_example(self): expected = [ ['regiontype_0', 'districttype_0', 'countytype_0', 'ea_0', 'parishtype_0', ''], ['regiontype_0', 'districttype_0', 'countytype_0', 'ea_0', 'parishtype_0_b', ''], ['regiontype_1', 'districttype_1', 'countytype_1', '', 'parishtype_1', 'ea_1'], ['regiontype_1', 'districttype_1', 'countytype_1', '', 'parishtype_1', 'ea_1_b']] self.assertEqual(expected, self.ea_csv_layout.table_layout_example()) def test_table_layout_when_no_location_type_or_location_yet(self): LocationType.objects.all().delete() LocationTypeDetails.objects.all().delete() Location.objects.all().delete() ea_csv_layout = UploadEACSVLayoutHelper() self.assertEqual([["No Location/LocationType added yet. Please add those first."]], ea_csv_layout.table_layout_example())
def test_table_layout_when_no_location_type_or_location_yet(self): LocationType.objects.all().delete() LocationTypeDetails.objects.all().delete() Location.objects.all().delete() ea_csv_layout = UploadEACSVLayoutHelper() self.assertEqual([["No Location/LocationType added yet. Please add those first."]], ea_csv_layout.table_layout_example())