tests: Fix tests

Author: pawamoy

tests/conftest.py

@@ -1,14 +1,56 @@
 """Configuration for the pytest test suite."""
 
+from __future__ import annotations
+
+from collections import ChainMap
+from typing import TYPE_CHECKING, Any
+
 import pytest
-from markdown import Markdown
-from mkdocstrings_handlers.python.handler import PythonHandler
+from markdown.core import Markdown
+from mkdocs.config.defaults import MkDocsConfig
+
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+    from pathlib import Path
+
+    from mkdocs import config
+    from mkdocstrings_handlers.python.handler import PythonHandler
+
+
+@pytest.fixture(name="mkdocs_conf")
+def fixture_mkdocs_conf(request: pytest.FixtureRequest, tmp_path: Path) -> Iterator[config.Config]:
+    """Yield a MkDocs configuration object."""
+    conf = MkDocsConfig()
+    while hasattr(request, "_parent_request") and hasattr(request._parent_request, "_parent_request"):
+        request = request._parent_request
+
+    conf_dict = {
+        "site_name": "foo",
+        "site_url": "https://example.org/",
+        "site_dir": str(tmp_path),
+        "plugins": [{"mkdocstrings": {"default_handler": "python"}}],
+        **getattr(request, "param", {}),
+    }
+    # Re-create it manually as a workaround for https://github.com/mkdocs/mkdocs/issues/2289
+    mdx_configs: dict[str, Any] = dict(ChainMap(*conf_dict.get("markdown_extensions", [])))
+
+    conf.load_dict(conf_dict)
+    assert conf.validate() == ([], [])
+
+    conf["mdx_configs"] = mdx_configs
+    conf["markdown_extensions"].insert(0, "toc")  # Guaranteed to be added by MkDocs.
+
+    conf = conf["plugins"]["mkdocstrings"].on_config(conf)
+    conf = conf["plugins"]["autorefs"].on_config(conf)
+    yield conf
+    conf["plugins"]["mkdocstrings"].on_post_build(conf)
 
 
 @pytest.fixture(name="python_handler")
-def fixture_python_handler() -> PythonHandler:
+def fixture_python_handler(mkdocs_conf: MkDocsConfig) -> PythonHandler:
     """Return a PythonHandler instance."""
-    handler = PythonHandler("python", "material")
-    handler.update_env(md=Markdown(extensions=["toc"]), config={})
+    handlers = mkdocs_conf.plugins["mkdocstrings"].handlers  # type: ignore[attr-defined]
+    handler = handlers.get_handler("python")
+    handler._update_env(md=Markdown(extensions=["toc"]))
     handler.env.filters["convert_markdown"] = lambda *args, **kwargs: str(args) + str(kwargs)
     return handler

tests/test_extension.py

@@ -121,7 +121,7 @@ class Model(BaseModel):
         modules={"__init__.py": code},
         extensions=Extensions(PydanticExtension(schema=False)),
     ) as package:
-        python_handler.render(package["Model"], {})  # Assert no errors.
+        python_handler.render(package["Model"], python_handler.get_options({}))  # Assert no errors.
 
 
 def test_not_crashing_on_dynamic_field_description(caplog: pytest.LogCaptureFixture) -> None: