Update custom_docs_ui/tutorial001 and tests

docs/de/docs/how-to/custom-docs-ui-assets.md

@@ -18,7 +18,7 @@ Der erste Schritt besteht darin, die automatischen Dokumentationen zu deaktivier
 
 Um diese zu deaktivieren, setzen Sie deren URLs beim Erstellen Ihrer `FastAPI`-App auf `None`:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[8] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[11] *}
 
 ### Die benutzerdefinierten Dokumentationen hinzufügen
 
@@ -34,7 +34,7 @@ Sie können die internen Funktionen von FastAPI wiederverwenden, um die HTML-Sei
 
 Und genau so für ReDoc ...
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[2:6,11:19,22:24,27:33] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[3:7,14:23,26:28,31:38] *}
 
 /// tip | Tipp
 
@@ -50,7 +50,7 @@ Swagger UI erledigt das hinter den Kulissen für Sie, benötigt aber diesen „U
 
 Um nun testen zu können, ob alles funktioniert, erstellen Sie eine *Pfadoperation*:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[36:38] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[41:43] *}
 
 ### Es ausprobieren
 

docs/en/docs/how-to/custom-docs-ui-assets.md

@@ -18,7 +18,7 @@ The first step is to disable the automatic docs, as by default, those use the de
 
 To disable them, set their URLs to `None` when creating your `FastAPI` app:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[8] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[11] *}
 
 ### Include the custom docs
 
@@ -34,7 +34,7 @@ You can reuse FastAPI's internal functions to create the HTML pages for the docs
 
 And similarly for ReDoc...
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[2:6,11:19,22:24,27:33] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[3:7,14:23,26:28,31:38] *}
 
 /// tip
 
@@ -50,7 +50,7 @@ Swagger UI will handle it behind the scenes for you, but it needs this "redirect
 
 Now, to be able to test that everything works, create a *path operation*:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[36:38] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[41:43] *}
 
 ### Test it
 

docs/es/docs/how-to/custom-docs-ui-assets.md

@@ -18,7 +18,7 @@ El primer paso es desactivar la documentación automática, ya que por defecto,
 
 Para desactivarlos, establece sus URLs en `None` cuando crees tu aplicación de `FastAPI`:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[8] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[11] *}
 
 ### Incluye la documentación personalizada
 
@@ -34,7 +34,7 @@ Puedes reutilizar las funciones internas de FastAPI para crear las páginas HTML
 
 Y de manera similar para ReDoc...
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[2:6,11:19,22:24,27:33] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[3:7,14:23,26:28,31:38] *}
 
 /// tip | Consejo
 
@@ -50,7 +50,7 @@ Swagger UI lo manejará detrás de escena para ti, pero necesita este auxiliar d
 
 Ahora, para poder probar que todo funciona, crea una *path operation*:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[36:38] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[41:43] *}
 
 ### Pruébalo
 

docs/pt/docs/how-to/custom-docs-ui-assets.md

@@ -18,7 +18,7 @@ O primeiro passo é desativar a documentação automática, pois por padrão, el
 
 Para desativá-los, defina suas URLs como `None` ao criar seu aplicativo `FastAPI`:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[8] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[11] *}
 
 ### Incluir a documentação personalizada
 
@@ -34,7 +34,7 @@ Você pode reutilizar as funções internas do FastAPI para criar as páginas HT
 
 E de forma semelhante para o ReDoc...
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[2:6,11:19,22:24,27:33] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[3:7,14:23,26:28,31:38] *}
 
 /// tip | Dica
 
@@ -50,7 +50,7 @@ Swagger UI lidará com isso nos bastidores para você, mas ele precisa desse aux
 
 Agora, para poder testar se tudo funciona, crie uma *operação de rota*:
 
-{* ../../docs_src/custom_docs_ui/tutorial001.py hl[36:38] *}
+{* ../../docs_src/custom_docs_ui/tutorial001.py hl[41:43] *}
 
 ### Teste
 

docs_src/custom_docs_ui/tutorial001.py

@@ -1,19 +1,24 @@
-from fastapi import FastAPI
+import os
+
+from fastapi import FastAPI, Request
 from fastapi.openapi.docs import (
     get_redoc_html,
     get_swagger_ui_html,
     get_swagger_ui_oauth2_redirect_html,
 )
 
-app = FastAPI(docs_url=None, redoc_url=None)
+root_path = os.getenv("ROOT_PATH", "")
+
+app = FastAPI(docs_url=None, redoc_url=None, root_path=root_path)
 
 
 @app.get("/docs", include_in_schema=False)
-async def custom_swagger_ui_html():
+async def custom_swagger_ui_html(req: Request):
+    root_path = req.scope.get("root_path", "").rstrip("/")
     return get_swagger_ui_html(
-        openapi_url=app.openapi_url,
+        openapi_url=f"{root_path}{app.openapi_url}",
         title=app.title + " - Swagger UI",
-        oauth2_redirect_url=app.swagger_ui_oauth2_redirect_url,
+        oauth2_redirect_url=f"{root_path}{app.swagger_ui_oauth2_redirect_url}",
         swagger_js_url="https://unpkg.com/swagger-ui-dist@5/swagger-ui-bundle.js",
         swagger_css_url="https://unpkg.com/swagger-ui-dist@5/swagger-ui.css",
     )
