Add tests for the new .get() and .setdefault() methods of rfc822.Message

objects.

1 files changed, 19 insertions, 1 deletions

diff --git a/Lib/test/test_rfc822.py b/Lib/test/test_rfc822.py
index 967b91b..e59d5b3 100644
--- a/Lib/test/test_rfc822.py
+++ b/Lib/test/test_rfc822.py
@@ -1,4 +1,4 @@
-from test_support import verbose
+from test_support import verbose, verify
 import rfc822, sys
 try:
     from cStringIO import StringIO
@@ -124,3 +124,21 @@ test''', [('', 'goit@lip.com')])
 test('''To: guido@[132.151.1.21]
 
 foo''', [('', 'guido@[132.151.1.21]')])
+
+
+msg = rfc822.Message(StringIO('''To: "last, first" <userid@foo.net>
+
+test
+'''))
+verify(msg.get("to") == '"last, first" <userid@foo.net>')
+verify(msg.get("TO") == '"last, first" <userid@foo.net>')
+verify(msg.get("No-Such-Header") is None)
+verify(msg.get("No-Such-Header", "No-Such-Value") == "No-Such-Value")
+
+verify(not msg.has_key("New-Header"))
+verify(msg.setdefault("New-Header", "New-Value") == "New-Value")
+verify(msg.setdefault("New-Header", "Different-Value") == "New-Value")
+verify(msg["new-header"] == "New-Value")
+
+verify(msg.setdefault("Another-Header") == "")
+verify(msg["another-header"] == "")