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"

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

        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"

        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"

        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"

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