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

freddrake · freddrake · commit cf71fefa59f5 · 2001-05-22T15:02:19.000Z
objects.
diff --git 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 @@ def test(msg, results):
 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"] == "")
