Coverage for tests/test_dependency_security_overrides.py: 100%

1from typing import List, Tuple 6 ctx1abcdef

2 

3from fastapi import Depends, FastAPI, Security 6 ctx1abcdef

4from fastapi.security import SecurityScopes 6 ctx1abcdef

5from fastapi.testclient import TestClient 6 ctx1abcdef

6 

7app = FastAPI() 6 ctx1abcdef

8 

9 

10def get_user(required_scopes: SecurityScopes): 6 ctx1abcdef

11 return "john", required_scopes.scopes 12 ctx1sgthuivjwkxl

12 

13 

14def get_user_override(required_scopes: SecurityScopes): 6 ctx1abcdef

15 return "alice", required_scopes.scopes 6 ctx1mnopqr

16 

17 

18def get_data(): 6 ctx1abcdef

19 return [1, 2, 3] 12 ctx1smtnuovpwqxr

20 

21 

22def get_data_override(): 6 ctx1abcdef

23 return [3, 4, 5] 6 ctx1ghijkl

24 

25 

26@app.get("/user") 6 ctx1abcdef

27def read_user( 6 ctx1abcdef

28 user_data: Tuple[str, List[str]] = Security(get_user, scopes=["foo", "bar"]), 

29 data: List[int] = Depends(get_data), 

30): 

31 return {"user": user_data[0], "scopes": user_data[1], "data": data} 18 ctx1sgmthnuiovjpwkqxlr

32 

33 

34client = TestClient(app) 6 ctx1abcdef

35 

36 

37def test_normal(): 6 ctx1abcdef

38 response = client.get("/user") 6 ctx1stuvwx

39 assert response.json() == { 6 ctx1stuvwx

40 "user": "john", 

41 "scopes": ["foo", "bar"], 

42 "data": [1, 2, 3], 

43 } 

44 

45 

46def test_override_data(): 6 ctx1abcdef

47 app.dependency_overrides[get_data] = get_data_override 6 ctx1ghijkl

48 response = client.get("/user") 6 ctx1ghijkl

49 assert response.json() == { 6 ctx1ghijkl

50 "user": "john", 

51 "scopes": ["foo", "bar"], 

52 "data": [3, 4, 5], 

53 } 

54 app.dependency_overrides = {} 6 ctx1ghijkl

55 

56 

57def test_override_security(): 6 ctx1abcdef

58 app.dependency_overrides[get_user] = get_user_override 6 ctx1mnopqr

59 response = client.get("/user") 6 ctx1mnopqr

60 assert response.json() == { 6 ctx1mnopqr

61 "user": "alice", 

62 "scopes": ["foo", "bar"], 

63 "data": [1, 2, 3], 

64 } 

65 app.dependency_overrides = {} 6 ctx1mnopqr