Coverage for tests / test_pydantic_v1_error.py: 100%

1import sys 4 ctx1afbc

2import warnings 4 ctx1afbc

3 

4import pytest 4 ctx1afbc

5 

6from tests.utils import skip_module_if_py_gte_314 4 ctx1afbc

7 

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

9 skip_module_if_py_gte_314() 1 ctx1f

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): 2 ctx1gh

21 name: str 2 ctx1gh

22 

23 app = FastAPI() 2 ctx1gh

24 

25 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1gh

26 

27 @app.post("/param") 2 ctx1gh

28 def endpoint(data: ParamModelV1): # pragma: no cover 2 ctx1gh

29 return data 

30 

31 

32def test_raises_pydantic_v1_model_in_return_type() -> None: 3 ctx1abc

33 class ReturnModelV1(BaseModel): 2 ctx1ij

34 name: str 2 ctx1ij

35 

36 app = FastAPI() 2 ctx1ij

37 

38 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1ij

39 

40 @app.get("/return") 2 ctx1ij

41 def endpoint() -> ReturnModelV1: # pragma: no cover 2 ctx1ij

42 return ReturnModelV1(name="test") 

43 

44 

45def test_raises_pydantic_v1_model_in_response_model() -> None: 3 ctx1abc

46 class ResponseModelV1(BaseModel): 2 ctx1kl

47 name: str 2 ctx1kl

48 

49 app = FastAPI() 2 ctx1kl

50 

51 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1kl

52 

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

54 def endpoint(): # pragma: no cover 2 ctx1kl

55 return {"name": "test"} 

56 

57 

58def test_raises_pydantic_v1_model_in_additional_responses_model() -> None: 3 ctx1abc

59 class ErrorModelV1(BaseModel): 2 ctx1de

60 detail: str 2 ctx1de

61 

62 app = FastAPI() 2 ctx1de

63 

64 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1de

65 

66 @app.get( 2 ctx1de

67 "/responses", response_model=None, responses={400: {"model": ErrorModelV1}} 2 ctx1de

68 ) 

69 def endpoint(): # pragma: no cover 2 ctx1de

70 return {"ok": True} 

71 

72 

73def test_raises_pydantic_v1_model_in_union() -> None: 3 ctx1abc

74 class ModelV1A(BaseModel): 2 ctx1mn

75 name: str 2 ctx1mn

76 

77 app = FastAPI() 2 ctx1mn

78 

79 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1mn

80 

81 @app.post("/union") 2 ctx1mn

82 def endpoint(data: dict | ModelV1A): # pragma: no cover 2 ctx1mn

83 return data 

84 

85 

86def test_raises_pydantic_v1_model_in_sequence() -> None: 3 ctx1abc

87 class ModelV1A(BaseModel): 2 ctx1op

88 name: str 2 ctx1op

89 

90 app = FastAPI() 2 ctx1op

91 

92 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1op

93 

94 @app.post("/sequence") 2 ctx1op

95 def endpoint(data: list[ModelV1A]): # pragma: no cover 2 ctx1op

96 return data