Coverage for tests/test_duplicate_models_openapi.py: 100%

1from fastapi import FastAPI 5 ctx1abcde

2from fastapi.testclient import TestClient 5 ctx1abcde

3from pydantic import BaseModel 5 ctx1abcde

4 

5app = FastAPI() 5 ctx1abcde

6 

7 

8class Model(BaseModel): 5 ctx1abcde

9 pass 5 ctx1abcde

10 

11 

12class Model2(BaseModel): 5 ctx1abcde

13 a: Model 5 ctx1abcde

14 

15 

16class Model3(BaseModel): 5 ctx1abcde

17 c: Model 5 ctx1abcde

18 d: Model2 5 ctx1abcde

19 

20 

21@app.get("/", response_model=Model3) 5 ctx1abcde

22def f(): 5 ctx1abcde

23 return {"c": {}, "d": {"a": {}}} 5 ctx1abcde

24 

25 

26client = TestClient(app) 5 ctx1abcde

27 

28 

29def test_get_api_route(): 5 ctx1abcde

30 response = client.get("/") 5 ctx1abcde

31 assert response.status_code == 200, response.text 5 ctx1abcde

32 assert response.json() == {"c": {}, "d": {"a": {}}} 5 ctx1abcde

33 

34 

35def test_openapi_schema(): 5 ctx1abcde

36 response = client.get("/openapi.json") 5 ctx1abcde

37 assert response.status_code == 200, response.text 5 ctx1abcde

38 assert response.json() == { 5 ctx1abcde

39 "openapi": "3.1.0", 

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

41 "paths": { 

42 "/": { 

43 "get": { 

44 "summary": "F", 

45 "operationId": "f__get", 

46 "responses": { 

47 "200": { 

48 "description": "Successful Response", 

49 "content": { 

50 "application/json": { 

51 "schema": {"$ref": "#/components/schemas/Model3"} 

52 } 

53 }, 

54 } 

55 }, 

56 } 

57 } 

58 }, 

59 "components": { 

60 "schemas": { 

61 "Model": {"title": "Model", "type": "object", "properties": {}}, 

62 "Model2": { 

63 "title": "Model2", 

64 "required": ["a"], 

65 "type": "object", 

66 "properties": {"a": {"$ref": "#/components/schemas/Model"}}, 

67 }, 

68 "Model3": { 

69 "title": "Model3", 

70 "required": ["c", "d"], 

71 "type": "object", 

72 "properties": { 

73 "c": {"$ref": "#/components/schemas/Model"}, 

74 "d": {"$ref": "#/components/schemas/Model2"}, 

75 }, 

76 }, 

77 } 

78 }, 

79 }