Coverage for tests / test_response_model_default_factory.py: 100%

1from fastapi import FastAPI 4 ctx1abcd

2from fastapi.testclient import TestClient 4 ctx1abcd

3from pydantic import BaseModel, Field 4 ctx1abcd

4 

5app = FastAPI() 4 ctx1abcd

6 

7 

8class ResponseModel(BaseModel): 4 ctx1abcd

9 code: int = 200 4 ctx1abcd

10 message: str = Field(default_factory=lambda: "Successful operation.") 11 ctx1aefbgchidjk

11 

12 

13@app.get( 4 ctx1abcd

14 "/response_model_has_default_factory_return_dict", 

15 response_model=ResponseModel, 

16) 

17async def response_model_has_default_factory_return_dict(): 4 ctx1abcd

18 return {"code": 200} 4 ctx1eghj

19 

20 

21@app.get( 4 ctx1abcd

22 "/response_model_has_default_factory_return_model", 

23 response_model=ResponseModel, 

24) 

25async def response_model_has_default_factory_return_model(): 4 ctx1abcd

26 return ResponseModel() 4 ctx1flik

27 

28 

29client = TestClient(app) 4 ctx1abcd

30 

31 

32def test_response_model_has_default_factory_return_dict(): 4 ctx1abcd

33 response = client.get("/response_model_has_default_factory_return_dict") 4 ctx1eghj

34 

35 assert response.status_code == 200, response.text 4 ctx1eghj

36 

37 assert response.json()["code"] == 200 4 ctx1eghj

38 assert response.json()["message"] == "Successful operation." 4 ctx1eghj

39 

40 

41def test_response_model_has_default_factory_return_model(): 4 ctx1abcd

42 response = client.get("/response_model_has_default_factory_return_model") 4 ctx1flik

43 

44 assert response.status_code == 200, response.text 4 ctx1flik

45 

46 assert response.json()["code"] == 200 4 ctx1flik

47 assert response.json()["message"] == "Successful operation." 4 ctx1flik