1
1
mirror of https://github.com/Kozea/WeasyPrint.git synced 2024-10-04 07:57:52 +03:00
WeasyPrint/tests/test_url.py
2024-01-16 13:33:10 +01:00

33 lines
920 B
Python

"""Test URLs."""
import re
import pytest
from .testing_utils import FakeHTML, capture_logs, resource_path
@pytest.mark.parametrize('url, base_url', (
('https://weasyprint.org]', resource_path('<inline HTML>')),
('https://weasyprint.org]', 'https://weasyprint.org]'),
('https://weasyprint.org/', 'https://weasyprint.org]'),
))
def test_malformed_url_link(url, base_url):
"""Test malformed URLs."""
with capture_logs() as logs:
pdf = FakeHTML(
string=f'<p><a href="{url}">My Link</a></p>',
base_url=base_url).write_pdf()
assert len(logs) == 1
assert "Malformed" in logs[0]
assert "]" in logs[0]
uris = re.findall(b'/URI \\((.*)\\)', pdf)
types = re.findall(b'/S (/\\w*)', pdf)
subtypes = re.findall(b'/Subtype (/\\w*)', pdf)
assert uris.pop(0) == url.encode()
assert subtypes.pop(0) == b'/Link'
assert types.pop(0) == b'/URI'