Produits Entr'ouvert » Passerelle

    @endpoint(

        name='district',

        description=_('Districts in Nancy Town'),

        parameters={

            'lat': {'description': _('Latitude')},

            'lon': {'description': _('Longitude')},

        },

        show=False,

    )

    def district(self, request, lon=None, lat=None):

        # deprecated endpoint

        if 'NANCY_Grc' in self.base_url:

            # Nancy URL used to contains folder, service and layer, remove them

            self.base_url = 'https://geoservices.grand-nancy.org/arcgis/rest/'

        features = self.mapservice_query(

            request,

            folder='public',

            service='NANCY_Grc',

            layer='0',

            template='{{ attributes.NOM }}',

            id_template='{{ attributes.NUMERO }}',

            lon=lon,

            lat=lat,

        )['data']

        if not features:

            raise APIError('No features found.')

        for feature in features:

            del feature['attributes']

        feature['id'] = int(feature['id'])

        if len(features) == 1:

            return {'data': features[0]}

        return {'data': features}

import os

import mock

import pytest

from django.contrib.contenttypes.models import ContentType

from django.urls import reverse

from passerelle.apps.arcgis.models import ArcGIS

from passerelle.base.models import AccessRight, ApiUser

from passerelle.compat import json_loads

TEST_BASE_DIR = os.path.join(os.path.dirname(__file__), 'data', 'nancy_arcgis')

def get_file_content(filename):

    with open(os.path.join(TEST_BASE_DIR, filename), 'rb') as fd:

        return fd.read()

class MockedRequestsResponse(mock.Mock):

    def json(self):

        return json_loads(self.content)

@pytest.fixture

def setup(db):

    api = ApiUser.objects.create(username='all', keytype='', key='')

    arcgis = ArcGIS.objects.create(base_url='https://example.net/layer/0', slug='test')

    obj_type = ContentType.objects.get_for_model(arcgis)

    AccessRight.objects.create(

        codename='can_access', apiuser=api, resource_type=obj_type, resource_pk=arcgis.pk

    )

    return arcgis

@pytest.fixture

def url():

    return reverse('generic-endpoint', kwargs={'connector': 'arcgis', 'slug': 'test', 'endpoint': 'district'})

def test_get_district_parameters_error(app, setup, url):

    resp = app.get(url, params={'lon': 'lon', 'lat': 'lat'}, status=400)

    assert resp.json['err_desc'] == '<lon> and <lat> must be floats'

@mock.patch('passerelle.utils.Request.get')

def test_get_district(mocked_get, app, setup, url):

    mocked_get.return_value = MockedRequestsResponse(

        content=get_file_content('sigresponse.json'), status_code=200

    )

    resp = app.get(url, params={'lon': 6.172122, 'lat': 48.673836}, status=200)

    data = resp.json['data']

    assert data['id'] == 4

    assert data['text'] == 'HAUSSONVILLE / BLANDAN / MON DESERT / SAURUPT'

@mock.patch('passerelle.utils.Request.get')

def test_get_all_district(mocked_get, app, setup, url):

    mocked_get.return_value = MockedRequestsResponse(

        content=get_file_content('all_districts.json'), status_code=200

    )

    resp = app.get(url, status=200)

    data = resp.json['data']

    assert len(data) == 7

@mock.patch('passerelle.utils.Request.get')

def test_no_district(mocked_get, app, setup, url):

    mocked_get.return_value = MockedRequestsResponse(content='{"features": []}', status_code=200)

    resp = app.get(url, status=200)

    assert resp.json['err'] == 1

    assert resp.json['err_class'] == 'passerelle.utils.jsonresponse.APIError'

    assert resp.json['err_desc'] == 'No features found.'

    resp = app.get('/arcgis/test/district', params={'lon': 6.172122, 'lat': 48.673836}, status=200)

    assert log.extra['connector_endpoint'] == 'district'

    assert '/arcgis/test/district?' in log.extra['connector_endpoint_url']

        assert u"endpoint GET /arcgis/test/district?" in record.message

    assert data['id'] == 4

    assert data['text'] == 'HAUSSONVILLE / BLANDAN / MON DESERT / SAURUPT'

    app.get('/arcgis/test/district', params={'lon': 6.172122, 'lat': 48.673836}, status=200)

    resp = app.get('/arcgis/test/district', params={'lon': 6.172122, 'lat': 48.673836}, status=200)

    resp = app.patch('/arcgis/test/district', status=405)

    assert log.message == 'endpoint PATCH /arcgis/test/district (=> 405)'

        assert record.message == 'endpoint PATCH /arcgis/test/district (=> 405)'

-