Coverage for tests / test_validate_response_dataclass.py: 100%

1import pytest 4 ctx1adbc

2from fastapi import FastAPI 4 ctx1adbc

3from fastapi.exceptions import ResponseValidationError 4 ctx1adbc

4from fastapi.testclient import TestClient 4 ctx1adbc

5from pydantic.dataclasses import dataclass 4 ctx1adbc

6 

7app = FastAPI() 4 ctx1adbc

8 

9 

10@dataclass 4 ctx1adbc

11class Item: 4 ctx1adbc

12 name: str 3 ctx1abc

13 price: float | None = None 4 ctx1adbc

14 owner_ids: list[int] | None = None 4 ctx1adbc

15 

16 

17@app.get("/items/invalid", response_model=Item) 4 ctx1adbc

18def get_invalid(): 4 ctx1adbc

19 return {"name": "invalid", "price": "foo"} 4 ctx1efgh

20 

21 

22@app.get("/items/innerinvalid", response_model=Item) 4 ctx1adbc

23def get_innerinvalid(): 4 ctx1adbc

24 return {"name": "double invalid", "price": "foo", "owner_ids": ["foo", "bar"]} 4 ctx1ijkl

25 

26 

27@app.get("/items/invalidlist", response_model=list[Item]) 4 ctx1adbc

28def get_invalidlist(): 4 ctx1adbc

29 return [ 4 ctx1mnop

30 {"name": "foo"}, 

31 {"name": "bar", "price": "bar"}, 

32 {"name": "baz", "price": "baz"}, 

33 ] 

34 

35 

36client = TestClient(app) 4 ctx1adbc

37 

38 

39def test_invalid(): 4 ctx1adbc

40 with pytest.raises(ResponseValidationError): 4 ctx1efgh

41 client.get("/items/invalid") 4 ctx1efgh

42 

43 

44def test_double_invalid(): 4 ctx1adbc

45 with pytest.raises(ResponseValidationError): 4 ctx1ijkl

46 client.get("/items/innerinvalid") 4 ctx1ijkl

47 

48 

49def test_invalid_list(): 4 ctx1adbc

50 with pytest.raises(ResponseValidationError): 4 ctx1mnop

51 client.get("/items/invalidlist") 4 ctx1mnop