fix tests

Author: vladak

tests/test_port_ssl.py

@@ -34,7 +34,7 @@ def test_default_port(self) -> None:
             ssl_mock = Mock()
             ssl_context.wrap_socket = ssl_mock
 
-            with self.assertRaises(OSError):
+            with self.assertRaises(MQTT.MMQTTException):
                 expected_port = port
                 mqtt_client.connect()
 
@@ -59,7 +59,7 @@ def test_connect_override(self):
                 connect_retries=1,
             )
 
-            with self.assertRaises(OSError):
+            with self.assertRaises(MQTT.MMQTTException):
                 expected_host = "127.0.0.2"
                 expected_port = 1884
                 self.assertNotEqual(
@@ -102,7 +102,7 @@ def test_tls_port(self) -> None:
                 socket_mock.connect = connect_mock
                 ssl_mock.wrap_socket = Mock(return_value=socket_mock)
 
-                with self.assertRaises(RuntimeError):
+                with self.assertRaises(MQTT.MMQTTException):
                     mqtt_client.connect()
 
                 ssl_mock.wrap_socket.assert_called()