Coverage for tests / test_openapi_route_extensions.py: 100%

1from fastapi import FastAPI 4 ctx1abcd

2from fastapi.testclient import TestClient 4 ctx1abcd

3from inline_snapshot import snapshot 4 ctx1abcd

4 

5app = FastAPI() 4 ctx1abcd

6 

7 

8@app.get("/", openapi_extra={"x-custom-extension": "value"}) 4 ctx1abcd

9def route_with_extras(): 4 ctx1abcd

10 return {} 3 ctx1efg

11 

12 

13client = TestClient(app) 4 ctx1abcd

14 

15 

16def test_get_route(): 4 ctx1abcd

17 response = client.get("/") 3 ctx1efg

18 assert response.status_code == 200, response.text 3 ctx1efg

19 assert response.json() == {} 3 ctx1efg

20 

21 

22def test_openapi_schema(): 4 ctx1abcd

23 response = client.get("/openapi.json") 3 ctx1hij

24 assert response.status_code == 200, response.text 3 ctx1hij

25 assert response.json() == snapshot( 3 ctx1hij

26 { 

27 "openapi": "3.1.0", 

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

29 "paths": { 

30 "/": { 

31 "get": { 

32 "responses": { 

33 "200": { 

34 "description": "Successful Response", 

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

36 }, 

37 }, 

38 "summary": "Route With Extras", 

39 "operationId": "route_with_extras__get", 

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

41 } 

42 }, 

43 }, 

44 } 

45 )