Coverage for tests/test_tutorial/test_custom_request_and_route/test_tutorial001.py: 100%

1import gzip 6 ctx1abcdef

2import json 6 ctx1abcdef

3 

4import pytest 6 ctx1abcdef

5from fastapi import Request 6 ctx1abcdef

6from fastapi.testclient import TestClient 6 ctx1abcdef

7 

8from docs_src.custom_request_and_route.tutorial001 import app 6 ctx1abcdef

9 

10 

11@app.get("/check-class") 6 ctx1abcdef

12async def check_gzip_request(request: Request): 6 ctx1abcdef

13 return {"request_class": type(request).__name__} 6 ctx1mnopqr

14 

15 

16client = TestClient(app) 6 ctx1abcdef

17 

18 

19@pytest.mark.parametrize("compress", [True, False]) 6 ctx1abcdef

20def test_gzip_request(compress): 6 ctx1abcdef

21 n = 1000 6 ctx1ghijkl

22 headers = {} 6 ctx1ghijkl

23 body = [1] * n 6 ctx1ghijkl

24 data = json.dumps(body).encode() 6 ctx1ghijkl

25 if compress: 6 ctx1ghijkl

26 data = gzip.compress(data) 6 ctx1ghijkl

27 headers["Content-Encoding"] = "gzip" 6 ctx1ghijkl

28 headers["Content-Type"] = "application/json" 6 ctx1ghijkl

29 response = client.post("/sum", content=data, headers=headers) 6 ctx1ghijkl

30 assert response.json() == {"sum": n} 6 ctx1ghijkl

31 

32 

33def test_request_class(): 6 ctx1abcdef

34 response = client.get("/check-class") 6 ctx1mnopqr

35 assert response.json() == {"request_class": "GzipRequest"} 6 ctx1mnopqr