~abompard/mailman/import21

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
# Copyright (C) 2013 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 conf subcommand."""

from __future__ import absolute_import, print_function, unicode_literals

__metaclass__ = type
__all__ = [
    'TestConf',
    ]


import os
import sys
import mock
import tempfile
import unittest

from StringIO import StringIO
from mailman.commands.cli_conf import Conf
from mailman.testing.layers import ConfigLayer



class FakeArgs:
    section = None
    key = None
    output = None
    sort = False


class FakeParser:
    def __init__(self):
        self.message = None

    def error(self, message):
        self.message = message
        sys.exit(1)



class TestConf(unittest.TestCase):
    """Test the conf subcommand."""

    layer = ConfigLayer

    def setUp(self):
        self.command = Conf()
        self.command.parser = FakeParser()
        self.args = FakeArgs()

    def test_cannot_access_nonexistent_section(self):
        self.args.section = 'thissectiondoesnotexist'
        self.args.key = None
        with self.assertRaises(SystemExit):
            self.command.process(self.args)
        self.assertEqual(self.command.parser.message,
                         'No such section: thissectiondoesnotexist')

    def test_cannot_access_nonexistent_key(self):
        self.args.section = "mailman"
        self.args.key = 'thiskeydoesnotexist'
        with self.assertRaises(SystemExit):
            self.command.process(self.args)
        self.assertEqual(self.command.parser.message,
                         'Section mailman: No such key: thiskeydoesnotexist')

    def test_output_to_explicit_stdout(self):
        self.args.output = '-'
        self.args.section = 'shell'
        self.args.key = 'use_ipython'
        with mock.patch('sys.stdout') as mock_object:
            self.command.process(self.args)
        mock_object.write.assert_has_calls(
            [mock.call('no'), mock.call('\n')])

    def test_output_to_file(self):
        self.args.section = 'shell'
        self.args.key = 'use_ipython'
        fd, filename = tempfile.mkstemp()
        try:
            self.args.output = filename
            self.command.process(self.args)
            with open(filename, 'r') as fp:
                contents = fp.read()
        finally:
            os.remove(filename)
        self.assertEqual(contents, 'no\n')

    def test_sort_by_section(self):
        self.args.output = '-'
        self.args.sort = True
        output = StringIO()
        with mock.patch('sys.stdout', output):
            self.command.process(self.args)
        last_line = ''
        for line in output.getvalue().splitlines():
            if not line.startswith('['):
                # This is a continuation line.  --sort doesn't sort these.
                continue
            self.assertTrue(line > last_line,
                            '{} !> {}'.format(line, last_line))
            last_line = line