Add some base tests

Author: SmileyChris

setup.cfg

@@ -90,3 +90,8 @@ ignore = W503
 [coverage:run]
 omit =
     .tox/*
+
+[coverage:report]
+exclude_lines =
+    pragma: no cover
+    @abstract

tests/test_base.py

@@ -0,0 +1,62 @@
+try:
+    from unittest import mock
+except ImportError:
+    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

tox.ini

@@ -31,5 +31,6 @@ skip_install = true
 deps = coverage
 commands =
   coverage html
+  coverage xml
   coverage report --include="tests/*" --fail-under=100 -m
   coverage report --omit="tests/*"  # --fail-under=90 -m