--- /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")