Coverage for tests/test_openapi_route_extensions.py: 100%

1from fastapi import FastAPI 5 ctx1abcde

2from fastapi.testclient import TestClient 5 ctx1abcde

3 

4app = FastAPI() 5 ctx1abcde

5 

6 

7@app.get("/", openapi_extra={"x-custom-extension": "value"}) 5 ctx1abcde

8def route_with_extras(): 5 ctx1abcde

9 return {} 5 ctx1abcde

10 

11 

12client = TestClient(app) 5 ctx1abcde

13 

14 

15def test_get_route(): 5 ctx1abcde

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

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

18 assert response.json() == {} 5 ctx1abcde

19 

20 

21def test_openapi(): 5 ctx1abcde

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

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

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

25 "openapi": "3.1.0", 

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

27 "paths": { 

28 "/": { 

29 "get": { 

30 "responses": { 

31 "200": { 

32 "description": "Successful Response", 

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

34 }, 

35 }, 

36 "summary": "Route With Extras", 

37 "operationId": "route_with_extras__get", 

38 "x-custom-extension": "value", 

39 } 

40 }, 

41 }, 

42 }