Issue #26309: Merge socketserver fix from 3.5

This commit is contained in:
Martin Panter 2016-02-18 11:01:32 +00:00
commit a2b64e630e
3 changed files with 30 additions and 0 deletions

View file

@ -319,6 +319,8 @@ class BaseServer:
except:
self.handle_error(request, client_address)
self.shutdown_request(request)
else:
self.shutdown_request(request)
def handle_timeout(self):
"""Called if no new request arrives within self.timeout.

View file

@ -280,6 +280,30 @@ class SocketServerTest(unittest.TestCase):
socketserver.TCPServer((HOST, -1),
socketserver.StreamRequestHandler)
def test_shutdown_request_called_if_verify_request_false(self):
# Issue #26309: BaseServer should call shutdown_request even if
# verify_request is False
shutdown_called = False
class MyServer(socketserver.TCPServer):
def verify_request(self, request, client_address):
return False
def shutdown_request(self, request):
nonlocal shutdown_called
shutdown_called = True
super().shutdown_request(request)
def connect_to_server(proto, addr):
s = socket.socket(proto, socket.SOCK_STREAM)
s.connect(addr)
s.close()
self.run_server(MyServer,
socketserver.StreamRequestHandler,
connect_to_server)
self.assertEqual(shutdown_called, True)
class MiscTestCase(unittest.TestCase):