From 2ffa605b52e253b609e79ed21938b0d318ea0dad Mon Sep 17 00:00:00 2001 From: davebshow <davebshow@gmail.com> Date: Sun, 17 May 2015 20:03:03 -0400 Subject: [PATCH] error if receive doesn't find termination condition --- aiogremlin/connection.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/aiogremlin/connection.py b/aiogremlin/connection.py index 7f265d1..95ab592 100644 --- a/aiogremlin/connection.py +++ b/aiogremlin/connection.py @@ -231,6 +231,8 @@ class GremlinClientWebSocketResponse(BaseConnection, ClientWebSocketResponse): self.parser.feed_data(msg.data.decode()) elif msg.tp == MsgType.text: self.parser.feed_data(msg.data.strip()) + else: + raise RuntimeError("Unknown message type.") break finally: self._waiting = False -- GitLab