Produits Entr'ouvert » Passerelle

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from django.db import models

from django.shortcuts import get_object_or_404

from django.urls import reverse

from django.utils.six.moves.urllib import parse as urlparse

from django.utils.translation import ugettext_lazy as _

from passerelle.base.models import BaseQuery, BaseResource

from passerelle.utils.api import endpoint

        for record in result_response.json().get('records'):

            data = {}

            for key, value in record.get('fields').items():

                data[key] = value

            data['id'] = record.get('recordid')

            data['text'] = render_to_string(text_template, data).strip()

            result.append(data)

        return {'data': result}

    @endpoint(

        name='q',

        description=_('Query'),

        pattern=r'^(?P<query_slug>[\w:_-]+)/$',

        perm='can_access',

        show=False,

        'text_template': '{{numero}} {{nom_rue}} {{nom_commun}}',

        'q': "rue de l'aubepine",

        'rows': 3,

    }

    mocked_get.return_value = utils.FakedResponse(content=FAKED_CONTENT_Q_SEARCH, status_code=200)

    resp = app.get(endpoint, params=params, status=200)

    assert not resp.json['err']

    assert len(resp.json['data']) == 3

    assert [x['id'] for x in resp.json['data']] == [

        '7cafcd5c692773e8b863587b2d38d6be82e023d8',

        '0984a5e1745701f71c91af73ce764e1f7132e0ff',

    ]

    # check text results

    assert [x['text'] for x in resp.json['data']] == [

        "19 RUE DE L'AUBEPINE Lipsheim",

        "29 RUE DE L'AUBEPINE Strasbourg",

    ]

    # check additional attributes

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

def test_search_using_id(mocked_get, app, connector):

    endpoint = utils.generic_endpoint_url('opendatasoft', 'search', slug=connector.slug)

    assert endpoint == '/opendatasoft/my_connector/search'

    params = {

        'dataset': 'referentiel-adresse-test',

    params = {

        'q': "rue de l'aubepine",

        'rows': 3,

    }

    mocked_get.return_value = utils.FakedResponse(content=FAKED_CONTENT_Q_SEARCH, status_code=200)

    resp = app.get(endpoint, params=params, status=200)

    assert not resp.json['err']

    assert len(resp.json['data']) == 3

    assert [x['id'] for x in resp.json['data']] == [

        '7cafcd5c692773e8b863587b2d38d6be82e023d8',

        '0984a5e1745701f71c91af73ce764e1f7132e0ff',

    ]

    # check text results

    assert [x['text'] for x in resp.json['data']] == [

        "19 RUE DE L'AUBEPINE Lipsheim",

        "29 RUE DE L'AUBEPINE Strasbourg",

    ]

    # check additional attributes

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

def test_query_q_using_id(mocked_get, app, query):

    endpoint = '/opendatasoft/my_connector/q/my_query/'

    params = {

        'id': '7cafcd5c692773e8b863587b2d38d6be82e023d8',

    }