Updating snapshots

blast-hardcheese · blast-hardcheese · commit cff5364768cf · 2025-03-25T22:55:51.000-07:00
diff --git a/tests/codegen/rpc/generated/test_service/__init__.py b/tests/codegen/rpc/generated/test_service/__init__.py
@@ -11,7 +11,6 @@
 
 from .rpc_method import (
     Rpc_MethodInput,
-    Rpc_MethodInputTypeAdapter,
     Rpc_MethodOutput,
     Rpc_MethodOutputTypeAdapter,
     encode_Rpc_MethodInput,
diff --git a/tests/codegen/rpc/generated/test_service/rpc_method.py b/tests/codegen/rpc/generated/test_service/rpc_method.py
@@ -40,9 +40,6 @@ class Rpc_MethodInput(TypedDict):
     data: str
 
 
-Rpc_MethodInputTypeAdapter: TypeAdapter[Rpc_MethodInput] = TypeAdapter(Rpc_MethodInput)
-
-
 class Rpc_MethodOutput(BaseModel):
     data: str
 
diff --git a/tests/codegen/snapshot/snapshots/test_basic_stream/test_service/__init__.py b/tests/codegen/snapshot/snapshots/test_basic_stream/test_service/__init__.py
@@ -11,17 +11,18 @@
 
 from .stream_method import (
     Stream_MethodInput,
-    Stream_MethodInputTypeAdapter,
     Stream_MethodOutput,
     Stream_MethodOutputTypeAdapter,
     encode_Stream_MethodInput,
 )
-from .emit_error import Emit_ErrorErrors, Emit_ErrorErrorsTypeAdapter
+from .emit_error import Emit_ErrorErrors
 
-intTypeAdapter: TypeAdapter[int] = TypeAdapter(int)
+boolTypeAdapter: TypeAdapter[bool] = TypeAdapter(bool)
 
 
-boolTypeAdapter: TypeAdapter[bool] = TypeAdapter(bool)
+Emit_ErrorErrorsTypeAdapter: TypeAdapter[Emit_ErrorErrors] = TypeAdapter(
+    Emit_ErrorErrors
+)
 
 
 class Test_ServiceService:
diff --git a/tests/codegen/snapshot/snapshots/test_basic_stream/test_service/emit_error.py b/tests/codegen/snapshot/snapshots/test_basic_stream/test_service/emit_error.py
@@ -38,8 +38,3 @@ class Emit_ErrorErrorsOneOf_UNEXPECTED_DISCONNECT(RiverError):
     | RiverUnknownError,
     WrapValidator(translate_unknown_error),
 ]
-
-
-Emit_ErrorErrorsTypeAdapter: TypeAdapter[Emit_ErrorErrors] = TypeAdapter(
-    Emit_ErrorErrors
-)
diff --git a/tests/codegen/snapshot/snapshots/test_basic_stream/test_service/stream_method.py b/tests/codegen/snapshot/snapshots/test_basic_stream/test_service/stream_method.py
@@ -40,11 +40,6 @@ class Stream_MethodInput(TypedDict):
     data: str
 
 
-Stream_MethodInputTypeAdapter: TypeAdapter[Stream_MethodInput] = TypeAdapter(
-    Stream_MethodInput
-)
-
-
 class Stream_MethodOutput(BaseModel):
     data: str
 
