diff options
| author | Chandler Abraham <cabraham@twitter.com> | 2015-04-13 13:36:09 -0700 |
|---|---|---|
| committer | Chandler Abraham <cabraham@twitter.com> | 2015-04-13 13:36:09 -0700 |
| commit | 2d72a1b6b56f1643cd1d8be59eee55aa7ca2f17f (patch) | |
| tree | 8e2831fdd659174703f00b4b1cd008cfb70185a2 /netlib/websockets/websockets.py | |
| parent | 0ed2a290639833d772b89cf333577820e84f8204 (diff) | |
| download | mitmproxy-2d72a1b6b56f1643cd1d8be59eee55aa7ca2f17f.tar.gz mitmproxy-2d72a1b6b56f1643cd1d8be59eee55aa7ca2f17f.tar.bz2 mitmproxy-2d72a1b6b56f1643cd1d8be59eee55aa7ca2f17f.zip | |
100% test coverage, though still need plenty more
Diffstat (limited to 'netlib/websockets/websockets.py')
| -rw-r--r-- | netlib/websockets/websockets.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/netlib/websockets/websockets.py b/netlib/websockets/websockets.py index cf9a68aa..ea3db21d 100644 --- a/netlib/websockets/websockets.py +++ b/netlib/websockets/websockets.py @@ -158,11 +158,10 @@ class WebSocketsFrame(object): ("actual_payload_length - " + str(self.actual_payload_length))]) def safe_to_bytes(self): - try: - assert self.is_valid() - return self.to_bytes() - except: - raise WebSocketFrameValidationException() + if self.is_valid(): + return self.to_bytes() + else: + raise WebSocketFrameValidationException() def to_bytes(self): """ |