PageRenderTime 665ms CodeModel.GetById 171ms app.highlight 111ms RepoModel.GetById 176ms app.codeStats 0ms

/Lib/test/test_rfc822.py

http://unladen-swallow.googlecode.com/
Python | 257 lines | 250 code | 7 blank | 0 comment | 0 complexity | 85dba0c1311615feaea6e7580c7acd77 MD5 | raw file
  1import unittest
  2from test import test_support
  3
  4rfc822 = test_support.import_module("rfc822", deprecated=True)
  5
  6try:
  7    from cStringIO import StringIO
  8except ImportError:
  9    from StringIO import StringIO
 10
 11
 12class MessageTestCase(unittest.TestCase):
 13    def create_message(self, msg):
 14        return rfc822.Message(StringIO(msg))
 15
 16    def test_get(self):
 17        msg = self.create_message(
 18            'To: "last, first" <userid@foo.net>\n\ntest\n')
 19        self.assert_(msg.get("to") == '"last, first" <userid@foo.net>')
 20        self.assert_(msg.get("TO") == '"last, first" <userid@foo.net>')
 21        self.assert_(msg.get("No-Such-Header") is None)
 22        self.assert_(msg.get("No-Such-Header", "No-Such-Value")
 23                     == "No-Such-Value")
 24
 25    def test_setdefault(self):
 26        msg = self.create_message(
 27            'To: "last, first" <userid@foo.net>\n\ntest\n')
 28        self.assert_(not msg.has_key("New-Header"))
 29        self.assert_(msg.setdefault("New-Header", "New-Value") == "New-Value")
 30        self.assert_(msg.setdefault("New-Header", "Different-Value")
 31                     == "New-Value")
 32        self.assert_(msg["new-header"] == "New-Value")
 33
 34        self.assert_(msg.setdefault("Another-Header") == "")
 35        self.assert_(msg["another-header"] == "")
 36
 37    def check(self, msg, results):
 38        """Check addresses and the date."""
 39        m = self.create_message(msg)
 40        i = 0
 41        for n, a in m.getaddrlist('to') + m.getaddrlist('cc'):
 42            try:
 43                mn, ma = results[i][0], results[i][1]
 44            except IndexError:
 45                print 'extra parsed address:', repr(n), repr(a)
 46                continue
 47            i = i + 1
 48            self.assertEqual(mn, n,
 49                             "Un-expected name: %s != %s" % (`mn`, `n`))
 50            self.assertEqual(ma, a,
 51                             "Un-expected address: %s != %s" % (`ma`, `a`))
 52            if mn == n and ma == a:
 53                pass
 54            else:
 55                print 'not found:', repr(n), repr(a)
 56
 57        out = m.getdate('date')
 58        if out:
 59            self.assertEqual(out,
 60                             (1999, 1, 13, 23, 57, 35, 0, 1, 0),
 61                             "date conversion failed")
 62
 63
 64    # Note: all test cases must have the same date (in various formats),
 65    # or no date!
 66
 67    def test_basic(self):
 68        self.check(
 69            'Date:    Wed, 13 Jan 1999 23:57:35 -0500\n'
 70            'From:    Guido van Rossum <guido@CNRI.Reston.VA.US>\n'
 71            'To:      "Guido van\n'
 72            '\t : Rossum" <guido@python.org>\n'
 73            'Subject: test2\n'
 74            '\n'
 75            'test2\n',
 76            [('Guido van\n\t : Rossum', 'guido@python.org')])
 77
 78        self.check(
 79            'From: Barry <bwarsaw@python.org\n'
 80            'To: guido@python.org (Guido: the Barbarian)\n'
 81            'Subject: nonsense\n'
 82            'Date: Wednesday, January 13 1999 23:57:35 -0500\n'
 83            '\n'
 84            'test',
 85            [('Guido: the Barbarian', 'guido@python.org')])
 86
 87        self.check(
 88            'From: Barry <bwarsaw@python.org\n'
 89            'To: guido@python.org (Guido: the Barbarian)\n'
 90            'Cc: "Guido: the Madman" <guido@python.org>\n'
 91            'Date:  13-Jan-1999 23:57:35 EST\n'
 92            '\n'
 93            'test',
 94            [('Guido: the Barbarian', 'guido@python.org'),
 95             ('Guido: the Madman', 'guido@python.org')
 96             ])
 97
 98        self.check(
 99            'To: "The monster with\n'
100            '     the very long name: Guido" <guido@python.org>\n'
101            'Date:    Wed, 13 Jan 1999 23:57:35 -0500\n'
102            '\n'
103            'test',
104            [('The monster with\n     the very long name: Guido',
105              'guido@python.org')])
106
107        self.check(
108            'To: "Amit J. Patel" <amitp@Theory.Stanford.EDU>\n'
109            'CC: Mike Fletcher <mfletch@vrtelecom.com>,\n'
110            '        "\'string-sig@python.org\'" <string-sig@python.org>\n'
111            'Cc: fooz@bat.com, bart@toof.com\n'
112            'Cc: goit@lip.com\n'
113            'Date:    Wed, 13 Jan 1999 23:57:35 -0500\n'
114            '\n'
115            'test',
116            [('Amit J. Patel', 'amitp@Theory.Stanford.EDU'),
117             ('Mike Fletcher', 'mfletch@vrtelecom.com'),
118             ("'string-sig@python.org'", 'string-sig@python.org'),
119             ('', 'fooz@bat.com'),
120             ('', 'bart@toof.com'),
121             ('', 'goit@lip.com'),
122             ])
123
124        self.check(
125            'To: Some One <someone@dom.ain>\n'
126            'From: Anudder Persin <subuddy.else@dom.ain>\n'
127            'Date:\n'
128            '\n'
129            'test',
130            [('Some One', 'someone@dom.ain')])
131
132        self.check(
133            'To: person@dom.ain (User J. Person)\n\n',
134            [('User J. Person', 'person@dom.ain')])
135
136    def test_doublecomment(self):
137        # The RFC allows comments within comments in an email addr
138        self.check(
139            'To: person@dom.ain ((User J. Person)), John Doe <foo@bar.com>\n\n',
140            [('User J. Person', 'person@dom.ain'), ('John Doe', 'foo@bar.com')])
141
142    def test_twisted(self):
143        # This one is just twisted.  I don't know what the proper
144        # result should be, but it shouldn't be to infloop, which is
145        # what used to happen!
146        self.check(
147            'To: <[smtp:dd47@mail.xxx.edu]_at_hmhq@hdq-mdm1-imgout.companay.com>\n'
148            'Date:    Wed, 13 Jan 1999 23:57:35 -0500\n'
149            '\n'
150            'test',
151            [('', ''),
152             ('', 'dd47@mail.xxx.edu'),
153             ('', '_at_hmhq@hdq-mdm1-imgout.companay.com'),
154             ])
155
156    def test_commas_in_full_name(self):
157        # This exercises the old commas-in-a-full-name bug, which
158        # should be doing the right thing in recent versions of the
159        # module.
160        self.check(
161            'To: "last, first" <userid@foo.net>\n'
162            '\n'
163            'test',
164            [('last, first', 'userid@foo.net')])
165
166    def test_quoted_name(self):
167        self.check(
168            'To: (Comment stuff) "Quoted name"@somewhere.com\n'
169            '\n'
170            'test',
171            [('Comment stuff', '"Quoted name"@somewhere.com')])
172
173    def test_bogus_to_header(self):
174        self.check(
175            'To: :\n'
176            'Cc: goit@lip.com\n'
177            'Date:    Wed, 13 Jan 1999 23:57:35 -0500\n'
178            '\n'
179            'test',
180            [('', 'goit@lip.com')])
181
182    def test_addr_ipquad(self):
183        self.check(
184            'To: guido@[132.151.1.21]\n'
185            '\n'
186            'foo',
187            [('', 'guido@[132.151.1.21]')])
188
189    def test_iter(self):
190        m = rfc822.Message(StringIO(
191            'Date:    Wed, 13 Jan 1999 23:57:35 -0500\n'
192            'From:    Guido van Rossum <guido@CNRI.Reston.VA.US>\n'
193            'To:      "Guido van\n'
194            '\t : Rossum" <guido@python.org>\n'
195            'Subject: test2\n'
196            '\n'
197            'test2\n' ))
198        self.assertEqual(sorted(m), ['date', 'from', 'subject', 'to'])
199
200    def test_rfc2822_phrases(self):
201        # RFC 2822 (the update to RFC 822) specifies that dots in phrases are
202        # obsolete syntax, which conforming programs MUST recognize but NEVER
203        # generate (see $4.1 Miscellaneous obsolete tokens).  This is a
204        # departure from RFC 822 which did not allow dots in non-quoted
205        # phrases.
206        self.check('To: User J. Person <person@dom.ain>\n\n',
207                   [('User J. Person', 'person@dom.ain')])
208
209    # This takes too long to add to the test suite
210##    def test_an_excrutiatingly_long_address_field(self):
211##        OBSCENELY_LONG_HEADER_MULTIPLIER = 10000
212##        oneaddr = ('Person' * 10) + '@' + ('.'.join(['dom']*10)) + '.com'
213##        addr = ', '.join([oneaddr] * OBSCENELY_LONG_HEADER_MULTIPLIER)
214##        lst = rfc822.AddrlistClass(addr).getaddrlist()
215##        self.assertEqual(len(lst), OBSCENELY_LONG_HEADER_MULTIPLIER)
216
217    def test_2getaddrlist(self):
218        eq = self.assertEqual
219        msg = self.create_message("""\
220To: aperson@dom.ain
221Cc: bperson@dom.ain
222Cc: cperson@dom.ain
223Cc: dperson@dom.ain
224
225A test message.
226""")
227        ccs = [('', a) for a in
228               ['bperson@dom.ain', 'cperson@dom.ain', 'dperson@dom.ain']]
229        addrs = msg.getaddrlist('cc')
230        addrs.sort()
231        eq(addrs, ccs)
232        # Try again, this one used to fail
233        addrs = msg.getaddrlist('cc')
234        addrs.sort()
235        eq(addrs, ccs)
236
237    def test_parseaddr(self):
238        eq = self.assertEqual
239        eq(rfc822.parseaddr('<>'), ('', ''))
240        eq(rfc822.parseaddr('aperson@dom.ain'), ('', 'aperson@dom.ain'))
241        eq(rfc822.parseaddr('bperson@dom.ain (Bea A. Person)'),
242           ('Bea A. Person', 'bperson@dom.ain'))
243        eq(rfc822.parseaddr('Cynthia Person <cperson@dom.ain>'),
244           ('Cynthia Person', 'cperson@dom.ain'))
245
246    def test_quote_unquote(self):
247        eq = self.assertEqual
248        eq(rfc822.quote('foo\\wacky"name'), 'foo\\\\wacky\\"name')
249        eq(rfc822.unquote('"foo\\\\wacky\\"name"'), 'foo\\wacky"name')
250
251
252def test_main():
253    test_support.run_unittest(MessageTestCase)
254
255
256if __name__ == "__main__":
257    test_main()