1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
|
import pathlib
import shutil
import tempfile
import textwrap
from xml.etree import ElementTree
import pytest
import responses
HERE = pathlib.Path(__file__).parent
REPO_ROOT = HERE.parent.parent.parent
#
# sphinx's intersphinx plugin loads an inventory from the docs.python.org docs for
# linkage each time it runs
# we use a cached copy so that:
# - tests don't require the network
# - we don't put unnecessary load on the python docs site
# - tests will be faster
#
@pytest.fixture(autouse=True)
def _mock_intersphinx_object_inventory_response(mocked_responses):
objects_inv = HERE / "objects.inv"
responses.get(
url="https://docs.python.org/3/objects.inv", body=objects_inv.read_bytes()
)
class SingleFileSphinxRunner:
"""
A SingleFileSphinxRunner runs sphinx with the full globus-sdk config loaded, against
some target RST source, and returns the parsed XML it produces.
For content with sphinx-specific features like `py:meth` and other sphinx domain
features, this is the best way of ensuring we get a realistic build with good
outputs.
Primarily, usage should use ``to_etree`` like so:
>>> def test_my_thing(sphinx_runner):
>>> etree_element = sphinx_runner.to_etree(
>>> ":py:class:`globus_sdk.TimersClient`"
>>> )
Because it's often difficult to know the structure of the produced XML, if
`debug=True` is passed, this will print the full XML to stdout, for you to
see in your test outputs.
"""
def to_etree(self, content, dedent=True, debug=False):
from sphinx.cmd.build import build_main as sphinx_main
if dedent:
content = textwrap.dedent(content)
source_dir = tempfile.TemporaryDirectory()
out_dir = tempfile.TemporaryDirectory()
with source_dir, out_dir:
self._prepare_file(source_dir.name, content, "index.rst")
self._prepare_sphinx_config(source_dir.name)
sphinx_rc = sphinx_main([source_dir.name, out_dir.name, "-b", "xml"])
assert sphinx_rc == 0
output_xml = pathlib.Path(out_dir.name) / "index.xml"
xml_text = output_xml.read_text(encoding="utf-8")
if debug:
print("--- debug from sphinx runner ---")
print()
print(xml_text)
print()
print("--- end debug from sphinx runner ---")
return ElementTree.fromstring(xml_text)
def ensure_failure(self, content, dedent=True, debug=False):
from sphinx.cmd.build import build_main as sphinx_main
if dedent:
content = textwrap.dedent(content)
source_dir = tempfile.TemporaryDirectory()
out_dir = tempfile.TemporaryDirectory()
with source_dir, out_dir:
self._prepare_file(source_dir.name, content, "index.rst")
self._prepare_sphinx_config(source_dir.name)
sphinx_rc = sphinx_main([source_dir.name, out_dir.name, "-b", "xml"])
assert sphinx_rc != 0
def _prepare_file(self, source_dir, content, filename):
source_path = pathlib.Path(source_dir) / filename
source_path.write_text(content, encoding="utf-8")
def _prepare_sphinx_config(self, source_dir):
shutil.copy(
REPO_ROOT / "docs" / "conf.py", pathlib.Path(source_dir) / "conf.py"
)
class DocutilsRunner:
"""
A DocutilsRunner is a direct user of docutils, with no special sphinx customizations
applied.
It can fail on sphinx-specific content, but it lets us directly test underlying
docutils behaviors.
Primarily, usage should use ``to_etree`` like so:
>>> def test_my_thing(docutils_runner):
>>> etree_element = docutils_runner.to_etree(
>>> ".. note:: a note directive"
>>> )
Because it's often difficult to know the structure of the produced XML, if
`debug=True` is passed, this will print the full XML to stdout, for you to
see in your test outputs.
"""
def __init__(self) -> None:
from docutils.frontend import get_default_settings
from docutils.parsers.rst import Parser
self.parser = Parser()
self.settings = get_default_settings(self.parser)
# set the halt_level so that docutils won't "power through" noncritical errors
# https://docutils.sourceforge.io/docs/user/config.html#halt-level
self.settings.halt_level = 2
def to_etree(self, content, dedent=True, debug=False):
if dedent:
content = textwrap.dedent(content)
doc = self.new_doc()
xml_string = self.make_xml(content, doc)
if debug:
print("--- debug from docutils runner ---")
print()
print(xml_string)
print()
print("--- end debug from docutils runner ---")
etree = ElementTree.fromstring(xml_string)
return etree
def new_doc(self, doc_source="TEST"):
from docutils.utils import new_document
return new_document(doc_source, self.settings.copy())
def make_xml(self, source, doc):
# docutils produces 'xml.dom.minidom'
# for a nicer API, convert to XML so we can reparse as etree
self.parser.parse(source, doc)
dom = doc.asdom()
xml_string = dom.toxml()
# also, unlink the dom object to ensure we don't get cyclic references
# which make GC fail to collect these (see xml.dom.minidom docs for detail)
dom.unlink()
return xml_string
@pytest.fixture
def sphinx_runner():
pytest.importorskip("sphinx", reason="testing sphinx extension needs sphinx")
return SingleFileSphinxRunner()
@pytest.fixture
def docutils_runner():
pytest.importorskip("docutils", reason="testing sphinx extension needs docutils")
return DocutilsRunner()
@pytest.fixture
def sphinxext():
"""
Provide the extension subpackage as a fixture so that we can properly capture
skip requirements and avoid awkward import requirements.
"""
pytest.importorskip("docutils", reason="testing sphinx extension needs docutils")
import globus_sdk._internal.extensions.sphinxext
return globus_sdk._internal.extensions.sphinxext
@pytest.fixture
def register_temporary_directive():
pytest.importorskip("docutils", reason="testing sphinx extension needs docutils")
from docutils.parsers.rst.directives import _directives, register_directive
registered_names = []
def func(name, directive):
registered_names.append(name)
register_directive(name, directive)
yield func
for name in registered_names:
del _directives[name]
|