Openstreetmap's unit test

searx/engines/openstreetmap.py

@@ -38,6 +38,9 @@ def response(resp):
 
     # parse results
     for r in json:
+        if 'display_name' not in r:
+            continue
+
         title = r['display_name']
         osm_type = r.get('osm_type', r.get('type'))
         url = result_base_url.format(osm_type=osm_type,
@@ -49,10 +52,8 @@ def response(resp):
         geojson = r.get('geojson')
 
         # if no geojson is found and osm_type is a node, add geojson Point
-        if not geojson and\
-           osm_type == 'node':
-            geojson = {u'type': u'Point',
-                       u'coordinates': [r['lon'], r['lat']]}
+        if not geojson and osm_type == 'node':
+            geojson = {u'type': u'Point', u'coordinates': [r['lon'], r['lat']]}
 
         address_raw = r.get('address')
         address = {}

searx/tests/engines/test_openstreetmap.py

@@ -0,0 +1,199 @@
+# -*- coding: utf-8 -*-
+from collections import defaultdict
+import mock
+from searx.engines import openstreetmap
+from searx.testing import SearxTestCase
+
+
+class TestOpenstreetmapEngine(SearxTestCase):
+
+    def test_request(self):
+        query = 'test_query'
+        dicto = defaultdict(dict)
+        dicto['pageno'] = 1
+        params = openstreetmap.request(query, dicto)
+        self.assertIn('url', params)
+        self.assertIn(query, params['url'])
+        self.assertIn('openstreetmap.org', params['url'])
+
+    def test_response(self):
+        self.assertRaises(AttributeError, openstreetmap.response, None)
+        self.assertRaises(AttributeError, openstreetmap.response, [])
+        self.assertRaises(AttributeError, openstreetmap.response, '')
+        self.assertRaises(AttributeError, openstreetmap.response, '[]')
+
+        response = mock.Mock(text='{}')
+        self.assertEqual(openstreetmap.response(response), [])
+
+        response = mock.Mock(text='{"data": []}')
+        self.assertEqual(openstreetmap.response(response), [])
+
+        json = """
+        [
+          {
+            "place_id": "127732055",
+            "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
+            "osm_type": "relation",
+            "osm_id": "7444",
+            "boundingbox": [
+              "48.8155755",
+              "48.902156",
+              "2.224122",
+              "2.4697602"
+            ],
+            "lat": "48.8565056",
+            "lon": "2.3521334",
+            "display_name": "This is the title",
+            "class": "place",
+            "type": "city",
+            "importance": 0.96893459932191,
+            "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
+            "address": {
+              "city": "Paris",
+              "county": "Paris",
+              "state": "Île-de-France",
+              "country": "France",
+              "country_code": "fr"
+            },
+            "geojson": {
+              "type": "Polygon",
+              "coordinates": [
+                [
+                  [
+                    2.224122,
+                    48.854199
+                  ]
+                ]
+              ]
+            }
+          }
+        ]
+        """
+        response = mock.Mock(text=json)
+        results = openstreetmap.response(response)
+        self.assertEqual(type(results), list)
+        self.assertEqual(len(results), 1)
+        self.assertEqual(results[0]['title'], 'This is the title')
+        self.assertEqual(results[0]['url'], 'https://openstreetmap.org/relation/7444')
+        self.assertIn('coordinates', results[0]['geojson'])
+        self.assertEqual(results[0]['geojson']['coordinates'][0][0][0], 2.224122)
+        self.assertEqual(results[0]['geojson']['coordinates'][0][0][1], 48.854199)
+        self.assertEqual(results[0]['address'], None)
+        self.assertIn('48.8155755', results[0]['boundingbox'])
+        self.assertIn('48.902156', results[0]['boundingbox'])
+        self.assertIn('2.224122', results[0]['boundingbox'])
+        self.assertIn('2.4697602', results[0]['boundingbox'])
+
+        json = """
+        [
+          {
+            "place_id": "127732055",
+            "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
+            "osm_type": "relation",
+            "osm_id": "7444",
+            "boundingbox": [
+              "48.8155755",
+              "48.902156",
+              "2.224122",
+              "2.4697602"
+            ],
+            "lat": "48.8565056",
+            "lon": "2.3521334",
+            "display_name": "This is the title",
+            "class": "tourism",
+            "type": "city",
+            "importance": 0.96893459932191,
+            "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
+            "address": {
+              "city": "Paris",
+              "county": "Paris",
+              "state": "Île-de-France",
+              "country": "France",
+              "country_code": "fr",
+              "address29": "Address"
+            },
+            "geojson": {
+              "type": "Polygon",
+              "coordinates": [
+                [
+                  [
+                    2.224122,
+                    48.854199
+                  ]
+                ]
+              ]
+            }
+          },
+          {
+            "place_id": "127732055",
+            "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
+            "osm_type": "relation",
+            "osm_id": "7444",
+            "boundingbox": [
+              "48.8155755",
+              "48.902156",
+              "2.224122",
+              "2.4697602"
+            ],
+            "lat": "48.8565056",
+            "lon": "2.3521334",
+            "display_name": "This is the title",
+            "class": "tourism",
+            "type": "city",
+            "importance": 0.96893459932191,
+            "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
+            "address": {
+              "city": "Paris",
+              "county": "Paris",
+              "state": "Île-de-France",
+              "country": "France",
+              "postcode": 75000,
+              "country_code": "fr"
+            },
+            "geojson": {
+              "type": "Polygon",
+              "coordinates": [
+                [
+                  [
+                    2.224122,
+                    48.854199
+                  ]
+                ]
+              ]
+            }
+          },
+          {
+            "place_id": "127732055",
+            "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
+            "osm_type": "node",
+            "osm_id": "7444",
+            "boundingbox": [
+              "48.8155755",
+              "48.902156",
+              "2.224122",
+              "2.4697602"
+            ],
+            "lat": "48.8565056",
+            "lon": "2.3521334",
+            "display_name": "This is the title",
+            "class": "tourism",
+            "type": "city",
+            "importance": 0.96893459932191,
+            "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
+            "address": {
+              "city": "Paris",
+              "county": "Paris",
+              "state": "Île-de-France",
+              "country": "France",
+              "country_code": "fr",
+              "address29": "Address"
+            }
+          }
+        ]
+        """
+        response = mock.Mock(text=json)
+        results = openstreetmap.response(response)
+        self.assertEqual(type(results), list)
+        self.assertEqual(len(results), 3)
+        self.assertIn('48.8565056', results[2]['geojson']['coordinates'])
+        self.assertIn('2.3521334', results[2]['geojson']['coordinates'])

searx/tests/test_engines.py

@@ -18,6 +18,7 @@ from searx.tests.engines.test_google_news import *  # noqa
 from searx.tests.engines.test_kickass import *  # noqa
 from searx.tests.engines.test_mediawiki import *  # noqa
 from searx.tests.engines.test_mixcloud import *  # noqa
+from searx.tests.engines.test_openstreetmap import *  # noqa
 from searx.tests.engines.test_piratebay import *  # noqa
 from searx.tests.engines.test_searchcode_code import *  # noqa
 from searx.tests.engines.test_searchcode_doc import *  # noqa