Coverage for tests/test_top_level_security_scheme_in_openapi.py: 100%

1# Test security scheme at the top level, including OpenAPI 

2# Ref: https://github.com/fastapi/fastapi/discussions/14263 

3# Ref: https://github.com/fastapi/fastapi/issues/14271 

4from fastapi import Depends, FastAPI 7 ctx1abcdefg

5from fastapi.security import HTTPBearer 7 ctx1abcdefg

6from fastapi.testclient import TestClient 7 ctx1abcdefg

7from inline_snapshot import snapshot 7 ctx1abcdefg

8 

9app = FastAPI() 7 ctx1abcdefg

10 

11bearer_scheme = HTTPBearer() 7 ctx1abcdefg

12 

13 

14@app.get("/", dependencies=[Depends(bearer_scheme)]) 7 ctx1abcdefg

15async def get_root(): 7 ctx1abcdefg

16 return {"message": "Hello, World!"} 7 ctx1hijklmn

17 

18 

19client = TestClient(app) 7 ctx1abcdefg

20 

21 

22def test_get_root(): 7 ctx1abcdefg

23 response = client.get("/", headers={"Authorization": "Bearer token"}) 7 ctx1hijklmn

24 assert response.status_code == 200, response.text 7 ctx1hijklmn

25 assert response.json() == {"message": "Hello, World!"} 7 ctx1hijklmn

26 

27 

28def test_get_root_no_token(): 7 ctx1abcdefg

29 response = client.get("/") 7 ctx1opqrstu

30 assert response.status_code == 401, response.text 7 ctx1opqrstu

31 assert response.json() == {"detail": "Not authenticated"} 7 ctx1opqrstu

32 

33 

34def test_openapi_schema(): 7 ctx1abcdefg

35 response = client.get("/openapi.json") 7 ctx1vwxyzAB

36 assert response.status_code == 200, response.text 7 ctx1vwxyzAB

37 assert response.json() == snapshot( 7 ctx1vwxyzAB

38 { 

39 "openapi": "3.1.0", 

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

41 "paths": { 

42 "/": { 

43 "get": { 

44 "summary": "Get Root", 

45 "operationId": "get_root__get", 

46 "responses": { 

47 "200": { 

48 "description": "Successful Response", 

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

50 } 

51 }, 

52 "security": [{"HTTPBearer": []}], 

53 } 

54 } 

55 }, 

56 "components": { 

57 "securitySchemes": {"HTTPBearer": {"type": "http", "scheme": "bearer"}} 

58 }, 

59 } 

60 )