# Copyright (c) 2012 netzguerilla.net <iro@netzguerilla.net>
# 
# This file is part of Iro.
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy of
# this software and associated documentation files (the "Software"), to deal in
# the Software without restriction, including without limitation the rights to use,
# copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
# #Software, and to permit persons to whom the Software is furnished to do so,
# subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

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

    @patch("iro.model.message.formatdate")
    def testSendMail(self,mock_f):
        mock_f.return_value="Wed, 21 Mar 2012 17:16:11 +0100"
        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(msg.get_all("Date"),["Wed, 21 Mar 2012 17:16:11 +0100"])
        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")