From ec13df75d01c95cfd0a4e021750dc39538af6b4f Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Mon, 22 Oct 2018 10:05:20 -0400
Subject: [PATCH] kinda test websocket support

---
 responder/models.py     | 34 +++++++++++++++++-----------------
 tests/test_responder.py | 10 ++++++++++
 2 files changed, 27 insertions(+), 17 deletions(-)

diff --git a/responder/models.py b/responder/models.py
index 17d29ce..20dc6b6 100644
--- a/responder/models.py
+++ b/responder/models.py
@@ -314,26 +314,26 @@ class WebSocket:
             return QueryDict({})
 
     async def accept(self):
-        return (await self._starlette.accept())
+        return await self._starlette.accept()
 
     async def receive(self):
-        return (await self._starlette.receive())
+        return await self._starlette.receive()
 
     async def content(self):
-        """Receive bytes"""
-        return (await self._starlette.receive_bytes()) 
+        """Receive bytes."""
+        return await self._starlette.receive_bytes()
 
     async def text(self):
-        """Receive text"""
-        return (await self._starlette.receive_text())
+        """Receive text."""
+        return await self._starlette.receive_text()
 
-    async def json(self):
-        """Receive json"""
-        return (await self._starlette.receive_json())
+    # async def json(self):
+    #     """Receive json"""
+    #     return await self._starlette.receive_json()
 
-    async def media(self):
-        """Receive json"""
-        return (await self.receive_json())
+    # async def media(self):
+    #     """Receive json"""
+    #     return (await self.receive_json())
 
     async def send(self):
         await self._starlette.send()
@@ -341,11 +341,11 @@ class WebSocket:
     async def send_text(self, data):
         await self._starlette.send_text(data)
 
-    async def send_json(self, data):
-        await self._starlette.send_json(data)
+    # async def send_json(self, data):
+    #     await self._starlette.send_json(data)
 
-    async def send_media(self, data):
-        await self.send_json(data)
+    # async def send_media(self, data):
+    #     await self.send_json(data)
 
-    async def close(self, code=1000) -> None:
+    async def close(self, code=1000):
         await self._starlette.close(code)
diff --git a/tests/test_responder.py b/tests/test_responder.py
index 35f4bfd..892c363 100644
--- a/tests/test_responder.py
+++ b/tests/test_responder.py
@@ -439,3 +439,13 @@ def test_404(session):
     r = session.get("/foo")
 
     assert r.status_code == responder.status_codes.HTTP_404
+
+
+def test_kinda_websockets(api):
+    @api.route("/ws", websocket=True)
+    async def websocket(ws):
+        await ws.accept()
+        await ws.send_text("Hello via websocket!")
+        await ws.close()
+
+    
\ No newline at end of file