from twisted.trial import unittest

import email

from email.header import decode_header

import base64

from smtp_helper import TestSMTPServer

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 testSendMail(self):

        c=[("send_from","send@t.de"),

            ("host",HOST),

            ("port",PORT),

            ("user",""),

            ("password",""),

            ]

        p=SMTP("test",c)

        content = "sadfadfgwertsdgsdf\n\nsdfgaerasdfsad\nadfasdf"

        status = p.send(Mail("sub", content, None), "t@t.de")

        self.assertEqual(status.status, status.GOOD)

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

        c=[("send_from","send@t.de"),

            ("host",HOST),

            ("port",PORT),

            ("user",""),

            ("password",""),

            ]

        p=SMTP("test",c)

        content = ""

        status = p.send(Mail("sub", content, "f@t.de"), "t@t.de")

        self.assertEqual(status.status, status.GOOD)

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

        c=[("send_from","send@t.de"),

            ("host",HOST),

            ("port",PORT-1),

            ("user",""),

            ("password",""),

            ]

        p=SMTP("test",c)

        content = ""

        status = p.send(Mail("sub", content, "f@t.de"), "t@t.de")

        self.assertEqual(status.status, status.ERROR)

        self.assertIsInstance(status.error,IOError)

        self.assertEqual(status.error.errno,111)

        self.assertEqual(len(self.smtp_server.rcvd), 0)

    def testNeededOption(self):

        pass

    testNeededOption.todo = "test NeededOption is not implemented"

    def testUnknownOption(self):

        pass

    testUnknownOption.todo = "test UnknownOption is not implemented"

    def testSendFunc(self):

        pass

    testSendFunc.todo = "test getSendFunc is not implemented"