fastapi

1
from fastapi import APIRouter, FastAPI
2
from fastapi.testclient import TestClient
3

4
app = FastAPI()
5

6
router = APIRouter()
7

8

9
@router.get("/users/{id}")
10
def read_user(segment: str, id: str):
11
    return {"segment": segment, "id": id}
12

13

14
app.include_router(router, prefix="/{segment}")
15

16

17
client = TestClient(app)
18

19

20
def test_get():
21
    response = client.get("/seg/users/foo")
22
    assert response.status_code == 200, response.text
23
    assert response.json() == {"segment": "seg", "id": "foo"}
24