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 6 ctx1heifjg

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 3 ctx1hij

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() 3 ctx1efg

24 await websocket.send_json(value) 3 ctx1efg

25 await websocket.close() 3 ctx1efg

26 

27 

28client = TestClient(app) 4 ctx1abcd

29 

30 

31def test_value_extracting_by_http(): 4 ctx1abcd

32 response = client.get("/http") 3 ctx1hij

33 assert response.status_code == 200 3 ctx1hij

34 assert response.json() == 42 3 ctx1hij

35 

36 

37def test_value_extracting_by_ws(): 4 ctx1abcd

38 with client.websocket_connect("/ws") as websocket: 3 ctx1efg

39 assert websocket.receive_json() == 42 3 ctx1efg