]> granicus.if.org Git - python/commitdiff
Issue #13895: fix test_ssl hanging under Ubuntu
authorAntoine Pitrou <solipsis@pitrou.net>
Sat, 28 Jan 2012 16:40:23 +0000 (17:40 +0100)
committerAntoine Pitrou <solipsis@pitrou.net>
Sat, 28 Jan 2012 16:40:23 +0000 (17:40 +0100)
1  2 
Lib/test/test_ssl.py

index 64394417833519e33662a6fc660dfc86459ad87e,1b08f2e7dc4cfbfce939c9bdab041d2de90d6f09..33f06d6cab891b8c568048f0858a416c97d1f69d
@@@ -1198,33 -1078,28 +1198,33 @@@ else
                                      chatty=chatty,
                                      connectionchatty=False)
          with server:
-             s = client_context.wrap_socket(socket.socket())
-             s.connect((HOST, server.port))
-             for arg in [indata, bytearray(indata), memoryview(indata)]:
-                 if connectionchatty:
-                     if support.verbose:
-                         sys.stdout.write(
-                             " client:  sending %r...\n" % indata)
-                 s.write(arg)
-                 outdata = s.read()
+             with client_context.wrap_socket(socket.socket()) as s:
+                 s.connect((HOST, server.port))
+                 for arg in [indata, bytearray(indata), memoryview(indata)]:
+                     if connectionchatty:
+                         if support.verbose:
+                             sys.stdout.write(
+                                 " client:  sending %r...\n" % indata)
+                     s.write(arg)
+                     outdata = s.read()
+                     if connectionchatty:
+                         if support.verbose:
+                             sys.stdout.write(" client:  read %r\n" % outdata)
+                     if outdata != indata.lower():
+                         raise AssertionError(
+                             "bad data <<%r>> (%d) received; expected <<%r>> (%d)\n"
+                             % (outdata[:20], len(outdata),
+                                indata[:20].lower(), len(indata)))
+                 s.write(b"over\n")
                  if connectionchatty:
                      if support.verbose:
-                         sys.stdout.write(" client:  read %r\n" % outdata)
-                 if outdata != indata.lower():
-                     raise AssertionError(
-                         "bad data <<%r>> (%d) received; expected <<%r>> (%d)\n"
-                         % (outdata[:20], len(outdata),
-                            indata[:20].lower(), len(indata)))
-             s.write(b"over\n")
-             if connectionchatty:
-                 if support.verbose:
-                     sys.stdout.write(" client:  closing connection.\n")
-             stats = {
-                 'compression': s.compression(),
-                 'cipher': s.cipher(),
-             }
-             s.close()
-             return stats
+                         sys.stdout.write(" client:  closing connection.\n")
++                stats = {
++                    'compression': s.compression(),
++                    'cipher': s.cipher(),
++                }
+                 s.close()
++                return stats
  
      def try_protocol_combo(server_protocol, client_protocol, expect_success,
                             certsreqs=None, server_options=0, client_options=0):