# -*- coding: utf-8 -*-
import contextlib
import socket
import ssl

import pytest

try:
    from urllib3.contrib.securetransport import WrappedSocket
except ImportError:
    pass


def setup_module():
    try:
        from urllib3.contrib.securetransport import inject_into_urllib3
        inject_into_urllib3()
    except ImportError as e:
        pytest.skip('Could not import SecureTransport: %r' % e)


def teardown_module():
    try:
        from urllib3.contrib.securetransport import extract_from_urllib3
        extract_from_urllib3()
    except ImportError:
        pass


from ..with_dummyserver.test_https import TestHTTPS, TestHTTPS_TLSv1  # noqa: F401
from ..with_dummyserver.test_socketlevel import (  # noqa: F401
    TestSNI, TestSocketClosing, TestClientCerts
)


def test_no_crash_with_empty_trust_bundle():
    with contextlib.closing(socket.socket()) as s:
        ws = WrappedSocket(s)
        with pytest.raises(ssl.SSLError):
ws._custom_validate(True, b"")

# -*- coding: utf-8 -*-
import contextlib
import socket
import ssl

import pytest

try:
    from urllib3.contrib.securetransport import (
        WrappedSocket, inject_into_urllib3, extract_from_urllib3
    )
except ImportError as e:
    pytestmark = pytest.mark.skip('Could not import SecureTransport: %r' % e)

from ..with_dummyserver.test_https import TestHTTPS, TestHTTPS_TLSv1  # noqa: F401
from ..with_dummyserver.test_socketlevel import (  # noqa: F401
    TestSNI, TestSocketClosing, TestClientCerts
)


def setup_module():
    inject_into_urllib3()


def teardown_module():
    extract_from_urllib3()


def test_no_crash_with_empty_trust_bundle():
    with contextlib.closing(socket.socket()) as s:
        ws = WrappedSocket(s)
        with pytest.raises(ssl.SSLError):
            ws._custom_validate(True, b"")

--- 
+++ 
@@ -6,11 +6,26 @@
 import pytest
 
 try:
-    from urllib3.contrib.securetransport import (
-        WrappedSocket, inject_into_urllib3, extract_from_urllib3
-    )
-except ImportError as e:
-    pytestmark = pytest.mark.skip('Could not import SecureTransport: %r' % e)
+    from urllib3.contrib.securetransport import WrappedSocket
+except ImportError:
+    pass
+
+
+def setup_module():
+    try:
+        from urllib3.contrib.securetransport import inject_into_urllib3
+        inject_into_urllib3()
+    except ImportError as e:
+        pytest.skip('Could not import SecureTransport: %r' % e)
+
+
+def teardown_module():
+    try:
+        from urllib3.contrib.securetransport import extract_from_urllib3
+        extract_from_urllib3()
+    except ImportError:
+        pass
+
 
 from ..with_dummyserver.test_https import TestHTTPS, TestHTTPS_TLSv1  # noqa: F401
 from ..with_dummyserver.test_socketlevel import (  # noqa: F401
@@ -18,16 +33,8 @@
 )
 
 
-def setup_module():
-    inject_into_urllib3()
-
-
-def teardown_module():
-    extract_from_urllib3()
-
-
 def test_no_crash_with_empty_trust_bundle():
     with contextlib.closing(socket.socket()) as s:
         ws = WrappedSocket(s)
         with pytest.raises(ssl.SSLError):
-            ws._custom_validate(True, b"")
+ws._custom_validate(True, b"")
Fix skip logic in SecureTransport tests