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

1import gzip 4 ctx1abdc

2import importlib 4 ctx1abdc

3import json 4 ctx1abdc

4 

5import pytest 4 ctx1abdc

6from fastapi import Request 4 ctx1abdc

7from fastapi.testclient import TestClient 4 ctx1abdc

8 

9from tests.utils import needs_py310 4 ctx1abdc

10 

11 

12@pytest.fixture( 4 ctx1abdc

13 name="client", 

14 params=[ 

15 pytest.param("tutorial001_py310", marks=needs_py310), 

16 pytest.param("tutorial001_an_py310", marks=needs_py310), 

17 ], 

18) 

19def get_client(request: pytest.FixtureRequest): 4 ctx1abdc

20 mod = importlib.import_module(f"docs_src.custom_request_and_route.{request.param}") 3 ctx1abc

21 

22 @mod.app.get("/check-class") 3 ctx1abc

23 async def check_gzip_request(request: Request): 3 ctx1abc

24 return {"request_class": type(request).__name__} 3 ctx1hij

25 

26 client = TestClient(mod.app) 3 ctx1abc

27 return client 3 ctx1abc

28 

29 

30@pytest.mark.parametrize("compress", [True, False]) 4 ctx1abdc

31def test_gzip_request(client: TestClient, compress): 4 ctx1abdc

32 n = 1000 3 ctx1efg

33 headers = {} 3 ctx1efg

34 body = [1] * n 3 ctx1efg

35 data = json.dumps(body).encode() 3 ctx1efg

36 if compress: 3 ctx1efg

37 data = gzip.compress(data) 3 ctx1efg

38 headers["Content-Encoding"] = "gzip" 3 ctx1efg

39 headers["Content-Type"] = "application/json" 3 ctx1efg

40 response = client.post("/sum", content=data, headers=headers) 3 ctx1efg

41 assert response.json() == {"sum": n} 3 ctx1efg

42 

43 

44def test_request_class(client: TestClient): 4 ctx1abdc

45 response = client.get("/check-class") 3 ctx1hij

46 assert response.json() == {"request_class": "GzipRequest"} 3 ctx1hij