test(websocket): separate ping and init

This commit is contained in:
Houkime 2024-05-15 18:36:17 +00:00
parent 02d337c3f0
commit 8348f11faf

View file

@ -29,7 +29,7 @@ def test_websocket_connection_bare(authorized_client):
assert websocket.scope is not None assert websocket.scope is not None
def test_websocket_graphql_ping(authorized_client): def test_websocket_graphql_init(authorized_client):
client = authorized_client client = authorized_client
with client.websocket_connect( with client.websocket_connect(
"/graphql", subprotocols=["graphql-transport-ws"] "/graphql", subprotocols=["graphql-transport-ws"]
@ -38,6 +38,12 @@ def test_websocket_graphql_ping(authorized_client):
ack = websocket.receive_json() ack = websocket.receive_json()
assert ack == {"type": "connection_ack"} assert ack == {"type": "connection_ack"}
def test_websocket_graphql_ping(authorized_client):
client = authorized_client
with client.websocket_connect(
"/graphql", subprotocols=["graphql-transport-ws"]
) as websocket:
# https://github.com/enisdenjo/graphql-ws/blob/master/PROTOCOL.md#ping # https://github.com/enisdenjo/graphql-ws/blob/master/PROTOCOL.md#ping
websocket.send_json({"type": "ping", "payload": {}}) websocket.send_json({"type": "ping", "payload": {}})
pong = websocket.receive_json() pong = websocket.receive_json()