Improve test so it does not test methods that do not exist in old Pythons.

This commit is contained in:
Hermann Höhne 2017-09-03 19:58:47 +02:00
parent dfd50db550
commit 8bad0cab7b
1 changed files with 6 additions and 4 deletions

View File

@ -280,10 +280,12 @@ class WebSockifyServerTestCase(unittest.TestCase):
raise ssl.SSLError(ssl.SSL_ERROR_EOF) raise ssl.SSLError(ssl.SSL_ERROR_EOF)
self.stubs.Set(select, 'select', fake_select) self.stubs.Set(select, 'select', fake_select)
# for recent versions of python if (hasattr(ssl, 'create_default_context')):
self.stubs.Set(ssl, 'create_default_context', fake_create_default_context) # for recent versions of python
# for fallback for old versions of python self.stubs.Set(ssl, 'create_default_context', fake_create_default_context)
self.stubs.Set(ssl, 'wrap_socket', fake_wrap_socket) else:
# for fallback for old versions of python
self.stubs.Set(ssl, 'wrap_socket', fake_wrap_socket)
self.assertRaises( self.assertRaises(
websockifyserver.WebSockifyServer.EClose, server.do_handshake, websockifyserver.WebSockifyServer.EClose, server.do_handshake,
sock, '127.0.0.1') sock, '127.0.0.1')