from twisted.trial import unittest

import email
from email.header import decode_header
import base64
from smtp_helper import TestSMTPServer

from mock import patch, Mock
import smtplib

from iro.error import NoRoute, NoTyp, NeededOption, UnknownOption
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):
        ret={"send_from":"send@t.de",
            "host":HOST,
            "port":PORT,
            "user":"",
            "password":"",
            }

        if c:
            ret.update(c)

        return SMTP("test",ret.items())

    def testSendMail(self):
        p=self.getSMTP()
        content = "sadfadfgwertsdgsdf\n\nsdfgaerasdfsad\nadfasdf"
        p.send(Mail("sub", content, None), "t@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,"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(Mail("sub", content, "f@t.de"), "t@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, Mail("sub", content, "f@t.de"), "t@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(Mail("sub", content, None), "t@t.de")

        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(Mail("sub", content, None), "t@t.de")
        
        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(Mail("sub", content, None), "t@t.de")
        
        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",
            }
        s=SMTP("test",c.items())
        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.bSSL,False)
        self.assertEqual(s.bTLS,False)

        c.update({"TLS":True, "SSL":True})
        s=SMTP("test", c.items())
        self.assertEqual(s.bSSL,True)
        self.assertEqual(s.bTLS,True)
        
        del c["host"]
        self.assertRaises(NeededOption,SMTP,"test",c.items())

        c = {"unknown":""}
        self.assertRaises(UnknownOption,SMTP,"test",c.items())

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