release 0.0.317 (#52)

* nicer ws errors

---------

Co-authored-by: nggit <[email protected]>

Author: nggit

setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name='tremolo',
-    version='0.0.316',
+    version='0.0.317',
     license='MIT',
     author='nggit',
     author_email='[email protected]',

tests/utils.py

@@ -1,6 +1,8 @@
-__all__ = ('function', 'getcontents', 'chunked_detected', 'read_chunked',
-           'valid_chunked', 'create_dummy_data', 'create_chunked_body',
-           'create_dummy_body', 'create_multipart_body', 'logger')
+__all__ = (
+    'function', 'getcontents', 'chunked_detected', 'read_chunked',
+    'valid_chunked', 'create_dummy_data', 'create_chunked_body',
+    'create_dummy_body', 'create_multipart_body', 'logger'
+)
 
 import asyncio  # noqa: E402
 import logging  # noqa: E402

tremolo/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.0.316'
+__version__ = '0.0.317'
 
 from .tremolo import Tremolo  # noqa: E402
 from . import exceptions  # noqa: E402,F401

tremolo/lib/websocket.py

@@ -35,7 +35,13 @@ async def accept(self):
         await self.response.write(None)
 
     async def recv(self):
-        first_byte, second_byte = await self.request.recv(2)
+        try:
+            first_byte, second_byte = await self.request.recv(2)
+        except ValueError as exc:
+            raise WebSocketClientClosed(
+                'connection closed: recv failed'
+            ) from exc
+
         # we don't use FIN
         # fin = (first_byte & 0x80) >> 7
         opcode = first_byte & 0x0f