Coverage for tests / test_security_oauth2_authorization_code_bearer.py: 100%

1from typing import Optional 4 ctx1abcd

2 

3from fastapi import FastAPI, Security 4 ctx1abcd

4from fastapi.security import OAuth2AuthorizationCodeBearer 4 ctx1abcd

5from fastapi.testclient import TestClient 4 ctx1abcd

6from inline_snapshot import snapshot 4 ctx1abcd

7 

8app = FastAPI() 4 ctx1abcd

9 

10oauth2_scheme = OAuth2AuthorizationCodeBearer( 4 ctx1abcd

11 authorizationUrl="authorize", tokenUrl="token", auto_error=True 

12) 

13 

14 

15@app.get("/items/") 4 ctx1abcd

16async def read_items(token: Optional[str] = Security(oauth2_scheme)): 4 ctx1abcd

17 return {"token": token} 6 ctx1efghij

18 

19 

20client = TestClient(app) 4 ctx1abcd

21 

22 

23def test_no_token(): 4 ctx1abcd

24 response = client.get("/items") 3 ctx1klm

25 assert response.status_code == 401, response.text 3 ctx1klm

26 assert response.json() == {"detail": "Not authenticated"} 3 ctx1klm

27 

28 

29def test_incorrect_token(): 4 ctx1abcd

30 response = client.get("/items", headers={"Authorization": "Non-existent testtoken"}) 3 ctx1nop

31 assert response.status_code == 401, response.text 3 ctx1nop

32 assert response.json() == {"detail": "Not authenticated"} 3 ctx1nop

33 

34 

35def test_token(): 4 ctx1abcd

36 response = client.get("/items", headers={"Authorization": "Bearer testtoken"}) 3 ctx1egi

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

38 assert response.json() == {"token": "testtoken"} 3 ctx1egi

39 

40 

41def test_token_with_whitespaces(): 4 ctx1abcd

42 response = client.get("/items", headers={"Authorization": "Bearer testtoken "}) 3 ctx1fhj

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

44 assert response.json() == {"token": "testtoken"} 3 ctx1fhj

45 

46 

47def test_openapi_schema(): 4 ctx1abcd

48 response = client.get("/openapi.json") 3 ctx1qrs

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

50 assert response.json() == snapshot( 3 ctx1qrs

51 { 

52 "openapi": "3.1.0", 

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

54 "paths": { 

55 "/items/": { 

56 "get": { 

57 "responses": { 

58 "200": { 

59 "description": "Successful Response", 

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

61 } 

62 }, 

63 "summary": "Read Items", 

64 "operationId": "read_items_items__get", 

65 "security": [{"OAuth2AuthorizationCodeBearer": []}], 

66 } 

67 } 

68 }, 

69 "components": { 

70 "securitySchemes": { 

71 "OAuth2AuthorizationCodeBearer": { 

72 "type": "oauth2", 

73 "flows": { 

74 "authorizationCode": { 

75 "authorizationUrl": "authorize", 

76 "tokenUrl": "token", 

77 "scopes": {}, 

78 } 

79 }, 

80 } 

81 } 

82 }, 

83 } 

84 )