Coverage for tests / test_response_code_no_body.py: 100%

28 statements  

« prev     ^ index     » next       coverage.py v7.13.3, created at 2026-02-12 18:15 +0000

1from fastapi import FastAPI 1abcd

2from fastapi.responses import JSONResponse 1abcd

3from fastapi.testclient import TestClient 1abcd

4from inline_snapshot import snapshot 1abcd

5from pydantic import BaseModel 1abcd

6 

7app = FastAPI() 1abcd

8 

9 

10class JsonApiResponse(JSONResponse): 1abcd

11 media_type = "application/vnd.api+json" 1abcd

12 

13 

14class Error(BaseModel): 1abcd

15 status: str 1abcd

16 title: str 1abcd

17 

18 

19class JsonApiError(BaseModel): 1abcd

20 errors: list[Error] 1abcd

21 

22 

23@app.get( 1abcd

24 "/a", 

25 status_code=204, 

26 response_class=JsonApiResponse, 

27 responses={500: {"description": "Error", "model": JsonApiError}}, 

28) 

29async def a(): 1abcd

30 pass 1efg

31 

32 

33@app.get("/b", responses={204: {"description": "No Content"}}) 1abcd

34async def b(): 1abcd

35 pass # pragma: no cover 

36 

37 

38client = TestClient(app) 1abcd

39 

40 

41def test_get_response(): 1abcd

42 response = client.get("/a") 1efg

43 assert response.status_code == 204, response.text 1efg

44 assert "content-length" not in response.headers 1efg

45 assert response.content == b"" 1efg

46 

47 

48def test_openapi_schema(): 1abcd

49 response = client.get("/openapi.json") 1hij

50 assert response.status_code == 200, response.text 1hij

51 assert response.json() == snapshot( 1hij

52 { 

53 "openapi": "3.1.0", 

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

55 "paths": { 

56 "/a": { 

57 "get": { 

58 "responses": { 

59 "500": { 

60 "description": "Error", 

61 "content": { 

62 "application/vnd.api+json": { 

63 "schema": { 

64 "$ref": "#/components/schemas/JsonApiError" 

65 } 

66 } 

67 }, 

68 }, 

69 "204": {"description": "Successful Response"}, 

70 }, 

71 "summary": "A", 

72 "operationId": "a_a_get", 

73 } 

74 }, 

75 "/b": { 

76 "get": { 

77 "responses": { 

78 "204": {"description": "No Content"}, 

79 "200": { 

80 "description": "Successful Response", 

81 "content": {"application/json": {"schema": {}}}, 

82 }, 

83 }, 

84 "summary": "B", 

85 "operationId": "b_b_get", 

86 } 

87 }, 

88 }, 

89 "components": { 

90 "schemas": { 

91 "Error": { 

92 "title": "Error", 

93 "required": ["status", "title"], 

94 "type": "object", 

95 "properties": { 

96 "status": {"title": "Status", "type": "string"}, 

97 "title": {"title": "Title", "type": "string"}, 

98 }, 

99 }, 

100 "JsonApiError": { 

101 "title": "JsonApiError", 

102 "required": ["errors"], 

103 "type": "object", 

104 "properties": { 

105 "errors": { 

106 "title": "Errors", 

107 "type": "array", 

108 "items": {"$ref": "#/components/schemas/Error"}, 

109 } 

110 }, 

111 }, 

112 } 

113 }, 

114 } 

115 )