Add tests

Author: justmobilize

tests/conftest.py

@@ -49,6 +49,19 @@ def adafruit_wiznet5k_socket_module():
     del sys.modules["adafruit_wiznet5k.adafruit_wiznet5k_socket"]
 
 
+@pytest.fixture
+def adafruit_wiznet5k_with_ssl_socket_module():
+    wiznet5k_module = type(sys)("adafruit_wiznet5k")
+    wiznet5k_socket_module = type(sys)("adafruit_wiznet5k_socket")
+    wiznet5k_socket_module.set_interface = set_interface
+    wiznet5k_socket_module.SOCK_STREAM = 1
+    sys.modules["adafruit_wiznet5k"] = wiznet5k_module
+    sys.modules["adafruit_wiznet5k.adafruit_wiznet5k_socket"] = wiznet5k_socket_module
+    yield
+    del sys.modules["adafruit_wiznet5k"]
+    del sys.modules["adafruit_wiznet5k.adafruit_wiznet5k_socket"]
+
+
 @pytest.fixture(autouse=True)
 def reset_connection_manager(monkeypatch):
     monkeypatch.setattr(

tests/ssl_context_test.py

@@ -11,6 +11,7 @@
 import pytest
 
 import adafruit_connection_manager
+from adafruit_connection_manager import WIZNET5K_SSL_SUPPORT_VERSION
 
 
 def test_connect_esp32spi_https(  # pylint: disable=unused-argument
@@ -51,7 +52,11 @@ def test_connect_wiznet5k_https_not_supported(  # pylint: disable=unused-argumen
 ):
     mock_pool = mocket.MocketPool()
     radio = mocket.MockRadio.WIZNET5K()
-    with mock.patch("sys.implementation", return_value=[9, 0, 0]):
+    old_version = (WIZNET5K_SSL_SUPPORT_VERSION[0] - 1, 0, 0)
+    with mock.patch(
+        "sys.implementation",
+        (None, old_version)
+    ):
         ssl_context = adafruit_connection_manager.get_radio_ssl_context(radio)
     connection_manager = adafruit_connection_manager.ConnectionManager(mock_pool)
 
@@ -61,3 +66,15 @@ def test_connect_wiznet5k_https_not_supported(  # pylint: disable=unused-argumen
             mocket.MOCK_HOST_1, 443, "https:", ssl_context=ssl_context
         )
     assert "This radio does not support TLS/HTTPS" in str(context)
+
+
+def test_connect_wiznet5k_https_supported(  # pylint: disable=unused-argument
+    adafruit_wiznet5k_with_ssl_socket_module,
+):
+    radio = mocket.MockRadio.WIZNET5K()
+    with mock.patch(
+        "sys.implementation",
+        (None, WIZNET5K_SSL_SUPPORT_VERSION)
+    ):
+        ssl_context = adafruit_connection_manager.get_radio_ssl_context(radio)
+    assert isinstance(ssl_context, ssl.SSLContext)