diff options
-rw-r--r-- | breezy/tests/test_test_server.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/breezy/tests/test_test_server.py b/breezy/tests/test_test_server.py index 8d42b02..3ec6e91 100644 --- a/breezy/tests/test_test_server.py +++ b/breezy/tests/test_test_server.py @@ -77,7 +77,12 @@ class TCPClient(object): return self.sock.sendall(s) def read(self, bufsize=4096): - return self.sock.recv(bufsize) + try: + return self.sock.recv(bufsize) + except socket.error as e: + if e.errno == errno.ECONNRESET: + return b"" + raise class TCPConnectionHandler(socketserver.BaseRequestHandler): |