2016-10-06 04:59:24 -05:00
|
|
|
"""Test the build process with manpage builder with the test root."""
|
|
|
|
|
|
2020-10-04 14:40:14 -05:00
|
|
|
from __future__ import annotations
|
2022-12-30 14:14:18 -06:00
|
|
|
|
2020-02-07 01:51:03 -06:00
|
|
|
import json
|
2020-10-04 14:40:14 -05:00
|
|
|
import re
|
2023-08-27 22:39:13 -05:00
|
|
|
import sys
|
2020-11-11 04:17:29 -06:00
|
|
|
import textwrap
|
2020-10-04 14:40:14 -05:00
|
|
|
import time
|
|
|
|
|
import wsgiref.handlers
|
2023-07-20 16:20:53 -05:00
|
|
|
from base64 import b64encode
|
2024-03-21 09:08:49 -05:00
|
|
|
from http.server import BaseHTTPRequestHandler
|
2021-02-06 12:07:40 -06:00
|
|
|
from queue import Queue
|
2024-05-14 21:13:57 -05:00
|
|
|
from typing import TYPE_CHECKING
|
2020-10-04 14:40:14 -05:00
|
|
|
from unittest import mock
|
2020-11-11 05:00:27 -06:00
|
|
|
|
2023-12-30 06:38:30 -06:00
|
|
|
import docutils
|
2017-01-05 10:14:47 -06:00
|
|
|
import pytest
|
2023-07-23 10:06:23 -05:00
|
|
|
from urllib3.poolmanager import PoolManager
|
2016-10-06 04:59:24 -05:00
|
|
|
|
2023-08-27 22:39:13 -05:00
|
|
|
import sphinx.util.http_date
|
2023-07-23 10:24:12 -05:00
|
|
|
from sphinx.builders.linkcheck import (
|
|
|
|
|
CheckRequest,
|
2024-05-14 21:13:57 -05:00
|
|
|
CheckResult,
|
2023-07-23 10:24:12 -05:00
|
|
|
Hyperlink,
|
|
|
|
|
HyperlinkAvailabilityCheckWorker,
|
|
|
|
|
RateLimit,
|
2024-04-04 04:25:53 -05:00
|
|
|
compile_linkcheck_allowed_redirects,
|
2023-07-23 10:24:12 -05:00
|
|
|
)
|
2024-03-14 05:45:45 -05:00
|
|
|
from sphinx.deprecation import RemovedInSphinx80Warning
|
2023-07-20 15:38:21 -05:00
|
|
|
from sphinx.util import requests
|
2024-03-23 18:43:54 -05:00
|
|
|
from sphinx.util.console import strip_colors
|
2020-10-04 14:40:14 -05:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
from tests.utils import CERT_FILE, serve_application
|
2020-11-08 14:57:06 -06:00
|
|
|
|
2020-10-04 14:40:14 -05:00
|
|
|
ts_re = re.compile(r".*\[(?P<ts>.*)\].*")
|
2022-09-10 11:26:41 -05:00
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
if TYPE_CHECKING:
|
|
|
|
|
from collections.abc import Callable
|
|
|
|
|
from io import StringIO
|
|
|
|
|
|
|
|
|
|
from sphinx.application import Sphinx
|
|
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
|
2024-03-21 09:08:49 -05:00
|
|
|
class DefaultsHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
def do_HEAD(self):
|
2023-07-22 18:01:41 -05:00
|
|
|
if self.path[1:].rstrip() in {"", "anchor.html"}:
|
2023-05-09 11:09:35 -05:00
|
|
|
self.send_response(200, "OK")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2023-05-09 11:09:35 -05:00
|
|
|
self.end_headers()
|
2022-09-10 11:26:41 -05:00
|
|
|
else:
|
|
|
|
|
self.send_response(404, "Not Found")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2022-09-10 11:26:41 -05:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def do_GET(self):
|
|
|
|
|
if self.path[1:].rstrip() == "":
|
2023-07-22 14:12:32 -05:00
|
|
|
content = b"ok\n\n"
|
2023-05-09 11:09:35 -05:00
|
|
|
elif self.path[1:].rstrip() == "anchor.html":
|
|
|
|
|
doc = '<!DOCTYPE html><html><body><a id="found"></a></body></html>'
|
2023-07-22 14:12:32 -05:00
|
|
|
content = doc.encode("utf-8")
|
|
|
|
|
else:
|
|
|
|
|
content = b""
|
|
|
|
|
|
|
|
|
|
if content:
|
|
|
|
|
self.send_response(200, "OK")
|
|
|
|
|
self.send_header("Content-Length", str(len(content)))
|
|
|
|
|
self.end_headers()
|
|
|
|
|
self.wfile.write(content)
|
|
|
|
|
else:
|
|
|
|
|
self.send_response(404, "Not Found")
|
|
|
|
|
self.send_header("Content-Length", "0")
|
|
|
|
|
self.end_headers()
|
2020-10-04 14:40:14 -05:00
|
|
|
|
2016-10-06 04:59:24 -05:00
|
|
|
|
2023-07-23 10:06:23 -05:00
|
|
|
class ConnectionMeasurement:
|
|
|
|
|
"""Measure the number of distinct host connections created during linkchecking"""
|
|
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
self.connections = set()
|
|
|
|
|
self.urllib3_connection_from_url = PoolManager.connection_from_url
|
|
|
|
|
self.patcher = mock.patch.object(
|
|
|
|
|
target=PoolManager,
|
|
|
|
|
attribute='connection_from_url',
|
|
|
|
|
new=self._collect_connections(),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
def _collect_connections(self):
|
|
|
|
|
def connection_collector(obj, url):
|
|
|
|
|
connection = self.urllib3_connection_from_url(obj, url)
|
|
|
|
|
self.connections.add(connection)
|
|
|
|
|
return connection
|
|
|
|
|
return connection_collector
|
|
|
|
|
|
|
|
|
|
def __enter__(self):
|
|
|
|
|
self.patcher.start()
|
|
|
|
|
return self
|
|
|
|
|
|
|
|
|
|
def __exit__(self, *args, **kwargs):
|
|
|
|
|
for connection in self.connections:
|
|
|
|
|
connection.close()
|
|
|
|
|
self.patcher.stop()
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def connection_count(self):
|
|
|
|
|
return len(self.connections)
|
|
|
|
|
|
|
|
|
|
|
2017-01-05 10:14:47 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_defaults(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, DefaultsHandler) as address:
|
2023-07-23 10:06:23 -05:00
|
|
|
with ConnectionMeasurement() as m:
|
|
|
|
|
app.build()
|
2023-07-23 16:23:08 -05:00
|
|
|
assert m.connection_count <= 5
|
2016-10-06 04:59:24 -05:00
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
# Text output
|
2016-10-06 04:59:24 -05:00
|
|
|
assert (app.outdir / 'output.txt').exists()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
2016-10-06 04:59:24 -05:00
|
|
|
|
2020-02-07 01:51:03 -06:00
|
|
|
# looking for '#top' and '#does-not-exist' not found should fail
|
2016-03-23 11:38:46 -05:00
|
|
|
assert "Anchor 'top' not found" in content
|
2018-07-16 16:17:44 -05:00
|
|
|
assert "Anchor 'does-not-exist' not found" in content
|
2019-02-05 09:21:52 -06:00
|
|
|
# images should fail
|
2024-04-04 04:25:53 -05:00
|
|
|
assert f"Not Found for url: http://{address}/image.png" in content
|
|
|
|
|
assert f"Not Found for url: http://{address}/image2.png" in content
|
2024-03-18 12:36:22 -05:00
|
|
|
# looking for missing local file should fail
|
2020-02-09 05:22:22 -06:00
|
|
|
assert "[broken] path/to/notfound" in content
|
2022-09-10 11:26:41 -05:00
|
|
|
assert len(content.splitlines()) == 5
|
2020-02-07 01:51:03 -06:00
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
# JSON output
|
2020-02-07 01:51:03 -06:00
|
|
|
assert (app.outdir / 'output.json').exists()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
2020-02-07 01:51:03 -06:00
|
|
|
|
|
|
|
|
rows = [json.loads(x) for x in content.splitlines()]
|
|
|
|
|
row = rows[0]
|
2022-09-10 11:26:41 -05:00
|
|
|
for attr in ("filename", "lineno", "status", "code", "uri", "info"):
|
2020-02-07 01:51:03 -06:00
|
|
|
assert attr in row
|
|
|
|
|
|
2023-05-09 11:09:35 -05:00
|
|
|
assert len(content.splitlines()) == 10
|
|
|
|
|
assert len(rows) == 10
|
2020-02-07 01:51:03 -06:00
|
|
|
# the output order of the rows is not stable
|
|
|
|
|
# due to possible variance in network latency
|
2020-11-15 02:03:26 -06:00
|
|
|
rowsby = {row["uri"]: row for row in rows}
|
2024-03-18 12:36:22 -05:00
|
|
|
# looking for local file that exists should succeed
|
|
|
|
|
assert rowsby["conf.py"]["status"] == "working"
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rowsby[f"http://{address}#!bar"] == {
|
2022-09-10 11:26:41 -05:00
|
|
|
'filename': 'links.rst',
|
|
|
|
|
'lineno': 5,
|
2020-02-07 01:51:03 -06:00
|
|
|
'status': 'working',
|
|
|
|
|
'code': 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
'uri': f'http://{address}#!bar',
|
2023-02-17 16:11:14 -06:00
|
|
|
'info': '',
|
2020-02-07 01:51:03 -06:00
|
|
|
}
|
2023-12-30 06:38:30 -06:00
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
def _missing_resource(filename: str, lineno: int) -> dict[str, str | int]:
|
2023-12-30 06:38:30 -06:00
|
|
|
return {
|
|
|
|
|
'filename': 'links.rst',
|
|
|
|
|
'lineno': lineno,
|
|
|
|
|
'status': 'broken',
|
|
|
|
|
'code': 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
'uri': f'http://{address}/{filename}',
|
|
|
|
|
'info': f'404 Client Error: Not Found for url: http://{address}/{filename}',
|
2023-12-30 06:38:30 -06:00
|
|
|
}
|
|
|
|
|
accurate_linenumbers = docutils.__version_info__[:2] >= (0, 21)
|
|
|
|
|
image2_lineno = 12 if accurate_linenumbers else 13
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rowsby[f'http://{address}/image2.png'] == _missing_resource("image2.png", image2_lineno)
|
2020-02-07 01:51:03 -06:00
|
|
|
# looking for '#top' and '#does-not-exist' not found should fail
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rowsby[f"http://{address}/#top"]["info"] == "Anchor 'top' not found"
|
|
|
|
|
assert rowsby[f"http://{address}/#top"]["status"] == "broken"
|
|
|
|
|
assert rowsby[f"http://{address}#does-not-exist"]["info"] == "Anchor 'does-not-exist' not found"
|
2020-02-07 01:51:03 -06:00
|
|
|
# images should fail
|
2024-04-04 04:25:53 -05:00
|
|
|
assert f"Not Found for url: http://{address}/image.png" in rowsby[f"http://{address}/image.png"]["info"]
|
2023-05-09 11:09:35 -05:00
|
|
|
# anchor should be found
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rowsby[f'http://{address}/anchor.html#found'] == {
|
2023-05-09 11:09:35 -05:00
|
|
|
'filename': 'links.rst',
|
|
|
|
|
'lineno': 14,
|
|
|
|
|
'status': 'working',
|
|
|
|
|
'code': 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
'uri': f'http://{address}/anchor.html#found',
|
2023-05-09 11:09:35 -05:00
|
|
|
'info': '',
|
|
|
|
|
}
|
2022-09-10 11:26:41 -05:00
|
|
|
|
|
|
|
|
|
2023-08-01 20:55:37 -05:00
|
|
|
@pytest.mark.sphinx(
|
|
|
|
|
'linkcheck', testroot='linkcheck', freshenv=True,
|
|
|
|
|
confoverrides={'linkcheck_anchors': False})
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_check_link_response_only(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, DefaultsHandler) as address:
|
2023-08-01 20:55:37 -05:00
|
|
|
app.build()
|
|
|
|
|
|
|
|
|
|
# JSON output
|
|
|
|
|
assert (app.outdir / 'output.json').exists()
|
|
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
|
|
|
|
|
|
|
|
|
rows = [json.loads(x) for x in content.splitlines()]
|
|
|
|
|
rowsby = {row["uri"]: row for row in rows}
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rowsby[f"http://{address}/#top"]["status"] == "working"
|
2023-08-01 20:55:37 -05:00
|
|
|
|
|
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-too-many-retries', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_too_many_retries(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, DefaultsHandler) as address:
|
2023-04-06 16:24:49 -05:00
|
|
|
app.build()
|
2022-09-10 11:26:41 -05:00
|
|
|
|
|
|
|
|
# Text output
|
|
|
|
|
assert (app.outdir / 'output.txt').exists()
|
|
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
|
|
|
|
|
|
|
|
|
# looking for non-existent URL should fail
|
|
|
|
|
assert " Max retries exceeded with url: /doesnotexist" in content
|
|
|
|
|
|
|
|
|
|
# JSON output
|
|
|
|
|
assert (app.outdir / 'output.json').exists()
|
|
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
|
|
|
|
|
|
|
|
|
assert len(content.splitlines()) == 1
|
|
|
|
|
row = json.loads(content)
|
|
|
|
|
# the output order of the rows is not stable
|
|
|
|
|
# due to possible variance in network latency
|
|
|
|
|
|
|
|
|
|
# looking for non-existent URL should fail
|
|
|
|
|
assert row['filename'] == 'index.rst'
|
|
|
|
|
assert row['lineno'] == 1
|
|
|
|
|
assert row['status'] == 'broken'
|
|
|
|
|
assert row['code'] == 0
|
2024-04-04 04:25:53 -05:00
|
|
|
assert row['uri'] == f'https://{address}/doesnotexist'
|
2022-09-10 11:26:41 -05:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-raw-node', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_raw_node(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler) as address:
|
|
|
|
|
# write an index file that contains a link back to this webserver's root
|
|
|
|
|
# URL. docutils will replace the raw node with the contents retrieved..
|
|
|
|
|
# ..and then the linkchecker will check that the root URL is available.
|
|
|
|
|
index = (app.srcdir / "index.rst")
|
|
|
|
|
index.write_text(
|
|
|
|
|
".. raw:: 'html'\n"
|
|
|
|
|
" :url: http://{address}/".format(address=address),
|
|
|
|
|
)
|
2022-09-10 11:26:41 -05:00
|
|
|
app.build()
|
|
|
|
|
|
|
|
|
|
# JSON output
|
|
|
|
|
assert (app.outdir / 'output.json').exists()
|
|
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
|
|
|
|
|
|
|
|
|
assert len(content.splitlines()) == 1
|
|
|
|
|
row = json.loads(content)
|
|
|
|
|
|
2022-08-07 04:02:45 -05:00
|
|
|
# raw nodes' url should be checked too
|
2022-09-10 11:26:41 -05:00
|
|
|
assert row == {
|
|
|
|
|
'filename': 'index.rst',
|
|
|
|
|
'lineno': 1,
|
|
|
|
|
'status': 'working',
|
|
|
|
|
'code': 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
'uri': f'http://{address}/', # the received rST contains a link to its' own URL
|
2022-09-10 11:26:41 -05:00
|
|
|
'info': '',
|
2022-08-07 04:02:45 -05:00
|
|
|
}
|
2020-02-07 01:51:03 -06:00
|
|
|
|
|
|
|
|
|
2017-01-05 10:14:47 -06:00
|
|
|
@pytest.mark.sphinx(
|
2022-09-10 11:26:41 -05:00
|
|
|
'linkcheck', testroot='linkcheck-anchors-ignore', freshenv=True,
|
|
|
|
|
confoverrides={'linkcheck_anchors_ignore': ["^!", "^top$"]})
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_anchors_ignored(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler):
|
2022-09-10 11:26:41 -05:00
|
|
|
app.build()
|
2016-10-06 04:59:24 -05:00
|
|
|
|
2016-03-23 11:38:46 -05:00
|
|
|
assert (app.outdir / 'output.txt').exists()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
2016-03-23 11:38:46 -05:00
|
|
|
|
|
|
|
|
# expect all ok when excluding #top
|
|
|
|
|
assert not content
|
2019-11-13 08:39:47 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2024-03-21 09:08:49 -05:00
|
|
|
class AnchorsIgnoreForUrlHandler(BaseHTTPRequestHandler):
|
2023-07-24 06:15:42 -05:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
if self.path in {'/valid', '/ignored'}:
|
|
|
|
|
self.send_response(200, "OK")
|
|
|
|
|
else:
|
|
|
|
|
self.send_response(404, "Not Found")
|
|
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def do_GET(self):
|
|
|
|
|
self.do_HEAD()
|
|
|
|
|
if self.path == '/valid':
|
|
|
|
|
self.wfile.write(b"<h1 id='valid-anchor'>valid anchor</h1>\n")
|
|
|
|
|
elif self.path == '/ignored':
|
|
|
|
|
self.wfile.write(b"no anchor but page exists\n")
|
|
|
|
|
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-anchors-ignore-for-url', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_anchors_ignored_for_url(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, AnchorsIgnoreForUrlHandler) as address:
|
2024-06-17 11:04:33 -05:00
|
|
|
app.config.linkcheck_anchors_ignore_for_url = [
|
2024-04-04 04:25:53 -05:00
|
|
|
f'http://{address}/ignored', # existing page
|
|
|
|
|
f'http://{address}/invalid', # unknown page
|
|
|
|
|
]
|
2023-07-24 06:15:42 -05:00
|
|
|
app.build()
|
|
|
|
|
|
|
|
|
|
assert (app.outdir / 'output.txt').exists()
|
|
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
|
|
|
|
|
|
|
|
|
attrs = ('filename', 'lineno', 'status', 'code', 'uri', 'info')
|
|
|
|
|
data = [json.loads(x) for x in content.splitlines()]
|
2024-04-24 13:07:31 -05:00
|
|
|
assert len(data) == 8
|
2023-07-24 06:15:42 -05:00
|
|
|
assert all(all(attr in row for attr in attrs) for row in data)
|
|
|
|
|
|
|
|
|
|
# rows may be unsorted due to network latency or
|
|
|
|
|
# the order the threads are processing the links
|
|
|
|
|
rows = {r['uri']: {'status': r['status'], 'info': r['info']} for r in data}
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rows[f'http://{address}/valid']['status'] == 'working'
|
|
|
|
|
assert rows[f'http://{address}/valid#valid-anchor']['status'] == 'working'
|
2024-04-24 13:36:36 -05:00
|
|
|
assert rows[f'http://{address}/valid#py:module::urllib.parse']['status'] == 'broken'
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rows[f'http://{address}/valid#invalid-anchor'] == {
|
2023-07-24 06:15:42 -05:00
|
|
|
'status': 'broken',
|
|
|
|
|
'info': "Anchor 'invalid-anchor' not found",
|
|
|
|
|
}
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rows[f'http://{address}/ignored']['status'] == 'working'
|
|
|
|
|
assert rows[f'http://{address}/ignored#invalid-anchor']['status'] == 'working'
|
2023-07-24 06:15:42 -05:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rows[f'http://{address}/invalid'] == {
|
2023-07-24 06:15:42 -05:00
|
|
|
'status': 'broken',
|
2024-04-04 04:25:53 -05:00
|
|
|
'info': f'404 Client Error: Not Found for url: http://{address}/invalid',
|
2023-07-24 06:15:42 -05:00
|
|
|
}
|
2024-04-04 04:25:53 -05:00
|
|
|
assert rows[f'http://{address}/invalid#anchor'] == {
|
2023-07-24 06:15:42 -05:00
|
|
|
'status': 'broken',
|
2024-04-04 04:25:53 -05:00
|
|
|
'info': f'404 Client Error: Not Found for url: http://{address}/invalid',
|
2023-07-24 06:15:42 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2020-11-11 03:50:45 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-anchor', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_raises_for_invalid_status(app: Sphinx) -> None:
|
2024-03-21 09:08:49 -05:00
|
|
|
class InternalServerErrorHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2020-11-08 14:57:06 -06:00
|
|
|
def do_GET(self):
|
|
|
|
|
self.send_error(500, "Internal Server Error")
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, InternalServerErrorHandler) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
2020-10-03 05:59:47 -05:00
|
|
|
assert content == (
|
2024-04-04 04:25:53 -05:00
|
|
|
f"index.rst:1: [broken] http://{address}/#anchor: "
|
2020-10-03 05:59:47 -05:00
|
|
|
"500 Server Error: Internal Server Error "
|
2024-04-04 04:25:53 -05:00
|
|
|
f"for url: http://{address}/\n"
|
2020-10-03 05:59:47 -05:00
|
|
|
)
|
|
|
|
|
|
2019-11-13 08:39:47 -06:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
def custom_handler(valid_credentials=(), success_criteria=lambda _: True):
|
|
|
|
|
"""
|
|
|
|
|
Returns an HTTP request handler that authenticates the client and then determines
|
|
|
|
|
an appropriate HTTP response code, based on caller-provided credentials and optional
|
|
|
|
|
success criteria, respectively.
|
|
|
|
|
"""
|
|
|
|
|
expected_token = None
|
|
|
|
|
if valid_credentials:
|
|
|
|
|
assert len(valid_credentials) == 2, "expected a pair of strings as credentials"
|
|
|
|
|
expected_token = b64encode(":".join(valid_credentials).encode()).decode("utf-8")
|
|
|
|
|
del valid_credentials
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
def authenticated(
|
|
|
|
|
method: Callable[[CustomHandler], None]
|
|
|
|
|
) -> Callable[[CustomHandler], None]:
|
|
|
|
|
def method_if_authenticated(self):
|
|
|
|
|
if expected_token is None:
|
|
|
|
|
return method(self)
|
|
|
|
|
elif not self.headers["Authorization"]:
|
|
|
|
|
self.send_response(401, "Unauthorized")
|
|
|
|
|
self.end_headers()
|
|
|
|
|
elif self.headers["Authorization"] == f"Basic {expected_token}":
|
|
|
|
|
return method(self)
|
|
|
|
|
else:
|
|
|
|
|
self.send_response(403, "Forbidden")
|
|
|
|
|
self.send_header("Content-Length", "0")
|
|
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
return method_if_authenticated
|
|
|
|
|
|
2024-03-21 09:08:49 -05:00
|
|
|
class CustomHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
@authenticated
|
2022-09-10 11:26:41 -05:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
self.do_GET()
|
2020-11-11 08:12:31 -06:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
@authenticated
|
2022-09-10 11:26:41 -05:00
|
|
|
def do_GET(self):
|
2023-07-20 16:20:53 -05:00
|
|
|
if success_criteria(self):
|
|
|
|
|
self.send_response(200, "OK")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2023-07-20 16:20:53 -05:00
|
|
|
else:
|
|
|
|
|
self.send_response(400, "Bad Request")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2022-09-10 11:26:41 -05:00
|
|
|
self.end_headers()
|
2023-07-20 16:20:53 -05:00
|
|
|
|
|
|
|
|
return CustomHandler
|
2020-11-11 08:12:31 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_auth_header_uses_first_match(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, custom_handler(valid_credentials=("user1", "password"))) as address:
|
2024-06-17 11:04:33 -05:00
|
|
|
app.config.linkcheck_auth = [
|
2024-04-04 04:25:53 -05:00
|
|
|
(r'^$', ('no', 'match')),
|
|
|
|
|
(fr'^http://{re.escape(address)}/$', ('user1', 'password')),
|
|
|
|
|
(r'.*local.*', ('user2', 'hunter2')),
|
|
|
|
|
]
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2022-09-10 11:26:41 -05:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
|
|
|
|
assert content["status"] == "working"
|
2020-11-11 08:12:31 -06:00
|
|
|
|
|
|
|
|
|
2024-01-09 06:33:40 -06:00
|
|
|
@pytest.mark.filterwarnings('ignore::sphinx.deprecation.RemovedInSphinx80Warning')
|
|
|
|
|
@pytest.mark.sphinx(
|
|
|
|
|
'linkcheck', testroot='linkcheck-localserver', freshenv=True,
|
|
|
|
|
confoverrides={'linkcheck_allow_unauthorized': False})
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_unauthorized_broken(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, custom_handler(valid_credentials=("user1", "password"))):
|
2024-01-09 06:33:40 -06:00
|
|
|
app.build()
|
|
|
|
|
|
|
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
|
|
|
|
assert content["info"] == "unauthorized"
|
|
|
|
|
assert content["status"] == "broken"
|
|
|
|
|
|
|
|
|
|
|
2020-11-11 08:12:31 -06:00
|
|
|
@pytest.mark.sphinx(
|
|
|
|
|
'linkcheck', testroot='linkcheck-localserver', freshenv=True,
|
|
|
|
|
confoverrides={'linkcheck_auth': [(r'^$', ('user1', 'password'))]})
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_auth_header_no_match(app: Sphinx) -> None:
|
2024-03-14 05:45:45 -05:00
|
|
|
with (
|
2024-04-04 04:25:53 -05:00
|
|
|
serve_application(app, custom_handler(valid_credentials=("user1", "password"))),
|
2024-03-14 05:45:45 -05:00
|
|
|
pytest.warns(RemovedInSphinx80Warning, match='linkcheck builder encountered an HTTP 401'),
|
|
|
|
|
):
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2022-09-10 11:26:41 -05:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
2024-01-09 06:33:40 -06:00
|
|
|
# This link is considered working based on the default linkcheck_allow_unauthorized=true
|
|
|
|
|
assert content["info"] == "unauthorized"
|
|
|
|
|
assert content["status"] == "working"
|
2020-05-31 11:37:15 -05:00
|
|
|
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_linkcheck_request_headers(app: Sphinx) -> None:
|
2023-07-20 16:20:53 -05:00
|
|
|
def check_headers(self):
|
|
|
|
|
if "X-Secret" in self.headers:
|
|
|
|
|
return False
|
2024-03-22 04:33:42 -05:00
|
|
|
return self.headers["Accept"] == "text/html"
|
2023-07-20 16:20:53 -05:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, custom_handler(success_criteria=check_headers)) as address:
|
2024-06-17 11:04:33 -05:00
|
|
|
app.config.linkcheck_request_headers = {
|
2024-04-04 04:25:53 -05:00
|
|
|
f"http://{address}/": {"Accept": "text/html"},
|
|
|
|
|
"*": {"X-Secret": "open sesami"},
|
|
|
|
|
}
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 09:26:22 -06:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
|
|
|
|
assert content["status"] == "working"
|
2020-11-11 09:26:22 -06:00
|
|
|
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_linkcheck_request_headers_no_slash(app: Sphinx) -> None:
|
2023-07-20 16:20:53 -05:00
|
|
|
def check_headers(self):
|
|
|
|
|
if "X-Secret" in self.headers:
|
|
|
|
|
return False
|
2024-03-22 04:33:42 -05:00
|
|
|
return self.headers["Accept"] == "application/json"
|
2023-07-20 16:20:53 -05:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, custom_handler(success_criteria=check_headers)) as address:
|
2024-06-17 11:04:33 -05:00
|
|
|
app.config.linkcheck_request_headers = {
|
2024-04-04 04:25:53 -05:00
|
|
|
f"http://{address}": {"Accept": "application/json"},
|
|
|
|
|
"*": {"X-Secret": "open sesami"},
|
|
|
|
|
}
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-05-31 11:37:15 -05:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
|
|
|
|
assert content["status"] == "working"
|
2020-11-11 09:26:22 -06:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.sphinx(
|
|
|
|
|
'linkcheck', testroot='linkcheck-localserver', freshenv=True,
|
|
|
|
|
confoverrides={'linkcheck_request_headers': {
|
|
|
|
|
"http://do.not.match.org": {"Accept": "application/json"},
|
2023-02-17 16:11:14 -06:00
|
|
|
"*": {"X-Secret": "open sesami"},
|
2020-11-11 09:26:22 -06:00
|
|
|
}})
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_linkcheck_request_headers_default(app: Sphinx) -> None:
|
2023-07-20 16:20:53 -05:00
|
|
|
def check_headers(self):
|
|
|
|
|
if self.headers["X-Secret"] != "open sesami":
|
|
|
|
|
return False
|
2024-03-22 04:33:42 -05:00
|
|
|
return self.headers["Accept"] != "application/json"
|
2023-07-20 16:20:53 -05:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, custom_handler(success_criteria=check_headers)):
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 09:26:22 -06:00
|
|
|
|
2023-07-20 16:20:53 -05:00
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
|
|
|
|
assert content["status"] == "working"
|
2020-11-11 04:17:29 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2020-11-11 04:17:29 -06:00
|
|
|
def make_redirect_handler(*, support_head):
|
2024-03-21 09:08:49 -05:00
|
|
|
class RedirectOnceHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2020-11-11 04:17:29 -06:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
if support_head:
|
|
|
|
|
self.do_GET()
|
|
|
|
|
else:
|
|
|
|
|
self.send_response(405, "Method Not Allowed")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2020-11-11 04:17:29 -06:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def do_GET(self):
|
|
|
|
|
if self.path == "/?redirected=1":
|
|
|
|
|
self.send_response(204, "No content")
|
|
|
|
|
else:
|
|
|
|
|
self.send_response(302, "Found")
|
2024-04-04 04:25:53 -05:00
|
|
|
self.send_header("Location", "/?redirected=1")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2020-11-11 04:17:29 -06:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def log_date_time_string(self):
|
|
|
|
|
"""Strip date and time from logged messages for assertions."""
|
|
|
|
|
return ""
|
|
|
|
|
|
|
|
|
|
return RedirectOnceHandler
|
|
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2020-11-11 04:17:29 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2021-04-29 06:20:01 -05:00
|
|
|
def test_follows_redirects_on_HEAD(app, capsys, warning):
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, make_redirect_handler(support_head=True)) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 04:17:29 -06:00
|
|
|
stdout, stderr = capsys.readouterr()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
2020-11-11 04:17:29 -06:00
|
|
|
assert content == (
|
|
|
|
|
"index.rst:1: [redirected with Found] "
|
2024-04-04 04:25:53 -05:00
|
|
|
f"http://{address}/ to http://{address}/?redirected=1\n"
|
2020-11-11 04:17:29 -06:00
|
|
|
)
|
|
|
|
|
assert stderr == textwrap.dedent(
|
|
|
|
|
"""\
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 302 -
|
|
|
|
|
127.0.0.1 - - [] "HEAD /?redirected=1 HTTP/1.1" 204 -
|
2023-02-17 16:11:14 -06:00
|
|
|
""",
|
2020-11-11 04:17:29 -06:00
|
|
|
)
|
2021-04-29 06:20:01 -05:00
|
|
|
assert warning.getvalue() == ''
|
2020-11-11 04:17:29 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2020-11-11 04:17:29 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2021-04-29 06:20:01 -05:00
|
|
|
def test_follows_redirects_on_GET(app, capsys, warning):
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, make_redirect_handler(support_head=False)) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 04:17:29 -06:00
|
|
|
stdout, stderr = capsys.readouterr()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
2020-11-11 04:17:29 -06:00
|
|
|
assert content == (
|
|
|
|
|
"index.rst:1: [redirected with Found] "
|
2024-04-04 04:25:53 -05:00
|
|
|
f"http://{address}/ to http://{address}/?redirected=1\n"
|
2020-11-11 04:17:29 -06:00
|
|
|
)
|
|
|
|
|
assert stderr == textwrap.dedent(
|
|
|
|
|
"""\
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 405 -
|
|
|
|
|
127.0.0.1 - - [] "GET / HTTP/1.1" 302 -
|
|
|
|
|
127.0.0.1 - - [] "GET /?redirected=1 HTTP/1.1" 204 -
|
2023-02-17 16:11:14 -06:00
|
|
|
""",
|
2020-11-11 04:17:29 -06:00
|
|
|
)
|
2021-04-29 06:20:01 -05:00
|
|
|
assert warning.getvalue() == ''
|
|
|
|
|
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-warn-redirects')
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_linkcheck_allowed_redirects(app: Sphinx, warning: StringIO) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, make_redirect_handler(support_head=False)) as address:
|
2024-06-17 11:04:33 -05:00
|
|
|
app.config.linkcheck_allowed_redirects = {f'http://{address}/.*1': '.*'}
|
2024-04-04 04:25:53 -05:00
|
|
|
compile_linkcheck_allowed_redirects(app, app.config)
|
2021-04-29 09:37:38 -05:00
|
|
|
app.build()
|
|
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2024-03-17 06:06:39 -05:00
|
|
|
rows = [json.loads(l) for l in fp]
|
2022-09-10 11:26:41 -05:00
|
|
|
|
|
|
|
|
assert len(rows) == 2
|
|
|
|
|
records = {row["uri"]: row for row in rows}
|
2024-04-04 04:25:53 -05:00
|
|
|
assert records[f"http://{address}/path1"]["status"] == "working"
|
|
|
|
|
assert records[f"http://{address}/path2"] == {
|
2022-09-10 11:26:41 -05:00
|
|
|
'filename': 'index.rst',
|
|
|
|
|
'lineno': 3,
|
|
|
|
|
'status': 'redirected',
|
|
|
|
|
'code': 302,
|
2024-04-04 04:25:53 -05:00
|
|
|
'uri': f'http://{address}/path2',
|
|
|
|
|
'info': f'http://{address}/?redirected=1',
|
2022-09-10 11:26:41 -05:00
|
|
|
}
|
2021-05-20 12:04:01 -05:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
assert (f"index.rst:3: WARNING: redirect http://{address}/path2 - with Found to "
|
|
|
|
|
f"http://{address}/?redirected=1\n" in strip_colors(warning.getvalue()))
|
2021-05-20 12:04:01 -05:00
|
|
|
assert len(warning.getvalue().splitlines()) == 1
|
2021-04-29 09:37:38 -05:00
|
|
|
|
|
|
|
|
|
2024-03-21 09:08:49 -05:00
|
|
|
class OKHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2020-11-11 11:31:02 -06:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
self.send_response(200, "OK")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2020-11-11 11:31:02 -06:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def do_GET(self):
|
2023-07-22 14:12:32 -05:00
|
|
|
content = b"ok\n"
|
|
|
|
|
self.send_response(200, "OK")
|
|
|
|
|
self.send_header("Content-Length", str(len(content)))
|
|
|
|
|
self.end_headers()
|
|
|
|
|
self.wfile.write(content)
|
2020-11-11 11:31:02 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2023-08-13 17:13:47 -05:00
|
|
|
@mock.patch("sphinx.builders.linkcheck.requests.get", wraps=requests.get)
|
2020-11-09 15:26:02 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-https', freshenv=True)
|
2023-08-13 17:13:47 -05:00
|
|
|
def test_invalid_ssl(get_request, app):
|
2020-11-09 15:26:02 -06:00
|
|
|
# Link indicates SSL should be used (https) but the server does not handle it.
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler) as address:
|
2023-08-13 17:13:47 -05:00
|
|
|
app.build()
|
|
|
|
|
assert not get_request.called
|
2020-11-09 15:26:02 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-09 15:26:02 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content["status"] == "broken"
|
|
|
|
|
assert content["filename"] == "index.rst"
|
|
|
|
|
assert content["lineno"] == 1
|
2024-04-04 04:25:53 -05:00
|
|
|
assert content["uri"] == f"https://{address}/"
|
2020-11-09 15:26:02 -06:00
|
|
|
assert "SSLError" in content["info"]
|
2020-11-11 11:31:02 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2020-11-11 11:31:02 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-https', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_connect_to_selfsigned_fails(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler, tls_enabled=True) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 11:31:02 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-11 11:31:02 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content["status"] == "broken"
|
|
|
|
|
assert content["filename"] == "index.rst"
|
|
|
|
|
assert content["lineno"] == 1
|
2024-04-04 04:25:53 -05:00
|
|
|
assert content["uri"] == f"https://{address}/"
|
2020-11-11 11:31:02 -06:00
|
|
|
assert "[SSL: CERTIFICATE_VERIFY_FAILED]" in content["info"]
|
|
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-https', freshenv=True,
|
|
|
|
|
confoverrides={'tls_verify': False})
|
|
|
|
|
def test_connect_to_selfsigned_with_tls_verify_false(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler, tls_enabled=True) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 11:31:02 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-11 11:31:02 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content == {
|
|
|
|
|
"code": 0,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'https://{address}/',
|
2020-11-11 11:31:02 -06:00
|
|
|
"info": "",
|
|
|
|
|
}
|
|
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-https', freshenv=True,
|
|
|
|
|
confoverrides={'tls_cacerts': CERT_FILE})
|
|
|
|
|
def test_connect_to_selfsigned_with_tls_cacerts(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler, tls_enabled=True) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 11:31:02 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-11 11:31:02 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content == {
|
|
|
|
|
"code": 0,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'https://{address}/',
|
2020-11-11 11:31:02 -06:00
|
|
|
"info": "",
|
|
|
|
|
}
|
|
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2020-11-11 11:31:02 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-https', freshenv=True)
|
2020-11-27 16:10:36 -06:00
|
|
|
def test_connect_to_selfsigned_with_requests_env_var(monkeypatch, app):
|
|
|
|
|
monkeypatch.setenv("REQUESTS_CA_BUNDLE", CERT_FILE)
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler, tls_enabled=True) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-11 11:31:02 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-11 11:31:02 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content == {
|
|
|
|
|
"code": 0,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'https://{address}/',
|
2020-11-11 11:31:02 -06:00
|
|
|
"info": "",
|
|
|
|
|
}
|
2020-11-14 12:56:15 -06:00
|
|
|
|
2020-11-15 02:03:26 -06:00
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver-https', freshenv=True,
|
|
|
|
|
confoverrides={'tls_cacerts': "does/not/exist"})
|
|
|
|
|
def test_connect_to_selfsigned_nonexistent_cert_file(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, OKHandler, tls_enabled=True) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-14 12:56:15 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-14 12:56:15 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content == {
|
|
|
|
|
"code": 0,
|
|
|
|
|
"status": "broken",
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'https://{address}/',
|
2020-11-14 12:56:15 -06:00
|
|
|
"info": "Could not find a suitable TLS CA certificate bundle, invalid path: does/not/exist",
|
|
|
|
|
}
|
2020-11-22 10:52:44 -06:00
|
|
|
|
|
|
|
|
|
2024-03-21 09:08:49 -05:00
|
|
|
class InfiniteRedirectOnHeadHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
self.send_response(302, "Found")
|
2024-04-04 04:25:53 -05:00
|
|
|
self.send_header("Location", "/")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2022-09-10 11:26:41 -05:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def do_GET(self):
|
2023-07-22 14:12:32 -05:00
|
|
|
content = b"ok\n"
|
2022-09-10 11:26:41 -05:00
|
|
|
self.send_response(200, "OK")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", str(len(content)))
|
2022-09-10 11:26:41 -05:00
|
|
|
self.end_headers()
|
2023-07-22 14:12:32 -05:00
|
|
|
self.wfile.write(content)
|
|
|
|
|
self.close_connection = True # we don't expect the client to read this response body
|
2022-09-10 11:26:41 -05:00
|
|
|
|
|
|
|
|
|
2020-11-22 10:52:44 -06:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2022-09-10 11:26:41 -05:00
|
|
|
def test_TooManyRedirects_on_HEAD(app, monkeypatch):
|
|
|
|
|
import requests.sessions
|
2020-11-22 10:52:44 -06:00
|
|
|
|
2022-09-10 11:26:41 -05:00
|
|
|
monkeypatch.setattr(requests.sessions, "DEFAULT_REDIRECT_LIMIT", 5)
|
2020-11-22 10:52:44 -06:00
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, InfiniteRedirectOnHeadHandler) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2020-11-22 10:52:44 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2020-11-22 10:52:44 -06:00
|
|
|
content = json.load(fp)
|
|
|
|
|
assert content == {
|
|
|
|
|
"code": 0,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'http://{address}/',
|
2020-11-22 10:52:44 -06:00
|
|
|
"info": "",
|
|
|
|
|
}
|
2020-10-04 14:40:14 -05:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def make_retry_after_handler(responses):
|
2024-03-21 09:08:49 -05:00
|
|
|
class RetryAfterHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2020-10-04 14:40:14 -05:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
status, retry_after = responses.pop(0)
|
|
|
|
|
self.send_response(status)
|
|
|
|
|
if retry_after:
|
|
|
|
|
self.send_header('Retry-After', retry_after)
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2020-10-04 14:40:14 -05:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
def log_date_time_string(self):
|
|
|
|
|
"""Strip date and time from logged messages for assertions."""
|
|
|
|
|
return ""
|
|
|
|
|
|
|
|
|
|
return RetryAfterHandler
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
|
|
|
|
def test_too_many_requests_retry_after_int_delay(app, capsys, status):
|
2024-04-04 04:25:53 -05:00
|
|
|
with (
|
|
|
|
|
serve_application(app, make_retry_after_handler([(429, "0"), (200, None)])) as address,
|
|
|
|
|
mock.patch("sphinx.builders.linkcheck.DEFAULT_DELAY", 0),
|
|
|
|
|
mock.patch("sphinx.builders.linkcheck.QUEUE_POLL_SECS", 0.01),
|
|
|
|
|
):
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
2020-10-04 14:40:14 -05:00
|
|
|
assert json.loads(content) == {
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"code": 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'http://{address}/',
|
2020-10-04 14:40:14 -05:00
|
|
|
"info": "",
|
|
|
|
|
}
|
2024-04-04 04:25:53 -05:00
|
|
|
rate_limit_log = f"-rate limited- http://{address}/ | sleeping...\n"
|
2024-03-23 18:43:54 -05:00
|
|
|
assert rate_limit_log in strip_colors(status.getvalue())
|
2020-10-04 14:40:14 -05:00
|
|
|
_stdout, stderr = capsys.readouterr()
|
|
|
|
|
assert stderr == textwrap.dedent(
|
|
|
|
|
"""\
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 429 -
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 200 -
|
2023-02-17 16:11:14 -06:00
|
|
|
""",
|
2020-10-04 14:40:14 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
2023-08-27 22:39:13 -05:00
|
|
|
@pytest.mark.parametrize('tz', [None, 'GMT', 'GMT+3', 'GMT-3'])
|
2020-10-04 14:40:14 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2023-08-27 22:39:13 -05:00
|
|
|
def test_too_many_requests_retry_after_HTTP_date(tz, app, monkeypatch, capsys):
|
2023-08-13 13:39:08 -05:00
|
|
|
retry_after = wsgiref.handlers.format_date_time(time.time())
|
2023-08-27 22:39:13 -05:00
|
|
|
|
|
|
|
|
with monkeypatch.context() as m:
|
|
|
|
|
if tz is not None:
|
|
|
|
|
m.setenv('TZ', tz)
|
|
|
|
|
if sys.platform != "win32":
|
|
|
|
|
time.tzset()
|
|
|
|
|
m.setattr(sphinx.util.http_date, '_GMT_OFFSET',
|
|
|
|
|
float(time.localtime().tm_gmtoff))
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, make_retry_after_handler([(429, retry_after), (200, None)])) as address:
|
2023-08-27 22:39:13 -05:00
|
|
|
app.build()
|
|
|
|
|
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
2020-10-04 14:40:14 -05:00
|
|
|
assert json.loads(content) == {
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"code": 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'http://{address}/',
|
2020-10-04 14:40:14 -05:00
|
|
|
"info": "",
|
|
|
|
|
}
|
|
|
|
|
_stdout, stderr = capsys.readouterr()
|
|
|
|
|
assert stderr == textwrap.dedent(
|
|
|
|
|
"""\
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 429 -
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 200 -
|
2023-02-17 16:11:14 -06:00
|
|
|
""",
|
2020-10-04 14:40:14 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
|
|
|
|
def test_too_many_requests_retry_after_without_header(app, capsys):
|
2024-04-04 04:25:53 -05:00
|
|
|
with (
|
|
|
|
|
serve_application(app, make_retry_after_handler([(429, None), (200, None)])) as address,
|
|
|
|
|
mock.patch("sphinx.builders.linkcheck.DEFAULT_DELAY", 0),
|
|
|
|
|
):
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
2020-10-04 14:40:14 -05:00
|
|
|
assert json.loads(content) == {
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"code": 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'http://{address}/',
|
2020-10-04 14:40:14 -05:00
|
|
|
"info": "",
|
|
|
|
|
}
|
|
|
|
|
_stdout, stderr = capsys.readouterr()
|
|
|
|
|
assert stderr == textwrap.dedent(
|
|
|
|
|
"""\
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 429 -
|
|
|
|
|
127.0.0.1 - - [] "HEAD / HTTP/1.1" 200 -
|
2023-02-17 16:11:14 -06:00
|
|
|
""",
|
2020-10-04 14:40:14 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
2024-04-12 17:18:22 -05:00
|
|
|
@pytest.mark.sphinx(
|
|
|
|
|
'linkcheck', testroot='linkcheck-localserver', freshenv=True,
|
|
|
|
|
confoverrides={
|
|
|
|
|
'linkcheck_report_timeouts_as_broken': False,
|
|
|
|
|
'linkcheck_timeout': 0.01,
|
|
|
|
|
}
|
|
|
|
|
)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_requests_timeout(app: Sphinx) -> None:
|
2024-03-21 09:08:49 -05:00
|
|
|
class DelayedResponseHandler(BaseHTTPRequestHandler):
|
2024-01-13 02:14:06 -06:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
|
|
|
|
def do_GET(self):
|
|
|
|
|
time.sleep(0.2) # wait before sending any response data
|
|
|
|
|
self.send_response(200, "OK")
|
|
|
|
|
self.send_header("Content-Length", "0")
|
|
|
|
|
self.end_headers()
|
|
|
|
|
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, DelayedResponseHandler):
|
2024-01-13 02:14:06 -06:00
|
|
|
app.build()
|
|
|
|
|
|
|
|
|
|
with open(app.outdir / "output.json", encoding="utf-8") as fp:
|
|
|
|
|
content = json.load(fp)
|
|
|
|
|
|
|
|
|
|
assert content["status"] == "timeout"
|
|
|
|
|
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True,
|
|
|
|
|
confoverrides={'linkcheck_rate_limit_timeout': 0.0})
|
|
|
|
|
def test_too_many_requests_user_timeout(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, make_retry_after_handler([(429, None)])) as address:
|
2021-01-27 11:52:00 -06:00
|
|
|
app.build()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
2020-10-04 14:40:14 -05:00
|
|
|
assert json.loads(content) == {
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
|
|
|
|
"status": "broken",
|
|
|
|
|
"code": 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'http://{address}/',
|
|
|
|
|
"info": f"429 Client Error: Too Many Requests for url: http://{address}/",
|
2020-10-04 14:40:14 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class FakeResponse:
|
2022-12-16 05:41:54 -06:00
|
|
|
headers: dict[str, str] = {}
|
2020-10-04 14:40:14 -05:00
|
|
|
url = "http://localhost/"
|
|
|
|
|
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_limit_rate_default_sleep(app: Sphinx) -> None:
|
2023-07-22 18:01:41 -05:00
|
|
|
worker = HyperlinkAvailabilityCheckWorker(app.config, Queue(), Queue(), {})
|
2020-10-04 14:40:14 -05:00
|
|
|
with mock.patch('time.time', return_value=0.0):
|
2023-07-20 15:14:00 -05:00
|
|
|
next_check = worker.limit_rate(FakeResponse.url, FakeResponse.headers.get("Retry-After"))
|
2020-10-04 14:40:14 -05:00
|
|
|
assert next_check == 60.0
|
|
|
|
|
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
@pytest.mark.sphinx(confoverrides={'linkcheck_rate_limit_timeout': 0.0})
|
|
|
|
|
def test_limit_rate_user_max_delay(app: Sphinx) -> None:
|
2023-07-22 18:01:41 -05:00
|
|
|
worker = HyperlinkAvailabilityCheckWorker(app.config, Queue(), Queue(), {})
|
2023-07-20 15:14:00 -05:00
|
|
|
next_check = worker.limit_rate(FakeResponse.url, FakeResponse.headers.get("Retry-After"))
|
2020-10-04 14:40:14 -05:00
|
|
|
assert next_check is None
|
|
|
|
|
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_limit_rate_doubles_previous_wait_time(app: Sphinx) -> None:
|
2021-02-06 12:07:40 -06:00
|
|
|
rate_limits = {"localhost": RateLimit(60.0, 0.0)}
|
2023-07-22 18:01:41 -05:00
|
|
|
worker = HyperlinkAvailabilityCheckWorker(app.config, Queue(), Queue(), rate_limits)
|
2020-10-04 14:40:14 -05:00
|
|
|
with mock.patch('time.time', return_value=0.0):
|
2023-07-20 15:14:00 -05:00
|
|
|
next_check = worker.limit_rate(FakeResponse.url, FakeResponse.headers.get("Retry-After"))
|
2020-10-04 14:40:14 -05:00
|
|
|
assert next_check == 120.0
|
|
|
|
|
|
|
|
|
|
|
2024-06-24 12:26:24 -05:00
|
|
|
@pytest.mark.sphinx(confoverrides={'linkcheck_rate_limit_timeout': 90})
|
|
|
|
|
def test_limit_rate_clips_wait_time_to_max_time(app: Sphinx, warning: StringIO) -> None:
|
2021-02-06 12:07:40 -06:00
|
|
|
rate_limits = {"localhost": RateLimit(60.0, 0.0)}
|
2023-07-22 18:01:41 -05:00
|
|
|
worker = HyperlinkAvailabilityCheckWorker(app.config, Queue(), Queue(), rate_limits)
|
2020-10-04 14:40:14 -05:00
|
|
|
with mock.patch('time.time', return_value=0.0):
|
2023-07-20 15:14:00 -05:00
|
|
|
next_check = worker.limit_rate(FakeResponse.url, FakeResponse.headers.get("Retry-After"))
|
2020-10-04 14:40:14 -05:00
|
|
|
assert next_check == 90.0
|
2024-06-24 12:26:24 -05:00
|
|
|
assert warning.getvalue() == ''
|
2020-10-04 14:40:14 -05:00
|
|
|
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
@pytest.mark.sphinx(confoverrides={'linkcheck_rate_limit_timeout': 90.0})
|
2024-06-24 12:26:24 -05:00
|
|
|
def test_limit_rate_bails_out_after_waiting_max_time(app: Sphinx, warning: StringIO) -> None:
|
2021-02-06 12:07:40 -06:00
|
|
|
rate_limits = {"localhost": RateLimit(90.0, 0.0)}
|
2023-07-22 18:01:41 -05:00
|
|
|
worker = HyperlinkAvailabilityCheckWorker(app.config, Queue(), Queue(), rate_limits)
|
2023-07-20 15:14:00 -05:00
|
|
|
next_check = worker.limit_rate(FakeResponse.url, FakeResponse.headers.get("Retry-After"))
|
2020-10-04 14:40:14 -05:00
|
|
|
assert next_check is None
|
2024-06-24 12:26:24 -05:00
|
|
|
assert warning.getvalue() == ''
|
2021-06-07 12:57:15 -05:00
|
|
|
|
|
|
|
|
|
2023-07-23 10:24:12 -05:00
|
|
|
@mock.patch('sphinx.util.requests.requests.Session.get_adapter')
|
|
|
|
|
def test_connection_contention(get_adapter, app, capsys):
|
|
|
|
|
# Create a shared, but limited-size, connection pool
|
|
|
|
|
import requests
|
|
|
|
|
get_adapter.return_value = requests.adapters.HTTPAdapter(pool_maxsize=1)
|
|
|
|
|
|
|
|
|
|
# Set an upper-bound on socket timeouts globally
|
|
|
|
|
import socket
|
|
|
|
|
socket.setdefaulttimeout(5)
|
|
|
|
|
|
|
|
|
|
# Create parallel consumer threads
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, make_redirect_handler(support_head=True)) as address:
|
|
|
|
|
|
|
|
|
|
# Place a workload into the linkcheck queue
|
|
|
|
|
link_count = 10
|
2024-05-14 21:13:57 -05:00
|
|
|
wqueue: Queue[CheckRequest] = Queue()
|
|
|
|
|
rqueue: Queue[CheckResult] = Queue()
|
2024-04-04 04:25:53 -05:00
|
|
|
for _ in range(link_count):
|
|
|
|
|
wqueue.put(CheckRequest(0, Hyperlink(f"http://{address}", "test", "test.rst", 1)))
|
|
|
|
|
|
2024-05-14 21:13:57 -05:00
|
|
|
begin = time.time()
|
|
|
|
|
checked: list[CheckResult] = []
|
2023-07-23 10:24:12 -05:00
|
|
|
threads = [
|
|
|
|
|
HyperlinkAvailabilityCheckWorker(
|
|
|
|
|
config=app.config,
|
|
|
|
|
rqueue=rqueue,
|
|
|
|
|
wqueue=wqueue,
|
|
|
|
|
rate_limits={},
|
|
|
|
|
)
|
|
|
|
|
for _ in range(10)
|
|
|
|
|
]
|
|
|
|
|
for thread in threads:
|
|
|
|
|
thread.start()
|
|
|
|
|
while time.time() < begin + 5 and len(checked) < link_count:
|
|
|
|
|
checked.append(rqueue.get(timeout=5))
|
|
|
|
|
for thread in threads:
|
|
|
|
|
thread.join(timeout=0)
|
|
|
|
|
|
|
|
|
|
# Ensure that all items were consumed within the time limit
|
|
|
|
|
_, stderr = capsys.readouterr()
|
|
|
|
|
assert len(checked) == link_count
|
|
|
|
|
assert "TimeoutError" not in stderr
|
|
|
|
|
|
|
|
|
|
|
2024-03-21 09:08:49 -05:00
|
|
|
class ConnectionResetHandler(BaseHTTPRequestHandler):
|
2023-07-22 14:12:32 -05:00
|
|
|
protocol_version = "HTTP/1.1"
|
|
|
|
|
|
2021-06-09 22:40:40 -05:00
|
|
|
def do_HEAD(self):
|
2023-07-22 14:12:32 -05:00
|
|
|
self.close_connection = True
|
2021-06-09 22:40:40 -05:00
|
|
|
|
|
|
|
|
def do_GET(self):
|
|
|
|
|
self.send_response(200, "OK")
|
2023-07-22 14:12:32 -05:00
|
|
|
self.send_header("Content-Length", "0")
|
2021-06-09 22:40:40 -05:00
|
|
|
self.end_headers()
|
|
|
|
|
|
|
|
|
|
|
2021-06-10 10:41:47 -05:00
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-localserver', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_get_after_head_raises_connection_error(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, ConnectionResetHandler) as address:
|
2021-06-09 22:40:40 -05:00
|
|
|
app.build()
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.txt').read_text(encoding='utf8')
|
2021-06-07 15:50:42 -05:00
|
|
|
assert not content
|
2022-04-26 21:04:19 -05:00
|
|
|
content = (app.outdir / 'output.json').read_text(encoding='utf8')
|
2021-06-10 10:43:14 -05:00
|
|
|
assert json.loads(content) == {
|
|
|
|
|
"filename": "index.rst",
|
|
|
|
|
"lineno": 1,
|
|
|
|
|
"status": "working",
|
|
|
|
|
"code": 0,
|
2024-04-04 04:25:53 -05:00
|
|
|
"uri": f'http://{address}/',
|
2021-06-10 10:43:14 -05:00
|
|
|
"info": "",
|
|
|
|
|
}
|
2021-11-26 11:14:29 -06:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.sphinx('linkcheck', testroot='linkcheck-documents_exclude', freshenv=True)
|
2024-05-14 21:13:57 -05:00
|
|
|
def test_linkcheck_exclude_documents(app: Sphinx) -> None:
|
2024-04-04 04:25:53 -05:00
|
|
|
with serve_application(app, DefaultsHandler):
|
2023-04-06 16:24:49 -05:00
|
|
|
app.build()
|
2021-11-26 11:14:29 -06:00
|
|
|
|
2022-04-17 20:32:06 -05:00
|
|
|
with open(app.outdir / 'output.json', encoding='utf-8') as fp:
|
2021-11-26 11:14:29 -06:00
|
|
|
content = [json.loads(record) for record in fp]
|
|
|
|
|
|
2024-03-25 05:48:03 -05:00
|
|
|
assert len(content) == 2
|
|
|
|
|
assert {
|
|
|
|
|
'filename': 'broken_link.rst',
|
|
|
|
|
'lineno': 4,
|
|
|
|
|
'status': 'ignored',
|
|
|
|
|
'code': 0,
|
|
|
|
|
'uri': 'https://www.sphinx-doc.org/this-is-a-broken-link',
|
|
|
|
|
'info': 'broken_link matched ^broken_link$ from linkcheck_exclude_documents',
|
|
|
|
|
} in content
|
|
|
|
|
assert {
|
|
|
|
|
'filename': 'br0ken_link.rst',
|
|
|
|
|
'lineno': 4,
|
|
|
|
|
'status': 'ignored',
|
|
|
|
|
'code': 0,
|
|
|
|
|
'uri': 'https://www.sphinx-doc.org/this-is-another-broken-link',
|
|
|
|
|
'info': 'br0ken_link matched br[0-9]ken_link from linkcheck_exclude_documents',
|
|
|
|
|
} in content
|
