mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-26 18:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Re-write the rfc822 tests to use PyUnit.

Browse source 
Update to reflect using "" as the default value for the second parameter
to the get() method.
This commit is contained in:
Fred Drake 2001-05-22 19:38:31 +00:00
parent 233226e977
commit 876dc70b2e
1 changed files with 152 additions and 128 deletions
Download patch file Download diff file Expand all files Collapse all files

222
Lib/test/test_rfc822.py
View file

@ -1,20 +1,44 @@
from test_support import verbose, verify import rfc822
import rfc822, sys import sys
import test_support
import unittest
try: try:
from cStringIO import StringIO from cStringIO import StringIO
except ImportError: except ImportError:
from StringIO import StringIO from StringIO import StringIO
def test(msg, results):
fp = StringIO()
fp.write(msg)
fp.seek(0)
m = rfc822.Message(fp)
i = 0
class MessageTestCase(unittest.TestCase):
def create_message(self, msg):
return rfc822.Message(StringIO(msg))
def test_get(self):
msg = self.create_message(
'To: "last, first" <userid@foo.net>\n\ntest\n')
self.assert_(msg.get("to") == '"last, first" <userid@foo.net>')
self.assert_(msg.get("TO") == '"last, first" <userid@foo.net>')
self.assert_(msg.get("No-Such-Header") == "")
self.assert_(msg.get("No-Such-Header", "No-Such-Value")
== "No-Such-Value")
def test_setdefault(self):
msg = self.create_message(
'To: "last, first" <userid@foo.net>\n\ntest\n')
self.assert_(not msg.has_key("New-Header"))
self.assert_(msg.setdefault("New-Header", "New-Value") == "New-Value")
self.assert_(msg.setdefault("New-Header", "Different-Value")
== "New-Value")
self.assert_(msg["new-header"] == "New-Value")
self.assert_(msg.setdefault("Another-Header") == "")
self.assert_(msg["another-header"] == "")
def check(self, msg, results):
"""Check addresses and the date."""
m = self.create_message(msg)
i = 0
for n, a in m.getaddrlist('to') + m.getaddrlist('cc'): for n, a in m.getaddrlist('to') + m.getaddrlist('cc'):
if verbose:
print 'name:', repr(n), 'addr:', repr(a)
try: try:
mn, ma = results[i][0], results[i][1] mn, ma = results[i][0], results[i][1]
except IndexError: except IndexError:
@ -22,69 +46,70 @@ def test(msg, results):
continue continue
i = i + 1 i = i + 1
if mn == n and ma == a: if mn == n and ma == a:
if verbose: pass
print ' [matched]'
else: else:
if verbose:
print ' [no match]'
print 'not found:', repr(n), repr(a) print 'not found:', repr(n), repr(a)
out = m.getdate('date') out = m.getdate('date')
if out: if out:
if verbose: self.assertEqual(out,
print 'Date:', m.getheader('date') (1999, 1, 13, 23, 57, 35, 0, 0, 0),
if out == (1999, 1, 13, 23, 57, 35, 0, 0, 0): "date conversion failed")
if verbose:
print ' [matched]'
else:
if verbose:
print ' [no match]'
print 'Date conversion failed:', out
# Note: all test cases must have the same date (in various formats), # Note: all test cases must have the same date (in various formats),
# or no date! # or no date!
test('''Date: Wed, 13 Jan 1999 23:57:35 -0500 def test_basic(self):
From: Guido van Rossum <guido@CNRI.Reston.VA.US> self.check(
To: "Guido van 'Date: Wed, 13 Jan 1999 23:57:35 -0500\n'
\t : Rossum" <guido@python.org> 'From: Guido van Rossum <guido@CNRI.Reston.VA.US>\n'
Subject: test2 'To: "Guido van\n'
'\t : Rossum" <guido@python.org>\n'
'Subject: test2\n'
'\n'
'test2\n',
[('Guido van\n\t : Rossum', 'guido@python.org')])
test2 self.check(
''', [('Guido van\n\t : Rossum', 'guido@python.org')]) 'From: Barry <bwarsaw@python.org\n'
'To: guido@python.org (Guido: the Barbarian)\n'
'Subject: nonsense\n'
'Date: Wednesday, January 13 1999 23:57:35 -0500\n'
'\n'
'test',
[('Guido: the Barbarian', 'guido@python.org')])
test('''From: Barry <bwarsaw@python.org self.check(
To: guido@python.org (Guido: the Barbarian) 'From: Barry <bwarsaw@python.org\n'
Subject: nonsense 'To: guido@python.org (Guido: the Barbarian)\n'
Date: Wednesday, January 13 1999 23:57:35 -0500 'Cc: "Guido: the Madman" <guido@python.org>\n'
'Date: 13-Jan-1999 23:57:35 EST\n'
test''', [('Guido: the Barbarian', 'guido@python.org'), '\n'
]) 'test',
[('Guido: the Barbarian', 'guido@python.org'),
test('''From: Barry <bwarsaw@python.org
To: guido@python.org (Guido: the Barbarian)
Cc: "Guido: the Madman" <guido@python.org>
Date: 13-Jan-1999 23:57:35 EST
test''', [('Guido: the Barbarian', 'guido@python.org'),
('Guido: the Madman', 'guido@python.org') ('Guido: the Madman', 'guido@python.org')
]) ])
test('''To: "The monster with self.check(
the very long name: Guido" <guido@python.org> 'To: "The monster with\n'
Date: Wed, 13 Jan 1999 23:57:35 -0500 ' the very long name: Guido" <guido@python.org>\n'
'Date: Wed, 13 Jan 1999 23:57:35 -0500\n'
test''', [('The monster with\n the very long name: Guido', '\n'
'test',
[('The monster with\n the very long name: Guido',
'guido@python.org')]) 'guido@python.org')])
test('''To: "Amit J. Patel" <amitp@Theory.Stanford.EDU> self.check(
CC: Mike Fletcher <mfletch@vrtelecom.com>, 'To: "Amit J. Patel" <amitp@Theory.Stanford.EDU>\n'
"'string-sig@python.org'" <string-sig@python.org> 'CC: Mike Fletcher <mfletch@vrtelecom.com>,\n'
Cc: fooz@bat.com, bart@toof.com ' "\'string-sig@python.org\'" <string-sig@python.org>\n'
Cc: goit@lip.com 'Cc: fooz@bat.com, bart@toof.com\n'
Date: Wed, 13 Jan 1999 23:57:35 -0500 'Cc: goit@lip.com\n'
'Date: Wed, 13 Jan 1999 23:57:35 -0500\n'
test''', [('Amit J. Patel', 'amitp@Theory.Stanford.EDU'), '\n'
'test',
[('Amit J. Patel', 'amitp@Theory.Stanford.EDU'),
('Mike Fletcher', 'mfletch@vrtelecom.com'), ('Mike Fletcher', 'mfletch@vrtelecom.com'),
("'string-sig@python.org'", 'string-sig@python.org'), ("'string-sig@python.org'", 'string-sig@python.org'),
('', 'fooz@bat.com'), ('', 'fooz@bat.com'),
@ -92,53 +117,52 @@ test''', [('Amit J. Patel', 'amitp@Theory.Stanford.EDU'),
('', 'goit@lip.com'), ('', 'goit@lip.com'),
]) ])
# This one is just twisted. I don't know what the proper result should be, def test_twisted(self):
# but it shouldn't be to infloop, which is what used to happen! # This one is just twisted. I don't know what the proper
test('''To: <[smtp:dd47@mail.xxx.edu]_at_hmhq@hdq-mdm1-imgout.companay.com> # result should be, but it shouldn't be to infloop, which is
Date: Wed, 13 Jan 1999 23:57:35 -0500 # what used to happen!
self.check(
test''', [('', ''), 'To: <[smtp:dd47@mail.xxx.edu]_at_hmhq@hdq-mdm1-imgout.companay.com>\n'
'Date: Wed, 13 Jan 1999 23:57:35 -0500\n'
'\n'
'test',
[('', ''),
('', 'dd47@mail.xxx.edu'), ('', 'dd47@mail.xxx.edu'),
('', '_at_hmhq@hdq-mdm1-imgout.companay.com') ('', '_at_hmhq@hdq-mdm1-imgout.companay.com'),
]) ])
# This exercises the old commas-in-a-full-name bug, which should be doing the def test_commas_in_full_name(self):
# right thing in recent versions of the module. # This exercises the old commas-in-a-full-name bug, which
test('''To: "last, first" <userid@foo.net> # should be doing the right thing in recent versions of the
# module.
self.check(
'To: "last, first" <userid@foo.net>\n'
'\n'
'test',
[('last, first', 'userid@foo.net')])
test''', [('last, first', 'userid@foo.net'), def test_quoted_name(self):
]) self.check(
'To: (Comment stuff) "Quoted name"@somewhere.com\n'
'\n'
'test',
[('Comment stuff', '"Quoted name"@somewhere.com')])
test('''To: (Comment stuff) "Quoted name"@somewhere.com def test_bogus_to_header(self):
self.check(
'To: :\n'
'Cc: goit@lip.com\n'
'Date: Wed, 13 Jan 1999 23:57:35 -0500\n'
'\n'
'test',
[('', 'goit@lip.com')])
test''', [('Comment stuff', '"Quoted name"@somewhere.com'), def test_addr_ipquad(self):
]) self.check(
'To: guido@[132.151.1.21]\n'
test('''To: : '\n'
Cc: goit@lip.com 'foo',
Date: Wed, 13 Jan 1999 23:57:35 -0500 [('', 'guido@[132.151.1.21]')])
test''', [('', 'goit@lip.com')])
test('''To: guido@[132.151.1.21] test_support.run_unittest(MessageTestCase)
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"] == "")