Coverage for tests/test_validate_response_dataclass.py: 100%

1from typing import List, Optional 7 ctx1abcdefg

2 

3import pytest 7 ctx1abcdefg

4from fastapi import FastAPI 7 ctx1abcdefg

5from fastapi.exceptions import ResponseValidationError 7 ctx1abcdefg

6from fastapi.testclient import TestClient 7 ctx1abcdefg

7from pydantic.dataclasses import dataclass 7 ctx1abcdefg

8 

9app = FastAPI() 7 ctx1abcdefg

10 

11 

12@dataclass 7 ctx1abcdefg

13class Item: 7 ctx1abcdefg

14 name: str 7 ctx1abcdefg

15 price: Optional[float] = None 7 ctx1abcdefg

16 owner_ids: Optional[List[int]] = None 7 ctx1abcdefg

17 

18 

19@app.get("/items/invalid", response_model=Item) 7 ctx1abcdefg

20def get_invalid(): 7 ctx1abcdefg

21 return {"name": "invalid", "price": "foo"} 7 ctx1hijklmn

22 

23 

24@app.get("/items/innerinvalid", response_model=Item) 7 ctx1abcdefg

25def get_innerinvalid(): 7 ctx1abcdefg

26 return {"name": "double invalid", "price": "foo", "owner_ids": ["foo", "bar"]} 7 ctx1opqrstu

27 

28 

29@app.get("/items/invalidlist", response_model=List[Item]) 7 ctx1abcdefg

30def get_invalidlist(): 7 ctx1abcdefg

31 return [ 7 ctx1vwxyzAB

32 {"name": "foo"}, 

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

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

35 ] 

36 

37 

38client = TestClient(app) 7 ctx1abcdefg

39 

40 

41def test_invalid(): 7 ctx1abcdefg

42 with pytest.raises(ResponseValidationError): 7 ctx1hijklmn

43 client.get("/items/invalid") 7 ctx1hijklmn

44 

45 

46def test_double_invalid(): 7 ctx1abcdefg

47 with pytest.raises(ResponseValidationError): 7 ctx1opqrstu

48 client.get("/items/innerinvalid") 7 ctx1opqrstu

49 

50 

51def test_invalid_list(): 7 ctx1abcdefg

52 with pytest.raises(ResponseValidationError): 7 ctx1vwxyzAB

53 client.get("/items/invalidlist") 7 ctx1vwxyzAB