diff options
author | Antoine Pitrou <solipsis@pitrou.net> | 2010-04-27 09:51:18 (GMT) |
---|---|---|
committer | Antoine Pitrou <solipsis@pitrou.net> | 2010-04-27 09:51:18 (GMT) |
commit | 435ba0cfb82ecadca693eaedd2a2806b8766afb1 (patch) | |
tree | 73b2bfe6672807ae5fab93cea7554fa1bd60c693 /Lib/test/test_ssl.py | |
parent | 150acda57fabfa941d2ed5d5e8ab3e49dac9a73e (diff) | |
download | cpython-435ba0cfb82ecadca693eaedd2a2806b8766afb1.zip cpython-435ba0cfb82ecadca693eaedd2a2806b8766afb1.tar.gz cpython-435ba0cfb82ecadca693eaedd2a2806b8766afb1.tar.bz2 |
replace custom code with standard library functionality (HTTPServer.shutdown())
+ enable test that was never run (!)
+ make tests faster by lowering polling timeout
Diffstat (limited to 'Lib/test/test_ssl.py')
-rw-r--r-- | Lib/test/test_ssl.py | 55 |
1 files changed, 7 insertions, 48 deletions
diff --git a/Lib/test/test_ssl.py b/Lib/test/test_ssl.py index ebb556b..435c198 100644 --- a/Lib/test/test_ssl.py +++ b/Lib/test/test_ssl.py @@ -34,7 +34,11 @@ def handle_error(prefix): if test_support.verbose: sys.stdout.write(prefix + exc_format) + +class BasicTests(unittest.TestCase): + def testSimpleSSLwrap(self): + # A crude test for the legacy API try: ssl.sslwrap_simple(socket.socket(socket.AF_INET)) except IOError, e: @@ -50,8 +54,6 @@ def handle_error(prefix): else: raise -class BasicTests(unittest.TestCase): - def testSSLconnect(self): if not test_support.is_resource_enabled('network'): return @@ -458,7 +460,7 @@ else: threading.Thread.start(self) def run (self): - self.sock.settimeout(0.5) + self.sock.settimeout(0.05) self.sock.listen(5) self.active = True if self.flag: @@ -605,49 +607,6 @@ else: certfile=self.certfile) return sslconn, addr - # The methods overridden below this are mainly so that we - # can run it in a thread and be able to stop it from another - # You probably wouldn't need them in other uses. - - def server_activate(self): - # We want to run this in a thread for testing purposes, - # so we override this to set timeout, so that we get - # a chance to stop the server - self.socket.settimeout(0.5) - HTTPServer.server_activate(self) - - def serve_forever(self): - # We want this to run in a thread, so we use a slightly - # modified version of "forever". - self.active = True - while 1: - try: - # We need to lock while handling the request. - # Another thread can close the socket after self.active - # has been checked and before the request is handled. - # This causes an exception when using the closed socket. - with self.active_lock: - if not self.active: - break - self.handle_request() - except socket.timeout: - pass - except KeyboardInterrupt: - self.server_close() - return - except: - sys.stdout.write(''.join(traceback.format_exception(*sys.exc_info()))) - break - time.sleep(0.1) - - def server_close(self): - # Again, we want this to run in a thread, so we need to override - # close to clear the "active" flag, so that serve_forever() will - # terminate. - with self.active_lock: - HTTPServer.server_close(self) - self.active = False - class RootedHTTPRequestHandler(SimpleHTTPRequestHandler): # need to override translate_path to get a known root, @@ -713,12 +672,12 @@ else: self.active = True if self.flag: self.flag.set() - self.server.serve_forever() + self.server.serve_forever(0.05) self.active = False def stop (self): self.active = False - self.server.server_close() + self.server.shutdown() def badCertTest (certfile): |