udata 10.0.9.dev33939__py2.py3-none-any.whl → 10.0.9.dev33959__py2.py3-none-any.whl

udata/core/spatial/api.py

@@ -54,7 +54,9 @@ class SuggestZonesAPI(API):
     def get(self):
         """Geospatial zones suggest endpoint using mongoDB contains"""
         args = suggest_parser.parse_args()
-        geozones = GeoZone.objects(Q(name__icontains=args["q"]) | Q(code__icontains=args["q"]))
+        geozones = GeoZone.objects(
+            Q(name__icontains=args["q"]) | Q(code__icontains=args["q"]) | Q(id__icontains=args["q"])
+        )
 
         # We're manually sorting based on zone level int (cause we don't have the int value directly in mongo document)
         level_id_to_int_level = {level.id: level.admin_level for level in GeoLevel.objects()}

udata/core/spatial/tests/test_api.py

@@ -75,6 +75,17 @@ class SpatialApiTest(APITestCase):
             self.assertIn("level", suggestion)
             self.assertIn("name-test", suggestion["name"])
 
+    def test_suggest_zones_on_id(self):
+        """It should suggest zones based on its id"""
+        zone = GeoZoneFactory()
+        for _ in range(2):
+            GeoZoneFactory()
+
+        response = self.get(url_for("api.suggest_zones"), qs={"q": zone.id})
+        self.assert200(response)
+
+        self.assertEqual(response.json[0]["id"], zone["id"])
+
     def test_suggest_zones_sorted(self):
         """It should suggest zones based on its name"""
         country_level = GeoLevelFactory(id="country", name="country", admin_level=10)