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 | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 28 | context = ssl.SSLContext(ssl.PROTOCOL_SSLv23) |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 29 | with support.transient_internet(self.testServer): |
| 30 | server = smtplib.SMTP(self.testServer, self.remotePort) |
Nadeem Vawda | 3fc5868 | 2011-07-30 23:46:54 +0200 | [diff] [blame] | 31 | try: |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 32 | server.starttls(context=context) |
Nadeem Vawda | 3fc5868 | 2011-07-30 23:46:54 +0200 | [diff] [blame] | 33 | except smtplib.SMTPException as e: |
| 34 | if e.args[0] == 'STARTTLS extension not supported by server.': |
| 35 | unittest.skip(e.args[0]) |
| 36 | else: |
| 37 | raise |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 38 | server.ehlo() |
| 39 | server.quit() |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 40 | |
| 41 | |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 42 | class SmtpSSLTest(unittest.TestCase): |
| 43 | testServer = 'smtp.gmail.com' |
| 44 | remotePort = 465 |
| 45 | |
| 46 | def test_connect(self): |
R. David Murray | eccb2ce | 2009-05-23 02:36:15 +0000 | [diff] [blame] | 47 | support.get_attribute(smtplib, 'SMTP_SSL') |
Antoine Pitrou | 6003ff7 | 2010-10-13 17:14:16 +0000 | [diff] [blame] | 48 | with support.transient_internet(self.testServer): |
| 49 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort) |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 50 | server.ehlo() |
| 51 | server.quit() |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 52 | |
Antoine Pitrou | c1d5206 | 2011-05-07 19:39:37 +0200 | [diff] [blame] | 53 | def test_connect_default_port(self): |
| 54 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 55 | with support.transient_internet(self.testServer): |
| 56 | server = smtplib.SMTP_SSL(self.testServer) |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 57 | server.ehlo() |
| 58 | server.quit() |
Antoine Pitrou | c1d5206 | 2011-05-07 19:39:37 +0200 | [diff] [blame] | 59 | |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 60 | def test_connect_using_sslcontext(self): |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 61 | context = ssl.SSLContext(ssl.PROTOCOL_SSLv23) |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 62 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 63 | with support.transient_internet(self.testServer): |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 64 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort, context=context) |
| 65 | server.ehlo() |
| 66 | server.quit() |
| 67 | |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 68 | def test_connect_using_sslcontext_verified(self): |
Victor Stinner | 19c899c | 2014-09-04 00:35:43 +0200 | [diff] [blame] | 69 | with support.transient_internet(self.testServer): |
| 70 | can_verify = check_ssl_verifiy(self.testServer, self.remotePort) |
| 71 | if not can_verify: |
| 72 | self.skipTest("SSL certificate can't be verified") |
| 73 | |
Christian Heimes | a5768f7 | 2013-12-02 20:44:17 +0100 | [diff] [blame] | 74 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 75 | context = ssl.create_default_context() |
| 76 | with support.transient_internet(self.testServer): |
| 77 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort, context=context) |
Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame] | 78 | server.ehlo() |
| 79 | server.quit() |
Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 80 | |
| 81 | |
R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 82 | if __name__ == "__main__": |
Zachary Ware | 38c707e | 2015-04-13 15:00:43 -0500 | [diff] [blame] | 83 | unittest.main() |