Add test for error after connection timeout

cole · May 16, 2019 · f90ff25 · f90ff25
1 parent b2119be
commit f90ff25
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/tests/test_timeouts.py b/tests/test_timeouts.py
@@ -5,7 +5,12 @@
 
 import pytest
 
-from aiosmtplib import SMTP, SMTPConnectTimeoutError, SMTPTimeoutError
+from aiosmtplib import (
+    SMTP,
+    SMTPConnectTimeoutError,
+    SMTPServerDisconnected,
+    SMTPTimeoutError,
+)
 from aiosmtplib.protocol import SMTPProtocol
 
 
@@ -125,3 +130,13 @@ async def test_connect_timeout_error(hostname, port):
         host=hostname, port=port
     )
     assert str(exc.value) == expected_message
+
+
+async def test_server_disconnected_error_after_connect_timeout(hostname, port, message):
+    client = SMTP(hostname=hostname, port=port)
+
+    with pytest.raises(SMTPConnectTimeoutError):
+        await client.connect(timeout=0.0)
+
+    with pytest.raises(SMTPServerDisconnected):
+        await client.sendmail(message["From"], [message["To"]], str(message))