diff --git a/tests/codegen/snapshot/snapshots/test_pathological_types/test_service/__init__.py b/tests/codegen/snapshot/snapshots/test_pathological_types/test_service/__init__.py
@@ -11,7 +11,6 @@
 
 from .pathological_method import (
     Pathological_MethodInput,
-    Pathological_MethodInputTypeAdapter,
     encode_Pathological_MethodInput,
     encode_Pathological_MethodInputObj_Boolean,
     encode_Pathological_MethodInputObj_Date,
diff --git a/tests/codegen/snapshot/snapshots/test_pathological_types/test_service/pathological_method.py b/tests/codegen/snapshot/snapshots/test_pathological_types/test_service/pathological_method.py
@@ -476,8 +476,3 @@ class Pathological_MethodInput(TypedDict):
     req_undefined: None
     string: NotRequired[str | None]
     undefined: NotRequired[None]
-
-
-Pathological_MethodInputTypeAdapter: TypeAdapter[Pathological_MethodInput] = (
-    TypeAdapter(Pathological_MethodInput)
-)
diff --git a/tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/__init__.py b/tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/__init__.py
@@ -13,7 +13,6 @@
     NeedsenumErrors,
     NeedsenumErrorsTypeAdapter,
     NeedsenumInput,
-    NeedsenumInputTypeAdapter,
     NeedsenumOutput,
     NeedsenumOutputTypeAdapter,
     encode_NeedsenumInput,
@@ -22,7 +21,6 @@
     NeedsenumobjectErrors,
     NeedsenumobjectErrorsTypeAdapter,
     NeedsenumobjectInput,
-    NeedsenumobjectInputTypeAdapter,
     NeedsenumobjectOutput,
     NeedsenumobjectOutputTypeAdapter,
     encode_NeedsenumobjectInput,
diff --git a/tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnum.py b/tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnum.py
@@ -29,8 +29,6 @@ def encode_NeedsenumInput(x: "NeedsenumInput") -> Any:
     return x
 
 
-NeedsenumInputTypeAdapter: TypeAdapter[NeedsenumInput] = TypeAdapter(NeedsenumInput)
-
 NeedsenumOutput = Annotated[
     Literal["out_first", "out_second"] | RiverUnknownValue,
     WrapValidator(translate_unknown_value),
diff --git a/tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnumObject.py b/tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnumObject.py
@@ -77,11 +77,6 @@ def encode_NeedsenumobjectInput(
     )
 
 
-NeedsenumobjectInputTypeAdapter: TypeAdapter[NeedsenumobjectInput] = TypeAdapter(
-    NeedsenumobjectInput
-)
-
-
 class NeedsenumobjectOutputFooOneOf_out_first(BaseModel):
     kind: Literal["out_first"] = Field(
         "out_first",
diff --git a/tests/codegen/test_rpc.py b/tests/codegen/test_rpc.py
@@ -19,8 +19,6 @@
 
 @pytest.fixture(scope="session", autouse=True)
 def generate_rpc_client() -> None:
-    import tests.codegen.rpc.generated
-
     shutil.rmtree("tests/codegen/rpc/generated")
 
     def file_opener(path: Path) -> TextIO:
@@ -35,12 +33,13 @@ def file_opener(path: Path) -> TextIO:
         protocol_version="v1.1",
         method_filter=None,
     )
-    importlib.reload(tests.codegen.rpc.generated)
 
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize("handlers", [{**basic_rpc_method}])
 async def test_basic_rpc(client: Client) -> None:
+    import tests.codegen.rpc.generated
+    importlib.reload(tests.codegen.rpc.generated)
     from tests.codegen.rpc.generated import RpcClient
 
     res = await RpcClient(client).test_service.rpc_method(
@@ -70,6 +69,8 @@ async def rpc_timeout_handler(request: str, context: grpc.aio.ServicerContext) -
 @pytest.mark.asyncio
 @pytest.mark.parametrize("handlers", [{**rpc_timeout_method}])
 async def test_rpc_timeout(client: Client) -> None:
+    import tests.codegen.rpc.generated
+    importlib.reload(tests.codegen.rpc.generated)
     from tests.codegen.rpc.generated import RpcClient
 
     with pytest.raises(RiverException):

Original file line number	Diff line number	Diff line change
`@@ -38,8 +38,3 @@ class Emit_ErrorErrorsOneOf_UNEXPECTED_DISCONNECT(RiverError):`
`38`	`38`	`\| RiverUnknownError,`
`39`	`39`	`WrapValidator(translate_unknown_error),`
`40`	`40`	`]`
`41`		`-`
`42`		`-`
`43`		`-Emit_ErrorErrorsTypeAdapter: TypeAdapter[Emit_ErrorErrors] = TypeAdapter(`
`44`		`- Emit_ErrorErrors`
`45`		`-)`