Coverage for tests/pydantic_specific/test_custom.py: 100%

1from typing import Annotated, Any 15 ctx1klabmcndoefghij

2 

3import pytest 15 ctx1klabmcndoefghij

4from annotated_types import Ge 15 ctx1klabmcndoefghij

5 

6from fast_depends import inject 15 ctx1klabmcndoefghij

7from fast_depends.exceptions import ValidationError 15 ctx1klabmcndoefghij

8from fast_depends.library import CustomField 15 ctx1klabmcndoefghij

9from tests.marks import pydanticV2 15 ctx1klabmcndoefghij

10 

11 

12class Header(CustomField): 15 ctx1klabmcndoefghij

13 def use(self, /, **kwargs: Any) -> dict[str, Any]: 15 ctx1klabmcndoefghij

14 kwargs = super().use(**kwargs) 10 ctx1abcdefghij

15 if v := kwargs.get("headers", {}).get(self.param_name): 10 ctx1abcdefghij

16 kwargs[self.param_name] = v 10 ctx1abcdefghij

17 return kwargs 10 ctx1abcdefghij

18 

19 

20@pydanticV2 15 ctx1klabmcndoefghij

21def test_annotated_header_with_meta() -> None: 15 ctx1klabmcndoefghij

22 @inject() 10 ctx1abcdefghij

23 def sync_catch(key: Annotated[int, Header(), Ge(3)] = 3) -> int: 10 ctx1abcdefghij

24 return key 10 ctx1abcdefghij

25 

26 assert sync_catch(headers={"key": "4"}) == 4 10 ctx1abcdefghij

27 

28 assert sync_catch(headers={}) == 3 10 ctx1abcdefghij

29 

30 with pytest.raises(ValidationError): 10 ctx1abcdefghij

31 sync_catch(headers={"key": "2"}) 10 ctx1abcdefghij