fastapi

1
from fastapi import Depends, FastAPI, Response
2
from fastapi.testclient import TestClient
3

4
app = FastAPI()
5

6

7
async def response_status_setter(response: Response):
8
    response.status_code = 201
9

10

11
async def parent_dep(result=Depends(response_status_setter)):
12
    return result
13

14

15
@app.get("/", dependencies=[Depends(parent_dep)])
16
async def get_main():
17
    return {"msg": "Hello World"}
18

19

20
client = TestClient(app)
21

22

23
def test_dependency_set_status_code():
24
    response = client.get("/")
25
    assert response.status_code == 201, response.text
26
    assert response.json() == {"msg": "Hello World"}
27