Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-08 02:34:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

Ignore EPIPE when sending inconsistent data to server and server closes connect before all data is sent.

Browse Source
This commit is contained in:
Kovid Goyal 2016-06-28 20:04:53 +05:30
parent 7ba62b9696
commit 177c22ce5b
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/calibre/srv/tests/web_sockets.py
View File

@ -4,7 +4,7 @@
from __future__ import (unicode_literals, division, absolute_import, from __future__ import (unicode_literals, division, absolute_import,
print_function) print_function)
import socket, os, struct import socket, os, struct, errno
from base64 import standard_b64encode from base64 import standard_b64encode
from collections import deque, namedtuple from collections import deque, namedtuple
from functools import partial from functools import partial
@ -245,7 +245,14 @@ class WebSocketTest(BaseTest):
frags, q = [], b'\xce\xba\xe1\xbd\xb9\xcf\x83\xce\xbc\xce\xb5\xed\xa0\x80\x80\x65\x64\x69\x74\x65\x64' frags, q = [], b'\xce\xba\xe1\xbd\xb9\xcf\x83\xce\xbc\xce\xb5\xed\xa0\x80\x80\x65\x64\x69\x74\x65\x64'
for i, b in enumerate(q): for i, b in enumerate(q):
frags.append({'opcode':(TEXT if i == 0 else CONTINUATION), 'fin':1 if i == len(q)-1 else 0, 'payload':b}) frags.append({'opcode':(TEXT if i == 0 else CONTINUATION), 'fin':1 if i == len(q)-1 else 0, 'payload':b})
try:
simple_test(frags, close_code=INCONSISTENT_DATA, send_close=False) simple_test(frags, close_code=INCONSISTENT_DATA, send_close=False)
except socket.error as err:
if err.errno != errno.EPIPE:
# This is raised on OS X occassionally, when the server
# closes the connection before the client has finished
# writing all data.
raise
for q in (b'\xce', b'\xce\xba\xe1'): for q in (b'\xce', b'\xce\xba\xe1'):
simple_test([{'opcode':TEXT, 'payload':q}], close_code=INCONSISTENT_DATA, send_close=False) simple_test([{'opcode':TEXT, 'payload':q}], close_code=INCONSISTENT_DATA, send_close=False)