~ulrith/mailman/russian-2.1.20

# Copyright (C) 2001-2010 by the Free Software Foundation, Inc.

#

# This program is free software; you can redistribute it and/or

# modify it under the terms of the GNU General Public License

# as published by the Free Software Foundation; either version 2

# of the License, or (at your option) any later version.

# 

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# 

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software 

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

"""Unit tests for the various Message class methods.

"""

import sys

import unittest

import email

try:

    from Mailman import __init__

except ImportError:

    import paths

from Mailman import Message

from Mailman import Version

from Mailman import Errors

from EmailBase import EmailBase

class TestSentMessage1(EmailBase):

    def test_user_notification(self):

        eq = self.assertEqual

        unless = self.failUnless

        msg = Message.UserNotification(

            'aperson@dom.ain',

            '_xtest@dom.ain',

            'Your Test List',

            'About your test list')

        msg.send(self._mlist)

        qmsg = email.message_from_string(self._readmsg())

        eq(qmsg['subject'], 'Your Test List')

        eq(qmsg['from'], '_xtest@dom.ain')

        eq(qmsg['to'], 'aperson@dom.ain')

        # The Message-ID: header has some time-variant information

        msgid = qmsg['message-id']

        unless(msgid.startswith('<mailman.'))

        unless(msgid.endswith('._xtest@dom.ain>'))

        eq(qmsg['sender'], '_xtest-bounces@dom.ain')

        eq(qmsg['errors-to'], '_xtest-bounces@dom.ain')

        eq(qmsg['x-beenthere'], '_xtest@dom.ain')

        eq(qmsg['x-mailman-version'], Version.VERSION)

        eq(qmsg['precedence'], 'bulk')

        eq(qmsg['list-id'], '<_xtest.dom.ain>')

        eq(qmsg['x-list-administrivia'], 'yes')

        eq(qmsg.get_payload(), 'About your test list')

class TestSentMessage2(EmailBase):

    def test_bounce_message(self):

        eq = self.assertEqual

        unless = self.failUnless

        msg = email.message_from_string("""\

To: _xtest@dom.ain

From: nobody@dom.ain

Subject: and another thing

yadda yadda yadda

""", Message.Message)

        self._mlist.BounceMessage(msg, {})

        qmsg = email.message_from_string(self._readmsg())

        unless(qmsg.is_multipart())

        eq(len(qmsg.get_payload()), 2)

        # The first payload is the details of the bounce action, and the

        # second message is the message/rfc822 attachment of the original

        # message.

        msg1 = qmsg.get_payload(0)

        eq(msg1.get_content_type(), 'text/plain')

        eq(msg1.get_payload(), '[No bounce details are available]')

        msg2 = qmsg.get_payload(1)

        eq(msg2.get_content_type(), 'message/rfc822')

        unless(msg2.is_multipart())

        msg3 = msg2.get_payload(0)

        eq(msg3.get_payload(), 'yadda yadda yadda\n')

def suite(x):

    suite = unittest.TestSuite()

    if x == '1':

        suite.addTest(unittest.makeSuite(TestSentMessage1))

    elif x == '2':

        suite.addTest(unittest.makeSuite(TestSentMessage2))

    return suite

if __name__ == '__main__':

    # There is some issue in asyncore.py that prevents successfully running more than

    # one test at a time, so specify which of the two tests as an argument.

    if len(sys.argv) == 1:

        x = '1'

    else:

        x = sys.argv[1]

    if x not in ('1', '2'):

        print >> sys.stderr, (

            'usage: python test_message.py [n] where n = 1, 2 is the sub-test to run.')

        sys.exit(1)

    unittest.TextTestRunner(verbosity=2).run(suite(x))