Coverage for tests/test_dependency_security_overrides.py: 100%

1from typing import List, Tuple 5 ctx1abcde

2 

3from fastapi import Depends, FastAPI, Security 5 ctx1abcde

4from fastapi.security import SecurityScopes 5 ctx1abcde

5from fastapi.testclient import TestClient 5 ctx1abcde

6 

7app = FastAPI() 5 ctx1abcde

8 

9 

10def get_user(required_scopes: SecurityScopes): 5 ctx1abcde

11 return "john", required_scopes.scopes 5 ctx1abcde

12 

13 

14def get_user_override(required_scopes: SecurityScopes): 5 ctx1abcde

15 return "alice", required_scopes.scopes 5 ctx1abcde

16 

17 

18def get_data(): 5 ctx1abcde

19 return [1, 2, 3] 5 ctx1abcde

20 

21 

22def get_data_override(): 5 ctx1abcde

23 return [3, 4, 5] 5 ctx1abcde

24 

25 

26@app.get("/user") 5 ctx1abcde

27def read_user( 5 ctx1abcde

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} 5 ctx1abcde

32 

33 

34client = TestClient(app) 5 ctx1abcde

35 

36 

37def test_normal(): 5 ctx1abcde

38 response = client.get("/user") 5 ctx1abcde

39 assert response.json() == { 5 ctx1abcde

40 "user": "john", 

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

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

43 } 

44 

45 

46def test_override_data(): 5 ctx1abcde

47 app.dependency_overrides[get_data] = get_data_override 5 ctx1abcde

48 response = client.get("/user") 5 ctx1abcde

49 assert response.json() == { 5 ctx1abcde

50 "user": "john", 

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

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

53 } 

54 app.dependency_overrides = {} 5 ctx1abcde

55 

56 

57def test_override_security(): 5 ctx1abcde

58 app.dependency_overrides[get_user] = get_user_override 5 ctx1abcde

59 response = client.get("/user") 5 ctx1abcde

60 assert response.json() == { 5 ctx1abcde

61 "user": "alice", 

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

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

64 } 

65 app.dependency_overrides = {} 5 ctx1abcde