~rashi007/mailman/docsfix

« back to all changes in this revision

Viewing changes to src/mailman/chains/tests/test_base.py

  • Committer: Barry Warsaw
  • Date: 2014-11-11 22:36:25 UTC
  • mfrom: (7257.1.2 coverage)
  • Revision ID: barry@list.org-20141111223625-uz607z1bhf6sflby
Add support for a 'coverage' environment.  e.g. `tox -e coverage` will run the
test suite and provide a code coverage report.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2014 by the Free Software Foundation, Inc.
 
2
#
 
3
# This file is part of GNU Mailman.
 
4
#
 
5
# GNU Mailman is free software: you can redistribute it and/or modify it under
 
6
# the terms of the GNU General Public License as published by the Free
 
7
# Software Foundation, either version 3 of the License, or (at your option)
 
8
# any later version.
 
9
#
 
10
# GNU Mailman is distributed in the hope that it will be useful, but WITHOUT
 
11
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 
12
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 
13
# more details.
 
14
#
 
15
# You should have received a copy of the GNU General Public License along with
 
16
# GNU Mailman.  If not, see <http://www.gnu.org/licenses/>.
 
17
 
 
18
"""Test the base chain stuff."""
 
19
 
 
20
from __future__ import absolute_import, print_function, unicode_literals
 
21
 
 
22
__metaclass__ = type
 
23
__all__ = [
 
24
    'TestMiscellaneous',
 
25
    ]
 
26
 
 
27
 
 
28
import unittest
 
29
 
 
30
from mailman.chains.accept import AcceptChain
 
31
from mailman.chains.base import Chain, Link
 
32
from mailman.interfaces.chain import LinkAction
 
33
from mailman.rules.any import Any
 
34
 
 
35
 
 
36
 
 
37
class TestMiscellaneous(unittest.TestCase):
 
38
    """Reach additional code coverage."""
 
39
 
 
40
    def test_link_repr(self):
 
41
        self.assertEqual(
 
42
            repr(Link(Any)), '<Link "if any then LinkAction.defer">')
 
43
 
 
44
    def test_link_repr_function(self):
 
45
        def function():
 
46
            pass
 
47
        self.assertEqual(
 
48
            repr(Link(Any, function=function)),
 
49
            '<Link "if any then LinkAction.defer" function()>')
 
50
 
 
51
    def test_link_repr_chain(self):
 
52
        self.assertEqual(
 
53
            repr(Link(Any, chain=AcceptChain)),
 
54
            '<Link "if any then LinkAction.defer" accept>')
 
55
 
 
56
    def test_link_repr_chain_and_function(self):
 
57
        def function():
 
58
            pass
 
59
        self.assertEqual(
 
60
            repr(Link(Any, chain=AcceptChain, function=function)),
 
61
            '<Link "if any then LinkAction.defer" accept function()>')
 
62
 
 
63
    def test_link_repr_chain_all(self):
 
64
        def function():
 
65
            pass
 
66
        self.assertEqual(
 
67
            repr(Link(Any, LinkAction.stop, AcceptChain, function)),
 
68
            '<Link "if any then LinkAction.stop" accept function()>')
 
69
 
 
70
    def test_flush(self):
 
71
        # Test that we can flush the links of a chain.
 
72
        chain = Chain('test', 'just a testing chain')
 
73
        chain.append_link(Link(Any))
 
74
        # Iterate over the links of the chain to prove there are some.
 
75
        count = sum(1 for link in chain.get_iterator())
 
76
        self.assertEqual(count, 1)
 
77
        # Flush the chain; then there will be no links.
 
78
        chain.flush()
 
79
        count = sum(1 for link in chain.get_iterator())
 
80
        self.assertEqual(count, 0)