-
Notifications
You must be signed in to change notification settings - Fork 10
/
test_rest.py
85 lines (75 loc) · 3.21 KB
/
test_rest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# -*- coding: utf-8 -*-
#
# This file is part of Invenio.
# Copyright (C) 2016 CERN.
#
# Invenio is free software; you can redistribute it
# and/or modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# Invenio 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
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Invenio; if not, write to the
# Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston,
# MA 02111-1307, USA.
#
# In applying this license, CERN does not
# waive the privileges and immunities granted to it by virtue of its status
# as an Intergovernmental Organization or submit itself to any jurisdiction.
"""Tests for OpenAIRE dataset loaders and resolvers."""
from __future__ import absolute_import, print_function
import json
from invenio_records_rest import InvenioRecordsREST
from invenio_records_rest.views import create_blueprint
from invenio_openaire.config import OPENAIRE_REST_ENDPOINTS
def _get_json(response, code=None):
"""Decode JSON from response."""
data = response.get_data(as_text=True)
if code is not None:
assert response.status_code == code, data
return json.loads(data)
def test_records_rest(app, db, es, grants):
"""Test Records REST."""
app.config['RECORDS_REST_ENDPOINTS'] = OPENAIRE_REST_ENDPOINTS
app.config['RECORDS_REST_DEFAULT_READ_PERMISSION_FACTORY'] = None
InvenioRecordsREST(app)
# invenio-records-rest >= 1.1.0 doesn't automatically register endpoints
if 'invenio_records_rest.frdoi_item' not in app.url_map._rules_by_endpoint:
app.register_blueprint(create_blueprint(OPENAIRE_REST_ENDPOINTS))
with app.test_client() as client:
# Item
res = client.get("/funders/10.13039/001")
assert res.status_code == 200
# List
res = client.get("/funders/")
assert res.status_code == 200
print(res.get_data(as_text=True))
# Suggest
res = client.get("/funders/_suggest?text=Uni")
data = _get_json(res, 200)
options = data['text'][0]['options']
assert len(options) == 2
assert [option['_source']['doi'] for option in options].sort() == [
'0.13039/501100000923', '10.13039/001'].sort()
res = client.get("/grants/10.13039/501100000923::LP0667725")
assert res.status_code == 200
# List
res = client.get("/grants/")
assert res.status_code == 200
# Suggest
funder = '10.13039/501100000923'
res = client.get(
"/grants/_suggest?text=LP&funder={}".format(funder))
data = _get_json(res, 200)
options = data['text'][0]['options']
assert len(options) == 3
assert [(option['_source']['code'],
option['_source']['funder']['doi']) for option in options
].sort() == [
('LP0989479', funder), ('LP0667725', funder),
('LP0215942', funder)].sort()