diff --git a/passerelle/apps/api_entreprise/models.py b/passerelle/apps/api_entreprise/models.py index 79f46e73..a215e1d8 100644 --- a/passerelle/apps/api_entreprise/models.py +++ b/passerelle/apps/api_entreprise/models.py @@ -434,63 +434,6 @@ class APIEntreprise(BaseResource): def exercices(self, request, siret, **kwargs): return self.get('v3/dgfip/etablissements/%s/chiffres_affaires' % siret, raw=True, **kwargs) - @endpoint( - perm='can_access', - pattern=r'(?P\w+)/$', - example_pattern='{siren}/', - description=_('Get firm\'s annual workforce data'), - parameters={ - 'siren': SIREN_PARAM, - 'object': OBJECT_PARAM, - 'context': CONTEXT_PARAM, - 'recipient': RECIPIENT_PARAM, - }, - ) - def effectifs_annuels_acoss_covid(self, request, siren, **kwargs): - if len(siren) != 9: - raise APIError(_('invalid SIREN length (must be 9 characters)')) - return self.get('v2/effectifs_annuels_acoss_covid/%s/' % siren, **kwargs) - - @endpoint( - perm='can_access', - pattern=r'(?P\w+)/(?P\w+)/(?P\w+)/$', - description=_('Get firm\'s monthly workforce data, by SIREN'), - parameters={ - 'year': YEAR_PARAM, - 'month': MONTH_PARAM, - 'siren': SIREN_PARAM, - 'object': OBJECT_PARAM, - 'context': CONTEXT_PARAM, - 'recipient': RECIPIENT_PARAM, - }, - ) - def entreprise_effectifs_mensuels_acoss_covid(self, request, year, month, siren, **kwargs): - if len(siren) != 9: - raise APIError(_('invalid SIREN length (must be 9 characters)')) - month = month.zfill(2) - return self.get( - 'v2/effectifs_mensuels_acoss_covid/%s/%s/entreprise/%s/' % (year, month, siren), **kwargs - ) - - @endpoint( - perm='can_access', - pattern=r'(?P\w+)/(?P\w+)/(?P\w+)/$', - description=_('Get firm\'s monthly workforce data, by SIRET'), - parameters={ - 'year': YEAR_PARAM, - 'month': MONTH_PARAM, - 'siret': SIRET_PARAM, - 'object': OBJECT_PARAM, - 'context': CONTEXT_PARAM, - 'recipient': RECIPIENT_PARAM, - }, - ) - def etablissement_effectifs_mensuels_acoss_covid(self, request, year, month, siret, **kwargs): - month = month.zfill(2) - return self.get( - 'v2/effectifs_mensuels_acoss_covid/%s/%s/etablissement/%s/' % (year, month, siret), **kwargs - ) - @endpoint( perm='can_access', pattern=r'(?P\w+)/$', diff --git a/tests/test_api_entreprise.py b/tests/test_api_entreprise.py index e09a4714..6cdd3573 100644 --- a/tests/test_api_entreprise.py +++ b/tests/test_api_entreprise.py @@ -293,25 +293,6 @@ DOCUMENTS_ASSOCIATION_RESPONSE = { } -EFFECTIFS_ANNUELS_ACOSS_COVID = {"siren": "418166096", "annee": "2019", "effectifs_annuels": 100.5} - - -ENTREPRISE_EFFECTIFS_MENSUELS_ACOSS_COVID = { - "siren": "418166096", - "annee": "2019", - "mois": "02", - "effectifs_mensuels": 100.5, -} - - -ETABLISSEMENT_EFFECTIFS_MENSUELS_ACOSS_COVID = { - "siret": "41816609600051", - "annee": "2019", - "mois": "02", - "effectifs_mensuels": 100.5, -} - - EXERCICES_RESPONSE = { "data": [ {"data": {"chiffre_affaires": 900001, "date_fin_exercice": "2015-12-01"}, "links": {}, "meta": {}} @@ -676,59 +657,6 @@ def test_document_association(app, resource, freezer): app.get(document['url'], status=404) -def test_effectifs_annuels_acoss_covid(app, resource): - with responses.RequestsMock() as rsps: - rsps.get( - 'https://entreprise.api.gouv.fr/v2/effectifs_annuels_acoss_covid/418166096/', - json=EFFECTIFS_ANNUELS_ACOSS_COVID, - ) - - response = app.get( - '/api-entreprise/test/effectifs_annuels_acoss_covid/418166096/', params=REQUEST_PARAMS - ) - data = response.json['data'] - assert data['siren'] == '418166096' - assert data['annee'] == '2019' - assert data['effectifs_annuels'] == 100.5 - - -def test_entreprise_effectifs_mensuels_acoss_covid(app, resource): - with responses.RequestsMock() as rsps: - rsps.get( - 'https://entreprise.api.gouv.fr/v2/effectifs_mensuels_acoss_covid/2019/02/entreprise/418166096/', - json=ENTREPRISE_EFFECTIFS_MENSUELS_ACOSS_COVID, - ) - - response = app.get( - '/api-entreprise/test/entreprise_effectifs_mensuels_acoss_covid/2019/02/418166096/', - params=REQUEST_PARAMS, - ) - - data = response.json['data'] - assert data['siren'] == '418166096' - assert data['annee'] == '2019' - assert data['mois'] == '02' - assert data['effectifs_mensuels'] == 100.5 - - -def test_etablissment_effectifs_mensuels_acoss_covid(app, resource): - with responses.RequestsMock() as rsps: - rsps.get( - 'https://entreprise.api.gouv.fr/v2/effectifs_mensuels_acoss_covid/2019/02/etablissement/418166096/', - json=ETABLISSEMENT_EFFECTIFS_MENSUELS_ACOSS_COVID, - ) - - response = app.get( - '/api-entreprise/test/etablissement_effectifs_mensuels_acoss_covid/2019/02/418166096/', - params=REQUEST_PARAMS, - ) - data = response.json['data'] - assert data['siret'] == '41816609600051' - assert data['annee'] == '2019' - assert data['mois'] == '02' - assert data['effectifs_mensuels'] == 100.5 - - def test_exercices(app, resource): with responses.RequestsMock() as rsps: rsps.get(