Handle folder headers with colons correctly. Based on patch from Johannes Berg.

[eoc.git] / eocTests.py

diff --git a/eocTests.py b/eocTests.py
index acf99b326b00fe2b2c32df608606e852201fbb12..1ecbfb1af9093d1638b2a8cde3650c4a138487c8 100644 (file)
--- a/eocTests.py
+++ b/eocTests.py
@@ -244,6 +244,30 @@ class DotDirTestCases(unittest.TestCase):
         mlm = eoc.MailingListManager(DOTDIR)
         self.dotdir_is_ok()
 
+
+class RemoveSomeHeadersTest(unittest.TestCase):
+
+    def testRemoveSomeHeaders(self):
+        mlm = eoc.MailingListManager(DOTDIR)
+        ml = eoc.MailingList(mlm, "list@example.com")
+        mail = """\
+Header-1: this is a simple header
+Header-2: this
+    is
+    a
+    complex header with a colon: yes it is
+Header-3: odd numbered headers are simple
+
+Body.
+"""
+        mail2 = ml.remove_some_headers(mail, ["Header-2"])
+        self.failUnlessEqual(mail2, """\
+Header-1: this is a simple header
+Header-3: odd numbered headers are simple
+
+Body.
+""")
+
 class ListBase(unittest.TestCase):
 
     def setUp(self):