1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
import json
from django.test import TestCase
try:
from django.urls import reverse
except ImportError:
from django.core.urlresolvers import reverse
from django.core.exceptions import ImproperlyConfigured
from rest_framework_gis import serializers as gis_serializers
from .models import BoxedLocation, Location
class TestRestFrameworkGisBBox(TestCase):
"""
unit tests for bbox support in restframework_gis
"""
def setUp(self):
self.geojson_boxedlocation_list_url = reverse('api_geojson_boxedlocation_list')
self.geojson_location_bbox_list_url = reverse('api_geojson_location_bbox_list')
def _create_locations(self):
self.bl1 = BoxedLocation.objects.create(
id=1,
name='l1',
slug='l1',
geometry='POINT (13.007 42.423)',
bbox_geometry='POLYGON((12.997 42.413,12.997 42.433,13.017 42.433,13.017 42.413,12.997 42.413))',
)
self.bl2 = BoxedLocation.objects.create(
id=2,
name='l2',
slug='l2',
geometry='POINT (12.007 43.423)',
bbox_geometry='POLYGON((11.997 43.413,11.997 43.433,12.017 43.433,12.017 43.413,11.997 43.413))',
)
self.l1 = Location.objects.create(
id=1,
name='l1',
slug='l1',
geometry='POLYGON((12.997 42.413,12.997 42.433,13.017 42.433,13.017 42.413,12.997 42.413))',
)
def test_list(self):
self._create_locations()
response = self.client.get(self.geojson_boxedlocation_list_url)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data['features']), 2)
for feature in response.data['features']:
self.assertIn('bbox', feature)
fid = feature['id']
if fid == 1:
self.assertEqual(feature['bbox'], self.bl1.bbox_geometry.extent)
elif fid == 2:
self.assertEqual(feature['bbox'], self.bl2.bbox_geometry.extent)
else:
self.fail("Unexpected id: {0}".format(fid))
BoxedLocation.objects.all().delete()
def test_post_location_list_geojson(self):
self.assertEqual(BoxedLocation.objects.count(), 0)
data = {
"properties": {"name": "geojson input test"},
"geometry": {"type": "Point", "coordinates": [12.49, 41.89]},
"bbox": [11.0, 40.0, 13.0, 42.0],
}
response = self.client.post(
self.geojson_boxedlocation_list_url,
data=json.dumps(data),
content_type='application/json',
)
self.assertEqual(response.status_code, 201)
self.assertEqual(BoxedLocation.objects.count(), 1)
self.assertEqual(
BoxedLocation.objects.all()[0].bbox_geometry.extent,
(11.0, 40.0, 13.0, 42.0),
)
def test_get_autogenerated_location_bbox_geojson(self):
self._create_locations()
response = self.client.get(self.geojson_location_bbox_list_url)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data['features']), 1)
self.assertEqual(response.data['features'][0]['bbox'], self.l1.geometry.extent)
def test_bbox_improperly_configured(self):
self._create_locations()
class LocationGeoFeatureSerializer(gis_serializers.GeoFeatureModelSerializer):
class Meta:
model = Location
geo_field = 'geometry'
bbox_geo_field = 'geometry'
auto_bbox = True
with self.assertRaises(ImproperlyConfigured):
LocationGeoFeatureSerializer(instance=self.l1)
|
