Coverage for tests / test_pydantic_v1_error.py: 100%

1import sys 4 ctx1adbc

2import warnings 4 ctx1adbc

3from typing import Union 4 ctx1adbc

4 

5import pytest 4 ctx1adbc

6 

7from tests.utils import skip_module_if_py_gte_314 4 ctx1adbc

8 

9if sys.version_info >= (3, 14): 4 ctx1adbc

10 skip_module_if_py_gte_314() 1 ctx1d

11 

12from fastapi import FastAPI 3 ctx1abc

13from fastapi.exceptions import PydanticV1NotSupportedError 3 ctx1abc

14 

15with warnings.catch_warnings(): 3 ctx1abc

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

17 from pydantic.v1 import BaseModel 3 ctx1abc

18 

19 

20def test_raises_pydantic_v1_model_in_endpoint_param() -> None: 3 ctx1abc

21 class ParamModelV1(BaseModel): 2 ctx1gh

22 name: str 2 ctx1gh

23 

24 app = FastAPI() 2 ctx1gh

25 

26 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1gh

27 

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

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

30 return data 

31 

32 

33def test_raises_pydantic_v1_model_in_return_type() -> None: 3 ctx1abc

34 class ReturnModelV1(BaseModel): 2 ctx1ij

35 name: str 2 ctx1ij

36 

37 app = FastAPI() 2 ctx1ij

38 

39 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1ij

40 

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

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

43 return ReturnModelV1(name="test") 

44 

45 

46def test_raises_pydantic_v1_model_in_response_model() -> None: 3 ctx1abc

47 class ResponseModelV1(BaseModel): 2 ctx1kl

48 name: str 2 ctx1kl

49 

50 app = FastAPI() 2 ctx1kl

51 

52 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1kl

53 

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

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

56 return {"name": "test"} 

57 

58 

59def test_raises_pydantic_v1_model_in_additional_responses_model() -> None: 3 ctx1abc

60 class ErrorModelV1(BaseModel): 2 ctx1ef

61 detail: str 2 ctx1ef

62 

63 app = FastAPI() 2 ctx1ef

64 

65 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1ef

66 

67 @app.get( 2 ctx1ef

68 "/responses", response_model=None, responses={400: {"model": ErrorModelV1}} 2 ctx1ef

69 ) 

70 def endpoint(): # pragma: no cover 2 ctx1ef

71 return {"ok": True} 

72 

73 

74def test_raises_pydantic_v1_model_in_union() -> None: 3 ctx1abc

75 class ModelV1A(BaseModel): 2 ctx1mn

76 name: str 2 ctx1mn

77 

78 app = FastAPI() 2 ctx1mn

79 

80 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1mn

81 

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

83 def endpoint(data: Union[dict, ModelV1A]): # pragma: no cover 2 ctx1mn

84 return data 

85 

86 

87def test_raises_pydantic_v1_model_in_sequence() -> None: 3 ctx1abc

88 class ModelV1A(BaseModel): 2 ctx1op

89 name: str 2 ctx1op

90 

91 app = FastAPI() 2 ctx1op

92 

93 with pytest.raises(PydanticV1NotSupportedError): 2 ctx1op

94 

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

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

97 return data