diff options
| author | Barry Warsaw | 2015-12-22 23:14:44 -0500 |
|---|---|---|
| committer | Barry Warsaw | 2015-12-22 23:14:44 -0500 |
| commit | a11e089cc1e0e5aff2502e584014295a414a43f9 (patch) | |
| tree | 2cd85ac193e1bcaf3d17582f0cbe8614b3932fe1 /src/mailman/commands/tests/test_members.py | |
| parent | ea1d7f360edd8f9ac70be5ed55caeaec27feb128 (diff) | |
| download | mailman-a11e089cc1e0e5aff2502e584014295a414a43f9.tar.gz mailman-a11e089cc1e0e5aff2502e584014295a414a43f9.tar.zst mailman-a11e089cc1e0e5aff2502e584014295a414a43f9.zip | |
Diffstat (limited to 'src/mailman/commands/tests/test_members.py')
| -rw-r--r-- | src/mailman/commands/tests/test_members.py | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/src/mailman/commands/tests/test_members.py b/src/mailman/commands/tests/test_members.py new file mode 100644 index 000000000..5afa2ca83 --- /dev/null +++ b/src/mailman/commands/tests/test_members.py @@ -0,0 +1,153 @@ +# Copyright (C) 2015 by the Free Software Foundation, Inc. +# +# This file is part of GNU Mailman. +# +# GNU Mailman 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 3 of the License, or (at your option) +# any later version. +# +# GNU Mailman 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 +# GNU Mailman. If not, see <http://www.gnu.org/licenses/>. + +"""Test the `mailman members` command.""" + +__all__ = [ + 'TestCLIMembers', + ] + + +import sys +import unittest + +from functools import partial +from io import StringIO +from mailman.app.lifecycle import create_list +from mailman.commands.cli_members import Members +from mailman.interfaces.member import MemberRole +from mailman.testing.helpers import subscribe +from mailman.testing.layers import ConfigLayer +from tempfile import NamedTemporaryFile +from unittest.mock import patch + + + +class FakeArgs: + input_filename = None + output_filename = None + role = None + regular = None + digest = None + nomail = None + list = None + + +class FakeParser: + def __init__(self): + self.message = None + + def error(self, message): + self.message = message + sys.exit(1) + + + +class TestCLIMembers(unittest.TestCase): + layer = ConfigLayer + + def setUp(self): + self._mlist = create_list('ant@example.com') + self.command = Members() + self.command.parser = FakeParser() + self.args = FakeArgs() + + def test_no_such_list(self): + self.args.list = ['bee.example.com'] + with self.assertRaises(SystemExit): + self.command.process(self.args) + self.assertEqual(self.command.parser.message, + 'No such list: bee.example.com') + + def test_bad_delivery_status(self): + self.args.list = ['ant.example.com'] + self.args.nomail = 'bogus' + with self.assertRaises(SystemExit): + self.command.process(self.args) + self.assertEqual(self.command.parser.message, + 'Unknown delivery status: bogus') + + def test_role_administrator(self): + subscribe(self._mlist, 'Anne', role=MemberRole.owner) + subscribe(self._mlist, 'Bart', role=MemberRole.moderator) + subscribe(self._mlist, 'Cate', role=MemberRole.nonmember) + subscribe(self._mlist, 'Dave', role=MemberRole.member) + self.args.list = ['ant.example.com'] + self.args.role = 'administrator' + with NamedTemporaryFile('w', encoding='utf-8') as outfp: + self.args.output_filename = outfp.name + self.command.process(self.args) + with open(outfp.name, 'r', encoding='utf-8') as infp: + lines = infp.readlines() + self.assertEqual(len(lines), 2) + self.assertEqual(lines[0], 'Anne Person <aperson@example.com>\n') + self.assertEqual(lines[1], 'Bart Person <bperson@example.com>\n') + + def test_role_any(self): + subscribe(self._mlist, 'Anne', role=MemberRole.owner) + subscribe(self._mlist, 'Bart', role=MemberRole.moderator) + subscribe(self._mlist, 'Cate', role=MemberRole.nonmember) + subscribe(self._mlist, 'Dave', role=MemberRole.member) + self.args.list = ['ant.example.com'] + self.args.role = 'any' + with NamedTemporaryFile('w', encoding='utf-8') as outfp: + self.args.output_filename = outfp.name + self.command.process(self.args) + with open(outfp.name, 'r', encoding='utf-8') as infp: + lines = infp.readlines() + self.assertEqual(len(lines), 4) + self.assertEqual(lines[0], 'Anne Person <aperson@example.com>\n') + self.assertEqual(lines[1], 'Bart Person <bperson@example.com>\n') + self.assertEqual(lines[2], 'Cate Person <cperson@example.com>\n') + self.assertEqual(lines[3], 'Dave Person <dperson@example.com>\n') + + def test_role_moderator(self): + subscribe(self._mlist, 'Anne', role=MemberRole.owner) + subscribe(self._mlist, 'Bart', role=MemberRole.moderator) + subscribe(self._mlist, 'Cate', role=MemberRole.nonmember) + subscribe(self._mlist, 'Dave', role=MemberRole.member) + self.args.list = ['ant.example.com'] + self.args.role = 'moderator' + with NamedTemporaryFile('w', encoding='utf-8') as outfp: + self.args.output_filename = outfp.name + self.command.process(self.args) + with open(outfp.name, 'r', encoding='utf-8') as infp: + lines = infp.readlines() + self.assertEqual(len(lines), 1) + self.assertEqual(lines[0], 'Bart Person <bperson@example.com>\n') + + def test_bad_role(self): + self.args.list = ['ant.example.com'] + self.args.role = 'bogus' + with self.assertRaises(SystemExit): + self.command.process(self.args) + self.assertEqual(self.command.parser.message, + 'Unknown member role: bogus') + + def test_already_subscribed_with_display_name(self): + subscribe(self._mlist, 'Anne') + outfp = StringIO() + with NamedTemporaryFile('w', buffering=1, encoding='utf-8') as infp: + print('Anne Person <aperson@example.com>', file=infp) + self.args.list = ['ant.example.com'] + self.args.input_filename = infp.name + with patch('builtins.print', partial(print, file=outfp)): + self.command.process(self.args) + self.assertEqual( + outfp.getvalue(), + 'Already subscribed (skipping): Anne Person <aperson@example.com>\n' + ) |
