iro/test_helpers/smtp_helper.py
author Sandro Knauß <knauss@netzguerilla.net>
Fri, 24 Aug 2012 01:05:06 +0200
branchdevel
changeset 294 0e75bd39767d
parent 230 448dd8d36839
permissions -rw-r--r--
adding LICENSE to all files
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
294
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     1
# Copyright (c) 2012 netzguerilla.net <iro@netzguerilla.net>
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     2
# 
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     3
# This file is part of Iro.
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     4
# 
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     5
# Permission is hereby granted, free of charge, to any person obtaining a copy of
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     6
# this software and associated documentation files (the "Software"), to deal in
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     7
# the Software without restriction, including without limitation the rights to use,
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     8
# copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
     9
# #Software, and to permit persons to whom the Software is furnished to do so,
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    10
# subject to the following conditions:
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    11
# 
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    12
# The above copyright notice and this permission notice shall be included in
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    13
# all copies or substantial portions of the Software.
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    14
# 
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    15
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    16
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    17
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    18
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    19
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    20
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
0e75bd39767d adding LICENSE to all files
Sandro Knauß <knauss@netzguerilla.net>
parents: 230
diff changeset
    21
164
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    22
"""
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    23
A helper class which allows test code to intercept and store sent email.
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    24
(from: http://lakin.weckers.net/thoughts/twisted/part1/threaded/)
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    25
"""
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    26
import smtpd
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    27
import asyncore
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    28
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    29
class TestSMTPServer(smtpd.SMTPServer):
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    30
    """
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    31
    An SMTP Server used to allow integration tests which ensure email is sent.
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    32
    """
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    33
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    34
    def __init__(self, localaddr):
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    35
        self.rcvd = []
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    36
        smtpd.SMTPServer.__init__(self, localaddr, None)
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    37
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    38
    def start(self):
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    39
        import threading
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    40
        self.poller = threading.Thread(target=asyncore.loop,
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    41
                kwargs={'timeout':0.1, 'use_poll':True})
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    42
        self.poller.start()
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    43
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    44
    def process_message(self, peer, mailfrom, rcpttos, data):
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    45
        self.rcvd.append((mailfrom, rcpttos, data))
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    46
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    47
    def close(self):
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    48
        smtpd.SMTPServer.close(self)
b634a8fed51f adding tests for smtp backend
Sandro Knauß <knauss@netzguerilla.net>
parents:
diff changeset
    49
        self.poller.join()