diff options
Diffstat (limited to 'src/mailman/chains/tests/test_headers.py')
| -rw-r--r-- | src/mailman/chains/tests/test_headers.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/mailman/chains/tests/test_headers.py b/src/mailman/chains/tests/test_headers.py index d42baa55e..c2e01e5e3 100644 --- a/src/mailman/chains/tests/test_headers.py +++ b/src/mailman/chains/tests/test_headers.py @@ -29,6 +29,7 @@ from mailman.chains.headers import HeaderMatchRule, make_link from mailman.config import config from mailman.core.chains import process from mailman.email.message import Message +from mailman.interfaces.action import Action from mailman.interfaces.chain import LinkAction, HoldEvent from mailman.interfaces.mailinglist import IHeaderMatchList from mailman.testing.helpers import ( @@ -160,9 +161,9 @@ class TestHeaderChain(unittest.TestCase): # properly. chain = config.chains['header-match'] header_matches = IHeaderMatchList(self._mlist) - header_matches.append('Foo', 'a+', 'reject') - header_matches.append('Bar', 'b+', 'discard') - header_matches.append('Baz', 'z+', 'accept') + header_matches.append('Foo', 'a+', Action.reject) + header_matches.append('Bar', 'b+', Action.discard) + header_matches.append('Baz', 'z+', Action.accept) links = [link for link in chain.get_links(self._mlist, Message(), {}) if link.rule.name != 'any'] self.assertEqual(len(links), 3) @@ -193,7 +194,7 @@ A message body. """) msgdata = {} header_matches = IHeaderMatchList(self._mlist) - header_matches.append('Foo', 'foo', 'accept') + header_matches.append('Foo', 'foo', Action.accept) # This event subscriber records the event that occurs when the message # is processed by the owner chain. events = [] |