diff -r 448dd8d36839 -r 3929338fd17f iro/tests/smtp.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/iro/tests/smtp.py Sun Mar 18 14:06:27 2012 +0100 @@ -0,0 +1,146 @@ +from twisted.trial import unittest + +import email +from email.header import decode_header +import base64 +from ..test_helpers.smtp_helper import TestSMTPServer + +from mock import patch, Mock +import smtplib + +from iro.error import NoRoute, NoTyp, NeededOption +from iro.model.message import Mail +from iro.offer.smtp import SMTP + +HOST = "localhost" +PORT = 9999 + +class TestSMTPProvider(unittest.TestCase): + def setUp(self): + self.smtp_server = TestSMTPServer((HOST, PORT)) + self.smtp_server.start() + + def tearDown(self): + self.smtp_server.close() + + def getSMTP(self, c=None): + _c={"send_from":"send@t.de", + "host":HOST, + "port":PORT, + "typ":"smtp", + } + + if c: + _c.update(c) + + ret = SMTP("test") + ret.load(_c.items()) + return ret + + def testSendMail(self): + p=self.getSMTP() + content = "sadfadfgwertsdgsdf\n\nsdfgaerasdfsad\nadfasdf" + p.send("t@t.de", Mail("sub", content, None)) + + + self.assertEqual(len(self.smtp_server.rcvd), 1) + fromaddr, toaddrs, message = self.smtp_server.rcvd[0] + msg = email.message_from_string(message) + + self.assertEqual(fromaddr,"send@t.de") + self.assertEqual(msg.get_all("From"),["send@t.de"]) + self.assertEqual(toaddrs,["t@t.de"]) + self.assertEqual(msg.get_all("To"),["t@t.de"]) + self.assertEqual(decode_header(msg.get("Subject")),[("sub","utf-8")]) + self.assertEqual(base64.b64decode(msg.get_payload()),content) + + def testSendMailExtraFrm(self): + p=self.getSMTP() + content = "" + p.send("t@t.de", Mail("sub", content, "f@t.de")) + + self.assertEqual(len(self.smtp_server.rcvd), 1) + fromaddr, toaddrs, message = self.smtp_server.rcvd[0] + msg = email.message_from_string(message) + + self.assertEqual(fromaddr,"f@t.de") + self.assertEqual(msg.get_all("From"),["f@t.de"]) + + def testSendMailException(self): + p=self.getSMTP({"port":PORT-1}) + content = "" + self.assertRaises(IOError, p.send, "t@t.de", Mail("sub", content, "f@t.de")) + + self.assertEqual(len(self.smtp_server.rcvd), 0) + + @patch("smtplib.SMTP_SSL") + def testSSLSendMail(self,mock_ssl): + def se(*args): + return smtplib.SMTP(*args) + mock_ssl.side_effect=se + + p=self.getSMTP({"SSL":True}) + content = "sadfadfgwertsdgsdf\n\nsdfgaerasdfsad\nadfasdf" + p.send("t@t.de", Mail("sub", content, None)) + + self.assertEqual(mock_ssl.call_count,1) + + self.assertEqual(len(self.smtp_server.rcvd), 1) + + @patch("smtplib.SMTP") + def testTLSSendMail(self,mock_smtp): + mock_s = Mock() + mock_smtp.return_value = mock_s + + p=self.getSMTP({"TLS":True}) + content = "sadfadfgwertsdgsdf\n\nsdfgaerasdfsad\nadfasdf" + p.send("t@t.de", Mail("sub", content, None)) + + mock_s.starttls.assert_called_once_with() + self.assertEqual(mock_s.sendmail.call_count,1) + self.assertEqual([i[0] for i in mock_s.method_calls],["starttls","sendmail","quit"]) + + @patch("smtplib.SMTP") + def testLoginSendMail(self,mock_smtp): + mock_s = Mock() + mock_smtp.return_value = mock_s + + p=self.getSMTP({"user":"user","password":"pw"}) + content = "sadfadfgwertsdgsdf\n\nsdfgaerasdfsad\nadfasdf" + p.send("t@t.de", Mail("sub", content, None)) + + mock_s.login.assert_called_once_with("user","pw") + self.assertEqual(mock_s.sendmail.call_count,1) + self.assertEqual([i[0] for i in mock_s.method_calls],["login","sendmail","quit"]) + + + def testNeededOption(self): + c={"send_from":"send@t.de", + "host":HOST, + "port":PORT, + "user":"u", + "password":"p", + "typ":"smtp", + } + s = self.getSMTP(c) + self.assertEqual(s.send_from, "send@t.de") + self.assertEqual(s.host, HOST) + self.assertEqual(s.port, PORT) + self.assertEqual(s.user, "u") + self.assertEqual(s.password, "p") + self.assertEqual(s.SSL,False) + self.assertEqual(s.TLS,False) + + c.update({"TLS":True, "SSL":True}) + s = self.getSMTP(c) + self.assertEqual(s.SSL,True) + self.assertEqual(s.TLS,True) + + del c["host"] + self.assertRaises(NeededOption, s.load, c) + + def testSendFunc(self): + s = self.getSMTP() + self.assertEqual(s.getSendFunc("mail",None), s.send) + self.assertRaises(NoRoute,s.getSendFunc,"mail","foo") + self.assertRaises(NoTyp,s.getSendFunc,"mail2","foo")