From 1037c14b97118693fba35d569ec5816852626b1e Mon Sep 17 00:00:00 2001 From: Justin Myers Date: Sat, 11 May 2024 09:35:46 -0700 Subject: [PATCH] Update test exceptions --- adafruit_minimqtt/adafruit_minimqtt.py | 2 +- tests/test_port_ssl.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/adafruit_minimqtt/adafruit_minimqtt.py b/adafruit_minimqtt/adafruit_minimqtt.py index 7ede9559..622c6e68 100644 --- 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 index 9ac154da..2aa877f5 100644 --- 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)