diff options
| author | bwarsaw | 2001-10-01 16:39:12 +0000 |
|---|---|---|
| committer | bwarsaw | 2001-10-01 16:39:12 +0000 |
| commit | e858c7afdd086b2dad99ee13a15ab8f06eb4d136 (patch) | |
| tree | 4800a990b602d96c6a49d2f7097bbc3da81aff5e /tests/test_handlers.py | |
| parent | 2825f393561f20c82103914c741bbc5ca88543eb (diff) | |
| download | mailman-e858c7afdd086b2dad99ee13a15ab8f06eb4d136.tar.gz mailman-e858c7afdd086b2dad99ee13a15ab8f06eb4d136.tar.zst mailman-e858c7afdd086b2dad99ee13a15ab8f06eb4d136.zip | |
Diffstat (limited to 'tests/test_handlers.py')
| -rw-r--r-- | tests/test_handlers.py | 1648 |
1 files changed, 1648 insertions, 0 deletions
diff --git a/tests/test_handlers.py b/tests/test_handlers.py new file mode 100644 index 000000000..894222611 --- /dev/null +++ b/tests/test_handlers.py @@ -0,0 +1,1648 @@ +# Copyright (C) 2001 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + +"""Unit tests for the various Mailman/Handlers/*.py modules. +""" + +import os +import time +import sha +import unittest +import cPickle +import errno +import email +from types import ListType + +from Mailman import mm_cfg +from Mailman.MailList import MailList +from Mailman import Message +from Mailman import Errors +from Mailman import Pending +from Mailman.Queue.Switchboard import Switchboard + +from Mailman.Handlers import Acknowledge +from Mailman.Handlers import AfterDelivery +from Mailman.Handlers import Approve +from Mailman.Handlers import CalcRecips +from Mailman.Handlers import Cleanse +from Mailman.Handlers import CookHeaders +from Mailman.Handlers import Decorate +from Mailman.Handlers import FileRecips +from Mailman.Handlers import Hold +from Mailman.Handlers import Personalize +from Mailman.Handlers import Replybot +from Mailman.Handlers import SMTPDirect +from Mailman.Handlers import Sendmail +from Mailman.Handlers import SpamDetect +from Mailman.Handlers import Tagger +from Mailman.Handlers import ToArchive +from Mailman.Handlers import ToDigest +from Mailman.Handlers import ToOutgoing +from Mailman.Handlers import ToUsenet + +from TestBase import TestBase + + + +def password(plaintext): + return sha.new(plaintext).hexdigest() + + + +class TestAcknowledge(TestBase): + def setUp(self): + TestBase.setUp(self) + # We're going to want to inspect this queue directory + self._sb = Switchboard(mm_cfg.VIRGINQUEUE_DIR) + # Add a member + self._mlist.addNewMember('aperson@dom.ain') + + def tearDown(self): + for f in os.listdir(mm_cfg.VIRGINQUEUE_DIR): + os.unlink(os.path.join(mm_cfg.VIRGINQUEUE_DIR, f)) + TestBase.tearDown(self) + + def test_no_ack_msgdata(self): + eq = self.assertEqual + # Make sure there are no files in the virgin queue already + eq(len(self._sb.files()), 0) + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + Acknowledge.process(self._mlist, msg, + {'original_sender': 'aperson@dom.ain'}) + eq(len(self._sb.files()), 0) + + def test_no_ack_not_a_member(self): + eq = self.assertEqual + # Make sure there are no files in the virgin queue already + eq(len(self._sb.files()), 0) + msg = email.message_from_string("""\ +From: bperson@dom.ain + +""", Message.Message) + Acknowledge.process(self._mlist, msg, + {'original_sender': 'bperson@dom.ain'}) + eq(len(self._sb.files()), 0) + + def test_no_ack_sender(self): + eq = self.assertEqual + eq(len(self._sb.files()), 0) + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + Acknowledge.process(self._mlist, msg, {}) + eq(len(self._sb.files()), 0) + + def test_ack_no_subject(self): + eq = self.assertEqual + self._mlist.host_name = 'dom.ain' + self._mlist.web_page_url = 'http://www.dom.ain/mailman/' + self._mlist.setMemberOption( + 'aperson@dom.ain', mm_cfg.AcknowledgePosts, 1) + eq(len(self._sb.files()), 0) + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + Acknowledge.process(self._mlist, msg, {}) + files = self._sb.files() + eq(len(files), 1) + qmsg, qdata = self._sb.dequeue(files[0]) + # Check the .db file + eq(qdata.get('listname'), '_xtest') + eq(qdata.get('recips'), ['aperson@dom.ain']) + eq(qdata.get('version'), 3) + # Check the .pck + eq(qmsg['subject'], '_xtest post acknowledgement') + eq(qmsg['to'], 'aperson@dom.ain') + eq(qmsg['from'], '_xtest-admin@dom.ain') + eq(qmsg.get_type(), 'text/plain') + eq(qmsg.get_param('charset'), 'us-ascii') + msgid = qmsg['message-id'] + self.failUnless(msgid.startswith('<mailman.')) + self.failUnless(msgid.endswith('._xtest@dom.ain>')) + eq(qmsg.get_payload(), """\ +Your message entitled + + (no subject) + +was successfully received by the _xtest mailing list. + +List info page: http://www.dom.ain/mailman/listinfo/_xtest +""") + # Make sure we dequeued the only message + eq(len(self._sb.files()), 0) + + def test_ack_with_subject(self): + eq = self.assertEqual + self._mlist.host_name = 'dom.ain' + self._mlist.web_page_url = 'http://www.dom.ain/mailman/' + self._mlist.setMemberOption( + 'aperson@dom.ain', mm_cfg.AcknowledgePosts, 1) + eq(len(self._sb.files()), 0) + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: Wish you were here + +""", Message.Message) + Acknowledge.process(self._mlist, msg, {}) + files = self._sb.files() + eq(len(files), 1) + qmsg, qdata = self._sb.dequeue(files[0]) + # Check the .db file + eq(qdata.get('listname'), '_xtest') + eq(qdata.get('recips'), ['aperson@dom.ain']) + eq(qdata.get('version'), 3) + # Check the .pck + eq(qmsg['subject'], '_xtest post acknowledgement') + eq(qmsg['to'], 'aperson@dom.ain') + eq(qmsg['from'], '_xtest-admin@dom.ain') + eq(qmsg.get_type(), 'text/plain') + eq(qmsg.get_param('charset'), 'us-ascii') + msgid = qmsg['message-id'] + self.failUnless(msgid.startswith('<mailman.')) + self.failUnless(msgid.endswith('._xtest@dom.ain>')) + eq(qmsg.get_payload(), """\ +Your message entitled + + Wish you were here + +was successfully received by the _xtest mailing list. + +List info page: http://www.dom.ain/mailman/listinfo/_xtest +""") + # Make sure we dequeued the only message + eq(len(self._sb.files()), 0) + + def test_ack_with_prefixed_subject(self): + eq = self.assertEqual + self._mlist.host_name = 'dom.ain' + self._mlist.web_page_url = 'http://www.dom.ain/mailman/' + self._mlist.subject_prefix = '[XTEST] ' + self._mlist.setMemberOption( + 'aperson@dom.ain', mm_cfg.AcknowledgePosts, 1) + eq(len(self._sb.files()), 0) + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: [XTEST] Wish you were here + +""", Message.Message) + Acknowledge.process(self._mlist, msg, {}) + files = self._sb.files() + eq(len(files), 1) + qmsg, qdata = self._sb.dequeue(files[0]) + # Check the .db file + eq(qdata.get('listname'), '_xtest') + eq(qdata.get('recips'), ['aperson@dom.ain']) + eq(qdata.get('version'), 3) + # Check the .pck + eq(qmsg['subject'], '_xtest post acknowledgement') + eq(qmsg['to'], 'aperson@dom.ain') + eq(qmsg['from'], '_xtest-admin@dom.ain') + eq(qmsg.get_type(), 'text/plain') + eq(qmsg.get_param('charset'), 'us-ascii') + msgid = qmsg['message-id'] + self.failUnless(msgid.startswith('<mailman.')) + self.failUnless(msgid.endswith('._xtest@dom.ain>')) + eq(qmsg.get_payload(), """\ +Your message entitled + + Wish you were here + +was successfully received by the _xtest mailing list. + +List info page: http://www.dom.ain/mailman/listinfo/_xtest +""") + # Make sure we dequeued the only message + eq(len(self._sb.files()), 0) + + + +class TestAfterDelivery(TestBase): + # Both msg and msgdata are ignored + def test_process(self): + mlist = self._mlist + last_post_time = mlist.last_post_time + post_id = mlist.post_id + AfterDelivery.process(mlist, None, None) + self.failUnless(mlist.last_post_time > last_post_time) + self.assertEqual(mlist.post_id, post_id + 1) + + + +class TestApprove(TestBase): + def test_short_circuit(self): + msgdata = {'approved': 1} + rtn = Approve.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + self.assertEqual(rtn, None) + + def test_approved_moderator(self): + mlist = self._mlist + mlist.mod_password = password('wazoo') + msg = email.message_from_string("""\ +Approved: wazoo + +""") + msgdata = {} + Approve.process(mlist, msg, msgdata) + self.failUnless(msgdata.has_key('approved')) + self.assertEqual(msgdata['approved'], 1) + + def test_approve_moderator(self): + mlist = self._mlist + mlist.mod_password = password('wazoo') + msg = email.message_from_string("""\ +Approve: wazoo + +""") + msgdata = {} + Approve.process(mlist, msg, msgdata) + self.failUnless(msgdata.has_key('approved')) + self.assertEqual(msgdata['approved'], 1) + + def test_approved_admin(self): + mlist = self._mlist + mlist.password = password('wazoo') + msg = email.message_from_string("""\ +Approved: wazoo + +""") + msgdata = {} + Approve.process(mlist, msg, msgdata) + self.failUnless(msgdata.has_key('approved')) + self.assertEqual(msgdata['approved'], 1) + + def test_approve_admin(self): + mlist = self._mlist + mlist.password = password('wazoo') + msg = email.message_from_string("""\ +Approve: wazoo + +""") + msgdata = {} + Approve.process(mlist, msg, msgdata) + self.failUnless(msgdata.has_key('approved')) + self.assertEqual(msgdata['approved'], 1) + + def test_unapproved(self): + mlist = self._mlist + mlist.password = password('zoowa') + msg = email.message_from_string("""\ +Approve: wazoo + +""") + msgdata = {} + Approve.process(mlist, msg, msgdata) + self.assertEqual(msgdata.get('approved'), None) + + def test_trip_beentheres(self): + mlist = self._mlist + mlist.host_name = 'dom.ain' + msg = email.message_from_string("""\ +X-BeenThere: %s + +""" % mlist.GetListEmail()) + self.assertRaises(Errors.LoopError, Approve.process, mlist, msg, {}) + + + +class TestCalcRecips(TestBase): + def setUp(self): + TestBase.setUp(self) + # Add a bunch of regular members + mlist = self._mlist + mlist.addNewMember('aperson@dom.ain') + mlist.addNewMember('bperson@dom.ain') + mlist.addNewMember('cperson@dom.ain') + # And a bunch of digest members + mlist.addNewMember('dperson@dom.ain', digest=1) + mlist.addNewMember('eperson@dom.ain', digest=1) + mlist.addNewMember('fperson@dom.ain', digest=1) + + def test_short_circuit(self): + msgdata = {'recips': 1} + rtn = CalcRecips.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + self.assertEqual(rtn, None) + + def test_simple_path(self): + msgdata = {} + msg = email.message_from_string("""\ +From: dperson@dom.ain + +""", Message.Message) + CalcRecips.process(self._mlist, msg, msgdata) + self.failUnless(msgdata.has_key('recips')) + recips = msgdata['recips'] + recips.sort() + self.assertEqual(recips, ['aperson@dom.ain', 'bperson@dom.ain', + 'cperson@dom.ain']) + + def test_exclude_sender(self): + msgdata = {} + msg = email.message_from_string("""\ +From: cperson@dom.ain + +""", Message.Message) + self._mlist.setMemberOption('cperson@dom.ain', + mm_cfg.DontReceiveOwnPosts, 1) + CalcRecips.process(self._mlist, msg, msgdata) + self.failUnless(msgdata.has_key('recips')) + recips = msgdata['recips'] + recips.sort() + self.assertEqual(recips, ['aperson@dom.ain', 'bperson@dom.ain']) + + def test_urgent_moderator(self): + self._mlist.mod_password = password('xxXXxx') + msgdata = {} + msg = email.message_from_string("""\ +From: dperson@dom.ain +Urgent: xxXXxx + +""", Message.Message) + CalcRecips.process(self._mlist, msg, msgdata) + self.failUnless(msgdata.has_key('recips')) + recips = msgdata['recips'] + recips.sort() + self.assertEqual(recips, ['aperson@dom.ain', 'bperson@dom.ain', + 'cperson@dom.ain', 'dperson@dom.ain', + 'eperson@dom.ain', 'fperson@dom.ain']) + + def test_urgent_admin(self): + self._mlist.mod_password = password('yyYYyy') + self._mlist.password = password('xxXXxx') + msgdata = {} + msg = email.message_from_string("""\ +From: dperson@dom.ain +Urgent: xxXXxx + +""", Message.Message) + CalcRecips.process(self._mlist, msg, msgdata) + self.failUnless(msgdata.has_key('recips')) + recips = msgdata['recips'] + recips.sort() + self.assertEqual(recips, ['aperson@dom.ain', 'bperson@dom.ain', + 'cperson@dom.ain', 'dperson@dom.ain', + 'eperson@dom.ain', 'fperson@dom.ain']) + + def test_urgent_reject(self): + self._mlist.mod_password = password('yyYYyy') + self._mlist.password = password('xxXXxx') + msgdata = {} + msg = email.message_from_string("""\ +From: dperson@dom.ain +Urgent: zzZZzz + +""", Message.Message) + self.assertRaises(CalcRecips.RejectUrgentMessage, + CalcRecips.process, + self._mlist, msg, msgdata) + + # BAW: must test the do_topic_filters() path... + + + +class TestCleanse(TestBase): + def setUp(self): + TestBase.setUp(self) + self._mlist.host_name = 'dom.ain' + + def test_simple_cleanse(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain +Approved: yes +Urgent: indeed +Reply-To: bperson@dom.ain +Sender: asystem@dom.ain +Return-Receipt-To: another@dom.ain +Disposition-Notification-To: athird@dom.ain +X-Confirm-Reading-To: afourth@dom.ain +X-PMRQC: afifth@dom.ain +Subject: a message to you + +""", Message.Message) + Cleanse.process(self._mlist, msg, {}) + eq(msg['approved'], None) + eq(msg['urgent'], None) + eq(msg['return-receipt-to'], None) + eq(msg['disposition-notification-to'], None) + eq(msg['x-confirm-reading-to'], None) + eq(msg['x-pmrqc'], None) + eq(msg['from'], 'aperson@dom.ain') + eq(msg['reply-to'], 'bperson@dom.ain') + eq(msg['sender'], 'asystem@dom.ain') + eq(msg['subject'], 'a message to you') + + def test_anon_cleanse(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain +Approved: yes +Urgent: indeed +Reply-To: bperson@dom.ain +Sender: asystem@dom.ain +Return-Receipt-To: another@dom.ain +Disposition-Notification-To: athird@dom.ain +X-Confirm-Reading-To: afourth@dom.ain +X-PMRQC: afifth@dom.ain +Subject: a message to you + +""", Message.Message) + self._mlist.anonymous_list = 1 + Cleanse.process(self._mlist, msg, {}) + eq(msg['approved'], None) + eq(msg['urgent'], None) + eq(msg['return-receipt-to'], None) + eq(msg['disposition-notification-to'], None) + eq(msg['x-confirm-reading-to'], None) + eq(msg['x-pmrqc'], None) + eq(len(msg.get_all('from')), 1) + eq(len(msg.get_all('reply-to')), 1) + eq(msg['from'], '_xtest-admin@dom.ain') + eq(msg['reply-to'], '_xtest@dom.ain') + eq(msg['sender'], None) + eq(msg['subject'], 'a message to you') + + + +class TestCookHeaders(TestBase): + def test_transform_noack_to_xack(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +X-Ack: yes + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {'noack': 1}) + eq(len(msg.get_all('x-ack')), 1) + eq(msg['x-ack'], 'no') + + def test_original_sender(self): + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + msgdata = {} + CookHeaders.process(self._mlist, msg, msgdata) + self.assertEqual(msgdata.get('original_sender'), 'aperson@dom.ain') + + def test_no_original_sender(self): + msg = email.message_from_string("""\ +Subject: about this message + +""", Message.Message) + msgdata = {} + CookHeaders.process(self._mlist, msg, msgdata) + self.assertEqual(msgdata.get('original_sender'), '') + + def test_explicit_sender_and_errors_to(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +Sender: asystem@dom.ain +Errors-To: bsystem@dom.ain + +""", Message.Message) + msgdata = {'errorsto': 'robohelper@dom.ain'} + CookHeaders.process(self._mlist, msg, msgdata) + eq(msg['sender'], 'robohelper@dom.ain') + eq(msg['errors-to'], 'robohelper@dom.ain') + + def test_implicit_sender_and_errors_to(self): + eq = self.assertEqual + self._mlist.host_name = 'dom.ain' + msg = email.message_from_string("""\ +Sender: asystem@dom.ain +Errors-To: bsystem@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['sender'], '_xtest-admin@dom.ain') + eq(msg['errors-to'], '_xtest-admin@dom.ain') + + def test_xbeenthere(self): + self._mlist.host_name = 'dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + self.assertEqual(msg['x-beenthere'], '_xtest@dom.ain') + + def test_multiple_xbeentheres(self): + eq = self.assertEqual + self._mlist.host_name = 'dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain +X-BeenThere: alist@another.dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(len(msg.get_all('x-beenthere')), 2) + beentheres = msg.get_all('x-beenthere') + beentheres.sort() + eq(beentheres, ['_xtest@dom.ain', 'alist@another.dom.ain']) + + def test_nonexisting_mmversion(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['x-mailman-version'], mm_cfg.VERSION) + + def test_existing_mmversion(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain +X-Mailman-Version: 3000 + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(len(msg.get_all('x-mailman-version')), 1) + eq(msg['x-mailman-version'], '3000') + + def test_nonexisting_precedence(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['precedence'], 'bulk') + + def test_existing_precedence(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain +Precedence: junk + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(len(msg.get_all('precedence')), 1) + eq(msg['precedence'], 'junk') + + def test_subject_munging_no_subject(self): + self._mlist.subject_prefix = '[XTEST] ' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + self.assertEqual(msg['subject'], '[XTEST] (no subject)') + + def test_subject_munging(self): + self._mlist.subject_prefix = '[XTEST] ' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: About Mailman... + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + self.assertEqual(msg['subject'], '[XTEST] About Mailman...') + + def test_no_subject_munging_for_digests(self): + self._mlist.subject_prefix = '[XTEST] ' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: About Mailman... + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {'isdigest': 1}) + self.assertEqual(msg['subject'], 'About Mailman...') + + def test_no_subject_munging_for_fasttrack(self): + self._mlist.subject_prefix = '[XTEST] ' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: About Mailman... + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {'_fasttrack': 1}) + self.assertEqual(msg['subject'], 'About Mailman...') + + def test_no_subject_munging_has_prefix(self): + self._mlist.subject_prefix = '[XTEST] ' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: Re: [XTEST] About Mailman... + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + self.assertEqual(msg['subject'], 'Re: [XTEST] About Mailman...') + + def test_reply_to_list(self): + eq = self.assertEqual + self._mlist.reply_goes_to_list = 1 + self._mlist.host_name = 'dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['reply-to'], '_xtest@dom.ain') + eq(msg['x-reply-to'], None) + + def test_reply_to_list_with_xreplyto(self): + eq = self.assertEqual + self._mlist.reply_goes_to_list = 1 + self._mlist.host_name = 'dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Reply-To: bperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['reply-to'], '_xtest@dom.ain') + eq(msg['x-reply-to'], 'bperson@dom.ain') + + def test_reply_to_explicit(self): + eq = self.assertEqual + self._mlist.reply_goes_to_list = 2 + self._mlist.host_name = 'dom.ain' + self._mlist.reply_to_address = 'mlist@dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['reply-to'], 'mlist@dom.ain') + eq(msg['x-reply-to'], None) + + def test_reply_to_list_with_xreplyto2(self): + eq = self.assertEqual + self._mlist.reply_goes_to_list = 2 + self._mlist.host_name = 'dom.ain' + self._mlist.reply_to_address = 'mlist@dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Reply-To: bperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['reply-to'], 'mlist@dom.ain') + eq(msg['x-reply-to'], 'bperson@dom.ain') + + def test_reply_to_explicit_with_bad_address(self): + eq = self.assertEqual + self._mlist.reply_goes_to_list = 2 + self._mlist.host_name = 'dom.ain' + # A bad address that is missing the domain part + self._mlist.reply_to_address = 'mlist' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Reply-To: bperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['reply-to'], 'bperson@dom.ain') + eq(msg['x-reply-to'], None) + + def test_list_headers_nolist(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {'_nolist': 1}) + eq(msg['list-id'], None) + eq(msg['list-help'], None) + eq(msg['list-unsubscribe'], None) + eq(msg['list-subscribe'], None) + eq(msg['list-post'], None) + eq(msg['list-archive'], None) + + def test_list_headers(self): + eq = self.assertEqual + self._mlist.archive = 1 + self._mlist.host_name = 'dom.ain' + self._mlist.web_page_url = 'http://www.dom.ain/mailman/' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['list-id'], '<_xtest.dom.ain>') + eq(msg['list-help'], '<mailto:_xtest-request@dom.ain?subject=help>') + eq(msg['list-unsubscribe'], + '<http://www.dom.ain/mailman/listinfo/_xtest>,' + '\n\t<mailto:_xtest-request@dom.ain?subject=unsubscribe>') + eq(msg['list-subscribe'], + '<http://www.dom.ain/mailman/listinfo/_xtest>,' + '\n\t<mailto:_xtest-request@dom.ain?subject=subscribe>') + eq(msg['list-post'], '<mailto:_xtest@dom.ain>') + eq(msg['list-archive'], '<http://www.dom.ain/pipermail/_xtest/>') + + def test_list_headers_with_description(self): + eq = self.assertEqual + self._mlist.archive = 1 + self._mlist.description = 'A Test List' + self._mlist.host_name = 'dom.ain' + self._mlist.web_page_url = 'http://www.dom.ain/mailman/' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + CookHeaders.process(self._mlist, msg, {}) + eq(msg['list-id'], 'A Test List <_xtest.dom.ain>') + eq(msg['list-help'], '<mailto:_xtest-request@dom.ain?subject=help>') + eq(msg['list-unsubscribe'], + '<http://www.dom.ain/mailman/listinfo/_xtest>,' + '\n\t<mailto:_xtest-request@dom.ain?subject=unsubscribe>') + eq(msg['list-subscribe'], + '<http://www.dom.ain/mailman/listinfo/_xtest>,' + '\n\t<mailto:_xtest-request@dom.ain?subject=subscribe>') + eq(msg['list-post'], '<mailto:_xtest@dom.ain>') + eq(msg['list-archive'], '<http://www.dom.ain/pipermail/_xtest/>') + + + +class TestDecorate(TestBase): + def test_short_circuit(self): + msgdata = {'isdigest': 1} + rtn = Decorate.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + self.assertEqual(rtn, None) + + def test_no_multipart(self): + mlist = self._mlist + mlist.msg_header = 'header\n' + mlist.msg_footer = 'footer' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.get_payload(), """\ +header +Here is a message. +footer""") + + def test_no_multipart_template(self): + mlist = self._mlist + mlist.msg_header = '%(real_name)s header\n' + mlist.msg_footer = '%(real_name)s footer' + mlist.real_name = 'XTest' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.get_payload(), """\ +XTest header +Here is a message. +XTest footer""") + + def test_no_multipart_type_error(self): + mlist = self._mlist + mlist.msg_header = '%(real_name) header\n' + mlist.msg_footer = '%(real_name) footer' + mlist.real_name = 'XTest' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.get_payload(), """\ +%(real_name) header +Here is a message. +%(real_name) footer""") + + def test_no_multipart_value_error(self): + mlist = self._mlist + mlist.msg_header = '%(real_name)p header\n' + mlist.msg_footer = '%(real_name)p footer' + mlist.real_name = 'XTest' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.get_payload(), """\ +%(real_name)p header +Here is a message. +%(real_name)p footer""") + + def test_no_multipart_missing_key(self): + mlist = self._mlist + mlist.msg_header = '%(spooge)s header\n' + mlist.msg_footer = '%(spooge)s footer' + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.get_payload(), """\ +%(spooge)s header +Here is a message. +%(spooge)s footer""") + + def test_multipart(self): + mlist = self._mlist + mlist.msg_header = 'header\n' + mlist.msg_footer = 'footer' + msg1 = email.message_from_string("""\ +From: aperson@dom.ain + +Here is the first message. +""") + msg2 = email.message_from_string("""\ +From: bperson@dom.ain + +Here is the second message. +""") + msg = Message.Message() + msg.add_payload(msg1) + msg.add_payload(msg2) + msg.add_header('Content-Type', 'multipart/mixed', boundary='BOUNDARY') + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.as_string(unixfrom=0), """\ +Content-Type: multipart/mixed; boundary="BOUNDARY" + +--BOUNDARY +Content-Type: text/plain; charset="us-ascii" +MIME-Version: 1.0 +Content-Transfer-Encoding: 7bit + +header + +--BOUNDARY +From: aperson@dom.ain + +Here is the first message. + +--BOUNDARY +From: bperson@dom.ain + +Here is the second message. + +--BOUNDARY +Content-Type: text/plain; charset="us-ascii" +MIME-Version: 1.0 +Content-Transfer-Encoding: 7bit + +footer + +--BOUNDARY--""") + + def test_image(self): + mlist = self._mlist + mlist.msg_header = 'header\n' + mlist.msg_footer = 'footer' + msg = email.message_from_string("""\ +From: aperson@dom.ain +Content-type: image/x-spooge + +IMAGEDATAIMAGEDATAIMAGEDATA +""") + Decorate.process(self._mlist, msg, {}) + self.assertEqual(msg.get_payload(), """\ +IMAGEDATAIMAGEDATAIMAGEDATA +""") + + def test_personalize_assert(self): + raises = self.assertRaises + raises(AssertionError, Decorate.process, + self._mlist, None, {'personalized': 1}) + raises(AssertionError, Decorate.process, + self._mlist, None, {'personalized': 1, + 'recips': [1, 2, 3]}) + + def test_personalize_not_a_member(self): + mlist = self._mlist + mlist.msg_header = """\ +%(user_address)s +%(user_delivered_to)s +%(user_language)s +%(user_name)s +%(user_optionsurl)s +""" + mlist.msg_footer = mlist.msg_header + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(mlist, msg, {'personalized': 1, + 'recips': ['aperson@dom.ain']}) + self.assertEqual(msg.get_payload(), """\ +aperson@dom.ain +%(user_delivered_to)s +%(user_language)s +%(user_name)s +%(user_optionsurl)s +Here is a message. +aperson@dom.ain +%(user_delivered_to)s +%(user_language)s +%(user_name)s +%(user_optionsurl)s +""") + + def test_personalize(self): + mlist = self._mlist + mlist.msg_header = """\ +%(user_address)s +%(user_delivered_to)s +%(user_language)s +%(user_name)s +%(user_optionsurl)s +""" + mlist.msg_footer = mlist.msg_header + mlist.web_page_url = 'http://www.dom.ain/mailman/' + mlist.addNewMember('APerson@dom.ain', language='xx', + password='xxXXxx', realname='A. Person') + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(mlist, msg, {'personalized': 1, + 'recips': ['aperson@dom.ain']}) + self.assertEqual(msg.get_payload(), """\ +aperson@dom.ain +APerson@dom.ain +xx +A. Person +http://www.dom.ain/mailman/options/_xtest/aperson%40dom.ain +Here is a message. +aperson@dom.ain +APerson@dom.ain +xx +A. Person +http://www.dom.ain/mailman/options/_xtest/aperson%40dom.ain +""") + + def test_personalize_no_name(self): + mlist = self._mlist + mlist.msg_header = """\ +%(user_address)s +%(user_delivered_to)s +%(user_language)s +%(user_name)s +%(user_optionsurl)s +""" + mlist.msg_footer = mlist.msg_header + mlist.web_page_url = 'http://www.dom.ain/mailman/' + mlist.addNewMember('APerson@dom.ain', language='xx', password='xxXXxx') + msg = email.message_from_string("""\ +From: aperson@dom.ain + +Here is a message. +""") + Decorate.process(mlist, msg, {'personalized': 1, + 'recips': ['aperson@dom.ain']}) + self.assertEqual(msg.get_payload(), """\ +aperson@dom.ain +APerson@dom.ain +xx +not available +http://www.dom.ain/mailman/options/_xtest/aperson%40dom.ain +Here is a message. +aperson@dom.ain +APerson@dom.ain +xx +not available +http://www.dom.ain/mailman/options/_xtest/aperson%40dom.ain +""") + + + +class TestFileRecips(TestBase): + def test_short_circuit(self): + msgdata = {'recips': 1} + rtn = FileRecips.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + self.assertEqual(rtn, None) + + def test_file_nonexistant(self): + msgdata = {} + FileRecips.process(self._mlist, None, msgdata) + self.assertEqual(msgdata.get('recips'), []) + + def test_file_exists_no_sender(self): + msg = email.message_from_string("""\ +To: yall@dom.ain + +""", Message.Message) + msgdata = {} + file = os.path.join(self._mlist.fullpath(), 'members.txt') + addrs = ['aperson@dom.ain', 'bperson@dom.ain', + 'cperson@dom.ain', 'dperson@dom.ain'] + fp = open(file, 'w') + try: + for addr in addrs: + print >> fp, addr + fp.close() + FileRecips.process(self._mlist, msg, msgdata) + self.assertEqual(msgdata.get('recips'), addrs) + finally: + try: + os.unlink(file) + except OSError, e: + if e.errno <> e.ENOENT: raise + + def test_file_exists_no_member(self): + msg = email.message_from_string("""\ +From: eperson@dom.ain +To: yall@dom.ain + +""", Message.Message) + msgdata = {} + file = os.path.join(self._mlist.fullpath(), 'members.txt') + addrs = ['aperson@dom.ain', 'bperson@dom.ain', + 'cperson@dom.ain', 'dperson@dom.ain'] + fp = open(file, 'w') + try: + for addr in addrs: + print >> fp, addr + fp.close() + FileRecips.process(self._mlist, msg, msgdata) + self.assertEqual(msgdata.get('recips'), addrs) + finally: + try: + os.unlink(file) + except OSError, e: + if e.errno <> e.ENOENT: raise + + def test_file_exists_is_member(self): + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: yall@dom.ain + +""", Message.Message) + msgdata = {} + file = os.path.join(self._mlist.fullpath(), 'members.txt') + addrs = ['aperson@dom.ain', 'bperson@dom.ain', + 'cperson@dom.ain', 'dperson@dom.ain'] + fp = open(file, 'w') + try: + for addr in addrs: + print >> fp, addr + self._mlist.addNewMember(addr) + fp.close() + FileRecips.process(self._mlist, msg, msgdata) + self.assertEqual(msgdata.get('recips'), addrs[1:]) + finally: + try: + os.unlink(file) + except OSError, e: + if e.errno <> e.ENOENT: raise + + + +class TestHold(TestBase): + def setUp(self): + TestBase.setUp(self) + self._mlist.administrivia = 1 + self._mlist.respond_to_post_requests = 0 + self._mlist.admin_immed_notify = 0 + # We're going to want to inspect this queue directory + self._sb = Switchboard(mm_cfg.VIRGINQUEUE_DIR) + + def tearDown(self): + for f in os.listdir(mm_cfg.VIRGINQUEUE_DIR): + os.unlink(os.path.join(mm_cfg.VIRGINQUEUE_DIR, f)) + TestBase.tearDown(self) + try: + os.unlink(os.path.join(mm_cfg.DATA_DIR, 'pending.db')) + except OSError, e: + if e.errno <> errno.ENOENT: raise + for f in [holdfile for holdfile in os.listdir(mm_cfg.DATA_DIR) + if holdfile.startswith('heldmsg-')]: + os.unlink(os.path.join(mm_cfg.DATA_DIR, f)) + + def test_short_circuit(self): + msgdata = {'approved': 1} + rtn = Hold.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + self.assertEqual(rtn, None) + + def test_administrivia(self): + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: unsubscribe + +""", Message.Message) + self.assertRaises(Hold.Administrivia, Hold.process, + self._mlist, msg, {}) + + def test_forbidden_posters(self): + self._mlist.forbidden_posters = ['aperson@dom.ain'] + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: The forbidden donut + +""", Message.Message) + self.assertRaises(Hold.ForbiddenPoster, Hold.process, + self._mlist, msg, {}) + + def test_moderated_no_posters(self): + self._mlist.moderated = 1 + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: An unapproved message + +""", Message.Message) + self.assertRaises(Hold.ModeratedPost, Hold.process, + self._mlist, msg, {}) + + def test_moderated_posters(self): + self._mlist.moderated = 1 + self._mlist.posters = ['aperson@dom.ain'] + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An unapproved message + +""", Message.Message) + rtn = Hold.process(self._mlist, msg, {}) + self.assertEqual(rtn, None) + + def test_member_posting_only(self): + self._mlist.member_posting_only = 1 + msg = email.message_from_string("""\ +From: bperson@dom.ain +To: _xtest@dom.ain +Subject: An unapproved message + +""", Message.Message) + self.assertRaises(Hold.NonMemberPost, Hold.process, + self._mlist, msg, {}) + + def test_member_posting_only_ok_member(self): + self._mlist.addNewMember('aperson@dom.ain') + self._mlist.member_posting_only = 1 + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An unapproved message + +""", Message.Message) + rtn = Hold.process(self._mlist, msg, {}) + self.assertEqual(rtn, None) + + def test_member_posting_only_ok_poster(self): + self._mlist.member_posting_only = 1 + self._mlist.posters = ['aperson@dom.ain'] + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An unapproved message + +""", Message.Message) + rtn = Hold.process(self._mlist, msg, {}) + self.assertEqual(rtn, None) + + def test_posters_ok(self): + self._mlist.posters = ['aperson@dom.ain'] + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An unapproved message + +""", Message.Message) + rtn = Hold.process(self._mlist, msg, {}) + self.assertEqual(rtn, None) + + def test_posters(self): + self._mlist.posters = ['bperson@dom.ain'] + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An unapproved message + +""", Message.Message) + self.assertRaises(Hold.NotExplicitlyAllowed, Hold.process, + self._mlist, msg, {}) + + def test_max_recips(self): + self._mlist.max_num_recipients = 5 + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain, bperson@dom.ain +Cc: cperson@dom.ain +Cc: dperson@dom.ain (Jimmy D. Person) +To: Billy E. Person <eperson@dom.ain> + +Hey folks! +""", Message.Message) + self.assertRaises(Hold.TooManyRecipients, Hold.process, + self._mlist, msg, {}) + + def test_implicit_destination(self): + self._mlist.require_explicit_destination = 1 + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: An implicit message + +""", Message.Message) + self.assertRaises(Hold.ImplicitDestination, Hold.process, + self._mlist, msg, {}) + + def test_implicit_destination_fromusenet(self): + self._mlist.require_explicit_destination = 1 + msg = email.message_from_string("""\ +From: aperson@dom.ain +Subject: An implicit message + +""", Message.Message) + rtn = Hold.process(self._mlist, msg, {'fromusenet': 1}) + self.assertEqual(rtn, None) + + def test_suspicious_header(self): + self._mlist.bounce_matching_headers = 'From: .*person@(blah.)?dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An implicit message + +""", Message.Message) + self.assertRaises(Hold.SuspiciousHeaders, Hold.process, + self._mlist, msg, {}) + + def test_suspicious_header_ok(self): + self._mlist.bounce_matching_headers = 'From: .*person@blah.dom.ain' + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain +Subject: An implicit message + +""", Message.Message) + rtn = Hold.process(self._mlist, msg, {}) + self.assertEqual(rtn, None) + + def test_max_message_size(self): + self._mlist.max_message_size = 1 + msg = email.message_from_string("""\ +From: aperson@dom.ain +To: _xtest@dom.ain + +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx +""", Message.Message) + self.assertRaises(Hold.MessageTooBig, Hold.process, + self._mlist, msg, {}) + + def test_hold_notifications(self): + eq = self.assertEqual + self._mlist.respond_to_post_requests = 1 + self._mlist.admin_immed_notify = 1 + self._mlist.host_name = 'dom.ain' + # Now cause an implicit destination hold + msg = email.message_from_string("""\ +From: aperson@dom.ain + +""", Message.Message) + self.assertRaises(Hold.ImplicitDestination, Hold.process, + self._mlist, msg, {}) + # Now we have to make sure there are two messages in the virgin queue, + # one to the sender and one to the list owners. + qfiles = {} + files = self._sb.files() + eq(len(files), 2) + for filebase in files: + qmsg, qdata = self._sb.dequeue(filebase) + to = qmsg['to'] + qfiles[to] = qmsg, qdata + # BAW: We could be testing many other attributes of either the + # messages or the metadata files... + keys = qfiles.keys() + keys.sort() + eq(keys, ['_xtest-owner@dom.ain', 'aperson@dom.ain']) + # Get the pending cookie from the message to the sender + pmsg, pdata = qfiles['aperson@dom.ain'] + confirmlines = pmsg.get_payload().split('\n') + cookie = confirmlines[-3].split('/')[-1] + # We also need to make sure there's an entry in the Pending database + # for the heold message. + data = Pending.confirm(cookie) + eq(data, ('H', 1)) + heldmsg = os.path.join(mm_cfg.DATA_DIR, 'heldmsg-_xtest-1.pck') + self.failUnless(os.path.exists(heldmsg)) + os.unlink(heldmsg) + holdfiles = [f for f in os.listdir(mm_cfg.DATA_DIR) + if f.startswith('heldmsg-')] + eq(len(holdfiles), 0) + + + +class TestPersonalize(TestBase): + def setUp(self): + TestBase.setUp(self) + # We're going to want to inspect this queue directory + self._sb = Switchboard(mm_cfg.INQUEUE_DIR) + # Add a member + self._mlist.addNewMember('aperson@dom.ain') + self._mlist.addNewMember('bperson@dom.ain', realname='B. Person') + + def tearDown(self): + for f in os.listdir(mm_cfg.INQUEUE_DIR): + os.unlink(os.path.join(mm_cfg.INQUEUE_DIR, f)) + TestBase.tearDown(self) + + def test_short_circuit(self): + eq = self.assertEqual + msgdata = {} + rtn = Personalize.process(self._mlist, None, {}) + # Not really a great test, but there's little else to assert + eq(rtn, None) + eq(msgdata, {}) + # Try short circuit via msgdata + self._mlist.personalize = 1 + msgdata['personalized'] = 1 + rtn = Personalize.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + eq(rtn, None) + eq(msgdata, {'personalized': 1}) + + def test_personalize(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +To: _xtest@dom.ain + +A message for you. +""") + self._mlist.personalize = 1 + msgdata = {'recips': ['aperson@dom.ain', 'bperson@dom.ain']} + Personalize.process(self._mlist, msg, msgdata) + eq(msg['to'], '_xtest@dom.ain') + eq(msgdata, {}) + # There should be two files in qfiles/in + files = self._sb.files() + eq(len(files), 2) + filedata = {} + for filebase in files: + msg, data = self._sb.dequeue(filebase) + key = data.get('recips') + eq(type(key), ListType) + eq(len(key), 1) + filedata[key[0]] = msg, data + eq(len(filedata), 2) + keys = filedata.keys() + keys.sort() + eq(keys, ['aperson@dom.ain', 'bperson@dom.ain']) + for person in keys: + msg, data = filedata[person] + eq(data['pipeline'], ['Decorate', 'ToOutgoing']) + eq(data['recips'], [person]) + eq(data['personalized'], 1) + realname = self._mlist.getMemberName(person) + if realname: + to = '%s (%s)' % (person, realname) + else: + to = person + eq(msg['to'], to) + + + +class TestReplybot(TestBase): + pass + + + +class TestSMTPDirect(TestBase): + pass + + + +class TestSendmail(TestBase): + pass + + + +class TestSpamDetect(TestBase): + def test_short_circuit(self): + msgdata = {'approved': 1} + rtn = SpamDetect.process(self._mlist, None, msgdata) + # Not really a great test, but there's little else to assert + self.assertEqual(rtn, None) + + def test_spam_detect(self): + msg1 = email.message_from_string("""\ +From: aperson@dom.ain + +A message. +""") + msg2 = email.message_from_string("""\ +To: xlist@dom.ain + +A message. +""") + spammers = mm_cfg.KNOWN_SPAMMERS[:] + try: + mm_cfg.KNOWN_SPAMMERS.append(('from', '.?person')) + self.assertRaises(SpamDetect.SpamDetected, + SpamDetect.process, self._mlist, msg1, {}) + rtn = SpamDetect.process(self._mlist, msg2, {}) + self.assertEqual(rtn, None) + finally: + mm_cfg.KNOWN_SPAMMERS = spammers + + + +class TestTagger(TestBase): + pass + + + +class TestToArchive(TestBase): + def setUp(self): + TestBase.setUp(self) + # We're going to want to inspect this queue directory + self._sb = Switchboard(mm_cfg.ARCHQUEUE_DIR) + + def tearDown(self): + for f in os.listdir(mm_cfg.ARCHQUEUE_DIR): + os.unlink(os.path.join(mm_cfg.ARCHQUEUE_DIR, f)) + TestBase.tearDown(self) + + def test_short_circuit(self): + eq = self.assertEqual + msgdata = {'isdigest': 1} + ToArchive.process(self._mlist, None, msgdata) + eq(len(self._sb.files()), 0) + # Try the other half of the or... + self._mlist.archive = 0 + ToArchive.process(self._mlist, None, msgdata) + eq(len(self._sb.files()), 0) + # Now try the various message header shortcuts + msg = email.message_from_string("""\ +X-No-Archive: YES + +""") + self._mlist.archive = 1 + ToArchive.process(self._mlist, msg, {}) + eq(len(self._sb.files()), 0) + # And for backwards compatibility + msg = email.message_from_string("""\ +X-Archive: NO + +""") + ToArchive.process(self._mlist, msg, {}) + eq(len(self._sb.files()), 0) + + def test_normal_archiving(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +Subject: About Mailman + +It rocks! +""") + ToArchive.process(self._mlist, msg, {}) + files = self._sb.files() + eq(len(files), 1) + msg2, data = self._sb.dequeue(files[0]) + eq(len(data), 2) + eq(data['version'], 3) + self.failUnless(data['received_time'] <= time.time()) + eq(msg.as_string(unixfrom=0), msg2.as_string(unixfrom=0)) + + + +class TestToDigest(TestBase): + pass + + + +class TestToOutgoing(TestBase): + def setUp(self): + TestBase.setUp(self) + # We're going to want to inspect this queue directory + self._sb = Switchboard(mm_cfg.OUTQUEUE_DIR) + + def tearDown(self): + for f in os.listdir(mm_cfg.OUTQUEUE_DIR): + os.unlink(os.path.join(mm_cfg.OUTQUEUE_DIR, f)) + TestBase.tearDown(self) + + def test_outgoing(self): + eq = self.assertEqual + msg = email.message_from_string("""\ +Subject: About Mailman + +It rocks! +""") + msgdata = {'foo': 1, 'bar': 2} + ToOutgoing.process(self._mlist, msg, msgdata) + files = self._sb.files() + eq(len(files), 1) + msg2, data = self._sb.dequeue(files[0]) + eq(msg.as_string(unixfrom=0), msg2.as_string(unixfrom=0)) + eq(len(data), 5) + eq(data['foo'], 1) + eq(data['bar'], 2) + eq(data['version'], 3) + eq(data['listname'], '_xtest') + self.failUnless(data['received_time'] <= time.time()) + + + +class TestToUsenet(TestBase): + def setUp(self): + TestBase.setUp(self) + # We're going to want to inspect this queue directory + self._sb = Switchboard(mm_cfg.NEWSQUEUE_DIR) + + def tearDown(self): + for f in os.listdir(mm_cfg.NEWSQUEUE_DIR): + os.unlink(os.path.join(mm_cfg.NEWSQUEUE_DIR, f)) + TestBase.tearDown(self) + + def test_short_circuit(self): + eq = self.assertEqual + mlist = self._mlist + mlist.gateway_to_news = 0 + ToUsenet.process(mlist, None, {}) + eq(len(self._sb.files()), 0) + mlist.gateway_to_news = 1 + ToUsenet.process(mlist, None, {'isdigest': 1}) + eq(len(self._sb.files()), 0) + ToUsenet.process(mlist, None, {'fromusenet': 1}) + eq(len(self._sb.files()), 0) + + def test_to_usenet(self): + # BAW: Should we, can we, test the error conditions that only log to a + # file instead of raising an exception? + eq = self.assertEqual + mlist = self._mlist + mlist.gateway_to_news = 1 + mlist.linked_newsgroup = 'foo' + mlist.nntp_host = 'bar' + msg = email.message_from_string("""\ +Subject: About Mailman + +Mailman rocks! +""") + ToUsenet.process(mlist, msg, {}) + files = self._sb.files() + eq(len(files), 1) + msg2, data = self._sb.dequeue(files[0]) + eq(msg.as_string(unixfrom=0), msg2.as_string(unixfrom=0)) + eq(data['version'], 3) + eq(data['listname'], '_xtest') + self.failUnless(data['received_time'] <= time.time()) + + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(unittest.makeSuite(TestAcknowledge)) + suite.addTest(unittest.makeSuite(TestAfterDelivery)) + suite.addTest(unittest.makeSuite(TestApprove)) + suite.addTest(unittest.makeSuite(TestCalcRecips)) + suite.addTest(unittest.makeSuite(TestCleanse)) + suite.addTest(unittest.makeSuite(TestCookHeaders)) + suite.addTest(unittest.makeSuite(TestDecorate)) + suite.addTest(unittest.makeSuite(TestFileRecips)) + suite.addTest(unittest.makeSuite(TestHold)) + suite.addTest(unittest.makeSuite(TestPersonalize)) + suite.addTest(unittest.makeSuite(TestReplybot)) + suite.addTest(unittest.makeSuite(TestSMTPDirect)) + suite.addTest(unittest.makeSuite(TestSendmail)) + suite.addTest(unittest.makeSuite(TestSpamDetect)) + suite.addTest(unittest.makeSuite(TestTagger)) + suite.addTest(unittest.makeSuite(TestToArchive)) + suite.addTest(unittest.makeSuite(TestToDigest)) + suite.addTest(unittest.makeSuite(TestToOutgoing)) + suite.addTest(unittest.makeSuite(TestToUsenet)) + return suite + + + +if __name__ == '__main__': + unittest.main(defaultTest='suite') |
