def test_should_return_the_country_in_use(self):
self.assertIsNone(LocationTypeDetails.the_country())
uganda = Location.objects.create(name="Uganda")
a_type = LocationType.objects.create(name="country", slug="country")
LocationTypeDetails.objects.create(location_type=a_type, country=uganda)
self.assertEqual(uganda, LocationTypeDetails.the_country())
def test_should_return_the_country_in_use(self):
self.assertIsNone(LocationTypeDetails.the_country())
uganda = Location.objects.create(name="Uganda")
a_type = LocationType.objects.create(name="country", slug="country")
LocationTypeDetails.objects.create(location_type=a_type,
country=uganda)
self.assertEqual(uganda, LocationTypeDetails.the_country())
def test_should_save_with_auto_incremented_order(self):
country = LocationType.objects.create(name='country', slug='country')
country_details = LocationTypeDetails.objects.create(required=True, has_code=False, location_type=country)
self.assertEqual(1, country_details.order)
district = LocationType.objects.create(name='district', slug='district')
data = {
'required': True,
'has_code': False,
'location_type': district,
}
location_type_details = LocationTypeDetails(**data)
location_type_details.save()
self.assertEqual(2, location_type_details.order)
def test_should_save_with_auto_incremented_order(self):
country = LocationType.objects.create(name='country', slug='country')
country_details = LocationTypeDetails.objects.create(
required=True, has_code=False, location_type=country)
self.assertEqual(1, country_details.order)
district = LocationType.objects.create(name='district',
slug='district')
data = {
'required': True,
'has_code': False,
'location_type': district,
}
location_type_details = LocationTypeDetails(**data)
location_type_details.save()
self.assertEqual(2, location_type_details.order)
def test_fields(self):
location_type_details = LocationTypeDetails()
fields = [
str(item.attname) for item in location_type_details._meta.fields
]
self.assertEqual(9, len(fields))
for field in [
'id', 'created', 'modified', 'location_type_id', 'required',
'has_code', 'length_of_code', 'country_id', 'order'
]:
self.assertIn(field, fields)
def test_should_return_location_type_objects_ordered_by_order(self):
country = LocationType.objects.create(name='country', slug='country')
location_type_details = LocationTypeDetails.objects.create(required=True, has_code=False, location_type=country)
district = LocationType.objects.create(name='district', slug='district')
location_type_details_1 = LocationTypeDetails.objects.create(required=True, has_code=False,
location_type=district)
county = LocationType.objects.create(name='county', slug='county')
location_type_details_1 = LocationTypeDetails.objects.create(required=True, has_code=False,
location_type=county)
ordered_types = LocationTypeDetails.get_ordered_types()
self.assertEqual(country, ordered_types[0])
self.assertEqual(district, ordered_types[1])
self.assertEqual(county, ordered_types[2])
def test_should_return_location_type_objects_ordered_by_order(self):
country = LocationType.objects.create(name='country', slug='country')
location_type_details = LocationTypeDetails.objects.create(
required=True, has_code=False, location_type=country)
district = LocationType.objects.create(name='district',
slug='district')
location_type_details_1 = LocationTypeDetails.objects.create(
required=True, has_code=False, location_type=district)
county = LocationType.objects.create(name='county', slug='county')
location_type_details_1 = LocationTypeDetails.objects.create(
required=True, has_code=False, location_type=county)
ordered_types = LocationTypeDetails.get_ordered_types()
self.assertEqual(country, ordered_types[0])
self.assertEqual(district, ordered_types[1])
self.assertEqual(county, ordered_types[2])
def test_should_return_location_type_objects_ordered_by_order(self):
country = LocationType.objects.create(name='country', slug='country')
uganda = Location.objects.create(name="Uganda", type=country)
location_type_details = LocationTypeDetails.objects.create(
required=True,
has_code=False,
length_of_code=6,
location_type=country,
country=uganda,
order=1)
district = LocationType.objects.create(name='district',
slug='district')
kampala = Location.objects.create(name="Kampala",
type=district,
parent=uganda)
location_type_details_1 = LocationTypeDetails.objects.create(
required=True,
has_code=False,
length_of_code=6,
location_type=district,
country=uganda,
order=2)
county = LocationType.objects.create(name='county', slug='county')
some_county = Location.objects.create(name="County",
type=county,
parent=kampala)
location_type_details_1 = LocationTypeDetails.objects.create(
required=True,
has_code=False,
length_of_code=6,
location_type=county,
country=uganda,
order=3)
ordered_types = LocationTypeDetails.get_ordered_types()
self.assertEqual(country, ordered_types[0])
self.assertEqual(district, ordered_types[1])
self.assertEqual(county, ordered_types[2])
