test_rfc.py

minimal python variant for small footprint apps like embedded apps

from test_support import verbose
import rfc822, sys
try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO

def test(msg, results):
    fp = StringIO()
    fp.write(msg)
    fp.seek(0)
    m = rfc822.Message(fp)
    i = 0
    for n, a in m.getaddrlist('to') + m.getaddrlist('cc'):
        if verbose:
            print 'name:', repr(n), 'addr:', repr(a)
        try:
            mn, ma = results[i][0], results[i][1]
        except IndexError:
            print 'extra parsed address:', repr(n), repr(a)
            continue
        i = i + 1
        if mn == n and ma == a:
            if verbose:
                print '    [matched]'
        else:
            if verbose:
                print '    [no match]'
            print 'not found:', repr(n), repr(a)

test('''Date:    Wed, 13 Jan 1999 23:57:35 -0500
From:    Guido van Rossum <[e-mail]>
To:      "Guido van
	 : Rossum" <[e-mail]>
Subject: test2

test2
''', [('Guido van\n	 : Rossum', '[e-mail]')])

test('''From: Barry <[e-mail]
To: [e-mail] (Guido: the Barbarian)
Subject: nonsense

test''', [('Guido: the Barbarian', '[e-mail]'),
          ])

test('''From: Barry <[e-mail]
To: [e-mail] (Guido: the Barbarian)
Cc: "Guido: the Madman" <[e-mail]>

test''', [('Guido: the Barbarian', '[e-mail]'),
          ('Guido: the Madman', '[e-mail]')
          ])

test('''To: "The monster with
     the very long name: Guido" <[e-mail]>

test''', [('The monster with\n     the very long name: Guido',
           '[e-mail]')])

test('''To: "Amit J. Patel" <[e-mail]>
CC: Mike Fletcher <[e-mail]>,
        "'[e-mail]'" <[e-mail]>
Cc: [e-mail], [e-mail]
Cc: [e-mail]

test''', [('Amit J. Patel', '[e-mail]'),
          ('Mike Fletcher', '[e-mail]'),
          ("'[e-mail]'", '[e-mail]'),
          ('', '[e-mail]'),
          ('', '[e-mail]'),
          ('', '[e-mail]'),
          ])

# This one is just twisted.  I don't know what the proper result should be,
# but it shouldn't be to infloop, which is what used to happen!
test('''To: <[smtp:[e-mail]][e-mail]>

test''', [('', ''),
          ('', '[e-mail]'),
          ('', '[e-mail]')
          ])