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

1import gzip 4 ctx1abcd

2import importlib 4 ctx1abcd

3import json 4 ctx1abcd

4 

5import pytest 4 ctx1abcd

6from fastapi import Request 4 ctx1abcd

7from fastapi.testclient import TestClient 4 ctx1abcd

8 

9from tests.utils import needs_py310 4 ctx1abcd

10 

11 

12@pytest.fixture( 4 ctx1abcd

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 ctx1abcd

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

21 

22 @mod.app.get("/check-class") 20 ctx1rstuvwxyzABCDEFGHIJK

23 async def check_gzip_request(request: Request): 20 ctx1rstuvwxyzABCDEFGHIJK

24 return {"request_class": type(request).__name__} 8 ctx1LMNOPQRS

25 

26 client = TestClient(mod.app) 20 ctx1rstuvwxyzABCDEFGHIJK

27 return client 20 ctx1rstuvwxyzABCDEFGHIJK

28 

29 

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

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

32 n = 1000 13 ctx1lemfgnhoipjqk

33 headers = {} 13 ctx1lemfgnhoipjqk

34 body = [1] * n 13 ctx1lemfgnhoipjqk

35 data = json.dumps(body).encode() 13 ctx1lemfgnhoipjqk

36 if compress: 13 ctx1lemfgnhoipjqk

37 data = gzip.compress(data) 7 ctx1efghijk

38 headers["Content-Encoding"] = "gzip" 7 ctx1efghijk

39 headers["Content-Type"] = "application/json" 13 ctx1lemfgnhoipjqk

40 response = client.post("/sum", content=data, headers=headers) 13 ctx1lemfgnhoipjqk

41 assert response.json() == {"sum": n} 13 ctx1lemfgnhoipjqk

42 

43 

44def test_request_class(client: TestClient): 4 ctx1abcd

45 response = client.get("/check-class") 8 ctx1LMNOPQRS

46 assert response.json() == {"request_class": "GzipRequest"} 8 ctx1LMNOPQRS