Coverage for tests/test_http_connection_injection.py: 100%

1from fastapi import Depends, FastAPI 5 ctx1abcde

2from fastapi.requests import HTTPConnection 5 ctx1abcde

3from fastapi.testclient import TestClient 5 ctx1abcde

4from starlette.websockets import WebSocket 5 ctx1abcde

5 

6app = FastAPI() 5 ctx1abcde

7app.state.value = 42 5 ctx1abcde

8 

9 

10async def extract_value_from_http_connection(conn: HTTPConnection): 5 ctx1abcde

11 return conn.app.state.value 5 ctx1abcde

12 

13 

14@app.get("/http") 5 ctx1abcde

15async def get_value_by_http(value: int = Depends(extract_value_from_http_connection)): 5 ctx1abcde

16 return value 5 ctx1abcde

17 

18 

19@app.websocket("/ws") 5 ctx1abcde

20async def get_value_by_ws( 5 ctx1abcde

21 websocket: WebSocket, value: int = Depends(extract_value_from_http_connection) 

22): 

23 await websocket.accept() 5 ctx1abcde

24 await websocket.send_json(value) 5 ctx1abcde

25 await websocket.close() 5 ctx1abcde

26 

27 

28client = TestClient(app) 5 ctx1abcde

29 

30 

31def test_value_extracting_by_http(): 5 ctx1abcde

32 response = client.get("/http") 5 ctx1abcde

33 assert response.status_code == 200 5 ctx1abcde

34 assert response.json() == 42 5 ctx1abcde

35 

36 

37def test_value_extracting_by_ws(): 5 ctx1abcde

38 with client.websocket_connect("/ws") as websocket: 5 ctx1abcde

39 assert websocket.receive_json() == 42 5 ctx1abcde