fix: serve_docs

BobTheBuidler · BobTheBuidler · commit a9106766d275 · 2024-11-02T18:41:58.000Z
diff --git a/sphinx_autodoc_vyper/server.py b/sphinx_autodoc_vyper/server.py
@@ -11,12 +11,14 @@
 def serve_docs(build_dir: Path, port: int = 8000) -> NoReturn:  # type: ignore [misc]
     """Serve the documentation on a local development server."""
 
-    if not build_dir.exists():
+    html_dir = build_dir / "html"
+
+    if not html_dir.exists():
         raise FileNotFoundError(
             "Documentation not found. Run 'sphinx-autodoc-vyper' first to generate the documentation."
         )
 
-    os.chdir(build_dir)
+    os.chdir(html_dir)
 
     handler = http.server.SimpleHTTPRequestHandler
 
diff --git a/tests/test_server.py b/tests/test_server.py
@@ -14,9 +14,11 @@
 def test_server_start(tmp_path: Path) -> None:
     """Test server startup and accessibility."""
     # Create mock build directory
-    build_dir = tmp_path / "docs" / "_build" / "html"
-    build_dir.mkdir(parents=True)
-    (build_dir / "index.html").write_text("<html><body>Test</body></html>")
+    build_dir = tmp_path / "docs" / "_build"
+
+    html_dir = build_dir / "html"
+    html_dir.mkdir(parents=True)
+    (html_dir / "index.html").write_text("<html><body>Test</body></html>")
 
     # Generate documentation
     cli._main(contracts_dir="", output_dir=str(build_dir))
