def test_duplicated_operation_ids(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi", delay: 5) }' ), ).as_dict() ) ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi", delay: 5) }' ), ).as_dict() ) data = ws.receive() assert data["type"] == "websocket.close" assert data["code"] == 4409
async def test_duplicated_operation_ids(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi", delay: 5) }'), ).as_dict()) await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi", delay: 5) }'), ).as_dict()) data = await ws.receive(timeout=2) assert ws.closed assert ws.close_code == 4409 assert data.extra == "Subscriber for sub1 already exists"
def test_subscription_cancellation(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi", delay: 99) }' ), ).as_dict() ) ws.send_json( SubscribeMessage( id="sub2", payload=SubscribeMessagePayload( query="subscription { debug { numActiveResultHandlers } }", ), ).as_dict() ) response = ws.receive_json() assert ( response == NextMessage( id="sub2", payload={"data": {"debug": {"numActiveResultHandlers": 2}}} ).as_dict() ) response = ws.receive_json() assert response == CompleteMessage(id="sub2").as_dict() ws.send_json(CompleteMessage(id="sub1").as_dict()) ws.send_json( SubscribeMessage( id="sub3", payload=SubscribeMessagePayload( query="subscription { debug { numActiveResultHandlers } }", ), ).as_dict() ) response = ws.receive_json() assert ( response == NextMessage( id="sub3", payload={"data": {"debug": {"numActiveResultHandlers": 1}}} ).as_dict() ) response = ws.receive_json() assert response == CompleteMessage(id="sub3").as_dict() ws.close()
async def test_simple_subscription(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi") }'), ).as_dict()) response = await ws.receive_json() assert (response == NextMessage(id="sub1", payload={ "data": { "echo": "Hi" } }).as_dict()) await ws.send_json(CompleteMessage(id="sub1").as_dict()) await ws.close() assert ws.closed
async def test_subscription_field_errors(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query="subscription { notASubscriptionField }", ), ).as_dict()) response = await ws.receive_json() assert response["type"] == ErrorMessage.type assert response["id"] == "sub1" assert len(response["payload"]) == 1 assert response["payload"][0]["path"] is None assert response["payload"][0]["locations"] == [{ "line": 1, "column": 16 }] assert ( response["payload"][0]["message"] == "The subscription field 'notASubscriptionField' is not defined.") await ws.close() assert ws.closed
async def test_subscription_exceptions(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { exception(message: "TEST EXC") }', ), ).as_dict()) response = await ws.receive_json() assert response["type"] == ErrorMessage.type assert response["id"] == "sub1" assert len(response["payload"]) == 1 assert response["payload"][0]["path"] is None assert response["payload"][0]["locations"] is None assert response["payload"][0]["message"] == "TEST EXC" await ws.close() assert ws.closed
def test_server_sent_ping(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload(query="subscription { requestPing }"), ).as_dict() ) response = ws.receive_json() assert response == PingMessage().as_dict() ws.send_json(PongMessage().as_dict()) response = ws.receive_json() assert ( response == NextMessage(id="sub1", payload={"data": {"requestPing": True}}).as_dict() ) response = ws.receive_json() assert response == CompleteMessage(id="sub1").as_dict() ws.close()
async def test_connection_init_timeout_cancellation(test_client): app = create_app(connection_init_wait_timeout=timedelta(milliseconds=500)) test_client = TestClient(app) with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() await asyncio.sleep(1) ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query="subscription { debug { isConnectionInitTimeoutTaskDone } }" ), ).as_dict() ) response = ws.receive_json() assert ( response == NextMessage( id="sub1", payload={"data": {"debug": {"isConnectionInitTimeoutTaskDone": True}}}, ).as_dict() ) ws.close()
def test_subscription_exceptions(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { exception(message: "TEST EXC") }', ), ).as_dict() ) response = ws.receive_json() assert response["type"] == ErrorMessage.type assert response["id"] == "sub1" assert len(response["payload"]) == 1 assert response["payload"][0]["path"] is None assert response["payload"][0]["locations"] is None assert response["payload"][0]["message"] == "TEST EXC" ws.close()
def test_subscription_field_errors(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query="subscription { notASubscriptionField }", ), ).as_dict() ) response = ws.receive_json() assert response["type"] == ErrorMessage.type assert response["id"] == "sub1" assert len(response["payload"]) == 1 assert response["payload"][0]["path"] is None assert response["payload"][0]["locations"] == [{"line": 1, "column": 16}] assert ( response["payload"][0]["message"] == "The subscription field 'notASubscriptionField' is not defined." ) ws.close()
def test_subscription_syntax_error(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload(query="subscription { INVALID_SYNTAX "), ).as_dict() ) response = ws.receive_json() assert response["type"] == ErrorMessage.type assert response["id"] == "sub1" assert len(response["payload"]) == 1 assert response["payload"][0]["path"] is None assert response["payload"][0]["locations"] == [{"line": 1, "column": 31}] assert ( response["payload"][0]["message"] == "Syntax Error: Expected Name, found <EOF>." ) ws.close()
def test_simple_subscription(test_client): with test_client.websocket_connect("/", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json(ConnectionInitMessage().as_dict()) response = ws.receive_json() assert response == ConnectionAckMessage().as_dict() ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi") }' ), ).as_dict() ) response = ws.receive_json() assert ( response == NextMessage(id="sub1", payload={"data": {"echo": "Hi"}}).as_dict() ) ws.send_json(CompleteMessage(id="sub1").as_dict()) ws.close()
async def test_subscription_syntax_error(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query="subscription { INVALID_SYNTAX "), ).as_dict()) response = await ws.receive_json() assert response["type"] == ErrorMessage.type assert response["id"] == "sub1" assert len(response["payload"]) == 1 assert response["payload"][0]["path"] is None assert response["payload"][0]["locations"] == [{ "line": 1, "column": 31 }] assert (response["payload"][0]["message"] == "Syntax Error: Expected Name, found <EOF>.") await ws.close() assert ws.closed
async def handle_message(self, message: dict): try: message_type = message.pop("type") if message_type == ConnectionInitMessage.type: await self.handle_connection_init( ConnectionInitMessage(**message)) elif message_type == PingMessage.type: await self.handle_ping(PingMessage(**message)) elif message_type == PongMessage.type: await self.handle_pong(PongMessage(**message)) elif message_type == SubscribeMessage.type: payload = SubscribeMessagePayload(**message.pop("payload")) await self.handle_subscribe( SubscribeMessage(payload=payload, **message)) elif message_type == CompleteMessage.type: await self.handle_complete(CompleteMessage(**message)) else: error_message = f"Unknown message type: {message_type}" await self.handle_invalid_message(error_message) except (KeyError, TypeError): error_message = "Failed to parse message" await self.handle_invalid_message(error_message)
def test_unauthorized_subscriptions(test_client): with test_client.websocket_connect("/graphql", [GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi") }'), ).as_dict()) data = ws.receive() assert data["type"] == "websocket.close" assert data["code"] == 4401
async def test_unauthorized_subscriptions(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi") }'), ).as_dict()) data = await ws.receive(timeout=2) assert ws.closed assert ws.close_code == 4401 assert data.extra == "Unauthorized"
async def test_connection_init_timeout_cancellation(aiohttp_client): app = create_app(connection_init_wait_timeout=timedelta(milliseconds=50)) aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await asyncio.sleep(0.1) await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query= "subscription { debug { isConnectionInitTimeoutTaskDone } }" ), ).as_dict()) response = await ws.receive_json() assert (response == NextMessage( id="sub1", payload={ "data": { "debug": { "isConnectionInitTimeoutTaskDone": True } } }, ).as_dict()) await ws.close() assert ws.closed
async def test_server_sent_ping(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query="subscription { requestPing }"), ).as_dict()) response = await ws.receive_json() assert response == PingMessage().as_dict() await ws.send_json(PongMessage().as_dict()) response = await ws.receive_json() assert (response == NextMessage(id="sub1", payload={ "data": { "requestPing": True } }).as_dict()) response = await ws.receive_json() assert response == CompleteMessage(id="sub1").as_dict() await ws.close() assert ws.closed
async def test_subscription_cancellation(aiohttp_client): app = create_app() aiohttp_app_client = await aiohttp_client(app) async with aiohttp_app_client.ws_connect( "/graphql", protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL]) as ws: await ws.send_json(ConnectionInitMessage().as_dict()) response = await ws.receive_json() assert response == ConnectionAckMessage().as_dict() await ws.send_json( SubscribeMessage( id="sub1", payload=SubscribeMessagePayload( query='subscription { echo(message: "Hi", delay: 99) }'), ).as_dict()) await ws.send_json( SubscribeMessage( id="sub2", payload=SubscribeMessagePayload( query="subscription { debug { numActiveResultHandlers } }", ), ).as_dict()) response = await ws.receive_json() assert (response == NextMessage(id="sub2", payload={ "data": { "debug": { "numActiveResultHandlers": 2 } } }).as_dict()) response = await ws.receive_json() assert response == CompleteMessage(id="sub2").as_dict() await ws.send_json(CompleteMessage(id="sub1").as_dict()) await ws.send_json( SubscribeMessage( id="sub3", payload=SubscribeMessagePayload( query="subscription { debug { numActiveResultHandlers } }", ), ).as_dict()) response = await ws.receive_json() assert (response == NextMessage(id="sub3", payload={ "data": { "debug": { "numActiveResultHandlers": 1 } } }).as_dict()) response = await ws.receive_json() assert response == CompleteMessage(id="sub3").as_dict() await ws.close() assert ws.closed