Changeset View
Changeset View
Standalone View
Standalone View
swh/web/tests/misc/test_iframe.py
# Copyright (C) 2021 The Software Heritage developers | # Copyright (C) 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 | ||||
from hypothesis import given | |||||
from swh.model.hashutil import hash_to_bytes | from swh.model.hashutil import hash_to_bytes | ||||
from swh.model.swhids import CoreSWHID, ObjectType | from swh.model.swhids import CoreSWHID, ObjectType | ||||
from swh.web.common.utils import reverse | from swh.web.common.utils import reverse | ||||
from swh.web.tests.strategies import unknown_directory | |||||
from swh.web.tests.utils import check_html_get_response | from swh.web.tests.utils import check_html_get_response | ||||
def test_content_swhid_iframe(client, content_swhid): | def test_content_swhid_iframe(client, content_swhid): | ||||
url = reverse("swhid-iframe", url_args={"swhid": str(content_swhid)}) | url = reverse("swhid-iframe", url_args={"swhid": str(content_swhid)}) | ||||
check_html_get_response( | check_html_get_response( | ||||
client, url, status_code=200, template_used="misc/iframe.html" | client, url, status_code=200, template_used="misc/iframe.html" | ||||
) | ) | ||||
Show All 28 Lines | |||||
def test_iframe_unsupported_object(client, revision_swhid): | def test_iframe_unsupported_object(client, revision_swhid): | ||||
url = reverse("swhid-iframe", url_args={"swhid": str(revision_swhid)}) | url = reverse("swhid-iframe", url_args={"swhid": str(revision_swhid)}) | ||||
check_html_get_response( | check_html_get_response( | ||||
client, url, status_code=400, template_used="misc/iframe.html" | client, url, status_code=400, template_used="misc/iframe.html" | ||||
) | ) | ||||
@given(unknown_directory()) | |||||
def test_iframe_object_not_found(client, unknown_directory): | def test_iframe_object_not_found(client, unknown_directory): | ||||
swhid = CoreSWHID( | swhid = CoreSWHID( | ||||
object_type=ObjectType.DIRECTORY, object_id=hash_to_bytes(unknown_directory) | object_type=ObjectType.DIRECTORY, object_id=hash_to_bytes(unknown_directory) | ||||
) | ) | ||||
url = reverse("swhid-iframe", url_args={"swhid": str(swhid)}) | url = reverse("swhid-iframe", url_args={"swhid": str(swhid)}) | ||||
check_html_get_response( | check_html_get_response( | ||||
client, url, status_code=404, template_used="misc/iframe.html" | client, url, status_code=404, template_used="misc/iframe.html" | ||||
) | ) | ||||
def test_swhid_iframe_unknown_error(client, mocker, content_swhid): | def test_swhid_iframe_unknown_error(client, mocker, content_swhid): | ||||
mocker.patch("swh.web.misc.iframe.get_swhid").side_effect = Exception("Error") | mocker.patch("swh.web.misc.iframe.get_swhid").side_effect = Exception("Error") | ||||
url = reverse("swhid-iframe", url_args={"swhid": str(content_swhid)}) | url = reverse("swhid-iframe", url_args={"swhid": str(content_swhid)}) | ||||
check_html_get_response( | check_html_get_response( | ||||
client, url, status_code=500, template_used="misc/iframe.html" | client, url, status_code=500, template_used="misc/iframe.html" | ||||
) | ) |