logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://hacktivis.me/git/searx.git

test_openstreetmap.py (6887B)

    

  1. # -*- coding: utf-8 -*-
  2. from collections import defaultdict
  3. import mock
  4. from searx.engines import openstreetmap
  5. from searx.testing import SearxTestCase
  8. class TestOpenstreetmapEngine(SearxTestCase):
  10.     def test_request(self):
  11.         query = 'test_query'
  12.         dicto = defaultdict(dict)
  13.         dicto['pageno'] = 1
  14.         params = openstreetmap.request(query, dicto)
  15.         self.assertIn('url', params)
  16.         self.assertIn(query, params['url'])
  17.         self.assertIn('openstreetmap.org', params['url'])
  19.     def test_response(self):
  20.         self.assertRaises(AttributeError, openstreetmap.response, None)
  21.         self.assertRaises(AttributeError, openstreetmap.response, [])
  22.         self.assertRaises(AttributeError, openstreetmap.response, '')
  23.         self.assertRaises(AttributeError, openstreetmap.response, '[]')
  25.         response = mock.Mock(text='{}')
  26.         self.assertEqual(openstreetmap.response(response), [])
  28.         response = mock.Mock(text='{"data": []}')
  29.         self.assertEqual(openstreetmap.response(response), [])
  31.         json = """
  32.         [
  33.           {
  34.             "place_id": "127732055",
  35.             "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
  36.             "osm_type": "relation",
  37.             "osm_id": "7444",
  38.             "boundingbox": [
  39.               "48.8155755",
  40.               "48.902156",
  41.               "2.224122",
  42.               "2.4697602"
  43.             ],
  44.             "lat": "48.8565056",
  45.             "lon": "2.3521334",
  46.             "display_name": "This is the title",
  47.             "class": "place",
  48.             "type": "city",
  49.             "importance": 0.96893459932191,
  50.             "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
  51.             "address": {
  52.               "city": "Paris",
  53.               "county": "Paris",
  54.               "state": "Île-de-France",
  55.               "country": "France",
  56.               "country_code": "fr"
  57.             },
  58.             "geojson": {
  59.               "type": "Polygon",
  60.               "coordinates": [
  61.                 [
  62.                   [
  63.                     2.224122,
  64.                     48.854199
  65.                   ]
  66.                 ]
  67.               ]
  68.             }
  69.           }
  70.         ]
  71.         """
  72.         response = mock.Mock(text=json)
  73.         results = openstreetmap.response(response)
  74.         self.assertEqual(type(results), list)
  75.         self.assertEqual(len(results), 1)
  76.         self.assertEqual(results[0]['title'], 'This is the title')
  77.         self.assertEqual(results[0]['url'], 'https://openstreetmap.org/relation/7444')
  78.         self.assertIn('coordinates', results[0]['geojson'])
  79.         self.assertEqual(results[0]['geojson']['coordinates'][0][0][0], 2.224122)
  80.         self.assertEqual(results[0]['geojson']['coordinates'][0][0][1], 48.854199)
  81.         self.assertEqual(results[0]['address'], None)
  82.         self.assertIn('48.8155755', results[0]['boundingbox'])
  83.         self.assertIn('48.902156', results[0]['boundingbox'])
  84.         self.assertIn('2.224122', results[0]['boundingbox'])
  85.         self.assertIn('2.4697602', results[0]['boundingbox'])
  87.         json = """
  88.         [
  89.           {
  90.             "place_id": "127732055",
  91.             "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
  92.             "osm_type": "relation",
  93.             "osm_id": "7444",
  94.             "boundingbox": [
  95.               "48.8155755",
  96.               "48.902156",
  97.               "2.224122",
  98.               "2.4697602"
  99.             ],
  100.             "lat": "48.8565056",
  101.             "lon": "2.3521334",
  102.             "display_name": "This is the title",
  103.             "class": "tourism",
  104.             "type": "city",
  105.             "importance": 0.96893459932191,
  106.             "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
  107.             "address": {
  108.               "city": "Paris",
  109.               "county": "Paris",
  110.               "state": "Île-de-France",
  111.               "country": "France",
  112.               "country_code": "fr",
  113.               "address29": "Address"
  114.             },
  115.             "geojson": {
  116.               "type": "Polygon",
  117.               "coordinates": [
  118.                 [
  119.                   [
  120.                     2.224122,
  121.                     48.854199
  122.                   ]
  123.                 ]
  124.               ]
  125.             }
  126.           },
  127.           {
  128.             "place_id": "127732055",
  129.             "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
  130.             "osm_type": "relation",
  131.             "osm_id": "7444",
  132.             "boundingbox": [
  133.               "48.8155755",
  134.               "48.902156",
  135.               "2.224122",
  136.               "2.4697602"
  137.             ],
  138.             "lat": "48.8565056",
  139.             "lon": "2.3521334",
  140.             "display_name": "This is the title",
  141.             "class": "tourism",
  142.             "type": "city",
  143.             "importance": 0.96893459932191,
  144.             "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
  145.             "address": {
  146.               "city": "Paris",
  147.               "county": "Paris",
  148.               "state": "Île-de-France",
  149.               "country": "France",
  150.               "postcode": 75000,
  151.               "country_code": "fr"
  152.             },
  153.             "geojson": {
  154.               "type": "Polygon",
  155.               "coordinates": [
  156.                 [
  157.                   [
  158.                     2.224122,
  159.                     48.854199
  160.                   ]
  161.                 ]
  162.               ]
  163.             }
  164.           },
  165.           {
  166.             "place_id": "127732055",
  167.             "licence": "Data © OpenStreetMap contributors, ODbL 1.0. http://www.openstreetmap.org/copyright",
  168.             "osm_type": "node",
  169.             "osm_id": "7444",
  170.             "boundingbox": [
  171.               "48.8155755",
  172.               "48.902156",
  173.               "2.224122",
  174.               "2.4697602"
  175.             ],
  176.             "lat": "48.8565056",
  177.             "lon": "2.3521334",
  178.             "display_name": "This is the title",
  179.             "class": "tourism",
  180.             "type": "city",
  181.             "importance": 0.96893459932191,
  182.             "icon": "https://nominatim.openstreetmap.org/images/mapicons/poi_place_city.p.20.png",
  183.             "address": {
  184.               "city": "Paris",
  185.               "county": "Paris",
  186.               "state": "Île-de-France",
  187.               "country": "France",
  188.               "country_code": "fr",
  189.               "address29": "Address"
  190.             }
  191.           }
  192.         ]
  193.         """
  194.         response = mock.Mock(text=json)
  195.         results = openstreetmap.response(response)
  196.         self.assertEqual(type(results), list)
  197.         self.assertEqual(len(results), 3)
  198.         self.assertIn('48.8565056', results[2]['geojson']['coordinates'])
  199.         self.assertIn('2.3521334', results[2]['geojson']['coordinates'])