Coverage for tests/test_http_connection_injection.py: 100%

1from fastapi import Depends, FastAPI 6 ctx1abcdef

2from fastapi.requests import HTTPConnection 6 ctx1abcdef

3from fastapi.testclient import TestClient 6 ctx1abcdef

4from starlette.websockets import WebSocket 6 ctx1abcdef

5 

6app = FastAPI() 6 ctx1abcdef

7app.state.value = 42 6 ctx1abcdef

8 

9 

10async def extract_value_from_http_connection(conn: HTTPConnection): 6 ctx1abcdef

11 return conn.app.state.value 12 ctx1mgnhoipjqkrl

12 

13 

14@app.get("/http") 6 ctx1abcdef

15async def get_value_by_http(value: int = Depends(extract_value_from_http_connection)): 6 ctx1abcdef

16 return value 6 ctx1mnopqr

17 

18 

19@app.websocket("/ws") 6 ctx1abcdef

20async def get_value_by_ws( 6 ctx1abcdef

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

22): 

23 await websocket.accept() 6 ctx1ghijkl

24 await websocket.send_json(value) 6 ctx1ghijkl

25 await websocket.close() 6 ctx1ghijkl

26 

27 

28client = TestClient(app) 6 ctx1abcdef

29 

30 

31def test_value_extracting_by_http(): 6 ctx1abcdef

32 response = client.get("/http") 6 ctx1mnopqr

33 assert response.status_code == 200 6 ctx1mnopqr

34 assert response.json() == 42 6 ctx1mnopqr

35 

36 

37def test_value_extracting_by_ws(): 6 ctx1abcdef

38 with client.websocket_connect("/ws") as websocket: 6 ctx1ghijkl

39 assert websocket.receive_json() == 42 6 ctx1ghijkl