Add base async tests

Author: SmileyChris

tests/conftest.py

@@ -3,6 +3,6 @@
 if sys.version_info > (3,):
     collect_ignore = ["test_django_channels.py"]
     if sys.version_info < (3, 6):
-        collect_ignore.append('test_gevent.py')
+        collect_ignore.append("test_gevent.py")
 else:
-    collect_ignore = ["test_aiohttp.py"]
+    collect_ignore = ["test_aiohttp.py", "test_base_async.py"]

tests/test_base_async.py

@@ -0,0 +1,59 @@
+from unittest import mock
+
+import json
+
+import pytest
+
+from graphql_ws import base
+
+
+def test_not_implemented():
+    server = base.BaseSubscriptionServer(schema=None)
+    with pytest.raises(NotImplementedError):
+        server.on_connection_init(connection_context=None, op_id=1, payload={})
+    with pytest.raises(NotImplementedError):
+        server.on_open(connection_context=None)
+    with pytest.raises(NotImplementedError):
+        server.on_stop(connection_context=None, op_id=1)
+
+
+def test_terminate():
+    server = base.BaseSubscriptionServer(schema=None)
+
+    context = mock.Mock()
+    server.on_connection_terminate(connection_context=context, op_id=1)
+    context.close.assert_called_with(1011)
+
+
+def test_send_error():
+    server = base.BaseSubscriptionServer(schema=None)
+    context = mock.Mock()
+    server.send_error(connection_context=context, op_id=1, error="test error")
+    context.send.assert_called_with(
+        {"id": 1, "type": "error", "payload": {"message": "test error"}}
+    )
+
+
+def test_message():
+    server = base.BaseSubscriptionServer(schema=None)
+    server.process_message = mock.Mock()
+    context = mock.Mock()
+    msg = {"id": 1, "type": base.GQL_CONNECTION_INIT, "payload": ""}
+    server.on_message(context, msg)
+    server.process_message.assert_called_with(context, msg)
+
+
+def test_message_str():
+    server = base.BaseSubscriptionServer(schema=None)
+    server.process_message = mock.Mock()
+    context = mock.Mock()
+    msg = {"id": 1, "type": base.GQL_CONNECTION_INIT, "payload": ""}
+    server.on_message(context, json.dumps(msg))
+    server.process_message.assert_called_with(context, msg)
+
+
+def test_message_invalid():
+    server = base.BaseSubscriptionServer(schema=None)
+    server.send_error = mock.Mock()
+    server.on_message(connection_context=None, message="'not-json")
+    assert server.send_error.called