Coverage for tests/test_security_scopes.py: 100%

1from typing import Dict 7 ctx1abcdefg

2 

3import pytest 7 ctx1abcdefg

4from fastapi import Depends, FastAPI, Security 7 ctx1abcdefg

5from fastapi.testclient import TestClient 7 ctx1abcdefg

6from typing_extensions import Annotated 7 ctx1abcdefg

7 

8 

9@pytest.fixture(name="call_counter") 7 ctx1abcdefg

10def call_counter_fixture(): 7 ctx1abcdefg

11 return {"count": 0} 7 ctx1abcdefg

12 

13 

14@pytest.fixture(name="app") 7 ctx1abcdefg

15def app_fixture(call_counter: Dict[str, int]): 7 ctx1abcdefg

16 def get_db(): 7 ctx1abcdefg

17 call_counter["count"] += 1 7 ctx1hijklmn

18 return f"db_{call_counter['count']}" 7 ctx1hijklmn

19 

20 def get_user(db: Annotated[str, Depends(get_db)]): 7 ctx1abcdefg

21 return "user" 7 ctx1hijklmn

22 

23 app = FastAPI() 7 ctx1abcdefg

24 

25 @app.get("/") 7 ctx1abcdefg

26 def endpoint( 7 ctx1abcdefg

27 db: Annotated[str, Depends(get_db)], 

28 user: Annotated[str, Security(get_user, scopes=["read"])], 

29 ): 

30 return {"db": db} 7 ctx1hijklmn

31 

32 return app 7 ctx1abcdefg

33 

34 

35@pytest.fixture(name="client") 7 ctx1abcdefg

36def client_fixture(app: FastAPI): 7 ctx1abcdefg

37 return TestClient(app) 7 ctx1abcdefg

38 

39 

40def test_security_scopes_dependency_called_once( 7 ctx1abcdefg

41 client: TestClient, call_counter: Dict[str, int] 

42): 

43 response = client.get("/") 7 ctx1hijklmn

44 

45 assert response.status_code == 200 7 ctx1hijklmn

46 assert call_counter["count"] == 1 7 ctx1hijklmn