Fixing SSL test

eandersson · May 3, 2016 · b36dd75 · b36dd75
1 parent bf7ed35
commit b36dd75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/amqpstorm/tests/io_tests.py b/amqpstorm/tests/io_tests.py
@@ -85,9 +85,9 @@ def test_io_simple_ssl_receive(self):
 
         self.assertTrue(io.use_ssl)
 
-        if compatibility.PYTHON3:
+        if hasattr(ssl, 'SSLObject'):
             io.socket = MagicMock(name='socket', spec=ssl.SSLObject)
-        else:
+        elif hasattr(ssl, 'SSLSocket'):
             io.socket = MagicMock(name='socket', spec=ssl.SSLSocket)
 
         io.socket.read.return_value = '12345'