Coverage for tests/test_computed_fields.py: 100%

1import pytest 5 ctx1abcde

2from fastapi import FastAPI 5 ctx1abcde

3from fastapi.testclient import TestClient 5 ctx1abcde

4 

5from .utils import needs_pydanticv2 5 ctx1abcde

6 

7 

8@pytest.fixture(name="client") 5 ctx1abcde

9def get_client(): 5 ctx1abcde

10 app = FastAPI() 5 ctx1abcde

11 

12 from pydantic import BaseModel, computed_field 5 ctx1abcde

13 

14 class Rectangle(BaseModel): 5 ctx1abcde

15 width: int 5 ctx1abcde

16 length: int 5 ctx1abcde

17 

18 @computed_field 5 ctx1abcde

19 @property 5 ctx1abcde

20 def area(self) -> int: 5 ctx1abcde

21 return self.width * self.length 5 ctx1fghij

22 

23 @app.get("/") 5 ctx1abcde

24 def read_root() -> Rectangle: 5 ctx1abcde

25 return Rectangle(width=3, length=4) 5 ctx1fghij

26 

27 @app.get("/responses", responses={200: {"model": Rectangle}}) 5 ctx1abcde

28 def read_responses() -> Rectangle: 5 ctx1abcde

29 return Rectangle(width=3, length=4) 5 ctx1fghij

30 

31 client = TestClient(app) 5 ctx1abcde

32 return client 5 ctx1abcde

33 

34 

35@pytest.mark.parametrize("path", ["/", "/responses"]) 5 ctx1abcde

36@needs_pydanticv2 5 ctx1abcde

37def test_get(client: TestClient, path: str): 5 ctx1abcde

38 response = client.get(path) 5 ctx1fghij

39 assert response.status_code == 200, response.text 5 ctx1fghij

40 assert response.json() == {"width": 3, "length": 4, "area": 12} 5 ctx1fghij

41 

42 

43@needs_pydanticv2 5 ctx1abcde

44def test_openapi_schema(client: TestClient): 5 ctx1abcde

45 response = client.get("/openapi.json") 5 ctx1klmno

46 assert response.status_code == 200, response.text 5 ctx1klmno

47 assert response.json() == { 5 ctx1klmno

48 "openapi": "3.1.0", 

49 "info": {"title": "FastAPI", "version": "0.1.0"}, 

50 "paths": { 

51 "/": { 

52 "get": { 

53 "summary": "Read Root", 

54 "operationId": "read_root__get", 

55 "responses": { 

56 "200": { 

57 "description": "Successful Response", 

58 "content": { 

59 "application/json": { 

60 "schema": {"$ref": "#/components/schemas/Rectangle"} 

61 } 

62 }, 

63 } 

64 }, 

65 } 

66 }, 

67 "/responses": { 

68 "get": { 

69 "summary": "Read Responses", 

70 "operationId": "read_responses_responses_get", 

71 "responses": { 

72 "200": { 

73 "description": "Successful Response", 

74 "content": { 

75 "application/json": { 

76 "schema": {"$ref": "#/components/schemas/Rectangle"} 

77 } 

78 }, 

79 } 

80 }, 

81 } 

82 }, 

83 }, 

84 "components": { 

85 "schemas": { 

86 "Rectangle": { 

87 "properties": { 

88 "width": {"type": "integer", "title": "Width"}, 

89 "length": {"type": "integer", "title": "Length"}, 

90 "area": {"type": "integer", "title": "Area", "readOnly": True}, 

91 }, 

92 "type": "object", 

93 "required": ["width", "length", "area"], 

94 "title": "Rectangle", 

95 } 

96 } 

97 }, 

98 }