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

1import gzip 5 ctx1abcde

2import json 5 ctx1abcde

3 

4import pytest 5 ctx1abcde

5from fastapi import Request 5 ctx1abcde

6from fastapi.testclient import TestClient 5 ctx1abcde

7 

8from docs_src.custom_request_and_route.tutorial001 import app 5 ctx1abcde

9 

10 

11@app.get("/check-class") 5 ctx1abcde

12async def check_gzip_request(request: Request): 5 ctx1abcde

13 return {"request_class": type(request).__name__} 5 ctx1abcde

14 

15 

16client = TestClient(app) 5 ctx1abcde

17 

18 

19@pytest.mark.parametrize("compress", [True, False]) 5 ctx1abcde

20def test_gzip_request(compress): 5 ctx1abcde

21 n = 1000 5 ctx1abcde

22 headers = {} 5 ctx1abcde

23 body = [1] * n 5 ctx1abcde

24 data = json.dumps(body).encode() 5 ctx1abcde

25 if compress: 5 ctx1abcde

26 data = gzip.compress(data) 5 ctx1abcde

27 headers["Content-Encoding"] = "gzip" 5 ctx1abcde

28 headers["Content-Type"] = "application/json" 5 ctx1abcde

29 response = client.post("/sum", content=data, headers=headers) 5 ctx1abcde

30 assert response.json() == {"sum": n} 5 ctx1abcde

31 

32 

33def test_request_class(): 5 ctx1abcde

34 response = client.get("/check-class") 5 ctx1abcde

35 assert response.json() == {"request_class": "GzipRequest"} 5 ctx1abcde