Initial tests for aiohttp and django_channels

Author: SmileyChris

setup.cfg

@@ -45,12 +45,22 @@ dev =
     flake8>=3.7,<4
     tox>=3,<4
     pytest>=3.2.5,<4
+    pytest-cov
+    pytest-asyncio; python_version>="3.4"
     Sphinx>=1.8,<2
     coverage>=5.0,<6
 test =
     pytest>=3.2.5,<4
     graphene>=2.0,<3
-
+    gevent
+    graphene>=2.0
+    graphene_django
+    mock; python_version<"3"
+    django==1.11.*; python_version<"3"
+    channels==1.*; python_version<"3"
+    django==2.*; python_version>="3"
+    channels==2.*; python_version>="3"
+    aiohttp; python_version>="3.5"
 
 [bdist_wheel]
 universal = 1

tests/conftest.py

@@ -0,0 +1,6 @@
+import sys
+
+if sys.version_info > (3,):
+    collect_ignore = ["test_django_channels.py"]
+else:
+    collect_ignore = ["test_aiohttp.py"]

tests/test_aiohttp.py

@@ -0,0 +1,73 @@
+from unittest import mock
+
+import pytest
+from aiohttp import WSMsgType
+
+from graphql_ws.aiohttp import AiohttpConnectionContext, AiohttpSubscriptionServer
+from graphql_ws.base import ConnectionClosedException
+
+
+class AsyncMock(mock.Mock):
+    def __call__(self, *args, **kwargs):
+
+        async def coro():
+            return super(AsyncMock, self).__call__(*args, **kwargs)
+
+        return coro()
+
+
+@pytest.fixture()
+def mock_ws():
+    ws = AsyncMock(spec=["receive", "send_str", "closed", "close"])
+    ws.closed = False
+    ws.receive.return_value = AsyncMock(spec=["type", "data"])
+    return ws
+
+
+@pytest.mark.asyncio
+class TestConnectionContext:
+    async def test_receive_good_data(self, mock_ws):
+        msg = mock_ws.receive.return_value
+        msg.type = WSMsgType.TEXT
+        msg.data = "test"
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        assert await connection_context.receive() == "test"
+
+    async def test_receive_error(self, mock_ws):
+        msg = mock_ws.receive.return_value
+        msg.type = WSMsgType.ERROR
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        with pytest.raises(ConnectionClosedException):
+            await connection_context.receive()
+
+    async def test_receive_closing(self, mock_ws):
+        mock_ws.receive.return_value.type = WSMsgType.CLOSING
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        with pytest.raises(ConnectionClosedException):
+            await connection_context.receive()
+
+    async def test_receive_closed(self, mock_ws):
+        mock_ws.receive.return_value.type = WSMsgType.CLOSED
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        with pytest.raises(ConnectionClosedException):
+            await connection_context.receive()
+
+    async def test_send(self, mock_ws):
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        await connection_context.send("test")
+        mock_ws.send_str.assert_called_with("test")
+
+    async def test_send_closed(self, mock_ws):
+        mock_ws.closed = True
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        await connection_context.send("test")
+        mock_ws.send_str.assert_not_called()
+
+    async def test_close(self, mock_ws):
+        connection_context = AiohttpConnectionContext(ws=mock_ws)
+        await connection_context.close(123)
+        mock_ws.close.assert_called_with(code=123)
+
+
+def test_subscription_server_smoke():
+    AiohttpSubscriptionServer(schema=None)

tests/test_django_channels.py

@@ -0,0 +1,27 @@
+import mock
+from django.conf import settings
+
+settings.configure()  # noqa
+
+from graphql_ws.django_channels import (
+    DjangoChannelConnectionContext,
+    DjangoChannelSubscriptionServer,
+)
+
+
+class TestConnectionContext:
+    def test_send(self):
+        msg = mock.Mock()
+        connection_context = DjangoChannelConnectionContext(message=msg)
+        connection_context.send("test")
+        msg.reply_channel.send.assert_called_with("test")
+
+    def test_close(self):
+        msg = mock.Mock()
+        connection_context = DjangoChannelConnectionContext(message=msg)
+        connection_context.close(123)
+        msg.reply_channel.send.assert_called_with({"close": True, "text": 123})
+
+
+def test_subscription_server_smoke():
+    DjangoChannelSubscriptionServer(schema=None)