Skip to content

Commit 8430149

Browse files
Use TypeAdapter to decode handshake request payload
1 parent eb40cac commit 8430149

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

tests/v2/test_v2_cancellation.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import msgpack
1111
import nanoid
1212
import pytest
13+
from pydantic import TypeAdapter
1314

1415
from replit_river.messages import parse_transport_msg
1516
from replit_river.rpc import (
@@ -280,9 +281,9 @@ async def test_subscription_cancel(ws_server: WsServerFixture) -> None:
280281

281282
assert not isinstance(request_msg, str)
282283
assert (serverconn := conn())
283-
handshake_request: ControlMessageHandshakeRequest[None] = (
284-
ControlMessageHandshakeRequest(**request_msg.payload)
285-
)
284+
handshake_request = TypeAdapter(
285+
ControlMessageHandshakeRequest[None]
286+
).validate_python(request_msg.payload)
286287

287288
handshake_resp = ControlMessageHandshakeResponse(
288289
status=HandShakeStatus(

0 commit comments

Comments
 (0)