def test_serializer_model_no_geofield(self):
self.serializer = ZipShapeSerializer()
response = HttpResponse()
with self.assertRaisesRegex(
ValueError,
"No geodjango geometry fields found in this model"):
self.serializer.serialize(Difficulty.objects.all(),
stream=response,
fields=['id', 'name'],
delete=False)
def setUp(self):
self.point1 = MushroomSpot.objects.create(serialized='SRID=%s;POINT(0 0)' % settings.SRID)
self.line1 = MushroomSpot.objects.create(serialized='SRID=%s;LINESTRING(0 0, 10 0)' % settings.SRID)
self.multipoint = MushroomSpot.objects.create(serialized='SRID=%s;MULTIPOINT((1 1), (2 2))' % settings.SRID)
MushroomSpot.geomfield = GeometryField(name='geom', srid=2154)
self.serializer = ZipShapeSerializer()
devnull = open(os.devnull, "wb")
self.serializer.serialize(MushroomSpot.objects.all(), stream=devnull,
fields=['id', 'name', 'number', 'size', 'boolean'], delete=False)
def test_serializer_model_geofield_do_not_exist(self):
app_settings['GEOM_FIELD_NAME'] = 'do_not_exist'
self.serializer = ZipShapeSerializer()
PathFactory.create()
response = HttpResponse()
with self.assertRaisesRegex(
ValueError,
"Geodjango geometry field not found with the name 'do_not_exist', "
"fields available are: 'geom_3d, geom, geom_cadastre'"):
self.serializer.serialize(Path.objects.all(),
stream=response,
fields=['id', 'name'],
delete=False)
app_settings['GEOM_FIELD_NAME'] = 'geom'
def test_serializer_model_geofield_multiple(self):
app_settings['GEOM_FIELD_NAME'] = None
self.serializer = ZipShapeSerializer()
PathFactory.create()
response = HttpResponse()
with self.assertRaisesRegex(
ValueError,
"More than one geodjango geometry field found, please specify which "
"to use by name using the 'geo_field' keyword. "
"Available fields are: 'geom_3d, geom, geom_cadastre'"):
self.serializer.serialize(Path.objects.all(),
stream=response,
fields=['id', 'name'],
delete=False)
app_settings['GEOM_FIELD_NAME'] = 'geom'
def setUp(self):
self.point1 = MushroomSpot.objects.create(
serialized='SRID=%s;POINT(0 0)' % settings.SRID)
self.point1.tags.add(Tag.objects.create(label="Tag1"))
self.point1.tags.add(Tag.objects.create(label="Tag2"))
self.line1 = MushroomSpot.objects.create(
serialized='SRID=%s;LINESTRING(0 0, 10 0)' % settings.SRID)
self.multipoint = MushroomSpot.objects.create(
serialized='SRID=%s;MULTIPOINT((1 1), (2 2))' % settings.SRID)
MushroomSpot.geomfield = GeometryField(name='geom', srid=settings.SRID)
self.serializer = ZipShapeSerializer()
response = HttpResponse()
self.serializer.serialize(
MushroomSpot.objects.all(),
stream=response,
fields=['id', 'name', 'number', 'size', 'boolean', 'tags'],
delete=False)
def setUp(self):
self.point1 = DiveFactory.create()
self.point1.themes.add(ThemeFactory.create(label="Tag1"))
self.point1.themes.add(ThemeFactory.create(label="Tag2"))
self.line1 = DiveFactory.create(geom='SRID=%s;LINESTRING(0 0, 10 0)' %
settings.SRID)
self.multiline = DiveFactory.create(
geom='SRID=%s;MULTILINESTRING((10 10, 20 20, 10 40),'
'(40 40, 30 30, 40 20, 30 10))' % settings.SRID)
self.multipoint = DiveFactory.create(
geom='SRID=%s;MULTIPOINT((1 1), (2 2))' % settings.SRID)
self.polygon = DiveFactory.create(
geom='SRID=%s;POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))' % settings.SRID)
self.multipolygon = DiveFactory.create(
geom='SRID=%s;MULTIPOLYGON(((0 0, 0 1, 1 1, 1 0, 0 0)),'
'((2 2, 2 3, 3 3, 3 2, 2 2)))' % settings.SRID)
self.serializer = ZipShapeSerializer()
response = HttpResponse()
self.serializer.serialize(Dive.objects.all(),
stream=response,
fields=['id', 'name'],
delete=False)
def test_shape_mixed(self):
"""
Test that a project made of intervention of different geom create multiple files.
Check that those files are each of a different type (Point/LineString) and that
the project and the intervention are correctly referenced in it.
"""
# Create topology line
topo_line = TopologyFactory.create(no_path=True)
line = PathFactory.create(
geom=LineString(Point(10, 10, 0), Point(11, 10, 0)))
PathAggregationFactory.create(topo_object=topo_line, path=line)
# Create a topology point
lng, lat = tuple(
Point(1, 1, srid=settings.SRID).transform(settings.API_SRID,
clone=True))
closest_path = PathFactory(geom=LineString(
Point(0, 0, 0), Point(1, 0, 0), srid=settings.SRID))
topo_point = TopologyHelper._topologypoint(lng, lat, None).reload()
self.assertEquals(topo_point.paths.get(), closest_path)
# Create one intervention by geometry (point/linestring)
it_point = InterventionFactory.create(topology=topo_point)
it_line = InterventionFactory.create(topology=topo_line)
# reload
it_point = type(it_point).objects.get(pk=it_point.pk)
it_line = type(it_line).objects.get(pk=it_line.pk)
proj = ProjectFactory.create()
proj.interventions.add(it_point)
proj.interventions.add(it_line)
# instanciate the class based view 'abnormally' to use create_shape directly
# to avoid making http request, authent and reading from a zip
pfl = ZipShapeSerializer()
devnull = open(os.devnull, "wb")
pfl.serialize(Project.objects.all(),
stream=devnull,
delete=False,
fields=ProjectFormatList.columns)
self.assertEquals(len(pfl.layers), 2)
ds_point = gdal.DataSource(pfl.layers.values()[0])
layer_point = ds_point[0]
ds_line = gdal.DataSource(pfl.layers.values()[1])
layer_line = ds_line[0]
self.assertEquals(layer_point.geom_type.name, 'MultiPoint')
self.assertEquals(layer_line.geom_type.name, 'LineString')
for layer in [layer_point, layer_line]:
self.assertEquals(layer.srs.name, 'RGF93_Lambert_93')
self.assertItemsEqual(layer.fields,
['domain', 'name', 'type', 'period', 'id'])
self.assertEquals(len(layer_point), 1)
self.assertEquals(len(layer_line), 1)
for feature in layer_point:
self.assertEquals(str(feature['id']), str(proj.pk))
self.assertTrue(feature.geom.geos.equals(it_point.geom))
for feature in layer_line:
self.assertEquals(str(feature['id']), str(proj.pk))
self.assertTrue(feature.geom.geos.equals(it_line.geom))
# Clean-up temporary shapefiles
for layer_file in pfl.layers.values():
for subfile in shapefile_files(layer_file):
os.remove(subfile)
