Coverage for tests / test_security_scopes.py: 100%

1from typing import Annotated 4 ctx1abdc

2 

3import pytest 4 ctx1abdc

4from fastapi import Depends, FastAPI, Security 4 ctx1abdc

5from fastapi.testclient import TestClient 4 ctx1abdc

6 

7 

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

9def call_counter_fixture(): 4 ctx1abdc

10 return {"count": 0} 3 ctx1abc

11 

12 

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

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

15 def get_db(): 3 ctx1abc

16 call_counter["count"] += 1 3 ctx1efg

17 return f"db_{call_counter['count']}" 3 ctx1efg

18 

19 def get_user(db: Annotated[str, Depends(get_db)]): 3 ctx1abc

20 return "user" 3 ctx1efg

21 

22 app = FastAPI() 3 ctx1abc

23 

24 @app.get("/") 3 ctx1abc

25 def endpoint( 3 ctx1abc

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

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

28 ): 

29 return {"db": db} 3 ctx1efg

30 

31 return app 3 ctx1abc

32 

33 

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

35def client_fixture(app: FastAPI): 4 ctx1abdc

36 return TestClient(app) 3 ctx1abc

37 

38 

39def test_security_scopes_dependency_called_once( 4 ctx1abdc

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

41): 

42 response = client.get("/") 3 ctx1efg

43 

44 assert response.status_code == 200 3 ctx1efg

45 assert call_counter["count"] == 1 3 ctx1efg