From 58fb61cce5f989ea4b3cc48ce104235b8eb5965d Mon Sep 17 00:00:00 2001 From: Barry Warsaw Date: Fri, 6 Sep 2002 03:39:59 +0000 Subject: [PATCH] test_replace_header(): New test for Message.replace_header(). --- Lib/email/test/test_email.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/Lib/email/test/test_email.py b/Lib/email/test/test_email.py index 5012d43703c..bd075ce5bf1 100644 --- a/Lib/email/test/test_email.py +++ b/Lib/email/test/test_email.py @@ -435,6 +435,23 @@ class TestMessageAPI(TestEmailBase): msg['Content-Type'] = 'no-slash-in-this-string' self.assertEqual(msg.get_content_subtype(), 'plain') + def test_replace_header(self): + eq = self.assertEqual + msg = Message() + msg.add_header('First', 'One') + msg.add_header('Second', 'Two') + msg.add_header('Third', 'Three') + eq(msg.keys(), ['First', 'Second', 'Third']) + eq(msg.values(), ['One', 'Two', 'Three']) + msg.replace_header('Second', 'Twenty') + eq(msg.keys(), ['First', 'Second', 'Third']) + eq(msg.values(), ['One', 'Twenty', 'Three']) + msg.add_header('First', 'Eleven') + msg.replace_header('First', 'One Hundred') + eq(msg.keys(), ['First', 'Second', 'Third', 'First']) + eq(msg.values(), ['One Hundred', 'Twenty', 'Three', 'Eleven']) + self.assertRaises(KeyError, msg.replace_header, 'Fourth', 'Missing') + # Test the email.Encoders module