Merge pull request #212 from justmobilize/exception-test-updates

Update test exceptions
adafruit · May 12, 2024 · ecfd228 · ecfd228
2 parents cbe2177 + 1037c14
commit ecfd228
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/adafruit_minimqtt/adafruit_minimqtt.py b/adafruit_minimqtt/adafruit_minimqtt.py
@@ -424,7 +424,7 @@ def connect(
                 )
                 self._reset_reconnect_backoff()
                 return ret
-            except RuntimeError as e:
+            except (MemoryError, OSError, RuntimeError) as e:
                 self.logger.warning(f"Socket error when connecting: {e}")
                 backoff = False
             except MMQTTException as e:

diff --git a/tests/test_port_ssl.py b/tests/test_port_ssl.py
@@ -120,6 +120,6 @@ def test_tls_without_ssl_context(self) -> None:
             connect_retries=1,
         )
 
-        with pytest.raises(AttributeError) as context:
+        with pytest.raises(ValueError) as context:
             mqtt_client.connect()
-        assert "ssl_context must be set" in str(context)
+        assert "ssl_context must be provided if using ssl" in str(context)