Set imap/smtp APLN in autotest

deltachat · Jul 17, 2024 · 11db1b9 · 11db1b9
1 parent f76f427
commit 11db1b9
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/validation/autotest.py b/validation/autotest.py
@@ -1,6 +1,7 @@
 import argparse
 import smtplib
 import os
+import ssl
 try:
     import yaml  # pip install pyyaml
 except ModuleNotFoundError:
@@ -19,10 +20,13 @@ def test_smtp(server: dict):
     host = server["hostname"]
     port = server["port"]
     if server["socket"] == "SSL":
-        smtplib.SMTP_SSL(host, port)
+        context = ssl.create_default_context()
+        context.set_alpn_protocols("smtp")
+        smtplib.SMTP_SSL(host, port, context=context)
     elif server["socket"] == "STARTTLS":
-        smtpconn = smtplib.SMTP(host, port)
+        smtpconn = smtplib.SMTP(host, port, context=context)
         context = ssl.create_default_context()
+        context.set_alpn_protocols("smtp")
         smtpconn.starttls(context=context)
         smtpconn.ehlo()
     elif server["socket"] == "PLAIN":
@@ -38,10 +42,13 @@ def test_imap(server: dict):
     host = server["hostname"]
     port = server["port"]
     if server["socket"] == "SSL":
-        imaplib.IMAP4_SSL(host, port=port)
+        context = ssl.create_default_context()
+        context.set_alpn_protocols("imap")
+        imaplib.IMAP4_SSL(host, port=port, context=context)
     elif server["socket"] == "STARTTLS":
         imapconn = imaplib.IMAP4(host, port=port)
         context = ssl.create_default_context()
+        context.set_alpn_protocols("imap")
         imapconn.starttls(ssl_context=context)
     elif server["socket"] == "PLAIN":
         imaplib.IMAP4(host, port)