From 8bad0cab7b0d2cfa6a206a40c31df9119ffe47c2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Hermann=20H=C3=B6hne?= Date: Sun, 3 Sep 2017 19:58:47 +0200 Subject: [PATCH] Improve test so it does not test methods that do not exist in old Pythons. --- tests/test_websockifyserver.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test_websockifyserver.py b/tests/test_websockifyserver.py index e27f2f7..63c9449 100644 --- a/tests/test_websockifyserver.py +++ b/tests/test_websockifyserver.py @@ -280,10 +280,12 @@ class WebSockifyServerTestCase(unittest.TestCase): raise ssl.SSLError(ssl.SSL_ERROR_EOF) self.stubs.Set(select, 'select', fake_select) - # for recent versions of python - self.stubs.Set(ssl, 'create_default_context', fake_create_default_context) - # for fallback for old versions of python - self.stubs.Set(ssl, 'wrap_socket', fake_wrap_socket) + if (hasattr(ssl, 'create_default_context')): + # for recent versions of python + self.stubs.Set(ssl, 'create_default_context', fake_create_default_context) + else: + # for fallback for old versions of python + self.stubs.Set(ssl, 'wrap_socket', fake_wrap_socket) self.assertRaises( websockifyserver.WebSockifyServer.EClose, server.do_handshake, sock, '127.0.0.1')