Consolidating

Author: blast-hardcheese

tests/v2/fixtures/raw_ws_server.py

@@ -3,6 +3,7 @@
     AsyncIterator,
     Awaitable,
     Callable,
+    Literal,
     TypeAlias,
     TypedDict,
 )
@@ -20,6 +21,11 @@
 ]
 
 
+class OuterPayload[A](TypedDict):
+    ok: Literal[True]
+    payload: A
+
+
 class _WsServerState(TypedDict):
     ipv4_laddr: tuple[str, int] | None
 

tests/v2/test_v2_cancellation.py

@@ -3,8 +3,6 @@
 from typing import (
     Any,
     AsyncIterator,
-    Literal,
-    TypedDict,
 )
 
 import msgpack
@@ -20,12 +18,7 @@
 from replit_river.transport_options import TransportOptions
 from replit_river.v2.client import Client
 from replit_river.v2.session import STREAM_CANCEL_BIT
-from tests.v2.fixtures.raw_ws_server import WsServerFixture
-
-
-class OuterPayload[A](TypedDict):
-    ok: Literal[True]
-    payload: A
+from tests.v2.fixtures.raw_ws_server import OuterPayload, WsServerFixture
 
 
 async def test_upload_cancel(ws_server: WsServerFixture) -> None:

tests/v2/test_v2_session_lifecycle.py

@@ -1,9 +1,5 @@
 import asyncio
 import logging
-from typing import (
-    Literal,
-    TypedDict,
-)
 
 import msgpack
 import nanoid
@@ -23,11 +19,6 @@
 from tests.v2.fixtures.raw_ws_server import WsServerFixture
 
 
-class OuterPayload[A](TypedDict):
-    ok: Literal[True]
-    payload: A
-
-
 class _PermissiveRateLimiter(RateLimiter):
     def start_restoring_budget(self, user: str) -> None:
         pass