blob: e9ad115d1475789258d71a77b1250ca13bc4c2de [file] [log] [blame] [edit]
#!/usr/bin/python
import unittest
import mail, email
class test_data:
mail_host = None
mail_port = None
mail_connect = False
mail_from_address = None
mail_to_address = None
mail_message = None
# we define our needed mock SMTP
class SMTP:
def __init__(self, host=None, port=25):
test_data.mail_host = host
test_data.mail_port = port
if test_data.mail_host:
self.connect(test_data.mail_host, test_data.mail_port)
def connect(self, host, port):
test_data.mail_connect = True
def quit(self):
test_data.mail_connect = False
def sendmail(self, from_address, to_address, message):
test_data.mail_from_address = from_address
test_data.mail_to_address = to_address
test_data.mail_message = message
class mail_test(unittest.TestCase):
cached_SMTP = None
def setUp(self):
# now perform the slip
self.cached_SMTP = mail.smtplib.SMTP
mail.smtplib.SMTP = SMTP
def tearDown(self):
# now put things back
mail.smtplib.SMTP = self.cached_SMTP
def test_send_message(self):
message = email.Message.Message()
message["To"] = "you"
message["Cc"] = "them"
message["From"] = "me"
message["Subject"] = "hello"
message.set_payload("Hello everybody!")
mail.send("me", "you", "them", "hello", "Hello everybody!")
self.assertEquals("me", test_data.mail_from_address)
self.assertEquals(["you","them"], test_data.mail_to_address)
self.assertEquals(message.as_string(), test_data.mail_message)
# this is so the test can be run in standalone mode
if __name__ == '__main__':
unittest.main()