66import pytest
77from pydantic import BaseModel , ValidationError
88
9- from repid import Header , Root
9+ from repid import FullPayload , Header
1010from repid .converter import BasicConverter , ConverterT , PydanticConverter
1111from repid .data import MessageData
1212from repid .dependencies import Depends
@@ -773,12 +773,12 @@ async def _fn_with_shared_header_in_two_depends(
773773 }
774774
775775
776- async def test_pydantic_converter_root_model_parse () -> None :
776+ async def test_pydantic_converter_full_payload_model_parse () -> None :
777777 class MyModel (BaseModel ):
778778 a : int
779779 b : float
780780
781- async def fn (model : Annotated [MyModel , Root ()]) -> None : ...
781+ async def fn (model : Annotated [MyModel , FullPayload ()]) -> None : ...
782782
783783 conv = PydanticConverter (fn , fn_locals = locals (), correlation_id = None )
784784 args , kwargs = await conv .convert_inputs (
@@ -795,12 +795,12 @@ async def fn(model: Annotated[MyModel, Root()]) -> None: ...
795795 assert kwargs == {"model" : MyModel (a = 1 , b = 2.5 )}
796796
797797
798- async def test_pydantic_converter_root_model_schema () -> None :
798+ async def test_pydantic_converter_full_payload_model_schema () -> None :
799799 class MyModel (BaseModel ):
800800 a : int
801801 b : float
802802
803- async def fn (model : Annotated [MyModel , Root ()]) -> None : ...
803+ async def fn (model : Annotated [MyModel , FullPayload ()]) -> None : ...
804804
805805 conv = PydanticConverter (fn , fn_locals = locals (), correlation_id = None )
806806 schema = conv .get_input_schema ()
@@ -817,12 +817,12 @@ async def fn(model: Annotated[MyModel, Root()]) -> None: ...
817817 assert schema .content_type == "application/json"
818818
819819
820- async def test_pydantic_converter_root_model_with_extra_kwargs () -> None :
820+ async def test_pydantic_converter_full_payload_model_with_extra_kwargs () -> None :
821821 class MyModel (BaseModel ):
822822 a : int
823823 b : float
824824
825- async def fn (model : Annotated [MyModel , Root ()], extra : str = "default" ) -> None : ...
825+ async def fn (model : Annotated [MyModel , FullPayload ()], extra : str = "default" ) -> None : ...
826826
827827 conv = PydanticConverter (fn , fn_locals = locals (), correlation_id = None )
828828 args , kwargs = await conv .convert_inputs (
@@ -842,11 +842,11 @@ async def fn(model: Annotated[MyModel, Root()], extra: str = "default") -> None:
842842 assert kwargs ["model" ].b == 2.5
843843
844844
845- async def test_pydantic_converter_root_model_with_extra_kwargs_schema () -> None :
845+ async def test_pydantic_converter_full_payload_model_with_extra_kwargs_schema () -> None :
846846 class MyModel (BaseModel ):
847847 a : int
848848
849- async def fn (model : Annotated [MyModel , Root ()], extra : str = "default" ) -> None : ...
849+ async def fn (model : Annotated [MyModel , FullPayload ()], extra : str = "default" ) -> None : ...
850850
851851 conv = PydanticConverter (fn , fn_locals = locals (), correlation_id = None )
852852 schema = conv .get_input_schema ()
@@ -857,27 +857,27 @@ async def fn(model: Annotated[MyModel, Root()], extra: str = "default") -> None:
857857 assert "extra" in props
858858
859859
860- def test_pydantic_converter_root_model_multiple_raises () -> None :
860+ def test_pydantic_converter_full_payload_model_multiple_raises () -> None :
861861 class M1 (BaseModel ):
862862 a : int
863863
864864 class M2 (BaseModel ):
865865 b : int
866866
867867 async def fn (
868- m1 : Annotated [M1 , Root ()],
869- m2 : Annotated [M2 , Root ()],
868+ m1 : Annotated [M1 , FullPayload ()],
869+ m2 : Annotated [M2 , FullPayload ()],
870870 ) -> None : ...
871871
872- with pytest .raises (ValueError , match = "Only one Root \\ (\\ ) marker is allowed per actor" ):
872+ with pytest .raises (ValueError , match = "Only one FullPayload \\ (\\ ) marker is allowed per actor" ):
873873 PydanticConverter (fn , fn_locals = locals (), correlation_id = None )
874874
875875
876- def test_basic_converter_root_model_raises () -> None :
876+ def test_basic_converter_full_payload_model_raises () -> None :
877877 class MyModel (BaseModel ):
878878 a : int
879879
880- async def fn (model : Annotated [MyModel , Root ()]) -> None : ...
880+ async def fn (model : Annotated [MyModel , FullPayload ()]) -> None : ...
881881
882- with pytest .raises (ValueError , match = "Root \\ (\\ ) requires PydanticConverter" ):
882+ with pytest .raises (ValueError , match = "FullPayload \\ (\\ ) requires PydanticConverter" ):
883883 BasicConverter (fn , fn_locals = locals (), correlation_id = None )
0 commit comments