1 changed files with 7 additions and 13 deletions
@ -1,25 +1,19 @@ |
|||
from fastapi import FastAPI, APIRouter |
|||
from fastapi.testclient import TestClient |
|||
from fastapi import APIRouter, FastAPI |
|||
|
|||
|
|||
def test_mount_on_router(): |
|||
app = FastAPI() |
|||
api_router = APIRouter(prefix="/api") |
|||
app = FastAPI() |
|||
api_router = APIRouter(prefix="/api") |
|||
|
|||
@api_router.get("/app") |
|||
def read_main(): |
|||
return {"message": "Hello World from main app"} |
|||
|
|||
subapi = FastAPI() |
|||
subapi = FastAPI() |
|||
|
|||
@subapi.get("/sub") |
|||
def read_sub(): |
|||
return {"message": "Hello World from sub API"} |
|||
|
|||
api_router.mount("/subapi", subapi) |
|||
app.include_router(api_router) |
|||
|
|||
client = TestClient(app) |
|||
api_router.mount("/subapi", subapi) # ← moved up |
|||
app.include_router(api_router) # ← now after |
|||
|
|||
assert client.get("/api/app").status_code == 200 |
|||
assert client.get("/api/subapi/sub").status_code == 200 |
|||
print("All tests passed.") |
|||
Loading…
Reference in new issue