| Benjamin Peterson | 90f5ba5 | 2010-03-11 22:53:45 +0000 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 2 | |
| 3 | import unittest |
| 4 | from test import support |
| 5 | import smtplib |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 6 | import 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 | |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 10 | |
| 11 | class SmtpTest(unittest.TestCase): |
| 12 | testServer = 'smtp.gmail.com' |
| 13 | remotePort = 25 |
| 14 | context = ssl.SSLContext(ssl.PROTOCOL_SSLv23) |
| 15 | |
| 16 | def test_connect_starttls(self): |
| 17 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 18 | with support.transient_internet(self.testServer): |
| 19 | server = smtplib.SMTP(self.testServer, self.remotePort) |
| 20 | server.starttls(context=self.context) |
| Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame^] | 21 | server.ehlo() |
| 22 | server.quit() |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 23 | |
| 24 | |
| R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 25 | class SmtpSSLTest(unittest.TestCase): |
| 26 | testServer = 'smtp.gmail.com' |
| 27 | remotePort = 465 |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 28 | context = ssl.SSLContext(ssl.PROTOCOL_SSLv23) |
| R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 29 | |
| 30 | def test_connect(self): |
| R. David Murray | eccb2ce | 2009-05-23 02:36:15 +0000 | [diff] [blame] | 31 | support.get_attribute(smtplib, 'SMTP_SSL') |
| Antoine Pitrou | 6003ff7 | 2010-10-13 17:14:16 +0000 | [diff] [blame] | 32 | with support.transient_internet(self.testServer): |
| 33 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort) |
| Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame^] | 34 | server.ehlo() |
| 35 | server.quit() |
| R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 36 | |
| Antoine Pitrou | c1d5206 | 2011-05-07 19:39:37 +0200 | [diff] [blame] | 37 | def test_connect_default_port(self): |
| 38 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 39 | with support.transient_internet(self.testServer): |
| 40 | server = smtplib.SMTP_SSL(self.testServer) |
| Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame^] | 41 | server.ehlo() |
| 42 | server.quit() |
| Antoine Pitrou | c1d5206 | 2011-05-07 19:39:37 +0200 | [diff] [blame] | 43 | |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 44 | def test_connect_using_sslcontext(self): |
| 45 | support.get_attribute(smtplib, 'SMTP_SSL') |
| 46 | with support.transient_internet(self.testServer): |
| 47 | server = smtplib.SMTP_SSL(self.testServer, self.remotePort, context=self.context) |
| Antoine Pitrou | cc9ca22 | 2011-05-18 18:04:04 +0200 | [diff] [blame^] | 48 | server.ehlo() |
| 49 | server.quit() |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 50 | |
| 51 | |
| R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 52 | def test_main(): |
| Antoine Pitrou | e065020 | 2011-05-18 18:03:09 +0200 | [diff] [blame] | 53 | support.run_unittest(SmtpTest, SmtpSSLTest) |
| R. David Murray | 87e2074 | 2009-05-23 01:30:26 +0000 | [diff] [blame] | 54 | |
| 55 | if __name__ == "__main__": |
| 56 | test_main() |