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

1import gzip 7 ctx1abcdefg

2import json 7 ctx1abcdefg

3 

4import pytest 7 ctx1abcdefg

5from fastapi import Request 7 ctx1abcdefg

6from fastapi.testclient import TestClient 7 ctx1abcdefg

7 

8from docs_src.custom_request_and_route.tutorial001 import app 7 ctx1abcdefg

9 

10 

11@app.get("/check-class") 7 ctx1abcdefg

12async def check_gzip_request(request: Request): 7 ctx1abcdefg

13 return {"request_class": type(request).__name__} 7 ctx1opqrstu

14 

15 

16client = TestClient(app) 7 ctx1abcdefg

17 

18 

19@pytest.mark.parametrize("compress", [True, False]) 7 ctx1abcdefg

20def test_gzip_request(compress): 7 ctx1abcdefg

21 n = 1000 7 ctx1hijklmn

22 headers = {} 7 ctx1hijklmn

23 body = [1] * n 7 ctx1hijklmn

24 data = json.dumps(body).encode() 7 ctx1hijklmn

25 if compress: 7 ctx1hijklmn

26 data = gzip.compress(data) 7 ctx1hijklmn

27 headers["Content-Encoding"] = "gzip" 7 ctx1hijklmn

28 headers["Content-Type"] = "application/json" 7 ctx1hijklmn

29 response = client.post("/sum", content=data, headers=headers) 7 ctx1hijklmn

30 assert response.json() == {"sum": n} 7 ctx1hijklmn

31 

32 

33def test_request_class(): 7 ctx1abcdefg

34 response = client.get("/check-class") 7 ctx1opqrstu

35 assert response.json() == {"request_class": "GzipRequest"} 7 ctx1opqrstu