def setUp(self):
super(GeoTestCase, self).setUp()
# Start with a clean database for each test.
Country.objects.all().delete()
SubRegion.objects.all().delete()
Region.objects.all().delete()
# A test Region that is needed by the test SubRegion.
region = Region()
region.name = "Test Region"
region.numeric_code = "001"
region.save()
# A test SubRegion that is needed by the test Country.
subregion = SubRegion()
subregion.name = "Test SubRegion"
subregion.numeric_code = "002"
subregion.region = region
subregion.save()
# Setup the test Country.
self.country = Country()
self.country.name = "Test"
self.country.subregion = subregion
self.country.save()
def create_country(self, name, numeric_code, sub_region=None, **kwargs):
if sub_region is None:
try:
sub_region = SubRegion.objects.get(name='test-subregion',
numeric_code='100')
except SubRegion.DoesNotExist:
sub_region = self.create_subregion(name='test-subregion',
numeric_code='100')
country = Country(name=name,
numeric_code=numeric_code,
subregion=sub_region)
for k, v in list(kwargs.items()):
setattr(country, k, v)
country.save()
return country
