1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
# Copyright (C) 2001-2017 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>'))
# The Sender: header is optional and addresses can be VERPed
if self._mlist.include_sender_header:
sender = qmsg['sender']
unless(sender.startswith('"_xtest" <_xtest-bounces'))
unless(sender.endswith('@dom.ain>'))
eto = qmsg['errors-to']
unless(eto.startswith('_xtest-bounces'))
unless(eto.endswith('@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))
|
