Coverage for tests/test_validate_response.py: 100%

1from typing import List, Optional, Union 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 import BaseModel 7 ctx1abcdefg

8 

9app = FastAPI() 7 ctx1abcdefg

10 

11 

12class Item(BaseModel): 7 ctx1abcdefg

13 name: str 7 ctx1abcdefg

14 price: Optional[float] = None 7 ctx1abcdefg

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

16 

17 

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

19def get_invalid(): 7 ctx1abcdefg

20 return {"name": "invalid", "price": "foo"} 7 ctx1vwxyzAB

21 

22 

23@app.get("/items/invalidnone", response_model=Item) 7 ctx1abcdefg

24def get_invalid_none(): 7 ctx1abcdefg

25 return None 7 ctx1CDEFGHI

26 

27 

28@app.get("/items/validnone", response_model=Union[Item, None]) 7 ctx1abcdefg

29def get_valid_none(send_none: bool = False): 7 ctx1abcdefg

30 if send_none: 14 ctx1hijklmnopqrstu

31 return None 7 ctx1ikmoqsu

32 else: 

33 return {"name": "invalid", "price": 3.2} 7 ctx1hjlnprt

34 

35 

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

37def get_innerinvalid(): 7 ctx1abcdefg

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

39 

40 

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

42def get_invalidlist(): 7 ctx1abcdefg

43 return [ 7 ctx1QRSTUVW

44 {"name": "foo"}, 

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

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

47 ] 

48 

49 

50client = TestClient(app) 7 ctx1abcdefg

51 

52 

53def test_invalid(): 7 ctx1abcdefg

54 with pytest.raises(ResponseValidationError): 7 ctx1vwxyzAB

55 client.get("/items/invalid") 7 ctx1vwxyzAB

56 

57 

58def test_invalid_none(): 7 ctx1abcdefg

59 with pytest.raises(ResponseValidationError): 7 ctx1CDEFGHI

60 client.get("/items/invalidnone") 7 ctx1CDEFGHI

61 

62 

63def test_valid_none_data(): 7 ctx1abcdefg

64 response = client.get("/items/validnone") 7 ctx1hjlnprt

65 data = response.json() 7 ctx1hjlnprt

66 assert response.status_code == 200 7 ctx1hjlnprt

67 assert data == {"name": "invalid", "price": 3.2, "owner_ids": None} 7 ctx1hjlnprt

68 

69 

70def test_valid_none_none(): 7 ctx1abcdefg

71 response = client.get("/items/validnone", params={"send_none": "true"}) 7 ctx1ikmoqsu

72 data = response.json() 7 ctx1ikmoqsu

73 assert response.status_code == 200 7 ctx1ikmoqsu

74 assert data is None 7 ctx1ikmoqsu

75 

76 

77def test_double_invalid(): 7 ctx1abcdefg

78 with pytest.raises(ResponseValidationError): 7 ctx1JKLMNOP

79 client.get("/items/innerinvalid") 7 ctx1JKLMNOP

80 

81 

82def test_invalid_list(): 7 ctx1abcdefg

83 with pytest.raises(ResponseValidationError): 7 ctx1QRSTUVW

84 client.get("/items/invalidlist") 7 ctx1QRSTUVW