"""
A helper class which allows test code to intercept and store sent email.
(from: http://lakin.weckers.net/thoughts/twisted/part1/threaded/)
"""
import smtpd
import asyncore

class TestSMTPServer(smtpd.SMTPServer):
    """
    An SMTP Server used to allow integration tests which ensure email is sent.
    """

    def __init__(self, localaddr):
        self.rcvd = []
        smtpd.SMTPServer.__init__(self, localaddr, None)

    def start(self):
        import threading
        self.poller = threading.Thread(target=asyncore.loop,
                kwargs={'timeout':0.1, 'use_poll':True})
        self.poller.start()

    def process_message(self, peer, mailfrom, rcpttos, data):
        self.rcvd.append((mailfrom, rcpttos, data))

    def close(self):
        smtpd.SMTPServer.close(self)
        self.poller.join()