dataservices-api/server/lib/python/cartodb_services/test/test_heremapsgeocoder.py

#!/usr/local/bin/python
# -*- coding: utf-8 -*-

import unittest
import requests_mock
from mock import Mock

from cartodb_services.here import HereMapsGeocoder
from cartodb_services.here.exceptions import BadGeocodingParams
from cartodb_services.here.exceptions import NoGeocodingParams
from cartodb_services.here.exceptions import MalformedResult

requests_mock.Mocker.TEST_PREFIX = 'test_'


@requests_mock.Mocker()
class HereMapsGeocoderTestCase(unittest.TestCase):
    EMPTY_RESPONSE = """{
        "Response": {
            "MetaInfo": {
                "Timestamp": "2015-11-04T16:31:57.273+0000"
            },
            "View": []
        }
    }"""

    GOOD_RESPONSE = unicode("""{
      "Response": {
        "MetaInfo": {
          "Timestamp": "2016-02-10T14:17:33.792+0000",
          "NextPageInformation": "2"
        },
        "View": [
          {
            "_type": "SearchResultsViewType",
            "ViewId": 0,
            "Result": [
              {
                "Relevance": 1,
                "MatchLevel": "houseNumber",
                "MatchQuality": {
                  "Street": [
                    1
                  ],
                  "HouseNumber": 1
                },
                "MatchType": "pointAddress",
                "Location": {
                  "LocationId": "NT_CKopMSB9JnBYAO11CMOrxB_zUD",
                  "LocationType": "address",
                  "DisplayPosition": {
                    "Latitude": 37.70246,
                    "Longitude": -5.2794
                  },
                  "NavigationPosition": [
                    {
                      "Latitude": 37.7024199,
                      "Longitude": -5.27939
                    }
                  ],
                  "MapView": {
                    "TopLeft": {
                      "Latitude": 37.7035842,
                      "Longitude": -5.2808208
                    },
                    "BottomRight": {
                      "Latitude": 37.7013358,
                      "Longitude": -5.2779792
                    }
                  },
                  "Address": {
                    "Label": "Calle Amor de Dios, 35, 14700 Palma del Río (Córdoba), España",
                    "Country": "ESP",
                    "State": "Andalucía",
                    "County": "Córdoba",
                    "City": "Palma del Río",
                    "Street": "Calle Amor de Dios",
                    "HouseNumber": "35",
                    "PostalCode": "14700",
                    "AdditionalData": [
                      {
                        "value": "España",
                        "key": "CountryName"
                      },
                      {
                        "value": "Andalucía",
                        "key": "StateName"
                      },
                      {
                        "value": "Córdoba",
                        "key": "CountyName"
                      }
                    ]
                  }
                }
              }
            ]
          }
        ]
      }
    }""", 'utf-8')

    MALFORMED_RESPONSE = """{"manolo": "escobar"}"""

    def setUp(self):
        logger = Mock()
        self.geocoder = HereMapsGeocoder(None, None, logger)

    def test_geocode_address_with_valid_params(self, req_mock):
        req_mock.register_uri('GET', HereMapsGeocoder.PRODUCTION_GEOCODE_JSON_URL,
                       text=self.GOOD_RESPONSE)
        response = self.geocoder.geocode(
            searchtext='Calle amor de dios',
            city='Cordoba',
            country='España')

        self.assertEqual(response[0], -5.2794)
        self.assertEqual(response[1], 37.70246)

    def test_geocode_address_with_invalid_params(self, req_mock):
        req_mock.register_uri('GET', HereMapsGeocoder.PRODUCTION_GEOCODE_JSON_URL,
                       text=self.GOOD_RESPONSE)
        with self.assertRaises(BadGeocodingParams):
            self.geocoder.geocode(
                searchtext='Calle Eloy Gonzalo 27',
                manolo='escobar')

    def test_geocode_address_with_no_params(self, req_mock):
        req_mock.register_uri('GET', HereMapsGeocoder.PRODUCTION_GEOCODE_JSON_URL,
                       text=self.GOOD_RESPONSE)
        with self.assertRaises(NoGeocodingParams):
            self.geocoder.geocode()

    def test_geocode_address_empty_response(self, req_mock):
        req_mock.register_uri('GET', HereMapsGeocoder.PRODUCTION_GEOCODE_JSON_URL,
                       text=self.EMPTY_RESPONSE)
        result = self.geocoder.geocode(searchtext='lkajfñlasjfñ')
        self.assertEqual(result, [])

    def test_geocode_with_malformed_result(self, req_mock):
        req_mock.register_uri('GET', HereMapsGeocoder.PRODUCTION_GEOCODE_JSON_URL,
                       text=self.MALFORMED_RESPONSE)
        with self.assertRaises(MalformedResult):
            self.geocoder.geocode(
                searchtext='Calle amor de dios',
                city='Cordoba',
                country='España')