Skip to content

Commit c77679e

Browse files
committed
resnapshotted the tests
1 parent 7be8c26 commit c77679e

File tree

4 files changed

+22
-10
lines changed

4 files changed

+22
-10
lines changed

tests/codegen/rpc/generated/test_service/rpc_method.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,13 @@ class Rpc_MethodInput(TypedDict):
3939
data: str
4040

4141

42-
Rpc_MethodInputTypeAdapter = TypeAdapter(Rpc_MethodInput) # type: ignore
42+
Rpc_MethodInputTypeAdapter: TypeAdapter[Rpc_MethodInput] = TypeAdapter(Rpc_MethodInput)
4343

4444

4545
class Rpc_MethodOutput(BaseModel):
4646
data: str
4747

4848

49-
Rpc_MethodOutputTypeAdapter = TypeAdapter(Rpc_MethodOutput) # type: ignore
49+
Rpc_MethodOutputTypeAdapter: TypeAdapter[Rpc_MethodOutput] = TypeAdapter(
50+
Rpc_MethodOutput
51+
)

tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnum.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,14 +26,14 @@
2626

2727
NeedsenumInput = Literal["in_first"] | Literal["in_second"]
2828
encode_NeedsenumInput: Callable[["NeedsenumInput"], Any] = lambda x: x
29-
NeedsenumInputTypeAdapter = TypeAdapter(NeedsenumInput) # type: ignore
29+
NeedsenumInputTypeAdapter: TypeAdapter[NeedsenumInput] = TypeAdapter(NeedsenumInput)
3030
NeedsenumOutput = Annotated[
3131
Literal["out_first"] | Literal["out_second"] | RiverUnknownValue,
3232
WrapValidator(translate_unknown_value),
3333
]
34-
NeedsenumOutputTypeAdapter = TypeAdapter(NeedsenumOutput) # type: ignore
34+
NeedsenumOutputTypeAdapter: TypeAdapter[NeedsenumOutput] = TypeAdapter(NeedsenumOutput)
3535
NeedsenumErrors = Annotated[
3636
Literal["err_first"] | Literal["err_second"] | RiverUnknownValue,
3737
WrapValidator(translate_unknown_value),
3838
]
39-
NeedsenumErrorsTypeAdapter = TypeAdapter(NeedsenumErrors) # type: ignore
39+
NeedsenumErrorsTypeAdapter: TypeAdapter[NeedsenumErrors] = TypeAdapter(NeedsenumErrors)

tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnumObject.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,9 @@ class NeedsenumobjectInputOneOf_in_second(TypedDict):
7171
if x["kind"] == "in_first"
7272
else encode_NeedsenumobjectInputOneOf_in_second(x)
7373
)
74-
NeedsenumobjectInputTypeAdapter = TypeAdapter(NeedsenumobjectInput) # type: ignore
74+
NeedsenumobjectInputTypeAdapter: TypeAdapter[NeedsenumobjectInput] = TypeAdapter(
75+
NeedsenumobjectInput
76+
)
7577

7678

7779
class NeedsenumobjectOutputFooOneOf_out_first(BaseModel):
@@ -104,7 +106,9 @@ class NeedsenumobjectOutput(BaseModel):
104106
foo: Optional[NeedsenumobjectOutputFoo] = None
105107

106108

107-
NeedsenumobjectOutputTypeAdapter = TypeAdapter(NeedsenumobjectOutput) # type: ignore
109+
NeedsenumobjectOutputTypeAdapter: TypeAdapter[NeedsenumobjectOutput] = TypeAdapter(
110+
NeedsenumobjectOutput
111+
)
108112

109113

110114
class NeedsenumobjectErrorsFooAnyOf_0(RiverError):
@@ -127,4 +131,6 @@ class NeedsenumobjectErrors(RiverError):
127131
foo: Optional[NeedsenumobjectErrorsFoo] = None
128132

129133

130-
NeedsenumobjectErrorsTypeAdapter = TypeAdapter(NeedsenumobjectErrors) # type: ignore
134+
NeedsenumobjectErrorsTypeAdapter: TypeAdapter[NeedsenumobjectErrors] = TypeAdapter(
135+
NeedsenumobjectErrors
136+
)

tests/codegen/stream/generated/test_service/stream_method.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,15 @@ class Stream_MethodInput(TypedDict):
3939
data: str
4040

4141

42-
Stream_MethodInputTypeAdapter = TypeAdapter(Stream_MethodInput) # type: ignore
42+
Stream_MethodInputTypeAdapter: TypeAdapter[Stream_MethodInput] = TypeAdapter(
43+
Stream_MethodInput
44+
)
4345

4446

4547
class Stream_MethodOutput(BaseModel):
4648
data: str
4749

4850

49-
Stream_MethodOutputTypeAdapter = TypeAdapter(Stream_MethodOutput) # type: ignore
51+
Stream_MethodOutputTypeAdapter: TypeAdapter[Stream_MethodOutput] = TypeAdapter(
52+
Stream_MethodOutput
53+
)

0 commit comments

Comments
 (0)