0007-api_entreprise-use-v3-in-etablissement-endpoint-7061.patch
passerelle/apps/api_entreprise/models.py | ||
---|---|---|
355 | 355 |
}, |
356 | 356 |
) |
357 | 357 |
def etablissements(self, request, siret, **kwargs): |
358 |
return self.get('v2/etablissements/%s/' % siret, **kwargs) |
|
358 |
raw_data = self.get('v3/insee/sirene/etablissements/diffusibles/%s' % siret, raw=True, **kwargs) |
|
359 |
res = {} |
|
360 |
res['etablissement'] = raw_data.get('data') |
|
361 |
return {'data': res} |
|
359 | 362 | |
360 | 363 |
@endpoint( |
361 | 364 |
perm='can_access', |
tests/test_api_entreprise.py | ||
---|---|---|
25 | 25 |
from tests.utils import FakedResponse, make_resource |
26 | 26 | |
27 | 27 |
ETABLISSEMENTS_RESPONSE = { |
28 |
"etablissement": {
|
|
28 |
"data": {
|
|
29 | 29 |
"siege_social": True, |
30 | 30 |
"siret": "41816609600051", |
31 | 31 |
"naf": "6202A", |
... | ... | |
409 | 409 |
def test_etablissements_endpoint(app, resource): |
410 | 410 |
with responses.RequestsMock() as rsps: |
411 | 411 |
rsps.get( |
412 |
'https://entreprise.api.gouv.fr/v2/etablissements/44317013900036/', json=ETABLISSEMENTS_RESPONSE |
|
412 |
'https://entreprise.api.gouv.fr/v3/insee/sirene/etablissements/diffusibles/44317013900036', |
|
413 |
json=ETABLISSEMENTS_RESPONSE, |
|
413 | 414 |
) |
414 | 415 |
response = app.get('/api-entreprise/test/etablissements/44317013900036/', params=REQUEST_PARAMS) |
415 | 416 |
assert 'data' in response.json |
416 |
- |