Coverage for tests / test_http_connection_injection.py: 100%

1from fastapi import Depends, FastAPI 4 ctx1abcd

2from fastapi.requests import HTTPConnection 4 ctx1abcd

3from fastapi.testclient import TestClient 4 ctx1abcd

4from starlette.websockets import WebSocket 4 ctx1abcd

5 

6app = FastAPI() 4 ctx1abcd

7app.state.value = 42 4 ctx1abcd

8 

9 

10async def extract_value_from_http_connection(conn: HTTPConnection): 4 ctx1abcd

11 return conn.app.state.value 7 ctx1heijfkg

12 

13 

14@app.get("/http") 4 ctx1abcd

15async def get_value_by_http(value: int = Depends(extract_value_from_http_connection)): 4 ctx1abcd

16 return value 4 ctx1hijk

17 

18 

19@app.websocket("/ws") 4 ctx1abcd

20async def get_value_by_ws( 4 ctx1abcd

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

22): 

23 await websocket.accept() 4 ctx1elfg

24 await websocket.send_json(value) 4 ctx1elfg

25 await websocket.close() 4 ctx1elfg

26 

27 

28client = TestClient(app) 4 ctx1abcd

29 

30 

31def test_value_extracting_by_http(): 4 ctx1abcd

32 response = client.get("/http") 4 ctx1hijk

33 assert response.status_code == 200 4 ctx1hijk

34 assert response.json() == 42 4 ctx1hijk

35 

36 

37def test_value_extracting_by_ws(): 4 ctx1abcd

38 with client.websocket_connect("/ws") as websocket: 4 ctx1elfg

39 assert websocket.receive_json() == 42 4 ctx1elfg