Changeset View
Changeset View
Standalone View
Standalone View
swh/web/tests/api/test_apiresponse.py
| # Copyright (C) 2015-2021 The Software Heritage developers | # Copyright (C) 2015-2021 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU Affero General Public License version 3, or any later version | # License: GNU Affero General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| import json | import json | ||||
| from corsheaders.middleware import ( | from corsheaders.middleware import ( | ||||
| ACCESS_CONTROL_ALLOW_HEADERS, | ACCESS_CONTROL_ALLOW_HEADERS, | ||||
| ACCESS_CONTROL_ALLOW_METHODS, | ACCESS_CONTROL_ALLOW_METHODS, | ||||
| ACCESS_CONTROL_ALLOW_ORIGIN, | ACCESS_CONTROL_ALLOW_ORIGIN, | ||||
| ) | ) | ||||
| from hypothesis import given | |||||
| from swh.model.swhids import ObjectType | from swh.model.swhids import ObjectType | ||||
| from swh.web.api.apiresponse import ( | from swh.web.api.apiresponse import ( | ||||
| compute_link_header, | compute_link_header, | ||||
| filter_by_fields, | filter_by_fields, | ||||
| make_api_response, | make_api_response, | ||||
| transform, | transform, | ||||
| ) | ) | ||||
| from swh.web.common.identifiers import gen_swhid | from swh.web.common.identifiers import gen_swhid | ||||
| from swh.web.common.utils import reverse | from swh.web.common.utils import reverse | ||||
| from swh.web.tests.django_asserts import assert_contains | from swh.web.tests.django_asserts import assert_contains | ||||
| from swh.web.tests.strategies import content, directory, revision | |||||
| from swh.web.tests.utils import check_http_get_response, check_http_post_response | from swh.web.tests.utils import check_http_get_response, check_http_post_response | ||||
| def test_compute_link_header(): | def test_compute_link_header(): | ||||
| next_link = "/api/endpoint/next" | next_link = "/api/endpoint/next" | ||||
| prev_link = "/api/endpoint/prev" | prev_link = "/api/endpoint/prev" | ||||
| rv = { | rv = { | ||||
| "headers": {"link-next": next_link, "link-prev": prev_link}, | "headers": {"link-next": next_link, "link-prev": prev_link}, | ||||
| ▲ Show 20 Lines • Show All 114 Lines • ▼ Show 20 Lines | def test_error_response_handler(mocker, api_client): | ||||
| mock_archive.stat_counters.side_effect = Exception("Something went wrong") | mock_archive.stat_counters.side_effect = Exception("Something went wrong") | ||||
| url = reverse("api-1-stat-counters") | url = reverse("api-1-stat-counters") | ||||
| resp = api_client.get(url) | resp = api_client.get(url) | ||||
| assert resp.status_code == 500 | assert resp.status_code == 500 | ||||
| assert "traceback" in resp.data | assert "traceback" in resp.data | ||||
| assert "Traceback" in resp.data["traceback"] | assert "Traceback" in resp.data["traceback"] | ||||
| @given(content(), directory(), revision()) | |||||
| def test_api_endpoints_have_cors_headers(client, content, directory, revision): | def test_api_endpoints_have_cors_headers(client, content, directory, revision): | ||||
| url = reverse("api-1-stat-counters") | url = reverse("api-1-stat-counters") | ||||
| resp = check_http_get_response( | resp = check_http_get_response( | ||||
| client, url, status_code=200, http_origin="https://example.org" | client, url, status_code=200, http_origin="https://example.org" | ||||
| ) | ) | ||||
| assert ACCESS_CONTROL_ALLOW_ORIGIN in resp | assert ACCESS_CONTROL_ALLOW_ORIGIN in resp | ||||
| Show All 26 Lines | |||||