Coverage for tests / test_security_openid_connect_optional.py: 100%

1from typing import Optional 4 ctx1abcd

2 

3from fastapi import Depends, FastAPI, Security 4 ctx1abcd

4from fastapi.security.open_id_connect_url import OpenIdConnect 4 ctx1abcd

5from fastapi.testclient import TestClient 4 ctx1abcd

6from inline_snapshot import snapshot 4 ctx1abcd

7from pydantic import BaseModel 4 ctx1abcd

8 

9app = FastAPI() 4 ctx1abcd

10 

11oid = OpenIdConnect(openIdConnectUrl="/openid", auto_error=False) 4 ctx1abcd

12 

13 

14class User(BaseModel): 4 ctx1abcd

15 username: str 4 ctx1abcd

16 

17 

18def get_current_user(oauth_header: Optional[str] = Security(oid)): 4 ctx1abcd

19 if oauth_header is None: 9 ctx1ekfglhimj

20 return None 3 ctx1klm

21 user = User(username=oauth_header) 6 ctx1efghij

22 return user 6 ctx1efghij

23 

24 

25@app.get("/users/me") 4 ctx1abcd

26def read_current_user(current_user: Optional[User] = Depends(get_current_user)): 4 ctx1abcd

27 if current_user is None: 9 ctx1ekfglhimj

28 return {"msg": "Create an account first"} 3 ctx1klm

29 return current_user 6 ctx1efghij

30 

31 

32client = TestClient(app) 4 ctx1abcd

33 

34 

35def test_security_oauth2(): 4 ctx1abcd

36 response = client.get("/users/me", headers={"Authorization": "Bearer footokenbar"}) 3 ctx1egi

37 assert response.status_code == 200, response.text 3 ctx1egi

38 assert response.json() == {"username": "Bearer footokenbar"} 3 ctx1egi

39 

40 

41def test_security_oauth2_password_other_header(): 4 ctx1abcd

42 response = client.get("/users/me", headers={"Authorization": "Other footokenbar"}) 3 ctx1fhj

43 assert response.status_code == 200, response.text 3 ctx1fhj

44 assert response.json() == {"username": "Other footokenbar"} 3 ctx1fhj

45 

46 

47def test_security_oauth2_password_bearer_no_header(): 4 ctx1abcd

48 response = client.get("/users/me") 3 ctx1klm

49 assert response.status_code == 200, response.text 3 ctx1klm

50 assert response.json() == {"msg": "Create an account first"} 3 ctx1klm

51 

52 

53def test_openapi_schema(): 4 ctx1abcd

54 response = client.get("/openapi.json") 3 ctx1nop

55 assert response.status_code == 200, response.text 3 ctx1nop

56 assert response.json() == snapshot( 3 ctx1nop

57 { 

58 "openapi": "3.1.0", 

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

60 "paths": { 

61 "/users/me": { 

62 "get": { 

63 "responses": { 

64 "200": { 

65 "description": "Successful Response", 

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

67 } 

68 }, 

69 "summary": "Read Current User", 

70 "operationId": "read_current_user_users_me_get", 

71 "security": [{"OpenIdConnect": []}], 

72 } 

73 } 

74 }, 

75 "components": { 

76 "securitySchemes": { 

77 "OpenIdConnect": { 

78 "type": "openIdConnect", 

79 "openIdConnectUrl": "/openid", 

80 } 

81 } 

82 }, 

83 } 

84 )