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