R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 1 | import unittest |
| 2 | from test import support |
| 3 | import smtplib |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 4 | import socket |
Antoine Pitrou | 0d38f2e | 2011-05-18 20:02:50 +0200 | [diff] [blame] | 5 | |
| 6 | ssl = support.import_module("ssl") |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 7 | |
R. David Murray | eccb2ce | 2009-05-23 02:36:15 +0000 | [diff] [blame] | 8 | support.requires("network") |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 9 | |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 10 | def check_ssl_verifiy(host, port): |
| 11 | context = ssl.create_default_context() |
| 12 | with socket.create_connection((host, port)) as sock: |
| 13 | try: |
| 14 | sock = context.wrap_socket(sock, server_hostname=host) |
| 15 | except Exception: |
| 16 | return False |
| 17 | else: |
| 18 | sock.close() |
| 19 | return True |
| 20 | |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 21 | |
| 22 | class SmtpTest(unittest.TestCase): |
| 23 | testServer = 'smtp.gmail.com' |
Berker Peksag | ed0392a | 2015-03-02 07:40:36 +0200 | [diff] [blame] | 24 | remotePort = 587 |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 25 | |
| 26 | def test_connect_starttls(self): |
| 27 | support.get_attribute(smtplib, 'SMTP_SSL') |
Christian Heimes | a170fa1 | 2017-09-15 20:27:30 +0200 | [diff] [blame] | 28 | context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT) |
| 29 | context.check_hostname = False |
| 30 | context.verify_mode = ssl.CERT_NONE |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 31 | with support.transient_internet(self.testServer): |
| 32 | server = smtplib.SMTP(self.testServer, self.remotePort) |
Nadeem Vawda | 3fc5868 | 2011-07-30 23:46:54 +0200 | [diff] [blame] | 33 | try: |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 34 | server.starttls(context=context) |
Nadeem Vawda | 3fc5868 | 2011-07-30 23:46:54 +0200 | [diff] [blame] | 35 | except smtplib.SMTPException as e: |
| 36 | if e.args[0] == 'STARTTLS extension not supported by server.': |
| 37 | unittest.skip(e.args[0]) |
| 38 | else: |
| 39 | raise |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 40 | server.ehlo() |
| 41 | server.quit() |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 42 | |
| 43 | |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 44 | class SmtpSSLTest(unittest.TestCase): |
| 45 | testServer = 'smtp.gmail.com' |
| 46 | remotePort = 465 |
| 47 | |
| 48 | def test_connect(self): |
R. David Murray | eccb2ce | 2009-05-23 02:36:15 +0000 | [diff] [blame] | 49 | support.get_attribute(smtplib, 'SMTP_SSL') |
Antoine Pitrou | 6003ff7 | 2010-10-13 17:14:16 +0000 | [diff] [blame] | 50 | with support.transient_internet(self.testServer): |
| 51 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort) |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 52 | server.ehlo() |
| 53 | server.quit() |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 54 | |
Antoine Pitrou | c1d5206 | 2011-05-07 19:39:37 +0200 | [diff] [blame] | 55 | def test_connect_default_port(self): |
| 56 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 57 | with support.transient_internet(self.testServer): |
| 58 | server = smtplib.SMTP_SSL(self.testServer) |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 59 | server.ehlo() |
| 60 | server.quit() |
Antoine Pitrou | c1d5206 | 2011-05-07 19:39:37 +0200 | [diff] [blame] | 61 | |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 62 | def test_connect_using_sslcontext(self): |
Christian Heimes | a170fa1 | 2017-09-15 20:27:30 +0200 | [diff] [blame] | 63 | context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT) |
| 64 | context.check_hostname = False |
| 65 | context.verify_mode = ssl.CERT_NONE |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 66 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 67 | with support.transient_internet(self.testServer): |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 68 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort, context=context) |
| 69 | server.ehlo() |
| 70 | server.quit() |
| 71 | |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 72 | def test_connect_using_sslcontext_verified(self): |
Victor Stinner | 19c899c | 2014-09-04 00:35:43 +0200 | [diff] [blame] | 73 | with support.transient_internet(self.testServer): |
| 74 | can_verify = check_ssl_verifiy(self.testServer, self.remotePort) |
| 75 | if not can_verify: |
| 76 | self.skipTest("SSL certificate can't be verified") |
| 77 | |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 78 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 79 | context = ssl.create_default_context() |
| 80 | with support.transient_internet(self.testServer): |
| 81 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort, context=context) |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 82 | server.ehlo() |
| 83 | server.quit() |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 84 | |
| 85 | |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 86 | if __name__ == "__main__": |
Zachary Ware | 38c707e | 2015-04-13 15:00:43 -0500 | [diff] [blame] | 87 | unittest.main() |