def setUp(self):
self.f = SnappedLineStringField()
self.wktgeom = ('LINESTRING(-0.77054223313507 -5.32573853776343,'
'-0.168053647782867 -4.66595028627023)')
self.geojson = ('{"type":"LineString","coordinates":['
' [-0.77054223313507,-5.32573853776343],'
' [-0.168053647782867,-4.66595028627023]]}')
class FieldsTest(TestCase):
def setUp(self):
self.f = SnappedLineStringField()
self.wktgeom = ('LINESTRING(-0.77054223313507 -5.32573853776343,'
'-0.168053647782867 -4.66595028627023)')
self.geojson = ('{"type":"LineString","coordinates":['
' [-0.77054223313507,-5.32573853776343],'
' [-0.168053647782867,-4.66595028627023]]}')
def test_dict_with_geom_is_mandatory(self):
self.assertRaises(ValidationError, self.f.clean,
('LINESTRING(0 0, 1 0)', ))
self.assertRaises(ValidationError, self.f.clean,
('{"geo": "LINESTRING(0 0, 1 0)"}', ))
def test_snaplist_is_mandatory(self):
self.assertRaises(ValidationError, self.f.clean,
('{"geom": "LINESTRING(0 0, 1 0)"}', ))
def test_snaplist_must_have_same_number_of_vertices(self):
self.assertRaises(
ValidationError, self.f.clean,
('{"geom": "LINESTRING(0 0, 1 0)", "snap": [null]}', ))
def test_geom_cannot_be_invalid_wkt(self):
self.assertRaises(ValidationError, self.f.clean,
('{"geom": "LINEPPRING(0 0, 1 0)", '
'"snap": [null, null]}', ))
def test_geom_can_be_geojson(self):
geojsonstr = self.geojson.replace('"', '\\"')
geom = self.f.clean('{"geom": "%s", '
' "snap": [null, null]}' % geojsonstr)
self.assertTrue(
geom.equals_exact(
LineString((100000, 100000), (200000, 200000),
srid=settings.SRID), 0.1))
def test_geom_is_not_snapped_if_snap_is_null(self):
value = '{"geom": "%s", "snap": [null, null]}' % self.wktgeom
self.assertTrue(
self.f.clean(value).equals_exact(
LineString((100000, 100000), (200000, 200000),
srid=settings.SRID), 0.1))
def test_geom_is_snapped_if_path_pk_is_provided(self):
path = PathFactory.create()
value = '{"geom": "%s", "snap": [null, %s]}' % (self.wktgeom, path.pk)
self.assertTrue(
self.f.clean(value).equals_exact(
LineString((100000, 100000), (2, 2), srid=settings.SRID), 0.1))
def test_snappedlinestring(self):
f = SnappedLineStringField()
self.assertRaises(ValidationError, f.clean, ('LINESTRING(0 0, 1 0)',))
self.assertRaises(ValidationError, f.clean, ('{"geom": "LINESTRING(0 0, 1 0)"}',))
self.assertRaises(ValidationError, f.clean, ('{"geom": "LINESTRING(0 0, 1 0)", "snap": [null]}',))
self.assertRaises(ValidationError, f.clean, ('{"geom": "LINEPPRING(0 0, 1 0)", "snap": [null, null]}',))
value = '{"geom": "LINESTRING(-0.77054223313507 -5.32573853776343,-0.168053647782867 -4.66595028627023)", "snap": [null, null]}'
self.assertTrue(f.clean(value).equals_exact(LineString((100000, 100000), (200000, 200000), srid=settings.SRID), 0.1))
path = PathFactory.create()
value = '{"geom": "LINESTRING(-0.77054223313507 -5.32573853776343,-0.168053647782867 -4.66595028627023)", "snap": [null, %s]}' % path.pk
self.assertTrue(f.clean(value).equals_exact(LineString((100000, 100000), (2, 2), srid=settings.SRID), 0.1))
class SnappedLineStringFieldTest(TestCase):
def setUp(self):
self.f = SnappedLineStringField()
self.wktgeom = ('LINESTRING(-0.77054223313507 -5.32573853776343,'
'-0.168053647782867 -4.66595028627023)')
self.geojson = ('{"type":"LineString","coordinates":['
' [-0.77054223313507,-5.32573853776343],'
' [-0.168053647782867,-4.66595028627023]]}')
def test_dict_with_geom_is_mandatory(self):
self.assertRaises(ValidationError, self.f.clean,
'LINESTRING(0 0, 1 0)')
self.assertRaises(ValidationError, self.f.clean,
'{"geo": "LINESTRING(0 0, 1 0)"}')
def test_snaplist_is_mandatory(self):
self.assertRaises(ValidationError, self.f.clean,
'{"geom": "LINESTRING(0 0, 1 0)"}')
def test_snaplist_must_have_same_number_of_vertices(self):
self.assertRaises(ValidationError, self.f.clean,
'{"geom": "LINESTRING(0 0, 1 0)", "snap": [null]}')
def test_geom_cannot_be_invalid_wkt(self):
self.assertRaises(ValidationError, self.f.clean,
'{"geom": "LINEPPRING(0 0, 1 0)", '
'"snap": [null, null]}')
def test_geom_can_be_geojson(self):
geojsonstr = self.geojson.replace('"', '\\"')
geom = self.f.clean('{"geom": "%s", '
' "snap": [null, null]}' % geojsonstr)
self.assertTrue(geom.equals_exact(
LineString((100000, 100000), (200000, 200000),
srid=settings.SRID), 0.1))
def test_geom_is_not_snapped_if_snap_is_null(self):
value = '{"geom": "%s", "snap": [null, null]}' % self.wktgeom
self.assertTrue(self.f.clean(value).equals_exact(
LineString((100000, 100000), (200000, 200000),
srid=settings.SRID), 0.1))
def test_geom_is_snapped_if_path_pk_is_provided(self):
path = PathFactory.create()
value = '{"geom": "%s", "snap": [null, %s]}' % (self.wktgeom, path.pk)
self.assertTrue(self.f.clean(value).equals_exact(
LineString((100000, 100000), (700000, 6600000),
srid=settings.SRID), 0.1))
def setUp(self):
self.f = SnappedLineStringField()
self.wktgeom = ('LINESTRING(-0.77054223313507 -5.32573853776343,'
'-0.168053647782867 -4.66595028627023)')
self.geojson = ('{"type":"LineString","coordinates":['
' [-0.77054223313507,-5.32573853776343],'
' [-0.168053647782867,-4.66595028627023]]}')
class PathForm(CommonForm):
geom = SnappedLineStringField()
reverse_geom = forms.BooleanField(required=False,
label=_("Reverse path"),
help_text=_("The path will be reversed once saved"))
geomfields = ['geom']
class Meta(CommonForm.Meta):
model = Path
fields = CommonForm.Meta.fields + \
['structure', 'name', 'stake', 'comfort', 'departure', 'arrival', 'comments',
'source', 'networks', 'usages', 'valid', 'draft', 'reverse_geom', 'geom']
def __init__(self, *args, **kwargs):
super(PathForm, self).__init__(*args, **kwargs)
if self.instance.pk:
if not self.instance.draft:
# Prevent to set a path as draft again (it could be used by a topology)
del self.fields['draft']
if not self.user.has_perm('core.change_path'):
del self.fields['draft']
else:
if not self.user.has_perm('core.add_draft_path') or not self.user.has_perm('core.add_path'):
del self.fields['draft']
self.fields['geom'].label = ''
def clean_geom(self):
pk = self.instance.pk if self.instance and self.instance.pk else -1
geom = self.cleaned_data['geom']
if geom is None:
raise forms.ValidationError(_("Invalid snapped geometry."))
if not geom.simple:
raise forms.ValidationError(_("Geometry is not simple."))
if not PathHelper.disjoint(geom, pk):
raise forms.ValidationError(_("Geometry overlaps another."))
return geom
def save(self, commit=True):
path = super(PathForm, self).save(commit=False)
if not self.instance.pk:
if self.user.has_perm('core.add_draft_path') and not self.user.has_perm('core.add_path'):
path.draft = True
if not self.user.has_perm('core.add_draft_path') and self.user.has_perm('core.add_path'):
path.draft = False
if self.cleaned_data.get('reverse_geom'):
path.reverse()
if commit:
path.save()
self.save_m2m()
return path
def test_snappedlinestring(self):
f = SnappedLineStringField()
self.assertRaises(ValidationError, f.clean, ('LINESTRING(0 0, 1 0)', ))
self.assertRaises(ValidationError, f.clean,
('{"geom": "LINESTRING(0 0, 1 0)"}', ))
self.assertRaises(
ValidationError, f.clean,
('{"geom": "LINESTRING(0 0, 1 0)", "snap": [null]}', ))
self.assertRaises(
ValidationError, f.clean,
('{"geom": "LINEPPRING(0 0, 1 0)", "snap": [null, null]}', ))
value = '{"geom": "LINESTRING(-0.77054223313507 -5.32573853776343,-0.168053647782867 -4.66595028627023)", "snap": [null, null]}'
self.assertTrue(
f.clean(value).equals_exact(
LineString((100000, 100000), (200000, 200000),
srid=settings.SRID), 0.1))
path = PathFactory.create()
value = '{"geom": "LINESTRING(-0.77054223313507 -5.32573853776343,-0.168053647782867 -4.66595028627023)", "snap": [null, %s]}' % path.pk
self.assertTrue(
f.clean(value).equals_exact(
LineString((100000, 100000), (2, 2), srid=settings.SRID), 0.1))
