add support for graphql-core 3 to graphql_ws.aiohttp

.travis.yml

@@ -18,15 +18,14 @@ deploy:
       MnY1TzdSMDJ0d2xyS3pXWlB6SGxLWHVLc0dwemxCS3RLZ0RVQW02aEFBaFZwdDBVbUhTWnVtQ0I1
       cnFkWndJWFdTcEQ4SU8rRHMzTTdwbGMzMThPT2ZkOFo2MXU1dVlRZkFlUklURkRpNjVLUHp4Y1U9
   on:
-    python: 2.7
+    python: 3.6
     repo: graphql-python/graphql-ws
     tags: true
-install: pip install -U tox-travis
+install: pip install -U tox-travis codecov
 language: python
 python:
 - 3.8
 - 3.7
 - 3.6
-- 3.5
-- 2.7
-script: tox
+script: tox -- --cov-branch --cov-report=term-missing --cov=graphql_ws
+after_success: codecov

README.md

@@ -7,6 +7,10 @@ Currently supports:
 * [Gevent](https://github.com/graphql-python/graphql-ws#gevent)
 * Sanic (uses [websockets](https://github.com/aaugustin/websockets/) library)
 
+[![PyPI version](https://badge.fury.io/py/graphql-ws.svg)](https://badge.fury.io/py/graphql-ws)
+[![TravisCI Build Status](https://travis-ci.org/graphql-python/graphql-ws.svg?branch=master)](https://travis-ci.org/graphql-python/graphql-ws)
+[![codecov](https://codecov.io/gh/graphql-python/graphql-ws/branch/master/graph/badge.svg)](https://codecov.io/gh/graphql-python/graphql-ws)
+
 # Installation instructions
 
 For instaling graphql-ws, just run this command in your shell
@@ -167,8 +171,8 @@ class Subscription(graphene.ObjectType):
 
 
     def resolve_count_seconds(
-        root, 
-        info, 
+        root,
+        info,
         up_to=5
     ):
         return Observable.interval(1000)\
@@ -202,4 +206,4 @@ from graphql_ws.django_channels import GraphQLSubscriptionConsumer
 channel_routing = [
     route_class(GraphQLSubscriptionConsumer, path=r"^/subscriptions"),
 ]
-```
+```

graphql_ws/aiohttp.py

@@ -1,20 +1,15 @@
-from inspect import isawaitable
-from asyncio import ensure_future, wait, shield
+from asyncio import ensure_future, shield, wait
 
 from aiohttp import WSMsgType
-from graphql.execution.executors.asyncio import AsyncioExecutor
+from graphql import subscribe
+from graphql.language import parse
 
 from .base import (
-    ConnectionClosedException, BaseConnectionContext, BaseSubscriptionServer)
-from .observable_aiter import setup_observable_extension
-
-from .constants import (
-    GQL_CONNECTION_ACK,
-    GQL_CONNECTION_ERROR,
-    GQL_COMPLETE
+    BaseConnectionContext,
+    BaseSubscriptionServer,
+    ConnectionClosedException,
 )
-
-setup_observable_extension()
+from .constants import GQL_COMPLETE, GQL_CONNECTION_ACK, GQL_CONNECTION_ERROR
 
 
 class AiohttpConnectionContext(BaseConnectionContext):
@@ -47,12 +42,6 @@ def __init__(self, schema, keep_alive=True, loop=None):
         self.loop = loop
         super().__init__(schema, keep_alive)
 
-    def get_graphql_params(self, *args, **kwargs):
-        params = super(AiohttpSubscriptionServer,
-                       self).get_graphql_params(*args, **kwargs)
-        return dict(params, return_promise=True,
-                    executor=AsyncioExecutor(loop=self.loop))
-
     async def _handle(self, ws, request_context=None):
         connection_context = AiohttpConnectionContext(ws, request_context)
         await self.on_open(connection_context)
@@ -69,7 +58,8 @@ async def _handle(self, ws, request_context=None):
                     (_, pending) = await wait(pending, timeout=0, loop=self.loop)
 
             task = ensure_future(
-                self.on_message(connection_context, message), loop=self.loop)
+                self.on_message(connection_context, message), loop=self.loop
+            )
             pending.add(task)
 
         self.on_close(connection_context)
@@ -99,23 +89,20 @@ async def on_connection_init(self, connection_context, op_id, payload):
             await connection_context.close(1011)
 
     async def on_start(self, connection_context, op_id, params):
-        execution_result = self.execute(
-            connection_context.request_context, params)
-
-        if isawaitable(execution_result):
-            execution_result = await execution_result
+        request_string = params.pop("request_string")
+        query = parse(request_string)
+        result = await subscribe(self.schema, query, **params)
 
-        if not hasattr(execution_result, '__aiter__'):
-            await self.send_execution_result(
-                connection_context, op_id, execution_result)
+        if not hasattr(result, "__aiter__"):
+            await self.send_execution_result(connection_context, op_id, result)
         else:
-            iterator = await execution_result.__aiter__()
-            connection_context.register_operation(op_id, iterator)
-            async for single_result in iterator:
+            connection_context.register_operation(op_id, result)
+            async for single_result in result:
                 if not connection_context.has_operation(op_id):
                     break
                 await self.send_execution_result(
-                    connection_context, op_id, single_result)
+                    connection_context, op_id, single_result
+                )
             await self.send_message(connection_context, op_id, GQL_COMPLETE)
 
     async def on_stop(self, connection_context, op_id):

graphql_ws/base.py

@@ -1,16 +1,16 @@
 import json
 from collections import OrderedDict
 
-from graphql import graphql, format_error
+from graphql import format_error, graphql
 
 from .constants import (
+    GQL_CONNECTION_ERROR,
     GQL_CONNECTION_INIT,
     GQL_CONNECTION_TERMINATE,
+    GQL_DATA,
+    GQL_ERROR,
     GQL_START,
     GQL_STOP,
-    GQL_ERROR,
-    GQL_CONNECTION_ERROR,
-    GQL_DATA
 )
 
 
@@ -51,7 +51,6 @@ def close(self, code):
 
 
 class BaseSubscriptionServer(object):
-
     def __init__(self, schema, keep_alive=True):
         self.schema = schema
         self.keep_alive = keep_alive
@@ -92,7 +91,8 @@ def process_message(self, connection_context, parsed_message):
             if not isinstance(params, dict):
                 error = Exception(
                     "Invalid params returned from get_graphql_params!"
-                    " Return values must be a dict.")
+                    " Return values must be a dict."
+                )
                 return self.send_error(connection_context, op_id, error)
 
             # If we already have a subscription with this id, unsubscribe from
@@ -106,8 +106,11 @@ def process_message(self, connection_context, parsed_message):
             return self.on_stop(connection_context, op_id)
 
         else:
-            return self.send_error(connection_context, op_id, Exception(
-                "Invalid message type: {}.".format(op_type)))
+            return self.send_error(
+                connection_context,
+                op_id,
+                Exception("Invalid message type: {}.".format(op_type)),
+            )
 
     def send_execution_result(self, connection_context, op_id, execution_result):
         result = self.execution_result_to_dict(execution_result)
@@ -118,8 +121,9 @@ def execution_result_to_dict(self, execution_result):
         if execution_result.data:
             result['data'] = execution_result.data
         if execution_result.errors:
-            result['errors'] = [format_error(error)
-                                for error in execution_result.errors]
+            result['errors'] = [
+                format_error(error) for error in execution_result.errors
+            ]
         return result
 
     def send_message(self, connection_context, op_id=None, op_type=None, payload=None):
@@ -144,16 +148,9 @@ def send_error(self, connection_context, op_id, error, error_type=None):
             ' GQL_CONNECTION_ERROR or GQL_ERROR'
         )
 
-        error_payload = {
-            'message': str(error)
-        }
+        error_payload = {'message': str(error)}
 
-        return self.send_message(
-            connection_context,
-            op_id,
-            error_type,
-            error_payload
-        )
+        return self.send_message(connection_context, op_id, error_type, error_payload)
 
     def unsubscribe(self, connection_context, op_id):
         if connection_context.has_operation(op_id):
@@ -170,8 +167,7 @@ def on_connection_terminate(self, connection_context, op_id):
         return connection_context.close(1011)
 
     def execute(self, request_context, params):
-        return graphql(
-            self.schema, **dict(params, allow_subscriptions=True))
+        return graphql(self.schema, params)
 
     def handle(self, ws, request_context=None):
         raise NotImplementedError("handle method not implemented")
@@ -180,8 +176,7 @@ def on_message(self, connection_context, message):
         try:
             if not isinstance(message, dict):
                 parsed_message = json.loads(message)
-                assert isinstance(
-                    parsed_message, dict), "Payload must be an object."
+                assert isinstance(parsed_message, dict), "Payload must be an object."
             else:
                 parsed_message = message
         except Exception as e:

requirements_dev.txt

@@ -6,9 +6,12 @@ tox>=3,<4
 coverage>=5.0,<6
 Sphinx>=1.8,<2
 PyYAML>=5.3,<6
-pytest==3.2.5
+pytest<=3.6,<6
 pytest-runner>=5.2,<6
 gevent>=1.1,<2
 graphene>=2.0,<3
 django>=1.5,<3
 channels>=1.0,<2
+pytest-aiohttp
+asyncmock; python_version<"3.8"
+pytest-cov

setup.py

@@ -3,7 +3,7 @@
 
 """The setup script."""
 
-from setuptools import setup, find_packages
+from setuptools import find_packages, setup
 
 with open("README.rst") as readme_file:
     readme = readme_file.read()
@@ -14,7 +14,7 @@
     history = history_file.read()
 
 requirements = [
-    "graphql-core>=2.0,<3",
+    "graphql-core>=3.0.0",
     # TODO: put package requirements here
 ]
 
@@ -26,7 +26,8 @@
 
 test_requirements = [
     "pytest",
-    # TODO: put package test requirements here
+    "pytest-aiohttp",
+    'asyncmock; python_version<"3.8"',
 ]
 
 setup(
@@ -57,7 +58,7 @@
         "Programming Language :: Python :: 3.5",
         "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
-        "Programming Language :: Python :: 3.8"
+        "Programming Language :: Python :: 3.8",
     ],
     test_suite="tests",
     tests_require=test_requirements,