Coverage for tests / test_security_scopes.py: 100%

1from typing import Annotated 4 ctx1abcd

2 

3import pytest 4 ctx1abcd

4from fastapi import Depends, FastAPI, Security 4 ctx1abcd

5from fastapi.testclient import TestClient 4 ctx1abcd

6 

7 

8@pytest.fixture(name="call_counter") 4 ctx1abcd

9def call_counter_fixture(): 4 ctx1abcd

10 return {"count": 0} 4 ctx1efgh

11 

12 

13@pytest.fixture(name="app") 4 ctx1abcd

14def app_fixture(call_counter: dict[str, int]): 4 ctx1abcd

15 def get_db(): 4 ctx1efgh

16 call_counter["count"] += 1 4 ctx1ijkl

17 return f"db_{call_counter['count']}" 4 ctx1ijkl

18 

19 def get_user(db: Annotated[str, Depends(get_db)]): 4 ctx1efgh

20 return "user" 4 ctx1ijkl

21 

22 app = FastAPI() 4 ctx1efgh

23 

24 @app.get("/") 4 ctx1efgh

25 def endpoint( 4 ctx1efgh

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

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

28 ): 

29 return {"db": db} 4 ctx1ijkl

30 

31 return app 4 ctx1efgh

32 

33 

34@pytest.fixture(name="client") 4 ctx1abcd

35def client_fixture(app: FastAPI): 4 ctx1abcd

36 return TestClient(app) 4 ctx1efgh

37 

38 

39def test_security_scopes_dependency_called_once( 4 ctx1abcd

40 client: TestClient, call_counter: dict[str, int] 

41): 

42 response = client.get("/") 4 ctx1ijkl

43 

44 assert response.status_code == 200 4 ctx1ijkl

45 assert call_counter["count"] == 1 4 ctx1ijkl