Coverage for tests / test_pydantic_v1_error.py: 100%

1import sys 4 ctx1agbc

2import warnings 4 ctx1agbc

3 

4import pytest 4 ctx1agbc

5 

6from tests.utils import skip_module_if_py_gte_314 4 ctx1agbc

7 

8if sys.version_info >= (3, 14): 4 ctx1agbc

9 skip_module_if_py_gte_314() 1 ctx1g

10 

11from fastapi import FastAPI 3 ctx1abc

12from fastapi.exceptions import PydanticV1NotSupportedError 3 ctx1abc

13 

14with warnings.catch_warnings(): 3 ctx1abc

15 warnings.simplefilter("ignore", UserWarning) 3 ctx1abc

16 from pydantic.v1 import BaseModel 3 ctx1abc

17 

18 

19def test_raises_pydantic_v1_model_in_endpoint_param() -> None: 3 ctx1abc

20 class ParamModelV1(BaseModel): 3 ctx1hij

21 name: str 3 ctx1hij

22 

23 app = FastAPI() 3 ctx1hij

24 

25 with pytest.raises(PydanticV1NotSupportedError): 3 ctx1hij

26 

27 @app.post("/param") 3 ctx1hij

28 def endpoint(data: ParamModelV1): # pragma: no cover 3 ctx1hij

29 return data 

30 

31 

32def test_raises_pydantic_v1_model_in_return_type() -> None: 3 ctx1abc

33 class ReturnModelV1(BaseModel): 3 ctx1klm

34 name: str 3 ctx1klm

35 

36 app = FastAPI() 3 ctx1klm

37 

38 with pytest.raises(PydanticV1NotSupportedError): 3 ctx1klm

39 

40 @app.get("/return") 3 ctx1klm

41 def endpoint() -> ReturnModelV1: # pragma: no cover 3 ctx1klm

42 return ReturnModelV1(name="test") 

43 

44 

45def test_raises_pydantic_v1_model_in_response_model() -> None: 3 ctx1abc

46 class ResponseModelV1(BaseModel): 3 ctx1nop

47 name: str 3 ctx1nop

48 

49 app = FastAPI() 3 ctx1nop

50 

51 with pytest.raises(PydanticV1NotSupportedError): 3 ctx1nop

52 

53 @app.get("/response-model", response_model=ResponseModelV1) 3 ctx1nop

54 def endpoint(): # pragma: no cover 3 ctx1nop

55 return {"name": "test"} 

56 

57 

58def test_raises_pydantic_v1_model_in_additional_responses_model() -> None: 3 ctx1abc

59 class ErrorModelV1(BaseModel): 3 ctx1def

60 detail: str 3 ctx1def

61 

62 app = FastAPI() 3 ctx1def

63 

64 with pytest.raises(PydanticV1NotSupportedError): 3 ctx1def

65 

66 @app.get( 3 ctx1def

67 "/responses", response_model=None, responses={400: {"model": ErrorModelV1}} 3 ctx1def

68 ) 

69 def endpoint(): # pragma: no cover 3 ctx1def

70 return {"ok": True} 

71 

72 

73def test_raises_pydantic_v1_model_in_union() -> None: 3 ctx1abc

74 class ModelV1A(BaseModel): 3 ctx1qrs

75 name: str 3 ctx1qrs

76 

77 app = FastAPI() 3 ctx1qrs

78 

79 with pytest.raises(PydanticV1NotSupportedError): 3 ctx1qrs

80 

81 @app.post("/union") 3 ctx1qrs

82 def endpoint(data: dict | ModelV1A): # pragma: no cover 3 ctx1qrs

83 return data 

84 

85 

86def test_raises_pydantic_v1_model_in_sequence() -> None: 3 ctx1abc

87 class ModelV1A(BaseModel): 3 ctx1tuv

88 name: str 3 ctx1tuv

89 

90 app = FastAPI() 3 ctx1tuv

91 

92 with pytest.raises(PydanticV1NotSupportedError): 3 ctx1tuv

93 

94 @app.post("/sequence") 3 ctx1tuv

95 def endpoint(data: list[ModelV1A]): # pragma: no cover 3 ctx1tuv

96 return data