@@ -25,9 +30,10 @@ async def swagger_ui_redirect():
 
 
 @app.get("/redoc", include_in_schema=False)
-async def redoc_html():
+async def redoc_html(req: Request):
+    root_path = req.scope.get("root_path", "").rstrip("/")
     return get_redoc_html(
-        openapi_url=app.openapi_url,
+        openapi_url=f"{root_path}{app.openapi_url}",
         title=app.title + " - ReDoc",
         redoc_js_url="https://unpkg.com/redoc@2/bundles/redoc.standalone.js",
     )

tests/test_tutorial/test_custom_docs_ui/test_tutorial001.py

@@ -1,42 +1,91 @@
+import importlib
 import os
 from pathlib import Path
+from typing import TypedDict
 
 import pytest
 from fastapi.testclient import TestClient
 
 
-@pytest.fixture(scope="module")
-def client():
+class Params(TypedDict):
+    app_root_path: str
+    asgi_root_path: str
+    request_prefix: str
+
+
+@pytest.fixture(
+    params=[
+        Params(app_root_path="", asgi_root_path="", request_prefix=""),
+        Params(app_root_path="/api", asgi_root_path="", request_prefix="/api"),
+        Params(app_root_path="/api", asgi_root_path="", request_prefix=""),
+        Params(app_root_path="", asgi_root_path="/api", request_prefix="/api"),
+        Params(app_root_path="", asgi_root_path="/api", request_prefix=""),
+    ],
+    ids=[
+        "Without root_path, request without prefix",
+        "FastAPI(root_path=root_path), request with prefix",
+        "FastAPI(root_path=root_path), request without prefix",
+        "TestClient(root_path=root_path), request with prefix",
+        "TestClient(root_path=root_path), request without prefix",
+    ],
+)
+def params(request: pytest.FixtureRequest):
+    return request.param
+
+
+@pytest.fixture
+def client(params: Params, monkeypatch):
     static_dir: Path = Path(os.getcwd()) / "static"
     print(static_dir)
     static_dir.mkdir(exist_ok=True)
-    from docs_src.custom_docs_ui.tutorial001 import app
+    monkeypatch.setenv("ROOT_PATH", params["app_root_path"])
+    from docs_src.custom_docs_ui import tutorial001
+
+    importlib.reload(tutorial001)
+    app = tutorial001.app
 
-    with TestClient(app) as client:
+    with TestClient(app, root_path=params["asgi_root_path"]) as client:
         yield client
+
     static_dir.rmdir()
 
 
-def test_swagger_ui_html(client: TestClient):
-    response = client.get("/docs")
+def test_swagger_ui_html(client: TestClient, params: Params):
+    request_prefix = params["request_prefix"]
+    root_path = params["app_root_path"] or params["asgi_root_path"]
+
+    response = client.get(f"{request_prefix}/docs")
     assert response.status_code == 200, response.text
     assert "https://unpkg.com/swagger-ui-dist@5/swagger-ui-bundle.js" in response.text
     assert "https://unpkg.com/swagger-ui-dist@5/swagger-ui.css" in response.text
+    assert f"{root_path}/docs/oauth2-redirect" in response.text
+
+    response = client.get(f"{request_prefix}/openapi.json")
+    assert response.status_code == 200
+
 
+def test_swagger_ui_oauth2_redirect_html(client: TestClient, params: Params):
+    request_prefix = params["request_prefix"]
 
-def test_swagger_ui_oauth2_redirect_html(client: TestClient):
-    response = client.get("/docs/oauth2-redirect")
+    response = client.get(f"{request_prefix}/docs/oauth2-redirect")
     assert response.status_code == 200, response.text
     assert "window.opener.swaggerUIRedirectOauth2" in response.text
 
 
-def test_redoc_html(client: TestClient):
-    response = client.get("/redoc")
+def test_redoc_html(client: TestClient, params: Params):
+    request_prefix = params["request_prefix"]
+
+    response = client.get(f"{request_prefix}/redoc")
     assert response.status_code == 200, response.text
     assert "https://unpkg.com/redoc@2/bundles/redoc.standalone.js" in response.text
 
+    response = client.get(f"{request_prefix}/openapi.json")
+    assert response.status_code == 200
+
+
+def test_api(client: TestClient, params: Params):
+    request_prefix = params["request_prefix"]
 
-def test_api(client: TestClient):
-    response = client.get("/users/john")
+    response = client.get(f"{request_prefix}/users/john")
     assert response.status_code == 200, response.text
     assert response.json()["message"] == "Hello john"

tests/test_tutorial/test_custom_docs_ui/test_tutorial002.py

@@ -6,8 +6,6 @@ from typing import TypedDict
 import pytest
 from fastapi.testclient import TestClient
 
-ROOT_PATH = "/api"
-
 
 class Params(TypedDict):
     app_root_path: str