def create(self, ident):
"""Create a AreaConfiguration for the Area with the given ident."""
try:
area = Area.objects.get(ident=ident)
except Area.DoesNotExist:
logger.warning("We cannot create a WaterBalanceAreaConfiguration for the non-existing Area with ident '%s'." % ident)
return None
area_config = AreaConfiguration(ident=area.ident,
name=area.name,
area=area,
data_set=area.data_set)
area_config.save()
return area_config
def create_areaconfiguration(self, area):
try:
self.area_configuration = AreaConfiguration(
ident=area.ident,
name=area.name,
area=area)
self.area_configuration.save()
return True
except:
return False
class StructureTest(TestCase):
def setUp(self):
self.area = None
self.area_configuration = None
self.create_area()
self.create_areaconfiguration(self.area)
self.create_structure_in_out()
def create_area(self):
user = User(username='******', password='******')
user.save()
geo_object_group = self.get_or_create_geoobjectgroup(
user.username)
self.area = Area(ident="test", name="test",
geo_object_group=geo_object_group,
geometry=GEOSGeometry(Point(0, 0), srid=4326),
data_administrator_id=1)
self.area.save()
def create_structure_in_out(self):
"""Add in/out objects into StructureInOut."""
StructureInOut(code='in', index=1, description="In default").save()
StructureInOut(code='uit', index=0, description="Uit default").save()
def create_areaconfiguration(self, area):
try:
self.area_configuration = AreaConfiguration(
ident=area.ident,
name=area.name,
area=area)
self.area_configuration.save()
return True
except:
return False
def test_create_structures(self):
"""Test creating of 10 structures."""
self.area_configuration.create_default_structures()
structures = Structure.objects.all()
self.assertEquals(len(structures), 10)
self.area_configuration.create_default_structures()
structures = Structure.objects.all()
self.assertEquals(len(structures), 10)
def get_or_create_geoobjectgroup(self, user_name):
from lizard_geo.models import GeoObjectGroup
user_obj = User.objects.get(username=user_name)
group_name = 'test'
group_slug = 'test'
geo_object_group, created = GeoObjectGroup.objects.get_or_create(
name=group_name, slug=group_slug, created_by=user_obj)
if created:
geo_object_group.source_log = 'test'
geo_object_group.save()
return geo_object_group
def tearDown(self):
"""Clear the Stuctures and AreaConfiguration."""
if self.area is not None:
Area.objects.all().delete
if self.area_configuration is not None:
AreaConfiguration.objects.all().delete()
