home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2010 November / maximum-cd-2010-11.iso / DiscContents / calibre-0.7.13.msi / file_449 (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2010-08-06  |  125.0 KB  |  2,741 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.6)
  3.  
  4. import os
  5. import sys
  6. import time
  7. import base64
  8. import difflib
  9. import unittest
  10. import warnings
  11. from cStringIO import StringIO
  12. import email
  13. from email.charset import Charset
  14. from email.header import Header, decode_header, make_header
  15. from email.parser import Parser, HeaderParser
  16. from email.generator import Generator, DecodedGenerator
  17. from email.message import Message
  18. from email.mime.application import MIMEApplication
  19. from email.mime.audio import MIMEAudio
  20. from email.mime.text import MIMEText
  21. from email.mime.image import MIMEImage
  22. from email.mime.base import MIMEBase
  23. from email.mime.message import MIMEMessage
  24. from email.mime.multipart import MIMEMultipart
  25. from email import utils
  26. from email import errors
  27. from email import encoders
  28. from email import iterators
  29. from email import base64mime
  30. from email import quoprimime
  31. from test.test_support import findfile, run_unittest
  32. from email.test import __file__ as landmark
  33. NL = '\n'
  34. EMPTYSTRING = ''
  35. SPACE = ' '
  36.  
  37. def openfile(filename, mode = 'r'):
  38.     path = os.path.join(os.path.dirname(landmark), 'data', filename)
  39.     return open(path, mode)
  40.  
  41.  
  42. class TestEmailBase(unittest.TestCase):
  43.     
  44.     def ndiffAssertEqual(self, first, second):
  45.         if first != second:
  46.             sfirst = str(first)
  47.             ssecond = str(second)
  48.             diff = difflib.ndiff(sfirst.splitlines(), ssecond.splitlines())
  49.             fp = StringIO()
  50.             print >>fp, NL, NL.join(diff)
  51.             raise self.failureException, fp.getvalue()
  52.         first != second
  53.  
  54.     
  55.     def _msgobj(self, filename):
  56.         fp = openfile(findfile(filename))
  57.         
  58.         try:
  59.             msg = email.message_from_file(fp)
  60.         finally:
  61.             fp.close()
  62.  
  63.         return msg
  64.  
  65.  
  66.  
  67. class TestMessageAPI(TestEmailBase):
  68.     
  69.     def test_get_all(self):
  70.         eq = self.assertEqual
  71.         msg = self._msgobj('msg_20.txt')
  72.         eq(msg.get_all('cc'), [
  73.             'ccc@zzz.org',
  74.             'ddd@zzz.org',
  75.             'eee@zzz.org'])
  76.         eq(msg.get_all('xx', 'n/a'), 'n/a')
  77.  
  78.     
  79.     def test_getset_charset(self):
  80.         eq = self.assertEqual
  81.         msg = Message()
  82.         eq(msg.get_charset(), None)
  83.         charset = Charset('iso-8859-1')
  84.         msg.set_charset(charset)
  85.         eq(msg['mime-version'], '1.0')
  86.         eq(msg.get_content_type(), 'text/plain')
  87.         eq(msg['content-type'], 'text/plain; charset="iso-8859-1"')
  88.         eq(msg.get_param('charset'), 'iso-8859-1')
  89.         eq(msg['content-transfer-encoding'], 'quoted-printable')
  90.         eq(msg.get_charset().input_charset, 'iso-8859-1')
  91.         msg.set_charset(None)
  92.         eq(msg.get_charset(), None)
  93.         eq(msg['content-type'], 'text/plain')
  94.         msg = Message()
  95.         msg['MIME-Version'] = '2.0'
  96.         msg['Content-Type'] = 'text/x-weird'
  97.         msg['Content-Transfer-Encoding'] = 'quinted-puntable'
  98.         msg.set_charset(charset)
  99.         eq(msg['mime-version'], '2.0')
  100.         eq(msg['content-type'], 'text/x-weird; charset="iso-8859-1"')
  101.         eq(msg['content-transfer-encoding'], 'quinted-puntable')
  102.  
  103.     
  104.     def test_set_charset_from_string(self):
  105.         eq = self.assertEqual
  106.         msg = Message()
  107.         msg.set_charset('us-ascii')
  108.         eq(msg.get_charset().input_charset, 'us-ascii')
  109.         eq(msg['content-type'], 'text/plain; charset="us-ascii"')
  110.  
  111.     
  112.     def test_set_payload_with_charset(self):
  113.         msg = Message()
  114.         charset = Charset('iso-8859-1')
  115.         msg.set_payload('This is a string payload', charset)
  116.         self.assertEqual(msg.get_charset().input_charset, 'iso-8859-1')
  117.  
  118.     
  119.     def test_get_charsets(self):
  120.         eq = self.assertEqual
  121.         msg = self._msgobj('msg_08.txt')
  122.         charsets = msg.get_charsets()
  123.         eq(charsets, [
  124.             None,
  125.             'us-ascii',
  126.             'iso-8859-1',
  127.             'iso-8859-2',
  128.             'koi8-r'])
  129.         msg = self._msgobj('msg_09.txt')
  130.         charsets = msg.get_charsets('dingbat')
  131.         eq(charsets, [
  132.             'dingbat',
  133.             'us-ascii',
  134.             'iso-8859-1',
  135.             'dingbat',
  136.             'koi8-r'])
  137.         msg = self._msgobj('msg_12.txt')
  138.         charsets = msg.get_charsets()
  139.         eq(charsets, [
  140.             None,
  141.             'us-ascii',
  142.             'iso-8859-1',
  143.             None,
  144.             'iso-8859-2',
  145.             'iso-8859-3',
  146.             'us-ascii',
  147.             'koi8-r'])
  148.  
  149.     
  150.     def test_get_filename(self):
  151.         eq = self.assertEqual
  152.         msg = self._msgobj('msg_04.txt')
  153.         filenames = [ p.get_filename() for p in msg.get_payload() ]
  154.         eq(filenames, [
  155.             'msg.txt',
  156.             'msg.txt'])
  157.         msg = self._msgobj('msg_07.txt')
  158.         subpart = msg.get_payload(1)
  159.         eq(subpart.get_filename(), 'dingusfish.gif')
  160.  
  161.     
  162.     def test_get_filename_with_name_parameter(self):
  163.         eq = self.assertEqual
  164.         msg = self._msgobj('msg_44.txt')
  165.         filenames = [ p.get_filename() for p in msg.get_payload() ]
  166.         eq(filenames, [
  167.             'msg.txt',
  168.             'msg.txt'])
  169.  
  170.     
  171.     def test_get_boundary(self):
  172.         eq = self.assertEqual
  173.         msg = self._msgobj('msg_07.txt')
  174.         eq(msg.get_boundary(), 'BOUNDARY')
  175.  
  176.     
  177.     def test_set_boundary(self):
  178.         eq = self.assertEqual
  179.         msg = self._msgobj('msg_01.txt')
  180.         msg.set_boundary('BOUNDARY')
  181.         (header, value) = msg.items()[4]
  182.         eq(header.lower(), 'content-type')
  183.         eq(value, 'text/plain; charset="us-ascii"; boundary="BOUNDARY"')
  184.         msg = self._msgobj('msg_04.txt')
  185.         msg.set_boundary('BOUNDARY')
  186.         (header, value) = msg.items()[4]
  187.         eq(header.lower(), 'content-type')
  188.         eq(value, 'multipart/mixed; boundary="BOUNDARY"')
  189.         msg = self._msgobj('msg_03.txt')
  190.         self.assertRaises(errors.HeaderParseError, msg.set_boundary, 'BOUNDARY')
  191.  
  192.     
  193.     def test_get_decoded_payload(self):
  194.         eq = self.assertEqual
  195.         msg = self._msgobj('msg_10.txt')
  196.         eq(msg.get_payload(decode = True), None)
  197.         eq(msg.get_payload(0).get_payload(decode = True), 'This is a 7bit encoded message.\n')
  198.         eq(msg.get_payload(1).get_payload(decode = True), '\xa1This is a Quoted Printable encoded message!\n')
  199.         eq(msg.get_payload(2).get_payload(decode = True), 'This is a Base64 encoded message.')
  200.         eq(msg.get_payload(3).get_payload(decode = True), 'This has no Content-Transfer-Encoding: header.\n')
  201.  
  202.     
  203.     def test_get_decoded_uu_payload(self):
  204.         eq = self.assertEqual
  205.         msg = Message()
  206.         msg.set_payload('begin 666 -\n+:&5L;&\\@=V]R;&0 \n \nend\n')
  207.         for cte in ('x-uuencode', 'uuencode', 'uue', 'x-uue'):
  208.             msg['content-transfer-encoding'] = cte
  209.             eq(msg.get_payload(decode = True), 'hello world')
  210.         
  211.         msg.set_payload('foo')
  212.         eq(msg.get_payload(decode = True), 'foo')
  213.  
  214.     
  215.     def test_decoded_generator(self):
  216.         eq = self.assertEqual
  217.         msg = self._msgobj('msg_07.txt')
  218.         fp = openfile('msg_17.txt')
  219.         
  220.         try:
  221.             text = fp.read()
  222.         finally:
  223.             fp.close()
  224.  
  225.         s = StringIO()
  226.         g = DecodedGenerator(s)
  227.         g.flatten(msg)
  228.         eq(s.getvalue(), text)
  229.  
  230.     
  231.     def test__contains__(self):
  232.         msg = Message()
  233.         msg['From'] = 'Me'
  234.         msg['to'] = 'You'
  235.         self.failUnless('from' in msg)
  236.         self.failUnless('From' in msg)
  237.         self.failUnless('FROM' in msg)
  238.         self.failUnless('to' in msg)
  239.         self.failUnless('To' in msg)
  240.         self.failUnless('TO' in msg)
  241.  
  242.     
  243.     def test_as_string(self):
  244.         eq = self.assertEqual
  245.         msg = self._msgobj('msg_01.txt')
  246.         fp = openfile('msg_01.txt')
  247.         
  248.         try:
  249.             text = fp.read()
  250.         finally:
  251.             fp.close()
  252.  
  253.         eq(text, msg.as_string())
  254.         fullrepr = str(msg)
  255.         lines = fullrepr.split('\n')
  256.         self.failUnless(lines[0].startswith('From '))
  257.         eq(text, NL.join(lines[1:]))
  258.  
  259.     
  260.     def test_bad_param(self):
  261.         msg = email.message_from_string('Content-Type: blarg; baz; boo\n')
  262.         self.assertEqual(msg.get_param('baz'), '')
  263.  
  264.     
  265.     def test_missing_filename(self):
  266.         msg = email.message_from_string('From: foo\n')
  267.         self.assertEqual(msg.get_filename(), None)
  268.  
  269.     
  270.     def test_bogus_filename(self):
  271.         msg = email.message_from_string('Content-Disposition: blarg; filename\n')
  272.         self.assertEqual(msg.get_filename(), '')
  273.  
  274.     
  275.     def test_missing_boundary(self):
  276.         msg = email.message_from_string('From: foo\n')
  277.         self.assertEqual(msg.get_boundary(), None)
  278.  
  279.     
  280.     def test_get_params(self):
  281.         eq = self.assertEqual
  282.         msg = email.message_from_string('X-Header: foo=one; bar=two; baz=three\n')
  283.         eq(msg.get_params(header = 'x-header'), [
  284.             ('foo', 'one'),
  285.             ('bar', 'two'),
  286.             ('baz', 'three')])
  287.         msg = email.message_from_string('X-Header: foo; bar=one; baz=two\n')
  288.         eq(msg.get_params(header = 'x-header'), [
  289.             ('foo', ''),
  290.             ('bar', 'one'),
  291.             ('baz', 'two')])
  292.         eq(msg.get_params(), None)
  293.         msg = email.message_from_string('X-Header: foo; bar="one"; baz=two\n')
  294.         eq(msg.get_params(header = 'x-header'), [
  295.             ('foo', ''),
  296.             ('bar', 'one'),
  297.             ('baz', 'two')])
  298.  
  299.     
  300.     def test_get_param_liberal(self):
  301.         msg = Message()
  302.         msg['Content-Type'] = 'Content-Type: Multipart/mixed; boundary = "CPIMSSMTPC06p5f3tG"'
  303.         self.assertEqual(msg.get_param('boundary'), 'CPIMSSMTPC06p5f3tG')
  304.  
  305.     
  306.     def test_get_param(self):
  307.         eq = self.assertEqual
  308.         msg = email.message_from_string('X-Header: foo=one; bar=two; baz=three\n')
  309.         eq(msg.get_param('bar', header = 'x-header'), 'two')
  310.         eq(msg.get_param('quuz', header = 'x-header'), None)
  311.         eq(msg.get_param('quuz'), None)
  312.         msg = email.message_from_string('X-Header: foo; bar="one"; baz=two\n')
  313.         eq(msg.get_param('foo', header = 'x-header'), '')
  314.         eq(msg.get_param('bar', header = 'x-header'), 'one')
  315.         eq(msg.get_param('baz', header = 'x-header'), 'two')
  316.  
  317.     
  318.     def test_get_param_funky_continuation_lines(self):
  319.         msg = self._msgobj('msg_22.txt')
  320.         self.assertEqual(msg.get_payload(1).get_param('name'), 'wibble.JPG')
  321.  
  322.     
  323.     def test_get_param_with_semis_in_quotes(self):
  324.         msg = email.message_from_string('Content-Type: image/pjpeg; name="Jim&&Jill"\n')
  325.         self.assertEqual(msg.get_param('name'), 'Jim&&Jill')
  326.         self.assertEqual(msg.get_param('name', unquote = False), '"Jim&&Jill"')
  327.  
  328.     
  329.     def test_has_key(self):
  330.         msg = email.message_from_string('Header: exists')
  331.         self.failUnless(msg.has_key('header'))
  332.         self.failUnless(msg.has_key('Header'))
  333.         self.failUnless(msg.has_key('HEADER'))
  334.         self.failIf(msg.has_key('headeri'))
  335.  
  336.     
  337.     def test_set_param(self):
  338.         eq = self.assertEqual
  339.         msg = Message()
  340.         msg.set_param('charset', 'iso-2022-jp')
  341.         eq(msg.get_param('charset'), 'iso-2022-jp')
  342.         msg.set_param('importance', 'high value')
  343.         eq(msg.get_param('importance'), 'high value')
  344.         eq(msg.get_param('importance', unquote = False), '"high value"')
  345.         eq(msg.get_params(), [
  346.             ('text/plain', ''),
  347.             ('charset', 'iso-2022-jp'),
  348.             ('importance', 'high value')])
  349.         eq(msg.get_params(unquote = False), [
  350.             ('text/plain', ''),
  351.             ('charset', '"iso-2022-jp"'),
  352.             ('importance', '"high value"')])
  353.         msg.set_param('charset', 'iso-9999-xx', header = 'X-Jimmy')
  354.         eq(msg.get_param('charset', header = 'X-Jimmy'), 'iso-9999-xx')
  355.  
  356.     
  357.     def test_del_param(self):
  358.         eq = self.assertEqual
  359.         msg = self._msgobj('msg_05.txt')
  360.         eq(msg.get_params(), [
  361.             ('multipart/report', ''),
  362.             ('report-type', 'delivery-status'),
  363.             ('boundary', 'D1690A7AC1.996856090/mail.example.com')])
  364.         old_val = msg.get_param('report-type')
  365.         msg.del_param('report-type')
  366.         eq(msg.get_params(), [
  367.             ('multipart/report', ''),
  368.             ('boundary', 'D1690A7AC1.996856090/mail.example.com')])
  369.         msg.set_param('report-type', old_val)
  370.         eq(msg.get_params(), [
  371.             ('multipart/report', ''),
  372.             ('boundary', 'D1690A7AC1.996856090/mail.example.com'),
  373.             ('report-type', old_val)])
  374.  
  375.     
  376.     def test_del_param_on_other_header(self):
  377.         msg = Message()
  378.         msg.add_header('Content-Disposition', 'attachment', filename = 'bud.gif')
  379.         msg.del_param('filename', 'content-disposition')
  380.         self.assertEqual(msg['content-disposition'], 'attachment')
  381.  
  382.     
  383.     def test_set_type(self):
  384.         eq = self.assertEqual
  385.         msg = Message()
  386.         self.assertRaises(ValueError, msg.set_type, 'text')
  387.         msg.set_type('text/plain')
  388.         eq(msg['content-type'], 'text/plain')
  389.         msg.set_param('charset', 'us-ascii')
  390.         eq(msg['content-type'], 'text/plain; charset="us-ascii"')
  391.         msg.set_type('text/html')
  392.         eq(msg['content-type'], 'text/html; charset="us-ascii"')
  393.  
  394.     
  395.     def test_set_type_on_other_header(self):
  396.         msg = Message()
  397.         msg['X-Content-Type'] = 'text/plain'
  398.         msg.set_type('application/octet-stream', 'X-Content-Type')
  399.         self.assertEqual(msg['x-content-type'], 'application/octet-stream')
  400.  
  401.     
  402.     def test_get_content_type_missing(self):
  403.         msg = Message()
  404.         self.assertEqual(msg.get_content_type(), 'text/plain')
  405.  
  406.     
  407.     def test_get_content_type_missing_with_default_type(self):
  408.         msg = Message()
  409.         msg.set_default_type('message/rfc822')
  410.         self.assertEqual(msg.get_content_type(), 'message/rfc822')
  411.  
  412.     
  413.     def test_get_content_type_from_message_implicit(self):
  414.         msg = self._msgobj('msg_30.txt')
  415.         self.assertEqual(msg.get_payload(0).get_content_type(), 'message/rfc822')
  416.  
  417.     
  418.     def test_get_content_type_from_message_explicit(self):
  419.         msg = self._msgobj('msg_28.txt')
  420.         self.assertEqual(msg.get_payload(0).get_content_type(), 'message/rfc822')
  421.  
  422.     
  423.     def test_get_content_type_from_message_text_plain_implicit(self):
  424.         msg = self._msgobj('msg_03.txt')
  425.         self.assertEqual(msg.get_content_type(), 'text/plain')
  426.  
  427.     
  428.     def test_get_content_type_from_message_text_plain_explicit(self):
  429.         msg = self._msgobj('msg_01.txt')
  430.         self.assertEqual(msg.get_content_type(), 'text/plain')
  431.  
  432.     
  433.     def test_get_content_maintype_missing(self):
  434.         msg = Message()
  435.         self.assertEqual(msg.get_content_maintype(), 'text')
  436.  
  437.     
  438.     def test_get_content_maintype_missing_with_default_type(self):
  439.         msg = Message()
  440.         msg.set_default_type('message/rfc822')
  441.         self.assertEqual(msg.get_content_maintype(), 'message')
  442.  
  443.     
  444.     def test_get_content_maintype_from_message_implicit(self):
  445.         msg = self._msgobj('msg_30.txt')
  446.         self.assertEqual(msg.get_payload(0).get_content_maintype(), 'message')
  447.  
  448.     
  449.     def test_get_content_maintype_from_message_explicit(self):
  450.         msg = self._msgobj('msg_28.txt')
  451.         self.assertEqual(msg.get_payload(0).get_content_maintype(), 'message')
  452.  
  453.     
  454.     def test_get_content_maintype_from_message_text_plain_implicit(self):
  455.         msg = self._msgobj('msg_03.txt')
  456.         self.assertEqual(msg.get_content_maintype(), 'text')
  457.  
  458.     
  459.     def test_get_content_maintype_from_message_text_plain_explicit(self):
  460.         msg = self._msgobj('msg_01.txt')
  461.         self.assertEqual(msg.get_content_maintype(), 'text')
  462.  
  463.     
  464.     def test_get_content_subtype_missing(self):
  465.         msg = Message()
  466.         self.assertEqual(msg.get_content_subtype(), 'plain')
  467.  
  468.     
  469.     def test_get_content_subtype_missing_with_default_type(self):
  470.         msg = Message()
  471.         msg.set_default_type('message/rfc822')
  472.         self.assertEqual(msg.get_content_subtype(), 'rfc822')
  473.  
  474.     
  475.     def test_get_content_subtype_from_message_implicit(self):
  476.         msg = self._msgobj('msg_30.txt')
  477.         self.assertEqual(msg.get_payload(0).get_content_subtype(), 'rfc822')
  478.  
  479.     
  480.     def test_get_content_subtype_from_message_explicit(self):
  481.         msg = self._msgobj('msg_28.txt')
  482.         self.assertEqual(msg.get_payload(0).get_content_subtype(), 'rfc822')
  483.  
  484.     
  485.     def test_get_content_subtype_from_message_text_plain_implicit(self):
  486.         msg = self._msgobj('msg_03.txt')
  487.         self.assertEqual(msg.get_content_subtype(), 'plain')
  488.  
  489.     
  490.     def test_get_content_subtype_from_message_text_plain_explicit(self):
  491.         msg = self._msgobj('msg_01.txt')
  492.         self.assertEqual(msg.get_content_subtype(), 'plain')
  493.  
  494.     
  495.     def test_get_content_maintype_error(self):
  496.         msg = Message()
  497.         msg['Content-Type'] = 'no-slash-in-this-string'
  498.         self.assertEqual(msg.get_content_maintype(), 'text')
  499.  
  500.     
  501.     def test_get_content_subtype_error(self):
  502.         msg = Message()
  503.         msg['Content-Type'] = 'no-slash-in-this-string'
  504.         self.assertEqual(msg.get_content_subtype(), 'plain')
  505.  
  506.     
  507.     def test_replace_header(self):
  508.         eq = self.assertEqual
  509.         msg = Message()
  510.         msg.add_header('First', 'One')
  511.         msg.add_header('Second', 'Two')
  512.         msg.add_header('Third', 'Three')
  513.         eq(msg.keys(), [
  514.             'First',
  515.             'Second',
  516.             'Third'])
  517.         eq(msg.values(), [
  518.             'One',
  519.             'Two',
  520.             'Three'])
  521.         msg.replace_header('Second', 'Twenty')
  522.         eq(msg.keys(), [
  523.             'First',
  524.             'Second',
  525.             'Third'])
  526.         eq(msg.values(), [
  527.             'One',
  528.             'Twenty',
  529.             'Three'])
  530.         msg.add_header('First', 'Eleven')
  531.         msg.replace_header('First', 'One Hundred')
  532.         eq(msg.keys(), [
  533.             'First',
  534.             'Second',
  535.             'Third',
  536.             'First'])
  537.         eq(msg.values(), [
  538.             'One Hundred',
  539.             'Twenty',
  540.             'Three',
  541.             'Eleven'])
  542.         self.assertRaises(KeyError, msg.replace_header, 'Fourth', 'Missing')
  543.  
  544.     
  545.     def test_broken_base64_payload(self):
  546.         x = 'AwDp0P7//y6LwKEAcPa/6Q=9'
  547.         msg = Message()
  548.         msg['content-type'] = 'audio/x-midi'
  549.         msg['content-transfer-encoding'] = 'base64'
  550.         msg.set_payload(x)
  551.         self.assertEqual(msg.get_payload(decode = True), x)
  552.  
  553.  
  554.  
  555. class TestEncoders(unittest.TestCase):
  556.     
  557.     def test_encode_empty_payload(self):
  558.         eq = self.assertEqual
  559.         msg = Message()
  560.         msg.set_charset('us-ascii')
  561.         eq(msg['content-transfer-encoding'], '7bit')
  562.  
  563.     
  564.     def test_default_cte(self):
  565.         eq = self.assertEqual
  566.         msg = MIMEText('hello world')
  567.         eq(msg['content-transfer-encoding'], '7bit')
  568.  
  569.     
  570.     def test_default_cte(self):
  571.         eq = self.assertEqual
  572.         msg = MIMEText('hello world')
  573.         eq(msg['content-transfer-encoding'], '7bit')
  574.         msg = MIMEText('hello \xf8 world')
  575.         eq(msg['content-transfer-encoding'], '8bit')
  576.         msg = MIMEText('hello \xf8 world', _charset = 'iso-8859-1')
  577.         eq(msg['content-transfer-encoding'], 'quoted-printable')
  578.  
  579.  
  580.  
  581. class TestLongHeaders(TestEmailBase):
  582.     
  583.     def test_split_long_continuation(self):
  584.         eq = self.ndiffAssertEqual
  585.         msg = email.message_from_string('Subject: bug demonstration\n\t12345678911234567892123456789312345678941234567895123456789612345678971234567898112345678911234567892123456789112345678911234567892123456789\n\tmore text\n\ntest\n')
  586.         sfp = StringIO()
  587.         g = Generator(sfp)
  588.         g.flatten(msg)
  589.         eq(sfp.getvalue(), 'Subject: bug demonstration\n\t12345678911234567892123456789312345678941234567895123456789612345678971234567898112345678911234567892123456789112345678911234567892123456789\n\tmore text\n\ntest\n')
  590.  
  591.     
  592.     def test_another_long_almost_unsplittable_header(self):
  593.         eq = self.ndiffAssertEqual
  594.         hstr = 'bug demonstration\n\t12345678911234567892123456789312345678941234567895123456789612345678971234567898112345678911234567892123456789112345678911234567892123456789\n\tmore text'
  595.         h = Header(hstr, continuation_ws = '\t')
  596.         eq(h.encode(), 'bug demonstration\n\t12345678911234567892123456789312345678941234567895123456789612345678971234567898112345678911234567892123456789112345678911234567892123456789\n\tmore text')
  597.         h = Header(hstr)
  598.         eq(h.encode(), 'bug demonstration\n 12345678911234567892123456789312345678941234567895123456789612345678971234567898112345678911234567892123456789112345678911234567892123456789\n more text')
  599.  
  600.     
  601.     def test_long_nonstring(self):
  602.         eq = self.ndiffAssertEqual
  603.         g = Charset('iso-8859-1')
  604.         cz = Charset('iso-8859-2')
  605.         utf8 = Charset('utf-8')
  606.         g_head = 'Die Mieter treten hier ein werden mit einem Foerderband komfortabel den Korridor entlang, an s\xfcdl\xfcndischen Wandgem\xe4lden vorbei, gegen die rotierenden Klingen bef\xf6rdert. '
  607.         cz_head = 'Finan\xe8ni metropole se hroutily pod tlakem jejich d\xf9vtipu.. '
  608.         utf8_head = u'µ¡úτó║πü½Φ¿Çπüåπü¿τ┐╗Φ¿│πü»πüòπéîπüªπüäπü╛πü¢πéôπÇéΣ╕ÇΘâ¿πü»πâëπéñπâäΦ¬₧πüºπüÖπüîπÇüπüéπü¿πü»πüºπüƒπéëπéüπüºπüÖπÇéσ«ƒΘÜ¢πü½πü»πÇîWenn ist das Nunstuck git und Slotermeyer? Ja! Beiherhund das Oder die Flipperwaldt gersput.πÇìπü¿Φ¿Çπüúπüªπüäπü╛πüÖπÇé'.encode('utf-8')
  609.         h = Header(g_head, g, header_name = 'Subject')
  610.         h.append(cz_head, cz)
  611.         h.append(utf8_head, utf8)
  612.         msg = Message()
  613.         msg['Subject'] = h
  614.         sfp = StringIO()
  615.         g = Generator(sfp)
  616.         g.flatten(msg)
  617.         eq(sfp.getvalue(), 'Subject: =?iso-8859-1?q?Die_Mieter_treten_hier_ein_werden_mit_einem_Foerd?=\n =?iso-8859-1?q?erband_komfortabel_den_Korridor_entlang=2C_an_s=FCdl=FCndi?=\n =?iso-8859-1?q?schen_Wandgem=E4lden_vorbei=2C_gegen_die_rotierenden_Kling?=\n =?iso-8859-1?q?en_bef=F6rdert=2E_?= =?iso-8859-2?q?Finan=E8ni_met?=\n =?iso-8859-2?q?ropole_se_hroutily_pod_tlakem_jejich_d=F9vtipu=2E=2E_?=\n =?utf-8?b?5q2j56K644Gr6KiA44GG44Go57+76Kiz44Gv44GV44KM44Gm44GE?=\n =?utf-8?b?44G+44Gb44KT44CC5LiA6YOo44Gv44OJ44Kk44OE6Kqe44Gn44GZ44GM44CB?=\n =?utf-8?b?44GC44Go44Gv44Gn44Gf44KJ44KB44Gn44GZ44CC5a6f6Zqb44Gr44Gv44CM?=\n =?utf-8?q?Wenn_ist_das_Nunstuck_git_und_Slotermeyer=3F_Ja!_Beiherhund_das?=\n =?utf-8?b?IE9kZXIgZGllIEZsaXBwZXJ3YWxkdCBnZXJzcHV0LuOAjeOBqOiogOOBow==?=\n =?utf-8?b?44Gm44GE44G+44GZ44CC?=\n\n')
  618.         eq(h.encode(), '=?iso-8859-1?q?Die_Mieter_treten_hier_ein_werden_mit_einem_Foerd?=\n =?iso-8859-1?q?erband_komfortabel_den_Korridor_entlang=2C_an_s=FCdl=FCndi?=\n =?iso-8859-1?q?schen_Wandgem=E4lden_vorbei=2C_gegen_die_rotierenden_Kling?=\n =?iso-8859-1?q?en_bef=F6rdert=2E_?= =?iso-8859-2?q?Finan=E8ni_met?=\n =?iso-8859-2?q?ropole_se_hroutily_pod_tlakem_jejich_d=F9vtipu=2E=2E_?=\n =?utf-8?b?5q2j56K644Gr6KiA44GG44Go57+76Kiz44Gv44GV44KM44Gm44GE?=\n =?utf-8?b?44G+44Gb44KT44CC5LiA6YOo44Gv44OJ44Kk44OE6Kqe44Gn44GZ44GM44CB?=\n =?utf-8?b?44GC44Go44Gv44Gn44Gf44KJ44KB44Gn44GZ44CC5a6f6Zqb44Gr44Gv44CM?=\n =?utf-8?q?Wenn_ist_das_Nunstuck_git_und_Slotermeyer=3F_Ja!_Beiherhund_das?=\n =?utf-8?b?IE9kZXIgZGllIEZsaXBwZXJ3YWxkdCBnZXJzcHV0LuOAjeOBqOiogOOBow==?=\n =?utf-8?b?44Gm44GE44G+44GZ44CC?=')
  619.  
  620.     
  621.     def test_long_header_encode(self):
  622.         eq = self.ndiffAssertEqual
  623.         h = Header('wasnipoop; giraffes="very-long-necked-animals"; spooge="yummy"; hippos="gargantuan"; marshmallows="gooey"', header_name = 'X-Foobar-Spoink-Defrobnit')
  624.         eq(h.encode(), 'wasnipoop; giraffes="very-long-necked-animals";\n spooge="yummy"; hippos="gargantuan"; marshmallows="gooey"')
  625.  
  626.     
  627.     def test_long_header_encode_with_tab_continuation(self):
  628.         eq = self.ndiffAssertEqual
  629.         h = Header('wasnipoop; giraffes="very-long-necked-animals"; spooge="yummy"; hippos="gargantuan"; marshmallows="gooey"', header_name = 'X-Foobar-Spoink-Defrobnit', continuation_ws = '\t')
  630.         eq(h.encode(), 'wasnipoop; giraffes="very-long-necked-animals";\n\tspooge="yummy"; hippos="gargantuan"; marshmallows="gooey"')
  631.  
  632.     
  633.     def test_header_splitter(self):
  634.         eq = self.ndiffAssertEqual
  635.         msg = MIMEText('')
  636.         msg['X-Foobar-Spoink-Defrobnit'] = 'wasnipoop; giraffes="very-long-necked-animals"; spooge="yummy"; hippos="gargantuan"; marshmallows="gooey"'
  637.         sfp = StringIO()
  638.         g = Generator(sfp)
  639.         g.flatten(msg)
  640.         eq(sfp.getvalue(), 'Content-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\nX-Foobar-Spoink-Defrobnit: wasnipoop; giraffes="very-long-necked-animals";\n\tspooge="yummy"; hippos="gargantuan"; marshmallows="gooey"\n\n')
  641.  
  642.     
  643.     def test_no_semis_header_splitter(self):
  644.         eq = self.ndiffAssertEqual
  645.         msg = Message()
  646.         msg['From'] = 'test@dom.ain'
  647.         msg['References'] = []([ '<%d@dom.ain>' % i for i in range(10) ])
  648.         msg.set_payload('Test')
  649.         sfp = StringIO()
  650.         g = Generator(sfp)
  651.         g.flatten(msg)
  652.         eq(sfp.getvalue(), 'From: test@dom.ain\nReferences: <0@dom.ain> <1@dom.ain> <2@dom.ain> <3@dom.ain> <4@dom.ain>\n\t<5@dom.ain> <6@dom.ain> <7@dom.ain> <8@dom.ain> <9@dom.ain>\n\nTest')
  653.  
  654.     
  655.     def test_no_split_long_header(self):
  656.         eq = self.ndiffAssertEqual
  657.         hstr = 'References: ' + 'x' * 80
  658.         h = Header(hstr, continuation_ws = '\t')
  659.         eq(h.encode(), 'References: xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')
  660.  
  661.     
  662.     def test_splitting_multiple_long_lines(self):
  663.         eq = self.ndiffAssertEqual
  664.         hstr = 'from babylon.socal-raves.org (localhost [127.0.0.1]); by babylon.socal-raves.org (Postfix) with ESMTP id B570E51B81; for <mailman-admin@babylon.socal-raves.org>; Sat, 2 Feb 2002 17:00:06 -0800 (PST)\n\tfrom babylon.socal-raves.org (localhost [127.0.0.1]); by babylon.socal-raves.org (Postfix) with ESMTP id B570E51B81; for <mailman-admin@babylon.socal-raves.org>; Sat, 2 Feb 2002 17:00:06 -0800 (PST)\n\tfrom babylon.socal-raves.org (localhost [127.0.0.1]); by babylon.socal-raves.org (Postfix) with ESMTP id B570E51B81; for <mailman-admin@babylon.socal-raves.org>; Sat, 2 Feb 2002 17:00:06 -0800 (PST)\n'
  665.         h = Header(hstr, continuation_ws = '\t')
  666.         eq(h.encode(), 'from babylon.socal-raves.org (localhost [127.0.0.1]);\n\tby babylon.socal-raves.org (Postfix) with ESMTP id B570E51B81;\n\tfor <mailman-admin@babylon.socal-raves.org>;\n\tSat, 2 Feb 2002 17:00:06 -0800 (PST)\n\tfrom babylon.socal-raves.org (localhost [127.0.0.1]);\n\tby babylon.socal-raves.org (Postfix) with ESMTP id B570E51B81;\n\tfor <mailman-admin@babylon.socal-raves.org>;\n\tSat, 2 Feb 2002 17:00:06 -0800 (PST)\n\tfrom babylon.socal-raves.org (localhost [127.0.0.1]);\n\tby babylon.socal-raves.org (Postfix) with ESMTP id B570E51B81;\n\tfor <mailman-admin@babylon.socal-raves.org>;\n\tSat, 2 Feb 2002 17:00:06 -0800 (PST)')
  667.  
  668.     
  669.     def test_splitting_first_line_only_is_long(self):
  670.         eq = self.ndiffAssertEqual
  671.         hstr = 'from modemcable093.139-201-24.que.mc.videotron.ca ([24.201.139.93] helo=cthulhu.gerg.ca)\n\tby kronos.mems-exchange.org with esmtp (Exim 4.05)\n\tid 17k4h5-00034i-00\n\tfor test@mems-exchange.org; Wed, 28 Aug 2002 11:25:20 -0400'
  672.         h = Header(hstr, maxlinelen = 78, header_name = 'Received', continuation_ws = '\t')
  673.         eq(h.encode(), 'from modemcable093.139-201-24.que.mc.videotron.ca ([24.201.139.93]\n\thelo=cthulhu.gerg.ca)\n\tby kronos.mems-exchange.org with esmtp (Exim 4.05)\n\tid 17k4h5-00034i-00\n\tfor test@mems-exchange.org; Wed, 28 Aug 2002 11:25:20 -0400')
  674.  
  675.     
  676.     def test_long_8bit_header(self):
  677.         eq = self.ndiffAssertEqual
  678.         msg = Message()
  679.         h = Header('Britische Regierung gibt', 'iso-8859-1', header_name = 'Subject')
  680.         h.append('gr\xfcnes Licht f\xfcr Offshore-Windkraftprojekte')
  681.         msg['Subject'] = h
  682.         eq(msg.as_string(), 'Subject: =?iso-8859-1?q?Britische_Regierung_gibt?= =?iso-8859-1?q?gr=FCnes?=\n =?iso-8859-1?q?_Licht_f=FCr_Offshore-Windkraftprojekte?=\n\n')
  683.  
  684.     
  685.     def test_long_8bit_header_no_charset(self):
  686.         eq = self.ndiffAssertEqual
  687.         msg = Message()
  688.         msg['Reply-To'] = 'Britische Regierung gibt gr\xfcnes Licht f\xfcr Offshore-Windkraftprojekte <a-very-long-address@example.com>'
  689.         eq(msg.as_string(), 'Reply-To: Britische Regierung gibt gr\xfcnes Licht f\xfcr Offshore-Windkraftprojekte <a-very-long-address@example.com>\n\n')
  690.  
  691.     
  692.     def test_long_to_header(self):
  693.         eq = self.ndiffAssertEqual
  694.         to = '"Someone Test #A" <someone@eecs.umich.edu>,<someone@eecs.umich.edu>,"Someone Test #B" <someone@umich.edu>, "Someone Test #C" <someone@eecs.umich.edu>, "Someone Test #D" <someone@eecs.umich.edu>'
  695.         msg = Message()
  696.         msg['To'] = to
  697.         eq(msg.as_string(0), 'To: "Someone Test #A" <someone@eecs.umich.edu>, <someone@eecs.umich.edu>,\n\t"Someone Test #B" <someone@umich.edu>,\n\t"Someone Test #C" <someone@eecs.umich.edu>,\n\t"Someone Test #D" <someone@eecs.umich.edu>\n\n')
  698.  
  699.     
  700.     def test_long_line_after_append(self):
  701.         eq = self.ndiffAssertEqual
  702.         s = 'This is an example of string which has almost the limit of header length.'
  703.         h = Header(s)
  704.         h.append('Add another line.')
  705.         eq(h.encode(), 'This is an example of string which has almost the limit of header length.\n Add another line.')
  706.  
  707.     
  708.     def test_shorter_line_with_append(self):
  709.         eq = self.ndiffAssertEqual
  710.         s = 'This is a shorter line.'
  711.         h = Header(s)
  712.         h.append('Add another sentence. (Surprise?)')
  713.         eq(h.encode(), 'This is a shorter line. Add another sentence. (Surprise?)')
  714.  
  715.     
  716.     def test_long_field_name(self):
  717.         eq = self.ndiffAssertEqual
  718.         fn = 'X-Very-Very-Very-Long-Header-Name'
  719.         gs = 'Die Mieter treten hier ein werden mit einem Foerderband komfortabel den Korridor entlang, an s\xfcdl\xfcndischen Wandgem\xe4lden vorbei, gegen die rotierenden Klingen bef\xf6rdert. '
  720.         h = Header(gs, 'iso-8859-1', header_name = fn)
  721.         eq(h.encode(), '=?iso-8859-1?q?Die_Mieter_treten_hier_?=\n =?iso-8859-1?q?ein_werden_mit_einem_Foerderband_komfortabel_den_Korridor_?=\n =?iso-8859-1?q?entlang=2C_an_s=FCdl=FCndischen_Wandgem=E4lden_vorbei=2C_g?=\n =?iso-8859-1?q?egen_die_rotierenden_Klingen_bef=F6rdert=2E_?=')
  722.  
  723.     
  724.     def test_long_received_header(self):
  725.         h = 'from FOO.TLD (vizworld.acl.foo.tld [123.452.678.9]) by hrothgar.la.mastaler.com (tmda-ofmipd) with ESMTP; Wed, 05 Mar 2003 18:10:18 -0700'
  726.         msg = Message()
  727.         msg['Received-1'] = Header(h, continuation_ws = '\t')
  728.         msg['Received-2'] = h
  729.         self.assertEqual(msg.as_string(), 'Received-1: from FOO.TLD (vizworld.acl.foo.tld [123.452.678.9]) by\n\throthgar.la.mastaler.com (tmda-ofmipd) with ESMTP;\n\tWed, 05 Mar 2003 18:10:18 -0700\nReceived-2: from FOO.TLD (vizworld.acl.foo.tld [123.452.678.9]) by\n\throthgar.la.mastaler.com (tmda-ofmipd) with ESMTP;\n\tWed, 05 Mar 2003 18:10:18 -0700\n\n')
  730.  
  731.     
  732.     def test_string_headerinst_eq(self):
  733.         h = '<15975.17901.207240.414604@sgigritzmann1.mathematik.tu-muenchen.de> (David Bremner\'s message of "Thu, 6 Mar 2003 13:58:21 +0100")'
  734.         msg = Message()
  735.         msg['Received-1'] = Header(h, header_name = 'Received-1', continuation_ws = '\t')
  736.         msg['Received-2'] = h
  737.         self.assertEqual(msg.as_string(), 'Received-1: <15975.17901.207240.414604@sgigritzmann1.mathematik.tu-muenchen.de>\n\t(David Bremner\'s message of "Thu, 6 Mar 2003 13:58:21 +0100")\nReceived-2: <15975.17901.207240.414604@sgigritzmann1.mathematik.tu-muenchen.de>\n\t(David Bremner\'s message of "Thu, 6 Mar 2003 13:58:21 +0100")\n\n')
  738.  
  739.     
  740.     def test_long_unbreakable_lines_with_continuation(self):
  741.         eq = self.ndiffAssertEqual
  742.         msg = Message()
  743.         t = ' iVBORw0KGgoAAAANSUhEUgAAADAAAAAwBAMAAAClLOS0AAAAGFBMVEUAAAAkHiJeRUIcGBi9\n locQDQ4zJykFBAXJfWDjAAACYUlEQVR4nF2TQY/jIAyFc6lydlG5x8Nyp1Y69wj1PN2I5gzp'
  744.         msg['Face-1'] = t
  745.         msg['Face-2'] = Header(t, header_name = 'Face-2')
  746.         eq(msg.as_string(), 'Face-1: iVBORw0KGgoAAAANSUhEUgAAADAAAAAwBAMAAAClLOS0AAAAGFBMVEUAAAAkHiJeRUIcGBi9\n\tlocQDQ4zJykFBAXJfWDjAAACYUlEQVR4nF2TQY/jIAyFc6lydlG5x8Nyp1Y69wj1PN2I5gzp\nFace-2: iVBORw0KGgoAAAANSUhEUgAAADAAAAAwBAMAAAClLOS0AAAAGFBMVEUAAAAkHiJeRUIcGBi9\n locQDQ4zJykFBAXJfWDjAAACYUlEQVR4nF2TQY/jIAyFc6lydlG5x8Nyp1Y69wj1PN2I5gzp\n\n')
  747.  
  748.     
  749.     def test_another_long_multiline_header(self):
  750.         eq = self.ndiffAssertEqual
  751.         m = 'Received: from siimage.com ([172.25.1.3]) by zima.siliconimage.com with Microsoft SMTPSVC(5.0.2195.4905);\n\tWed, 16 Oct 2002 07:41:11 -0700'
  752.         msg = email.message_from_string(m)
  753.         eq(msg.as_string(), 'Received: from siimage.com ([172.25.1.3]) by zima.siliconimage.com with\n\tMicrosoft SMTPSVC(5.0.2195.4905); Wed, 16 Oct 2002 07:41:11 -0700\n\n')
  754.  
  755.     
  756.     def test_long_lines_with_different_header(self):
  757.         eq = self.ndiffAssertEqual
  758.         h = 'List-Unsubscribe: <https://lists.sourceforge.net/lists/listinfo/spamassassin-talk>,\n        <mailto:spamassassin-talk-request@lists.sourceforge.net?subject=unsubscribe>'
  759.         msg = Message()
  760.         msg['List'] = h
  761.         msg['List'] = Header(h, header_name = 'List')
  762.         eq(msg.as_string(), 'List: List-Unsubscribe: <https://lists.sourceforge.net/lists/listinfo/spamassassin-talk>,\n\t<mailto:spamassassin-talk-request@lists.sourceforge.net?subject=unsubscribe>\nList: List-Unsubscribe: <https://lists.sourceforge.net/lists/listinfo/spamassassin-talk>,\n <mailto:spamassassin-talk-request@lists.sourceforge.net?subject=unsubscribe>\n\n')
  763.  
  764.  
  765.  
  766. class TestFromMangling(unittest.TestCase):
  767.     
  768.     def setUp(self):
  769.         self.msg = Message()
  770.         self.msg['From'] = 'aaa@bbb.org'
  771.         self.msg.set_payload('From the desk of A.A.A.:\nBlah blah blah\n')
  772.  
  773.     
  774.     def test_mangled_from(self):
  775.         s = StringIO()
  776.         g = Generator(s, mangle_from_ = True)
  777.         g.flatten(self.msg)
  778.         self.assertEqual(s.getvalue(), 'From: aaa@bbb.org\n\n>From the desk of A.A.A.:\nBlah blah blah\n')
  779.  
  780.     
  781.     def test_dont_mangle_from(self):
  782.         s = StringIO()
  783.         g = Generator(s, mangle_from_ = False)
  784.         g.flatten(self.msg)
  785.         self.assertEqual(s.getvalue(), 'From: aaa@bbb.org\n\nFrom the desk of A.A.A.:\nBlah blah blah\n')
  786.  
  787.  
  788.  
  789. class TestMIMEAudio(unittest.TestCase):
  790.     
  791.     def setUp(self):
  792.         datadir = os.path.join(os.path.dirname(landmark), 'data', '')
  793.         fp = open(findfile('audiotest.au', datadir), 'rb')
  794.         
  795.         try:
  796.             self._audiodata = fp.read()
  797.         finally:
  798.             fp.close()
  799.  
  800.         self._au = MIMEAudio(self._audiodata)
  801.  
  802.     
  803.     def test_guess_minor_type(self):
  804.         self.assertEqual(self._au.get_content_type(), 'audio/basic')
  805.  
  806.     
  807.     def test_encoding(self):
  808.         payload = self._au.get_payload()
  809.         self.assertEqual(base64.decodestring(payload), self._audiodata)
  810.  
  811.     
  812.     def test_checkSetMinor(self):
  813.         au = MIMEAudio(self._audiodata, 'fish')
  814.         self.assertEqual(au.get_content_type(), 'audio/fish')
  815.  
  816.     
  817.     def test_add_header(self):
  818.         eq = self.assertEqual
  819.         unless = self.failUnless
  820.         self._au.add_header('Content-Disposition', 'attachment', filename = 'audiotest.au')
  821.         eq(self._au['content-disposition'], 'attachment; filename="audiotest.au"')
  822.         eq(self._au.get_params(header = 'content-disposition'), [
  823.             ('attachment', ''),
  824.             ('filename', 'audiotest.au')])
  825.         eq(self._au.get_param('filename', header = 'content-disposition'), 'audiotest.au')
  826.         missing = []
  827.         eq(self._au.get_param('attachment', header = 'content-disposition'), '')
  828.         unless(self._au.get_param('foo', failobj = missing, header = 'content-disposition') is missing)
  829.         unless(self._au.get_param('foobar', missing) is missing)
  830.         unless(self._au.get_param('attachment', missing, header = 'foobar') is missing)
  831.  
  832.  
  833.  
  834. class TestMIMEImage(unittest.TestCase):
  835.     
  836.     def setUp(self):
  837.         fp = openfile('PyBanner048.gif')
  838.         
  839.         try:
  840.             self._imgdata = fp.read()
  841.         finally:
  842.             fp.close()
  843.  
  844.         self._im = MIMEImage(self._imgdata)
  845.  
  846.     
  847.     def test_guess_minor_type(self):
  848.         self.assertEqual(self._im.get_content_type(), 'image/gif')
  849.  
  850.     
  851.     def test_encoding(self):
  852.         payload = self._im.get_payload()
  853.         self.assertEqual(base64.decodestring(payload), self._imgdata)
  854.  
  855.     
  856.     def test_checkSetMinor(self):
  857.         im = MIMEImage(self._imgdata, 'fish')
  858.         self.assertEqual(im.get_content_type(), 'image/fish')
  859.  
  860.     
  861.     def test_add_header(self):
  862.         eq = self.assertEqual
  863.         unless = self.failUnless
  864.         self._im.add_header('Content-Disposition', 'attachment', filename = 'dingusfish.gif')
  865.         eq(self._im['content-disposition'], 'attachment; filename="dingusfish.gif"')
  866.         eq(self._im.get_params(header = 'content-disposition'), [
  867.             ('attachment', ''),
  868.             ('filename', 'dingusfish.gif')])
  869.         eq(self._im.get_param('filename', header = 'content-disposition'), 'dingusfish.gif')
  870.         missing = []
  871.         eq(self._im.get_param('attachment', header = 'content-disposition'), '')
  872.         unless(self._im.get_param('foo', failobj = missing, header = 'content-disposition') is missing)
  873.         unless(self._im.get_param('foobar', missing) is missing)
  874.         unless(self._im.get_param('attachment', missing, header = 'foobar') is missing)
  875.  
  876.  
  877.  
  878. class TestMIMEApplication(unittest.TestCase):
  879.     
  880.     def test_headers(self):
  881.         eq = self.assertEqual
  882.         msg = MIMEApplication('\xfa\xfb\xfc\xfd\xfe\xff')
  883.         eq(msg.get_content_type(), 'application/octet-stream')
  884.         eq(msg['content-transfer-encoding'], 'base64')
  885.  
  886.     
  887.     def test_body(self):
  888.         eq = self.assertEqual
  889.         bytes = '\xfa\xfb\xfc\xfd\xfe\xff'
  890.         msg = MIMEApplication(bytes)
  891.         eq(msg.get_payload(), '+vv8/f7/')
  892.         eq(msg.get_payload(decode = True), bytes)
  893.  
  894.  
  895.  
  896. class TestMIMEText(unittest.TestCase):
  897.     
  898.     def setUp(self):
  899.         self._msg = MIMEText('hello there')
  900.  
  901.     
  902.     def test_types(self):
  903.         eq = self.assertEqual
  904.         unless = self.failUnless
  905.         eq(self._msg.get_content_type(), 'text/plain')
  906.         eq(self._msg.get_param('charset'), 'us-ascii')
  907.         missing = []
  908.         unless(self._msg.get_param('foobar', missing) is missing)
  909.         unless(self._msg.get_param('charset', missing, header = 'foobar') is missing)
  910.  
  911.     
  912.     def test_payload(self):
  913.         self.assertEqual(self._msg.get_payload(), 'hello there')
  914.         self.failUnless(not self._msg.is_multipart())
  915.  
  916.     
  917.     def test_charset(self):
  918.         eq = self.assertEqual
  919.         msg = MIMEText('hello there', _charset = 'us-ascii')
  920.         eq(msg.get_charset().input_charset, 'us-ascii')
  921.         eq(msg['content-type'], 'text/plain; charset="us-ascii"')
  922.  
  923.  
  924.  
  925. class TestMultipart(TestEmailBase):
  926.     
  927.     def setUp(self):
  928.         fp = openfile('PyBanner048.gif')
  929.         
  930.         try:
  931.             data = fp.read()
  932.         finally:
  933.             fp.close()
  934.  
  935.         container = MIMEBase('multipart', 'mixed', boundary = 'BOUNDARY')
  936.         image = MIMEImage(data, name = 'dingusfish.gif')
  937.         image.add_header('content-disposition', 'attachment', filename = 'dingusfish.gif')
  938.         intro = MIMEText('Hi there,\n\nThis is the dingus fish.\n')
  939.         container.attach(intro)
  940.         container.attach(image)
  941.         container['From'] = 'Barry <barry@digicool.com>'
  942.         container['To'] = 'Dingus Lovers <cravindogs@cravindogs.com>'
  943.         container['Subject'] = 'Here is your dingus fish'
  944.         now = 9.8781e+08
  945.         timetuple = time.localtime(now)
  946.         if timetuple[-1] == 0:
  947.             tzsecs = time.timezone
  948.         else:
  949.             tzsecs = time.altzone
  950.         if tzsecs > 0:
  951.             sign = '-'
  952.         else:
  953.             sign = '+'
  954.         tzoffset = ' %s%04d' % (sign, tzsecs / 36)
  955.         container['Date'] = time.strftime('%a, %d %b %Y %H:%M:%S', time.localtime(now)) + tzoffset
  956.         self._msg = container
  957.         self._im = image
  958.         self._txt = intro
  959.  
  960.     
  961.     def test_hierarchy(self):
  962.         eq = self.assertEqual
  963.         unless = self.failUnless
  964.         raises = self.assertRaises
  965.         m = self._msg
  966.         unless(m.is_multipart())
  967.         eq(m.get_content_type(), 'multipart/mixed')
  968.         eq(len(m.get_payload()), 2)
  969.         raises(IndexError, m.get_payload, 2)
  970.         m0 = m.get_payload(0)
  971.         m1 = m.get_payload(1)
  972.         unless(m0 is self._txt)
  973.         unless(m1 is self._im)
  974.         eq(m.get_payload(), [
  975.             m0,
  976.             m1])
  977.         unless(not m0.is_multipart())
  978.         unless(not m1.is_multipart())
  979.  
  980.     
  981.     def test_empty_multipart_idempotent(self):
  982.         text = 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n\n--BOUNDARY\n\n\n--BOUNDARY--\n'
  983.         msg = Parser().parsestr(text)
  984.         self.ndiffAssertEqual(text, msg.as_string())
  985.  
  986.     
  987.     def test_no_parts_in_a_multipart_with_none_epilogue(self):
  988.         outer = MIMEBase('multipart', 'mixed')
  989.         outer['Subject'] = 'A subject'
  990.         outer['To'] = 'aperson@dom.ain'
  991.         outer['From'] = 'bperson@dom.ain'
  992.         outer.set_boundary('BOUNDARY')
  993.         self.ndiffAssertEqual(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n--BOUNDARY\n\n--BOUNDARY--')
  994.  
  995.     
  996.     def test_no_parts_in_a_multipart_with_empty_epilogue(self):
  997.         outer = MIMEBase('multipart', 'mixed')
  998.         outer['Subject'] = 'A subject'
  999.         outer['To'] = 'aperson@dom.ain'
  1000.         outer['From'] = 'bperson@dom.ain'
  1001.         outer.preamble = ''
  1002.         outer.epilogue = ''
  1003.         outer.set_boundary('BOUNDARY')
  1004.         self.ndiffAssertEqual(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n\n--BOUNDARY\n\n--BOUNDARY--\n')
  1005.  
  1006.     
  1007.     def test_one_part_in_a_multipart(self):
  1008.         eq = self.ndiffAssertEqual
  1009.         outer = MIMEBase('multipart', 'mixed')
  1010.         outer['Subject'] = 'A subject'
  1011.         outer['To'] = 'aperson@dom.ain'
  1012.         outer['From'] = 'bperson@dom.ain'
  1013.         outer.set_boundary('BOUNDARY')
  1014.         msg = MIMEText('hello world')
  1015.         outer.attach(msg)
  1016.         eq(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nhello world\n--BOUNDARY--')
  1017.  
  1018.     
  1019.     def test_seq_parts_in_a_multipart_with_empty_preamble(self):
  1020.         eq = self.ndiffAssertEqual
  1021.         outer = MIMEBase('multipart', 'mixed')
  1022.         outer['Subject'] = 'A subject'
  1023.         outer['To'] = 'aperson@dom.ain'
  1024.         outer['From'] = 'bperson@dom.ain'
  1025.         outer.preamble = ''
  1026.         msg = MIMEText('hello world')
  1027.         outer.attach(msg)
  1028.         outer.set_boundary('BOUNDARY')
  1029.         eq(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nhello world\n--BOUNDARY--')
  1030.  
  1031.     
  1032.     def test_seq_parts_in_a_multipart_with_none_preamble(self):
  1033.         eq = self.ndiffAssertEqual
  1034.         outer = MIMEBase('multipart', 'mixed')
  1035.         outer['Subject'] = 'A subject'
  1036.         outer['To'] = 'aperson@dom.ain'
  1037.         outer['From'] = 'bperson@dom.ain'
  1038.         outer.preamble = None
  1039.         msg = MIMEText('hello world')
  1040.         outer.attach(msg)
  1041.         outer.set_boundary('BOUNDARY')
  1042.         eq(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nhello world\n--BOUNDARY--')
  1043.  
  1044.     
  1045.     def test_seq_parts_in_a_multipart_with_none_epilogue(self):
  1046.         eq = self.ndiffAssertEqual
  1047.         outer = MIMEBase('multipart', 'mixed')
  1048.         outer['Subject'] = 'A subject'
  1049.         outer['To'] = 'aperson@dom.ain'
  1050.         outer['From'] = 'bperson@dom.ain'
  1051.         outer.epilogue = None
  1052.         msg = MIMEText('hello world')
  1053.         outer.attach(msg)
  1054.         outer.set_boundary('BOUNDARY')
  1055.         eq(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nhello world\n--BOUNDARY--')
  1056.  
  1057.     
  1058.     def test_seq_parts_in_a_multipart_with_empty_epilogue(self):
  1059.         eq = self.ndiffAssertEqual
  1060.         outer = MIMEBase('multipart', 'mixed')
  1061.         outer['Subject'] = 'A subject'
  1062.         outer['To'] = 'aperson@dom.ain'
  1063.         outer['From'] = 'bperson@dom.ain'
  1064.         outer.epilogue = ''
  1065.         msg = MIMEText('hello world')
  1066.         outer.attach(msg)
  1067.         outer.set_boundary('BOUNDARY')
  1068.         eq(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nhello world\n--BOUNDARY--\n')
  1069.  
  1070.     
  1071.     def test_seq_parts_in_a_multipart_with_nl_epilogue(self):
  1072.         eq = self.ndiffAssertEqual
  1073.         outer = MIMEBase('multipart', 'mixed')
  1074.         outer['Subject'] = 'A subject'
  1075.         outer['To'] = 'aperson@dom.ain'
  1076.         outer['From'] = 'bperson@dom.ain'
  1077.         outer.epilogue = '\n'
  1078.         msg = MIMEText('hello world')
  1079.         outer.attach(msg)
  1080.         outer.set_boundary('BOUNDARY')
  1081.         eq(outer.as_string(), 'Content-Type: multipart/mixed; boundary="BOUNDARY"\nMIME-Version: 1.0\nSubject: A subject\nTo: aperson@dom.ain\nFrom: bperson@dom.ain\n\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nhello world\n--BOUNDARY--\n\n')
  1082.  
  1083.     
  1084.     def test_message_external_body(self):
  1085.         eq = self.assertEqual
  1086.         msg = self._msgobj('msg_36.txt')
  1087.         eq(len(msg.get_payload()), 2)
  1088.         msg1 = msg.get_payload(1)
  1089.         eq(msg1.get_content_type(), 'multipart/alternative')
  1090.         eq(len(msg1.get_payload()), 2)
  1091.         for subpart in msg1.get_payload():
  1092.             eq(subpart.get_content_type(), 'message/external-body')
  1093.             eq(len(subpart.get_payload()), 1)
  1094.             subsubpart = subpart.get_payload(0)
  1095.             eq(subsubpart.get_content_type(), 'text/plain')
  1096.         
  1097.  
  1098.     
  1099.     def test_double_boundary(self):
  1100.         msg = self._msgobj('msg_37.txt')
  1101.         self.assertEqual(len(msg.get_payload()), 3)
  1102.  
  1103.     
  1104.     def test_nested_inner_contains_outer_boundary(self):
  1105.         eq = self.ndiffAssertEqual
  1106.         msg = self._msgobj('msg_38.txt')
  1107.         sfp = StringIO()
  1108.         iterators._structure(msg, sfp)
  1109.         eq(sfp.getvalue(), 'multipart/mixed\n    multipart/mixed\n        multipart/alternative\n            text/plain\n        text/plain\n    text/plain\n    text/plain\n')
  1110.  
  1111.     
  1112.     def test_nested_with_same_boundary(self):
  1113.         eq = self.ndiffAssertEqual
  1114.         msg = self._msgobj('msg_39.txt')
  1115.         sfp = StringIO()
  1116.         iterators._structure(msg, sfp)
  1117.         eq(sfp.getvalue(), 'multipart/mixed\n    multipart/mixed\n        multipart/alternative\n        application/octet-stream\n        application/octet-stream\n    text/plain\n')
  1118.  
  1119.     
  1120.     def test_boundary_in_non_multipart(self):
  1121.         msg = self._msgobj('msg_40.txt')
  1122.         self.assertEqual(msg.as_string(), 'MIME-Version: 1.0\nContent-Type: text/html; boundary="--961284236552522269"\n\n----961284236552522269\nContent-Type: text/html;\nContent-Transfer-Encoding: 7Bit\n\n<html></html>\n\n----961284236552522269--\n')
  1123.  
  1124.     
  1125.     def test_boundary_with_leading_space(self):
  1126.         eq = self.assertEqual
  1127.         msg = email.message_from_string('MIME-Version: 1.0\nContent-Type: multipart/mixed; boundary="    XXXX"\n\n--    XXXX\nContent-Type: text/plain\n\n\n--    XXXX\nContent-Type: text/plain\n\n--    XXXX--\n')
  1128.         self.failUnless(msg.is_multipart())
  1129.         eq(msg.get_boundary(), '    XXXX')
  1130.         eq(len(msg.get_payload()), 2)
  1131.  
  1132.     
  1133.     def test_boundary_without_trailing_newline(self):
  1134.         m = Parser().parsestr('Content-Type: multipart/mixed; boundary="===============0012394164=="\nMIME-Version: 1.0\n\n--===============0012394164==\nContent-Type: image/file1.jpg\nMIME-Version: 1.0\nContent-Transfer-Encoding: base64\n\nYXNkZg==\n--===============0012394164==--')
  1135.         self.assertEquals(m.get_payload(0).get_payload(), 'YXNkZg==')
  1136.  
  1137.  
  1138.  
  1139. class TestNonConformant(TestEmailBase):
  1140.     
  1141.     def test_parse_missing_minor_type(self):
  1142.         eq = self.assertEqual
  1143.         msg = self._msgobj('msg_14.txt')
  1144.         eq(msg.get_content_type(), 'text/plain')
  1145.         eq(msg.get_content_maintype(), 'text')
  1146.         eq(msg.get_content_subtype(), 'plain')
  1147.  
  1148.     
  1149.     def test_same_boundary_inner_outer(self):
  1150.         unless = self.failUnless
  1151.         msg = self._msgobj('msg_15.txt')
  1152.         inner = msg.get_payload(0)
  1153.         unless(hasattr(inner, 'defects'))
  1154.         self.assertEqual(len(inner.defects), 1)
  1155.         unless(isinstance(inner.defects[0], errors.StartBoundaryNotFoundDefect))
  1156.  
  1157.     
  1158.     def test_multipart_no_boundary(self):
  1159.         unless = self.failUnless
  1160.         msg = self._msgobj('msg_25.txt')
  1161.         unless(isinstance(msg.get_payload(), str))
  1162.         self.assertEqual(len(msg.defects), 2)
  1163.         unless(isinstance(msg.defects[0], errors.NoBoundaryInMultipartDefect))
  1164.         unless(isinstance(msg.defects[1], errors.MultipartInvariantViolationDefect))
  1165.  
  1166.     
  1167.     def test_invalid_content_type(self):
  1168.         eq = self.assertEqual
  1169.         neq = self.ndiffAssertEqual
  1170.         msg = Message()
  1171.         msg['Content-Type'] = 'text'
  1172.         eq(msg.get_content_maintype(), 'text')
  1173.         eq(msg.get_content_subtype(), 'plain')
  1174.         eq(msg.get_content_type(), 'text/plain')
  1175.         del msg['content-type']
  1176.         msg['Content-Type'] = 'foo'
  1177.         eq(msg.get_content_maintype(), 'text')
  1178.         eq(msg.get_content_subtype(), 'plain')
  1179.         eq(msg.get_content_type(), 'text/plain')
  1180.         s = StringIO()
  1181.         g = Generator(s)
  1182.         g.flatten(msg)
  1183.         neq(s.getvalue(), 'Content-Type: foo\n\n')
  1184.  
  1185.     
  1186.     def test_no_start_boundary(self):
  1187.         eq = self.ndiffAssertEqual
  1188.         msg = self._msgobj('msg_31.txt')
  1189.         eq(msg.get_payload(), '--BOUNDARY\nContent-Type: text/plain\n\nmessage 1\n\n--BOUNDARY\nContent-Type: text/plain\n\nmessage 2\n\n--BOUNDARY--\n')
  1190.  
  1191.     
  1192.     def test_no_separating_blank_line(self):
  1193.         eq = self.ndiffAssertEqual
  1194.         msg = self._msgobj('msg_35.txt')
  1195.         eq(msg.as_string(), "From: aperson@dom.ain\nTo: bperson@dom.ain\nSubject: here's something interesting\n\ncounter to RFC 2822, there's no separating newline here\n")
  1196.  
  1197.     
  1198.     def test_lying_multipart(self):
  1199.         unless = self.failUnless
  1200.         msg = self._msgobj('msg_41.txt')
  1201.         unless(hasattr(msg, 'defects'))
  1202.         self.assertEqual(len(msg.defects), 2)
  1203.         unless(isinstance(msg.defects[0], errors.NoBoundaryInMultipartDefect))
  1204.         unless(isinstance(msg.defects[1], errors.MultipartInvariantViolationDefect))
  1205.  
  1206.     
  1207.     def test_missing_start_boundary(self):
  1208.         outer = self._msgobj('msg_42.txt')
  1209.         bad = outer.get_payload(1).get_payload(0)
  1210.         self.assertEqual(len(bad.defects), 1)
  1211.         self.failUnless(isinstance(bad.defects[0], errors.StartBoundaryNotFoundDefect))
  1212.  
  1213.     
  1214.     def test_first_line_is_continuation_header(self):
  1215.         eq = self.assertEqual
  1216.         m = ' Line 1\nLine 2\nLine 3'
  1217.         msg = email.message_from_string(m)
  1218.         eq(msg.keys(), [])
  1219.         eq(msg.get_payload(), 'Line 2\nLine 3')
  1220.         eq(len(msg.defects), 1)
  1221.         self.failUnless(isinstance(msg.defects[0], errors.FirstHeaderLineIsContinuationDefect))
  1222.         eq(msg.defects[0].line, ' Line 1\n')
  1223.  
  1224.  
  1225.  
  1226. class TestRFC2047(unittest.TestCase):
  1227.     
  1228.     def test_rfc2047_multiline(self):
  1229.         eq = self.assertEqual
  1230.         s = 'Re: =?mac-iceland?q?r=8Aksm=9Arg=8Cs?= baz\n foo bar =?mac-iceland?q?r=8Aksm=9Arg=8Cs?='
  1231.         dh = decode_header(s)
  1232.         eq(dh, [
  1233.             ('Re:', None),
  1234.             ('r\x8aksm\x9arg\x8cs', 'mac-iceland'),
  1235.             ('baz foo bar', None),
  1236.             ('r\x8aksm\x9arg\x8cs', 'mac-iceland')])
  1237.         eq(str(make_header(dh)), 'Re: =?mac-iceland?q?r=8Aksm=9Arg=8Cs?= baz foo bar\n =?mac-iceland?q?r=8Aksm=9Arg=8Cs?=')
  1238.  
  1239.     
  1240.     def test_whitespace_eater_unicode(self):
  1241.         eq = self.assertEqual
  1242.         s = '=?ISO-8859-1?Q?Andr=E9?= Pirard <pirard@dom.ain>'
  1243.         dh = decode_header(s)
  1244.         eq(dh, [
  1245.             ('Andr\xe9', 'iso-8859-1'),
  1246.             ('Pirard <pirard@dom.ain>', None)])
  1247.         hu = unicode(make_header(dh)).encode('latin-1')
  1248.         eq(hu, 'Andr\xe9 Pirard <pirard@dom.ain>')
  1249.  
  1250.     
  1251.     def test_whitespace_eater_unicode_2(self):
  1252.         eq = self.assertEqual
  1253.         s = 'The =?iso-8859-1?b?cXVpY2sgYnJvd24gZm94?= jumped over the =?iso-8859-1?b?bGF6eSBkb2c=?='
  1254.         dh = decode_header(s)
  1255.         eq(dh, [
  1256.             ('The', None),
  1257.             ('quick brown fox', 'iso-8859-1'),
  1258.             ('jumped over the', None),
  1259.             ('lazy dog', 'iso-8859-1')])
  1260.         hu = make_header(dh).__unicode__()
  1261.         eq(hu, u'The quick brown fox jumped over the lazy dog')
  1262.  
  1263.     
  1264.     def test_rfc2047_missing_whitespace(self):
  1265.         s = 'Sm=?ISO-8859-1?B?9g==?=rg=?ISO-8859-1?B?5Q==?=sbord'
  1266.         dh = decode_header(s)
  1267.         self.assertEqual(dh, [
  1268.             (s, None)])
  1269.  
  1270.     
  1271.     def test_rfc2047_with_whitespace(self):
  1272.         s = 'Sm =?ISO-8859-1?B?9g==?= rg =?ISO-8859-1?B?5Q==?= sbord'
  1273.         dh = decode_header(s)
  1274.         self.assertEqual(dh, [
  1275.             ('Sm', None),
  1276.             ('\xf6', 'iso-8859-1'),
  1277.             ('rg', None),
  1278.             ('\xe5', 'iso-8859-1'),
  1279.             ('sbord', None)])
  1280.  
  1281.  
  1282.  
  1283. class TestMIMEMessage(TestEmailBase):
  1284.     
  1285.     def setUp(self):
  1286.         fp = openfile('msg_11.txt')
  1287.         
  1288.         try:
  1289.             self._text = fp.read()
  1290.         finally:
  1291.             fp.close()
  1292.  
  1293.  
  1294.     
  1295.     def test_type_error(self):
  1296.         self.assertRaises(TypeError, MIMEMessage, 'a plain string')
  1297.  
  1298.     
  1299.     def test_valid_argument(self):
  1300.         eq = self.assertEqual
  1301.         unless = self.failUnless
  1302.         subject = 'A sub-message'
  1303.         m = Message()
  1304.         m['Subject'] = subject
  1305.         r = MIMEMessage(m)
  1306.         eq(r.get_content_type(), 'message/rfc822')
  1307.         payload = r.get_payload()
  1308.         unless(isinstance(payload, list))
  1309.         eq(len(payload), 1)
  1310.         subpart = payload[0]
  1311.         unless(subpart is m)
  1312.         eq(subpart['subject'], subject)
  1313.  
  1314.     
  1315.     def test_bad_multipart(self):
  1316.         eq = self.assertEqual
  1317.         msg1 = Message()
  1318.         msg1['Subject'] = 'subpart 1'
  1319.         msg2 = Message()
  1320.         msg2['Subject'] = 'subpart 2'
  1321.         r = MIMEMessage(msg1)
  1322.         self.assertRaises(errors.MultipartConversionError, r.attach, msg2)
  1323.  
  1324.     
  1325.     def test_generate(self):
  1326.         m = Message()
  1327.         m['Subject'] = 'An enclosed message'
  1328.         m.set_payload('Here is the body of the message.\n')
  1329.         r = MIMEMessage(m)
  1330.         r['Subject'] = 'The enclosing message'
  1331.         s = StringIO()
  1332.         g = Generator(s)
  1333.         g.flatten(r)
  1334.         self.assertEqual(s.getvalue(), 'Content-Type: message/rfc822\nMIME-Version: 1.0\nSubject: The enclosing message\n\nSubject: An enclosed message\n\nHere is the body of the message.\n')
  1335.  
  1336.     
  1337.     def test_parse_message_rfc822(self):
  1338.         eq = self.assertEqual
  1339.         unless = self.failUnless
  1340.         msg = self._msgobj('msg_11.txt')
  1341.         eq(msg.get_content_type(), 'message/rfc822')
  1342.         payload = msg.get_payload()
  1343.         unless(isinstance(payload, list))
  1344.         eq(len(payload), 1)
  1345.         submsg = payload[0]
  1346.         self.failUnless(isinstance(submsg, Message))
  1347.         eq(submsg['subject'], 'An enclosed message')
  1348.         eq(submsg.get_payload(), 'Here is the body of the message.\n')
  1349.  
  1350.     
  1351.     def test_dsn(self):
  1352.         eq = self.assertEqual
  1353.         unless = self.failUnless
  1354.         msg = self._msgobj('msg_16.txt')
  1355.         eq(msg.get_content_type(), 'multipart/report')
  1356.         unless(msg.is_multipart())
  1357.         eq(len(msg.get_payload()), 3)
  1358.         subpart = msg.get_payload(0)
  1359.         eq(subpart.get_content_type(), 'text/plain')
  1360.         eq(subpart.get_payload(), 'This report relates to a message you sent with the following header fields:\n\n  Message-id: <002001c144a6$8752e060$56104586@oxy.edu>\n  Date: Sun, 23 Sep 2001 20:10:55 -0700\n  From: "Ian T. Henry" <henryi@oxy.edu>\n  To: SoCal Raves <scr@socal-raves.org>\n  Subject: [scr] yeah for Ians!!\n\nYour message cannot be delivered to the following recipients:\n\n  Recipient address: jangel1@cougar.noc.ucla.edu\n  Reason: recipient reached disk quota\n\n')
  1361.         subpart = msg.get_payload(1)
  1362.         eq(subpart.get_content_type(), 'message/delivery-status')
  1363.         eq(len(subpart.get_payload()), 2)
  1364.         dsn1 = subpart.get_payload(0)
  1365.         unless(isinstance(dsn1, Message))
  1366.         eq(dsn1['original-envelope-id'], '0GK500B4HD0888@cougar.noc.ucla.edu')
  1367.         eq(dsn1.get_param('dns', header = 'reporting-mta'), '')
  1368.         eq(dsn1.get_param('nsd', header = 'reporting-mta'), None)
  1369.         dsn2 = subpart.get_payload(1)
  1370.         unless(isinstance(dsn2, Message))
  1371.         eq(dsn2['action'], 'failed')
  1372.         eq(dsn2.get_params(header = 'original-recipient'), [
  1373.             ('rfc822', ''),
  1374.             ('jangel1@cougar.noc.ucla.edu', '')])
  1375.         eq(dsn2.get_param('rfc822', header = 'final-recipient'), '')
  1376.         subpart = msg.get_payload(2)
  1377.         eq(subpart.get_content_type(), 'message/rfc822')
  1378.         payload = subpart.get_payload()
  1379.         unless(isinstance(payload, list))
  1380.         eq(len(payload), 1)
  1381.         subsubpart = payload[0]
  1382.         unless(isinstance(subsubpart, Message))
  1383.         eq(subsubpart.get_content_type(), 'text/plain')
  1384.         eq(subsubpart['message-id'], '<002001c144a6$8752e060$56104586@oxy.edu>')
  1385.  
  1386.     
  1387.     def test_epilogue(self):
  1388.         eq = self.ndiffAssertEqual
  1389.         fp = openfile('msg_21.txt')
  1390.         
  1391.         try:
  1392.             text = fp.read()
  1393.         finally:
  1394.             fp.close()
  1395.  
  1396.         msg = Message()
  1397.         msg['From'] = 'aperson@dom.ain'
  1398.         msg['To'] = 'bperson@dom.ain'
  1399.         msg['Subject'] = 'Test'
  1400.         msg.preamble = 'MIME message'
  1401.         msg.epilogue = 'End of MIME message\n'
  1402.         msg1 = MIMEText('One')
  1403.         msg2 = MIMEText('Two')
  1404.         msg.add_header('Content-Type', 'multipart/mixed', boundary = 'BOUNDARY')
  1405.         msg.attach(msg1)
  1406.         msg.attach(msg2)
  1407.         sfp = StringIO()
  1408.         g = Generator(sfp)
  1409.         g.flatten(msg)
  1410.         eq(sfp.getvalue(), text)
  1411.  
  1412.     
  1413.     def test_no_nl_preamble(self):
  1414.         eq = self.ndiffAssertEqual
  1415.         msg = Message()
  1416.         msg['From'] = 'aperson@dom.ain'
  1417.         msg['To'] = 'bperson@dom.ain'
  1418.         msg['Subject'] = 'Test'
  1419.         msg.preamble = 'MIME message'
  1420.         msg.epilogue = ''
  1421.         msg1 = MIMEText('One')
  1422.         msg2 = MIMEText('Two')
  1423.         msg.add_header('Content-Type', 'multipart/mixed', boundary = 'BOUNDARY')
  1424.         msg.attach(msg1)
  1425.         msg.attach(msg2)
  1426.         eq(msg.as_string(), 'From: aperson@dom.ain\nTo: bperson@dom.ain\nSubject: Test\nContent-Type: multipart/mixed; boundary="BOUNDARY"\n\nMIME message\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nOne\n--BOUNDARY\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nTwo\n--BOUNDARY--\n')
  1427.  
  1428.     
  1429.     def test_default_type(self):
  1430.         eq = self.assertEqual
  1431.         fp = openfile('msg_30.txt')
  1432.         
  1433.         try:
  1434.             msg = email.message_from_file(fp)
  1435.         finally:
  1436.             fp.close()
  1437.  
  1438.         container1 = msg.get_payload(0)
  1439.         eq(container1.get_default_type(), 'message/rfc822')
  1440.         eq(container1.get_content_type(), 'message/rfc822')
  1441.         container2 = msg.get_payload(1)
  1442.         eq(container2.get_default_type(), 'message/rfc822')
  1443.         eq(container2.get_content_type(), 'message/rfc822')
  1444.         container1a = container1.get_payload(0)
  1445.         eq(container1a.get_default_type(), 'text/plain')
  1446.         eq(container1a.get_content_type(), 'text/plain')
  1447.         container2a = container2.get_payload(0)
  1448.         eq(container2a.get_default_type(), 'text/plain')
  1449.         eq(container2a.get_content_type(), 'text/plain')
  1450.  
  1451.     
  1452.     def test_default_type_with_explicit_container_type(self):
  1453.         eq = self.assertEqual
  1454.         fp = openfile('msg_28.txt')
  1455.         
  1456.         try:
  1457.             msg = email.message_from_file(fp)
  1458.         finally:
  1459.             fp.close()
  1460.  
  1461.         container1 = msg.get_payload(0)
  1462.         eq(container1.get_default_type(), 'message/rfc822')
  1463.         eq(container1.get_content_type(), 'message/rfc822')
  1464.         container2 = msg.get_payload(1)
  1465.         eq(container2.get_default_type(), 'message/rfc822')
  1466.         eq(container2.get_content_type(), 'message/rfc822')
  1467.         container1a = container1.get_payload(0)
  1468.         eq(container1a.get_default_type(), 'text/plain')
  1469.         eq(container1a.get_content_type(), 'text/plain')
  1470.         container2a = container2.get_payload(0)
  1471.         eq(container2a.get_default_type(), 'text/plain')
  1472.         eq(container2a.get_content_type(), 'text/plain')
  1473.  
  1474.     
  1475.     def test_default_type_non_parsed(self):
  1476.         eq = self.assertEqual
  1477.         neq = self.ndiffAssertEqual
  1478.         container = MIMEMultipart('digest', 'BOUNDARY')
  1479.         container.epilogue = ''
  1480.         subpart1a = MIMEText('message 1\n')
  1481.         subpart2a = MIMEText('message 2\n')
  1482.         subpart1 = MIMEMessage(subpart1a)
  1483.         subpart2 = MIMEMessage(subpart2a)
  1484.         container.attach(subpart1)
  1485.         container.attach(subpart2)
  1486.         eq(subpart1.get_content_type(), 'message/rfc822')
  1487.         eq(subpart1.get_default_type(), 'message/rfc822')
  1488.         eq(subpart2.get_content_type(), 'message/rfc822')
  1489.         eq(subpart2.get_default_type(), 'message/rfc822')
  1490.         neq(container.as_string(0), 'Content-Type: multipart/digest; boundary="BOUNDARY"\nMIME-Version: 1.0\n\n--BOUNDARY\nContent-Type: message/rfc822\nMIME-Version: 1.0\n\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nmessage 1\n\n--BOUNDARY\nContent-Type: message/rfc822\nMIME-Version: 1.0\n\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nmessage 2\n\n--BOUNDARY--\n')
  1491.         del subpart1['content-type']
  1492.         del subpart1['mime-version']
  1493.         del subpart2['content-type']
  1494.         del subpart2['mime-version']
  1495.         eq(subpart1.get_content_type(), 'message/rfc822')
  1496.         eq(subpart1.get_default_type(), 'message/rfc822')
  1497.         eq(subpart2.get_content_type(), 'message/rfc822')
  1498.         eq(subpart2.get_default_type(), 'message/rfc822')
  1499.         neq(container.as_string(0), 'Content-Type: multipart/digest; boundary="BOUNDARY"\nMIME-Version: 1.0\n\n--BOUNDARY\n\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nmessage 1\n\n--BOUNDARY\n\nContent-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\n\nmessage 2\n\n--BOUNDARY--\n')
  1500.  
  1501.     
  1502.     def test_mime_attachments_in_constructor(self):
  1503.         eq = self.assertEqual
  1504.         text1 = MIMEText('')
  1505.         text2 = MIMEText('')
  1506.         msg = MIMEMultipart(_subparts = (text1, text2))
  1507.         eq(len(msg.get_payload()), 2)
  1508.         eq(msg.get_payload(0), text1)
  1509.         eq(msg.get_payload(1), text2)
  1510.  
  1511.  
  1512.  
  1513. class TestIdempotent(TestEmailBase):
  1514.     
  1515.     def _msgobj(self, filename):
  1516.         fp = openfile(filename)
  1517.         
  1518.         try:
  1519.             data = fp.read()
  1520.         finally:
  1521.             fp.close()
  1522.  
  1523.         msg = email.message_from_string(data)
  1524.         return (msg, data)
  1525.  
  1526.     
  1527.     def _idempotent(self, msg, text):
  1528.         eq = self.ndiffAssertEqual
  1529.         s = StringIO()
  1530.         g = Generator(s, maxheaderlen = 0)
  1531.         g.flatten(msg)
  1532.         eq(text, s.getvalue())
  1533.  
  1534.     
  1535.     def test_parse_text_message(self):
  1536.         eq = self.assertEquals
  1537.         (msg, text) = self._msgobj('msg_01.txt')
  1538.         eq(msg.get_content_type(), 'text/plain')
  1539.         eq(msg.get_content_maintype(), 'text')
  1540.         eq(msg.get_content_subtype(), 'plain')
  1541.         eq(msg.get_params()[1], ('charset', 'us-ascii'))
  1542.         eq(msg.get_param('charset'), 'us-ascii')
  1543.         eq(msg.preamble, None)
  1544.         eq(msg.epilogue, None)
  1545.         self._idempotent(msg, text)
  1546.  
  1547.     
  1548.     def test_parse_untyped_message(self):
  1549.         eq = self.assertEquals
  1550.         (msg, text) = self._msgobj('msg_03.txt')
  1551.         eq(msg.get_content_type(), 'text/plain')
  1552.         eq(msg.get_params(), None)
  1553.         eq(msg.get_param('charset'), None)
  1554.         self._idempotent(msg, text)
  1555.  
  1556.     
  1557.     def test_simple_multipart(self):
  1558.         (msg, text) = self._msgobj('msg_04.txt')
  1559.         self._idempotent(msg, text)
  1560.  
  1561.     
  1562.     def test_MIME_digest(self):
  1563.         (msg, text) = self._msgobj('msg_02.txt')
  1564.         self._idempotent(msg, text)
  1565.  
  1566.     
  1567.     def test_long_header(self):
  1568.         (msg, text) = self._msgobj('msg_27.txt')
  1569.         self._idempotent(msg, text)
  1570.  
  1571.     
  1572.     def test_MIME_digest_with_part_headers(self):
  1573.         (msg, text) = self._msgobj('msg_28.txt')
  1574.         self._idempotent(msg, text)
  1575.  
  1576.     
  1577.     def test_mixed_with_image(self):
  1578.         (msg, text) = self._msgobj('msg_06.txt')
  1579.         self._idempotent(msg, text)
  1580.  
  1581.     
  1582.     def test_multipart_report(self):
  1583.         (msg, text) = self._msgobj('msg_05.txt')
  1584.         self._idempotent(msg, text)
  1585.  
  1586.     
  1587.     def test_dsn(self):
  1588.         (msg, text) = self._msgobj('msg_16.txt')
  1589.         self._idempotent(msg, text)
  1590.  
  1591.     
  1592.     def test_preamble_epilogue(self):
  1593.         (msg, text) = self._msgobj('msg_21.txt')
  1594.         self._idempotent(msg, text)
  1595.  
  1596.     
  1597.     def test_multipart_one_part(self):
  1598.         (msg, text) = self._msgobj('msg_23.txt')
  1599.         self._idempotent(msg, text)
  1600.  
  1601.     
  1602.     def test_multipart_no_parts(self):
  1603.         (msg, text) = self._msgobj('msg_24.txt')
  1604.         self._idempotent(msg, text)
  1605.  
  1606.     
  1607.     def test_no_start_boundary(self):
  1608.         (msg, text) = self._msgobj('msg_31.txt')
  1609.         self._idempotent(msg, text)
  1610.  
  1611.     
  1612.     def test_rfc2231_charset(self):
  1613.         (msg, text) = self._msgobj('msg_32.txt')
  1614.         self._idempotent(msg, text)
  1615.  
  1616.     
  1617.     def test_more_rfc2231_parameters(self):
  1618.         (msg, text) = self._msgobj('msg_33.txt')
  1619.         self._idempotent(msg, text)
  1620.  
  1621.     
  1622.     def test_text_plain_in_a_multipart_digest(self):
  1623.         (msg, text) = self._msgobj('msg_34.txt')
  1624.         self._idempotent(msg, text)
  1625.  
  1626.     
  1627.     def test_nested_multipart_mixeds(self):
  1628.         (msg, text) = self._msgobj('msg_12a.txt')
  1629.         self._idempotent(msg, text)
  1630.  
  1631.     
  1632.     def test_message_external_body_idempotent(self):
  1633.         (msg, text) = self._msgobj('msg_36.txt')
  1634.         self._idempotent(msg, text)
  1635.  
  1636.     
  1637.     def test_content_type(self):
  1638.         eq = self.assertEquals
  1639.         unless = self.failUnless
  1640.         (msg, text) = self._msgobj('msg_05.txt')
  1641.         eq(msg.get_content_type(), 'multipart/report')
  1642.         params = { }
  1643.         for pk, pv in msg.get_params():
  1644.             params[pk] = pv
  1645.         
  1646.         eq(params['report-type'], 'delivery-status')
  1647.         eq(params['boundary'], 'D1690A7AC1.996856090/mail.example.com')
  1648.         eq(msg.preamble, 'This is a MIME-encapsulated message.\n')
  1649.         eq(msg.epilogue, '\n')
  1650.         eq(len(msg.get_payload()), 3)
  1651.         msg1 = msg.get_payload(0)
  1652.         eq(msg1.get_content_type(), 'text/plain')
  1653.         eq(msg1.get_payload(), 'Yadda yadda yadda\n')
  1654.         msg2 = msg.get_payload(1)
  1655.         eq(msg2.get_content_type(), 'text/plain')
  1656.         eq(msg2.get_payload(), 'Yadda yadda yadda\n')
  1657.         msg3 = msg.get_payload(2)
  1658.         eq(msg3.get_content_type(), 'message/rfc822')
  1659.         self.failUnless(isinstance(msg3, Message))
  1660.         payload = msg3.get_payload()
  1661.         unless(isinstance(payload, list))
  1662.         eq(len(payload), 1)
  1663.         msg4 = payload[0]
  1664.         unless(isinstance(msg4, Message))
  1665.         eq(msg4.get_payload(), 'Yadda yadda yadda\n')
  1666.  
  1667.     
  1668.     def test_parser(self):
  1669.         eq = self.assertEquals
  1670.         unless = self.failUnless
  1671.         (msg, text) = self._msgobj('msg_06.txt')
  1672.         eq(msg.get_content_type(), 'message/rfc822')
  1673.         payload = msg.get_payload()
  1674.         unless(isinstance(payload, list))
  1675.         eq(len(payload), 1)
  1676.         msg1 = payload[0]
  1677.         self.failUnless(isinstance(msg1, Message))
  1678.         eq(msg1.get_content_type(), 'text/plain')
  1679.         self.failUnless(isinstance(msg1.get_payload(), str))
  1680.         eq(msg1.get_payload(), '\n')
  1681.  
  1682.  
  1683.  
  1684. class TestMiscellaneous(TestEmailBase):
  1685.     
  1686.     def test_message_from_string(self):
  1687.         fp = openfile('msg_01.txt')
  1688.         
  1689.         try:
  1690.             text = fp.read()
  1691.         finally:
  1692.             fp.close()
  1693.  
  1694.         msg = email.message_from_string(text)
  1695.         s = StringIO()
  1696.         g = Generator(s, maxheaderlen = 0)
  1697.         g.flatten(msg)
  1698.         self.assertEqual(text, s.getvalue())
  1699.  
  1700.     
  1701.     def test_message_from_file(self):
  1702.         fp = openfile('msg_01.txt')
  1703.         
  1704.         try:
  1705.             text = fp.read()
  1706.             fp.seek(0)
  1707.             msg = email.message_from_file(fp)
  1708.             s = StringIO()
  1709.             g = Generator(s, maxheaderlen = 0)
  1710.             g.flatten(msg)
  1711.             self.assertEqual(text, s.getvalue())
  1712.         finally:
  1713.             fp.close()
  1714.  
  1715.  
  1716.     
  1717.     def test_message_from_string_with_class(self):
  1718.         unless = self.failUnless
  1719.         fp = openfile('msg_01.txt')
  1720.         
  1721.         try:
  1722.             text = fp.read()
  1723.         finally:
  1724.             fp.close()
  1725.  
  1726.         
  1727.         class MyMessage(Message):
  1728.             pass
  1729.  
  1730.         msg = email.message_from_string(text, MyMessage)
  1731.         unless(isinstance(msg, MyMessage))
  1732.         fp = openfile('msg_02.txt')
  1733.         
  1734.         try:
  1735.             text = fp.read()
  1736.         finally:
  1737.             fp.close()
  1738.  
  1739.         msg = email.message_from_string(text, MyMessage)
  1740.         for subpart in msg.walk():
  1741.             unless(isinstance(subpart, MyMessage))
  1742.         
  1743.  
  1744.     
  1745.     def test_message_from_file_with_class(self):
  1746.         unless = self.failUnless
  1747.         
  1748.         class MyMessage(Message):
  1749.             pass
  1750.  
  1751.         fp = openfile('msg_01.txt')
  1752.         
  1753.         try:
  1754.             msg = email.message_from_file(fp, MyMessage)
  1755.         finally:
  1756.             fp.close()
  1757.  
  1758.         unless(isinstance(msg, MyMessage))
  1759.         fp = openfile('msg_02.txt')
  1760.         
  1761.         try:
  1762.             msg = email.message_from_file(fp, MyMessage)
  1763.         finally:
  1764.             fp.close()
  1765.  
  1766.         for subpart in msg.walk():
  1767.             unless(isinstance(subpart, MyMessage))
  1768.         
  1769.  
  1770.     
  1771.     def test__all__(self):
  1772.         module = __import__('email')
  1773.         all = module.__all__[:]
  1774.         all.sort()
  1775.         self.assertEqual(all, [
  1776.             'Charset',
  1777.             'Encoders',
  1778.             'Errors',
  1779.             'Generator',
  1780.             'Header',
  1781.             'Iterators',
  1782.             'MIMEAudio',
  1783.             'MIMEBase',
  1784.             'MIMEImage',
  1785.             'MIMEMessage',
  1786.             'MIMEMultipart',
  1787.             'MIMENonMultipart',
  1788.             'MIMEText',
  1789.             'Message',
  1790.             'Parser',
  1791.             'Utils',
  1792.             'base64MIME',
  1793.             'base64mime',
  1794.             'charset',
  1795.             'encoders',
  1796.             'errors',
  1797.             'generator',
  1798.             'header',
  1799.             'iterators',
  1800.             'message',
  1801.             'message_from_file',
  1802.             'message_from_string',
  1803.             'mime',
  1804.             'parser',
  1805.             'quopriMIME',
  1806.             'quoprimime',
  1807.             'utils'])
  1808.  
  1809.     
  1810.     def test_formatdate(self):
  1811.         now = time.time()
  1812.         self.assertEqual(utils.parsedate(utils.formatdate(now))[:6], time.gmtime(now)[:6])
  1813.  
  1814.     
  1815.     def test_formatdate_localtime(self):
  1816.         now = time.time()
  1817.         self.assertEqual(utils.parsedate(utils.formatdate(now, localtime = True))[:6], time.localtime(now)[:6])
  1818.  
  1819.     
  1820.     def test_formatdate_usegmt(self):
  1821.         now = time.time()
  1822.         self.assertEqual(utils.formatdate(now, localtime = False), time.strftime('%a, %d %b %Y %H:%M:%S -0000', time.gmtime(now)))
  1823.         self.assertEqual(utils.formatdate(now, localtime = False, usegmt = True), time.strftime('%a, %d %b %Y %H:%M:%S GMT', time.gmtime(now)))
  1824.  
  1825.     
  1826.     def test_parsedate_none(self):
  1827.         self.assertEqual(utils.parsedate(''), None)
  1828.  
  1829.     
  1830.     def test_parsedate_compact(self):
  1831.         self.assertEqual(utils.parsedate('Wed,3 Apr 2002 14:58:26 +0800'), utils.parsedate('Wed, 3 Apr 2002 14:58:26 +0800'))
  1832.  
  1833.     
  1834.     def test_parsedate_no_dayofweek(self):
  1835.         eq = self.assertEqual
  1836.         eq(utils.parsedate_tz('25 Feb 2003 13:47:26 -0800'), (2003, 2, 25, 13, 47, 26, 0, 1, -1, -28800))
  1837.  
  1838.     
  1839.     def test_parsedate_compact_no_dayofweek(self):
  1840.         eq = self.assertEqual
  1841.         eq(utils.parsedate_tz('5 Feb 2003 13:47:26 -0800'), (2003, 2, 5, 13, 47, 26, 0, 1, -1, -28800))
  1842.  
  1843.     
  1844.     def test_parsedate_acceptable_to_time_functions(self):
  1845.         eq = self.assertEqual
  1846.         timetup = utils.parsedate('5 Feb 2003 13:47:26 -0800')
  1847.         t = int(time.mktime(timetup))
  1848.         eq(time.localtime(t)[:6], timetup[:6])
  1849.         eq(int(time.strftime('%Y', timetup)), 2003)
  1850.         timetup = utils.parsedate_tz('5 Feb 2003 13:47:26 -0800')
  1851.         t = int(time.mktime(timetup[:9]))
  1852.         eq(time.localtime(t)[:6], timetup[:6])
  1853.         eq(int(time.strftime('%Y', timetup[:9])), 2003)
  1854.  
  1855.     
  1856.     def test_parseaddr_empty(self):
  1857.         self.assertEqual(utils.parseaddr('<>'), ('', ''))
  1858.         self.assertEqual(utils.formataddr(utils.parseaddr('<>')), '')
  1859.  
  1860.     
  1861.     def test_noquote_dump(self):
  1862.         self.assertEqual(utils.formataddr(('A Silly Person', 'person@dom.ain')), 'A Silly Person <person@dom.ain>')
  1863.  
  1864.     
  1865.     def test_escape_dump(self):
  1866.         self.assertEqual(utils.formataddr(('A (Very) Silly Person', 'person@dom.ain')), '"A \\(Very\\) Silly Person" <person@dom.ain>')
  1867.         a = 'A \\(Special\\) Person'
  1868.         b = 'person@dom.ain'
  1869.         self.assertEqual(utils.parseaddr(utils.formataddr((a, b))), (a, b))
  1870.  
  1871.     
  1872.     def test_escape_backslashes(self):
  1873.         self.assertEqual(utils.formataddr(('Arthur \\Backslash\\ Foobar', 'person@dom.ain')), '"Arthur \\\\Backslash\\\\ Foobar" <person@dom.ain>')
  1874.         a = 'Arthur \\Backslash\\ Foobar'
  1875.         b = 'person@dom.ain'
  1876.         self.assertEqual(utils.parseaddr(utils.formataddr((a, b))), (a, b))
  1877.  
  1878.     
  1879.     def test_name_with_dot(self):
  1880.         x = 'John X. Doe <jxd@example.com>'
  1881.         y = '"John X. Doe" <jxd@example.com>'
  1882.         (a, b) = ('John X. Doe', 'jxd@example.com')
  1883.         self.assertEqual(utils.parseaddr(x), (a, b))
  1884.         self.assertEqual(utils.parseaddr(y), (a, b))
  1885.         self.assertEqual(utils.formataddr((a, b)), y)
  1886.  
  1887.     
  1888.     def test_multiline_from_comment(self):
  1889.         x = 'Foo\n\tBar <foo@example.com>'
  1890.         self.assertEqual(utils.parseaddr(x), ('Foo Bar', 'foo@example.com'))
  1891.  
  1892.     
  1893.     def test_quote_dump(self):
  1894.         self.assertEqual(utils.formataddr(('A Silly; Person', 'person@dom.ain')), '"A Silly; Person" <person@dom.ain>')
  1895.  
  1896.     
  1897.     def test_fix_eols(self):
  1898.         eq = self.assertEqual
  1899.         eq(utils.fix_eols('hello'), 'hello')
  1900.         eq(utils.fix_eols('hello\n'), 'hello\r\n')
  1901.         eq(utils.fix_eols('hello\r'), 'hello\r\n')
  1902.         eq(utils.fix_eols('hello\r\n'), 'hello\r\n')
  1903.         eq(utils.fix_eols('hello\n\r'), 'hello\r\n\r\n')
  1904.  
  1905.     
  1906.     def test_charset_richcomparisons(self):
  1907.         eq = self.assertEqual
  1908.         ne = self.failIfEqual
  1909.         cset1 = Charset()
  1910.         cset2 = Charset()
  1911.         eq(cset1, 'us-ascii')
  1912.         eq(cset1, 'US-ASCII')
  1913.         eq(cset1, 'Us-AsCiI')
  1914.         eq('us-ascii', cset1)
  1915.         eq('US-ASCII', cset1)
  1916.         eq('Us-AsCiI', cset1)
  1917.         ne(cset1, 'usascii')
  1918.         ne(cset1, 'USASCII')
  1919.         ne(cset1, 'UsAsCiI')
  1920.         ne('usascii', cset1)
  1921.         ne('USASCII', cset1)
  1922.         ne('UsAsCiI', cset1)
  1923.         eq(cset1, cset2)
  1924.         eq(cset2, cset1)
  1925.  
  1926.     
  1927.     def test_getaddresses(self):
  1928.         eq = self.assertEqual
  1929.         eq(utils.getaddresses([
  1930.             'aperson@dom.ain (Al Person)',
  1931.             'Bud Person <bperson@dom.ain>']), [
  1932.             ('Al Person', 'aperson@dom.ain'),
  1933.             ('Bud Person', 'bperson@dom.ain')])
  1934.  
  1935.     
  1936.     def test_getaddresses_nasty(self):
  1937.         eq = self.assertEqual
  1938.         eq(utils.getaddresses([
  1939.             'foo: ;']), [
  1940.             ('', '')])
  1941.         eq(utils.getaddresses([
  1942.             '[]*-- =~$']), [
  1943.             ('', ''),
  1944.             ('', ''),
  1945.             ('', '*--')])
  1946.         eq(utils.getaddresses([
  1947.             'foo: ;',
  1948.             '"Jason R. Mastaler" <jason@dom.ain>']), [
  1949.             ('', ''),
  1950.             ('Jason R. Mastaler', 'jason@dom.ain')])
  1951.  
  1952.     
  1953.     def test_getaddresses_embedded_comment(self):
  1954.         eq = self.assertEqual
  1955.         addrs = utils.getaddresses([
  1956.             'User ((nested comment)) <foo@bar.com>'])
  1957.         eq(addrs[0][1], 'foo@bar.com')
  1958.  
  1959.     
  1960.     def test_utils_quote_unquote(self):
  1961.         eq = self.assertEqual
  1962.         msg = Message()
  1963.         msg.add_header('content-disposition', 'attachment', filename = 'foo\\wacky"name')
  1964.         eq(msg.get_filename(), 'foo\\wacky"name')
  1965.  
  1966.     
  1967.     def test_get_body_encoding_with_bogus_charset(self):
  1968.         charset = Charset('not a charset')
  1969.         self.assertEqual(charset.get_body_encoding(), 'base64')
  1970.  
  1971.     
  1972.     def test_get_body_encoding_with_uppercase_charset(self):
  1973.         eq = self.assertEqual
  1974.         msg = Message()
  1975.         msg['Content-Type'] = 'text/plain; charset=UTF-8'
  1976.         eq(msg['content-type'], 'text/plain; charset=UTF-8')
  1977.         charsets = msg.get_charsets()
  1978.         eq(len(charsets), 1)
  1979.         eq(charsets[0], 'utf-8')
  1980.         charset = Charset(charsets[0])
  1981.         eq(charset.get_body_encoding(), 'base64')
  1982.         msg.set_payload('hello world', charset = charset)
  1983.         eq(msg.get_payload(), 'aGVsbG8gd29ybGQ=\n')
  1984.         eq(msg.get_payload(decode = True), 'hello world')
  1985.         eq(msg['content-transfer-encoding'], 'base64')
  1986.         msg = Message()
  1987.         msg['Content-Type'] = 'text/plain; charset="US-ASCII"'
  1988.         charsets = msg.get_charsets()
  1989.         eq(len(charsets), 1)
  1990.         eq(charsets[0], 'us-ascii')
  1991.         charset = Charset(charsets[0])
  1992.         eq(charset.get_body_encoding(), encoders.encode_7or8bit)
  1993.         msg.set_payload('hello world', charset = charset)
  1994.         eq(msg.get_payload(), 'hello world')
  1995.         eq(msg['content-transfer-encoding'], '7bit')
  1996.  
  1997.     
  1998.     def test_charsets_case_insensitive(self):
  1999.         lc = Charset('us-ascii')
  2000.         uc = Charset('US-ASCII')
  2001.         self.assertEqual(lc.get_body_encoding(), uc.get_body_encoding())
  2002.  
  2003.     
  2004.     def test_partial_falls_inside_message_delivery_status(self):
  2005.         eq = self.ndiffAssertEqual
  2006.         msg = self._msgobj('msg_43.txt')
  2007.         sfp = StringIO()
  2008.         iterators._structure(msg, sfp)
  2009.         eq(sfp.getvalue(), 'multipart/report\n    text/plain\n    message/delivery-status\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n        text/plain\n    text/rfc822-headers\n')
  2010.  
  2011.  
  2012.  
  2013. class TestIterators(TestEmailBase):
  2014.     
  2015.     def test_body_line_iterator(self):
  2016.         eq = self.assertEqual
  2017.         neq = self.ndiffAssertEqual
  2018.         msg = self._msgobj('msg_01.txt')
  2019.         it = iterators.body_line_iterator(msg)
  2020.         lines = list(it)
  2021.         eq(len(lines), 6)
  2022.         neq(EMPTYSTRING.join(lines), msg.get_payload())
  2023.         msg = self._msgobj('msg_02.txt')
  2024.         it = iterators.body_line_iterator(msg)
  2025.         lines = list(it)
  2026.         eq(len(lines), 43)
  2027.         fp = openfile('msg_19.txt')
  2028.         
  2029.         try:
  2030.             neq(EMPTYSTRING.join(lines), fp.read())
  2031.         finally:
  2032.             fp.close()
  2033.  
  2034.  
  2035.     
  2036.     def test_typed_subpart_iterator(self):
  2037.         eq = self.assertEqual
  2038.         msg = self._msgobj('msg_04.txt')
  2039.         it = iterators.typed_subpart_iterator(msg, 'text')
  2040.         lines = []
  2041.         subparts = 0
  2042.         for subpart in it:
  2043.             subparts += 1
  2044.             lines.append(subpart.get_payload())
  2045.         
  2046.         eq(subparts, 2)
  2047.         eq(EMPTYSTRING.join(lines), 'a simple kind of mirror\nto reflect upon our own\na simple kind of mirror\nto reflect upon our own\n')
  2048.  
  2049.     
  2050.     def test_typed_subpart_iterator_default_type(self):
  2051.         eq = self.assertEqual
  2052.         msg = self._msgobj('msg_03.txt')
  2053.         it = iterators.typed_subpart_iterator(msg, 'text', 'plain')
  2054.         lines = []
  2055.         subparts = 0
  2056.         for subpart in it:
  2057.             subparts += 1
  2058.             lines.append(subpart.get_payload())
  2059.         
  2060.         eq(subparts, 1)
  2061.         eq(EMPTYSTRING.join(lines), '\nHi,\n\nDo you like this message?\n\n-Me\n')
  2062.  
  2063.  
  2064.  
  2065. class TestParsers(TestEmailBase):
  2066.     
  2067.     def test_header_parser(self):
  2068.         eq = self.assertEqual
  2069.         fp = openfile('msg_02.txt')
  2070.         
  2071.         try:
  2072.             msg = HeaderParser().parse(fp)
  2073.         finally:
  2074.             fp.close()
  2075.  
  2076.         eq(msg['from'], 'ppp-request@zzz.org')
  2077.         eq(msg['to'], 'ppp@zzz.org')
  2078.         eq(msg.get_content_type(), 'multipart/mixed')
  2079.         self.failIf(msg.is_multipart())
  2080.         self.failUnless(isinstance(msg.get_payload(), str))
  2081.  
  2082.     
  2083.     def test_whitespace_continuation(self):
  2084.         eq = self.assertEqual
  2085.         msg = email.message_from_string("From: aperson@dom.ain\nTo: bperson@dom.ain\nSubject: the next line has a space on it\n \nDate: Mon, 8 Apr 2002 15:09:19 -0400\nMessage-ID: spam\n\nHere's the message body\n")
  2086.         eq(msg['subject'], 'the next line has a space on it\n ')
  2087.         eq(msg['message-id'], 'spam')
  2088.         eq(msg.get_payload(), "Here's the message body\n")
  2089.  
  2090.     
  2091.     def test_whitespace_continuation_last_header(self):
  2092.         eq = self.assertEqual
  2093.         msg = email.message_from_string("From: aperson@dom.ain\nTo: bperson@dom.ain\nDate: Mon, 8 Apr 2002 15:09:19 -0400\nMessage-ID: spam\nSubject: the next line has a space on it\n \n\nHere's the message body\n")
  2094.         eq(msg['subject'], 'the next line has a space on it\n ')
  2095.         eq(msg['message-id'], 'spam')
  2096.         eq(msg.get_payload(), "Here's the message body\n")
  2097.  
  2098.     
  2099.     def test_crlf_separation(self):
  2100.         eq = self.assertEqual
  2101.         fp = openfile('msg_26.txt', mode = 'rb')
  2102.         
  2103.         try:
  2104.             msg = Parser().parse(fp)
  2105.         finally:
  2106.             fp.close()
  2107.  
  2108.         eq(len(msg.get_payload()), 2)
  2109.         part1 = msg.get_payload(0)
  2110.         eq(part1.get_content_type(), 'text/plain')
  2111.         eq(part1.get_payload(), 'Simple email with attachment.\r\n\r\n')
  2112.         part2 = msg.get_payload(1)
  2113.         eq(part2.get_content_type(), 'application/riscos')
  2114.  
  2115.     
  2116.     def test_multipart_digest_with_extra_mime_headers(self):
  2117.         eq = self.assertEqual
  2118.         neq = self.ndiffAssertEqual
  2119.         fp = openfile('msg_28.txt')
  2120.         
  2121.         try:
  2122.             msg = email.message_from_file(fp)
  2123.         finally:
  2124.             fp.close()
  2125.  
  2126.         eq(msg.is_multipart(), 1)
  2127.         eq(len(msg.get_payload()), 2)
  2128.         part1 = msg.get_payload(0)
  2129.         eq(part1.get_content_type(), 'message/rfc822')
  2130.         eq(part1.is_multipart(), 1)
  2131.         eq(len(part1.get_payload()), 1)
  2132.         part1a = part1.get_payload(0)
  2133.         eq(part1a.is_multipart(), 0)
  2134.         eq(part1a.get_content_type(), 'text/plain')
  2135.         neq(part1a.get_payload(), 'message 1\n')
  2136.         part2 = msg.get_payload(1)
  2137.         eq(part2.get_content_type(), 'message/rfc822')
  2138.         eq(part2.is_multipart(), 1)
  2139.         eq(len(part2.get_payload()), 1)
  2140.         part2a = part2.get_payload(0)
  2141.         eq(part2a.is_multipart(), 0)
  2142.         eq(part2a.get_content_type(), 'text/plain')
  2143.         neq(part2a.get_payload(), 'message 2\n')
  2144.  
  2145.     
  2146.     def test_three_lines(self):
  2147.         lines = [
  2148.             'From: Andrew Person <aperson@dom.ain',
  2149.             'Subject: Test',
  2150.             'Date: Tue, 20 Aug 2002 16:43:45 +1000']
  2151.         msg = email.message_from_string(NL.join(lines))
  2152.         self.assertEqual(msg['date'], 'Tue, 20 Aug 2002 16:43:45 +1000')
  2153.  
  2154.     
  2155.     def test_strip_line_feed_and_carriage_return_in_headers(self):
  2156.         eq = self.assertEqual
  2157.         value1 = 'text'
  2158.         value2 = 'more text'
  2159.         m = 'Header: %s\r\nNext-Header: %s\r\n\r\nBody\r\n\r\n' % (value1, value2)
  2160.         msg = email.message_from_string(m)
  2161.         eq(msg.get('Header'), value1)
  2162.         eq(msg.get('Next-Header'), value2)
  2163.  
  2164.     
  2165.     def test_rfc2822_header_syntax(self):
  2166.         eq = self.assertEqual
  2167.         m = '>From: foo\nFrom: bar\n!"#QUX;~: zoo\n\nbody'
  2168.         msg = email.message_from_string(m)
  2169.         eq(len(msg.keys()), 3)
  2170.         keys = msg.keys()
  2171.         keys.sort()
  2172.         eq(keys, [
  2173.             '!"#QUX;~',
  2174.             '>From',
  2175.             'From'])
  2176.         eq(msg.get_payload(), 'body')
  2177.  
  2178.     
  2179.     def test_rfc2822_space_not_allowed_in_header(self):
  2180.         eq = self.assertEqual
  2181.         m = '>From foo@example.com 11:25:53\nFrom: bar\n!"#QUX;~: zoo\n\nbody'
  2182.         msg = email.message_from_string(m)
  2183.         eq(len(msg.keys()), 0)
  2184.  
  2185.     
  2186.     def test_rfc2822_one_character_header(self):
  2187.         eq = self.assertEqual
  2188.         m = 'A: first header\nB: second header\nCC: third header\n\nbody'
  2189.         msg = email.message_from_string(m)
  2190.         headers = msg.keys()
  2191.         headers.sort()
  2192.         eq(headers, [
  2193.             'A',
  2194.             'B',
  2195.             'CC'])
  2196.         eq(msg.get_payload(), 'body')
  2197.  
  2198.  
  2199.  
  2200. class TestBase64(unittest.TestCase):
  2201.     
  2202.     def test_len(self):
  2203.         eq = self.assertEqual
  2204.         eq(base64mime.base64_len('hello'), len(base64mime.encode('hello', eol = '')))
  2205.         for size in range(15):
  2206.             if size == 0:
  2207.                 bsize = 0
  2208.             elif size <= 3:
  2209.                 bsize = 4
  2210.             elif size <= 6:
  2211.                 bsize = 8
  2212.             elif size <= 9:
  2213.                 bsize = 12
  2214.             elif size <= 12:
  2215.                 bsize = 16
  2216.             else:
  2217.                 bsize = 20
  2218.             eq(base64mime.base64_len('x' * size), bsize)
  2219.         
  2220.  
  2221.     
  2222.     def test_decode(self):
  2223.         eq = self.assertEqual
  2224.         eq(base64mime.decode(''), '')
  2225.         eq(base64mime.decode('aGVsbG8='), 'hello')
  2226.         eq(base64mime.decode('aGVsbG8=', 'X'), 'hello')
  2227.         eq(base64mime.decode('aGVsbG8NCndvcmxk\n', 'X'), 'helloXworld')
  2228.  
  2229.     
  2230.     def test_encode(self):
  2231.         eq = self.assertEqual
  2232.         eq(base64mime.encode(''), '')
  2233.         eq(base64mime.encode('hello'), 'aGVsbG8=\n')
  2234.         eq(base64mime.encode('hello\n'), 'aGVsbG8K\n')
  2235.         eq(base64mime.encode('hello\n', 0), 'aGVsbG8NCg==\n')
  2236.         eq(base64mime.encode('xxxx ' * 20, maxlinelen = 40), 'eHh4eCB4eHh4IHh4eHggeHh4eCB4eHh4IHh4eHgg\neHh4eCB4eHh4IHh4eHggeHh4eCB4eHh4IHh4eHgg\neHh4eCB4eHh4IHh4eHggeHh4eCB4eHh4IHh4eHgg\neHh4eCB4eHh4IA==\n')
  2237.         eq(base64mime.encode('xxxx ' * 20, maxlinelen = 40, eol = '\r\n'), 'eHh4eCB4eHh4IHh4eHggeHh4eCB4eHh4IHh4eHgg\r\neHh4eCB4eHh4IHh4eHggeHh4eCB4eHh4IHh4eHgg\r\neHh4eCB4eHh4IHh4eHggeHh4eCB4eHh4IHh4eHgg\r\neHh4eCB4eHh4IA==\r\n')
  2238.  
  2239.     
  2240.     def test_header_encode(self):
  2241.         eq = self.assertEqual
  2242.         he = base64mime.header_encode
  2243.         eq(he('hello'), '=?iso-8859-1?b?aGVsbG8=?=')
  2244.         eq(he('hello\nworld'), '=?iso-8859-1?b?aGVsbG8NCndvcmxk?=')
  2245.         eq(he('hello', charset = 'iso-8859-2'), '=?iso-8859-2?b?aGVsbG8=?=')
  2246.         eq(he('hello\nworld', keep_eols = True), '=?iso-8859-1?b?aGVsbG8Kd29ybGQ=?=')
  2247.         eq(he('xxxx ' * 20, maxlinelen = 40), '=?iso-8859-1?b?eHh4eCB4eHh4IHh4eHggeHg=?=\n =?iso-8859-1?b?eHggeHh4eCB4eHh4IHh4eHg=?=\n =?iso-8859-1?b?IHh4eHggeHh4eCB4eHh4IHg=?=\n =?iso-8859-1?b?eHh4IHh4eHggeHh4eCB4eHg=?=\n =?iso-8859-1?b?eCB4eHh4IHh4eHggeHh4eCA=?=\n =?iso-8859-1?b?eHh4eCB4eHh4IHh4eHgg?=')
  2248.         eq(he('xxxx ' * 20, maxlinelen = 40, eol = '\r\n'), '=?iso-8859-1?b?eHh4eCB4eHh4IHh4eHggeHg=?=\r\n =?iso-8859-1?b?eHggeHh4eCB4eHh4IHh4eHg=?=\r\n =?iso-8859-1?b?IHh4eHggeHh4eCB4eHh4IHg=?=\r\n =?iso-8859-1?b?eHh4IHh4eHggeHh4eCB4eHg=?=\r\n =?iso-8859-1?b?eCB4eHh4IHh4eHggeHh4eCA=?=\r\n =?iso-8859-1?b?eHh4eCB4eHh4IHh4eHgg?=')
  2249.  
  2250.  
  2251.  
  2252. class TestQuopri(unittest.TestCase):
  2253.     
  2254.     def setUp(self):
  2255.         self.hlit = [] + [ chr(x) for x in range(ord('0'), ord('9') + 1) ] + [
  2256.             '!',
  2257.             '*',
  2258.             '+',
  2259.             '-',
  2260.             '/',
  2261.             ' ']
  2262.         self.hnon = _[4]
  2263.         self.blit = [ chr(x) for x in range(ord(' '), ord('~') + 1) ] + [
  2264.             '\t']
  2265.         self.blit.remove('=')
  2266.         self.bnon = _[6]
  2267.  
  2268.     
  2269.     def test_header_quopri_check(self):
  2270.         for c in self.hlit:
  2271.             self.failIf(quoprimime.header_quopri_check(c))
  2272.         
  2273.         for c in self.hnon:
  2274.             self.failUnless(quoprimime.header_quopri_check(c))
  2275.         
  2276.  
  2277.     
  2278.     def test_body_quopri_check(self):
  2279.         for c in self.blit:
  2280.             self.failIf(quoprimime.body_quopri_check(c))
  2281.         
  2282.         for c in self.bnon:
  2283.             self.failUnless(quoprimime.body_quopri_check(c))
  2284.         
  2285.  
  2286.     
  2287.     def test_header_quopri_len(self):
  2288.         eq = self.assertEqual
  2289.         hql = quoprimime.header_quopri_len
  2290.         enc = quoprimime.header_encode
  2291.         for s in ('hello', 'h@e@l@l@o@'):
  2292.             eq(hql(s), len(enc(s, charset = '', eol = '')) - 7)
  2293.         
  2294.         for c in self.hlit:
  2295.             eq(hql(c), 1)
  2296.         
  2297.         for c in self.hnon:
  2298.             eq(hql(c), 3)
  2299.         
  2300.  
  2301.     
  2302.     def test_body_quopri_len(self):
  2303.         eq = self.assertEqual
  2304.         bql = quoprimime.body_quopri_len
  2305.         for c in self.blit:
  2306.             eq(bql(c), 1)
  2307.         
  2308.         for c in self.bnon:
  2309.             eq(bql(c), 3)
  2310.         
  2311.  
  2312.     
  2313.     def test_quote_unquote_idempotent(self):
  2314.         for x in range(256):
  2315.             c = chr(x)
  2316.             self.assertEqual(quoprimime.unquote(quoprimime.quote(c)), c)
  2317.         
  2318.  
  2319.     
  2320.     def test_header_encode(self):
  2321.         eq = self.assertEqual
  2322.         he = quoprimime.header_encode
  2323.         eq(he('hello'), '=?iso-8859-1?q?hello?=')
  2324.         eq(he('hello\nworld'), '=?iso-8859-1?q?hello=0D=0Aworld?=')
  2325.         eq(he('hello', charset = 'iso-8859-2'), '=?iso-8859-2?q?hello?=')
  2326.         eq(he('hello\nworld', keep_eols = True), '=?iso-8859-1?q?hello=0Aworld?=')
  2327.         eq(he('hello\xc7there'), '=?iso-8859-1?q?hello=C7there?=')
  2328.         eq(he('xxxx ' * 20, maxlinelen = 40), '=?iso-8859-1?q?xxxx_xxxx_xxxx_xxxx_xx?=\n =?iso-8859-1?q?xx_xxxx_xxxx_xxxx_xxxx?=\n =?iso-8859-1?q?_xxxx_xxxx_xxxx_xxxx_x?=\n =?iso-8859-1?q?xxx_xxxx_xxxx_xxxx_xxx?=\n =?iso-8859-1?q?x_xxxx_xxxx_?=')
  2329.         eq(he('xxxx ' * 20, maxlinelen = 40, eol = '\r\n'), '=?iso-8859-1?q?xxxx_xxxx_xxxx_xxxx_xx?=\r\n =?iso-8859-1?q?xx_xxxx_xxxx_xxxx_xxxx?=\r\n =?iso-8859-1?q?_xxxx_xxxx_xxxx_xxxx_x?=\r\n =?iso-8859-1?q?xxx_xxxx_xxxx_xxxx_xxx?=\r\n =?iso-8859-1?q?x_xxxx_xxxx_?=')
  2330.  
  2331.     
  2332.     def test_decode(self):
  2333.         eq = self.assertEqual
  2334.         eq(quoprimime.decode(''), '')
  2335.         eq(quoprimime.decode('hello'), 'hello')
  2336.         eq(quoprimime.decode('hello', 'X'), 'hello')
  2337.         eq(quoprimime.decode('hello\nworld', 'X'), 'helloXworld')
  2338.  
  2339.     
  2340.     def test_encode(self):
  2341.         eq = self.assertEqual
  2342.         eq(quoprimime.encode(''), '')
  2343.         eq(quoprimime.encode('hello'), 'hello')
  2344.         eq(quoprimime.encode('hello\r\nworld'), 'hello\nworld')
  2345.         eq(quoprimime.encode('hello\r\nworld', 0), 'hello\nworld')
  2346.         eq(quoprimime.encode('xxxx ' * 20, maxlinelen = 40), 'xxxx xxxx xxxx xxxx xxxx xxxx xxxx xxxx=\n xxxx xxxx xxxx xxxx xxxx xxxx xxxx xxx=\nx xxxx xxxx xxxx xxxx=20')
  2347.         eq(quoprimime.encode('xxxx ' * 20, maxlinelen = 40, eol = '\r\n'), 'xxxx xxxx xxxx xxxx xxxx xxxx xxxx xxxx=\r\n xxxx xxxx xxxx xxxx xxxx xxxx xxxx xxx=\r\nx xxxx xxxx xxxx xxxx=20')
  2348.         eq(quoprimime.encode('one line\n\ntwo line'), 'one line\n\ntwo line')
  2349.  
  2350.  
  2351.  
  2352. class TestCharset(unittest.TestCase):
  2353.     
  2354.     def tearDown(self):
  2355.         CharsetModule = charset
  2356.         import email
  2357.         
  2358.         try:
  2359.             del CharsetModule.CHARSETS['fake']
  2360.         except KeyError:
  2361.             pass
  2362.  
  2363.  
  2364.     
  2365.     def test_idempotent(self):
  2366.         eq = self.assertEqual
  2367.         c = Charset('us-ascii')
  2368.         s = 'Hello World!'
  2369.         sp = c.to_splittable(s)
  2370.         eq(s, c.from_splittable(sp))
  2371.         s = '\xa4\xa2\xa4\xa4\xa4\xa6\xa4\xa8\xa4\xaa'
  2372.         sp = c.to_splittable(s)
  2373.         eq(s, c.from_splittable(sp))
  2374.  
  2375.     
  2376.     def test_body_encode(self):
  2377.         eq = self.assertEqual
  2378.         c = Charset('iso-8859-1')
  2379.         eq('hello w=F6rld', c.body_encode('hello w\xf6rld'))
  2380.         c = Charset('utf-8')
  2381.         eq('aGVsbG8gd29ybGQ=\n', c.body_encode('hello world'))
  2382.         c = Charset('us-ascii')
  2383.         eq('hello world', c.body_encode('hello world'))
  2384.         c = Charset('euc-jp')
  2385.         
  2386.         try:
  2387.             eq('\x1b$B5FCO;~IW\x1b(B', c.body_encode('\xb5\xc6\xc3\xcf\xbb\xfe\xc9\xd7'))
  2388.             eq('\xb5\xc6\xc3\xcf\xbb\xfe\xc9\xd7', c.body_encode('\xb5\xc6\xc3\xcf\xbb\xfe\xc9\xd7', False))
  2389.         except LookupError:
  2390.             pass
  2391.  
  2392.         CharsetModule = charset
  2393.         import email
  2394.         CharsetModule.add_charset('fake', CharsetModule.QP, None)
  2395.         c = Charset('fake')
  2396.         eq('hello w\xf6rld', c.body_encode('hello w\xf6rld'))
  2397.  
  2398.     
  2399.     def test_unicode_charset_name(self):
  2400.         charset = Charset(u'us-ascii')
  2401.         self.assertEqual(str(charset), 'us-ascii')
  2402.         self.assertRaises(errors.CharsetError, Charset, 'asc\xffii')
  2403.  
  2404.  
  2405.  
  2406. class TestHeader(TestEmailBase):
  2407.     
  2408.     def test_simple(self):
  2409.         eq = self.ndiffAssertEqual
  2410.         h = Header('Hello World!')
  2411.         eq(h.encode(), 'Hello World!')
  2412.         h.append(' Goodbye World!')
  2413.         eq(h.encode(), 'Hello World!  Goodbye World!')
  2414.  
  2415.     
  2416.     def test_simple_surprise(self):
  2417.         eq = self.ndiffAssertEqual
  2418.         h = Header('Hello World!')
  2419.         eq(h.encode(), 'Hello World!')
  2420.         h.append('Goodbye World!')
  2421.         eq(h.encode(), 'Hello World! Goodbye World!')
  2422.  
  2423.     
  2424.     def test_header_needs_no_decoding(self):
  2425.         h = 'no decoding needed'
  2426.         self.assertEqual(decode_header(h), [
  2427.             (h, None)])
  2428.  
  2429.     
  2430.     def test_long(self):
  2431.         h = Header("I am the very model of a modern Major-General; I've information vegetable, animal, and mineral; I know the kings of England, and I quote the fights historical from Marathon to Waterloo, in order categorical; I'm very well acquainted, too, with matters mathematical; I understand equations, both the simple and quadratical; about binomial theorem I'm teeming with a lot o' news, with many cheerful facts about the square of the hypotenuse.", maxlinelen = 76)
  2432.         for l in h.encode(splitchars = ' ').split('\n '):
  2433.             self.failUnless(len(l) <= 76)
  2434.         
  2435.  
  2436.     
  2437.     def test_multilingual(self):
  2438.         eq = self.ndiffAssertEqual
  2439.         g = Charset('iso-8859-1')
  2440.         cz = Charset('iso-8859-2')
  2441.         utf8 = Charset('utf-8')
  2442.         g_head = 'Die Mieter treten hier ein werden mit einem Foerderband komfortabel den Korridor entlang, an s\xfcdl\xfcndischen Wandgem\xe4lden vorbei, gegen die rotierenden Klingen bef\xf6rdert. '
  2443.         cz_head = 'Finan\xe8ni metropole se hroutily pod tlakem jejich d\xf9vtipu.. '
  2444.         utf8_head = u'µ¡úτó║πü½Φ¿Çπüåπü¿τ┐╗Φ¿│πü»πüòπéîπüªπüäπü╛πü¢πéôπÇéΣ╕ÇΘâ¿πü»πâëπéñπâäΦ¬₧πüºπüÖπüîπÇüπüéπü¿πü»πüºπüƒπéëπéüπüºπüÖπÇéσ«ƒΘÜ¢πü½πü»πÇîWenn ist das Nunstuck git und Slotermeyer? Ja! Beiherhund das Oder die Flipperwaldt gersput.πÇìπü¿Φ¿Çπüúπüªπüäπü╛πüÖπÇé'.encode('utf-8')
  2445.         h = Header(g_head, g)
  2446.         h.append(cz_head, cz)
  2447.         h.append(utf8_head, utf8)
  2448.         enc = h.encode()
  2449.         eq(enc, '=?iso-8859-1?q?Die_Mieter_treten_hier_ein_werden_mit_einem_Foerderband_ko?=\n =?iso-8859-1?q?mfortabel_den_Korridor_entlang=2C_an_s=FCdl=FCndischen_Wan?=\n =?iso-8859-1?q?dgem=E4lden_vorbei=2C_gegen_die_rotierenden_Klingen_bef=F6?=\n =?iso-8859-1?q?rdert=2E_?= =?iso-8859-2?q?Finan=E8ni_metropole_se_hroutily?=\n =?iso-8859-2?q?_pod_tlakem_jejich_d=F9vtipu=2E=2E_?= =?utf-8?b?5q2j56K6?=\n =?utf-8?b?44Gr6KiA44GG44Go57+76Kiz44Gv44GV44KM44Gm44GE44G+44Gb44KT44CC?=\n =?utf-8?b?5LiA6YOo44Gv44OJ44Kk44OE6Kqe44Gn44GZ44GM44CB44GC44Go44Gv44Gn?=\n =?utf-8?b?44Gf44KJ44KB44Gn44GZ44CC5a6f6Zqb44Gr44Gv44CMV2VubiBpc3QgZGFz?=\n =?utf-8?q?_Nunstuck_git_und_Slotermeyer=3F_Ja!_Beiherhund_das_Oder_die_Fl?=\n =?utf-8?b?aXBwZXJ3YWxkdCBnZXJzcHV0LuOAjeOBqOiogOOBo+OBpuOBhOOBvuOBmQ==?=\n =?utf-8?b?44CC?=')
  2450.         eq(decode_header(enc), [
  2451.             (g_head, 'iso-8859-1'),
  2452.             (cz_head, 'iso-8859-2'),
  2453.             (utf8_head, 'utf-8')])
  2454.         ustr = unicode(h)
  2455.         eq(ustr.encode('utf-8'), 'Die Mieter treten hier ein werden mit einem Foerderband komfortabel den Korridor entlang, an s\xc3\xbcdl\xc3\xbcndischen Wandgem\xc3\xa4lden vorbei, gegen die rotierenden Klingen bef\xc3\xb6rdert. Finan\xc4\x8dni metropole se hroutily pod tlakem jejich d\xc5\xafvtipu.. \xe6\xad\xa3\xe7\xa2\xba\xe3\x81\xab\xe8\xa8\x80\xe3\x81\x86\xe3\x81\xa8\xe7\xbf\xbb\xe8\xa8\xb3\xe3\x81\xaf\xe3\x81\x95\xe3\x82\x8c\xe3\x81\xa6\xe3\x81\x84\xe3\x81\xbe\xe3\x81\x9b\xe3\x82\x93\xe3\x80\x82\xe4\xb8\x80\xe9\x83\xa8\xe3\x81\xaf\xe3\x83\x89\xe3\x82\xa4\xe3\x83\x84\xe8\xaa\x9e\xe3\x81\xa7\xe3\x81\x99\xe3\x81\x8c\xe3\x80\x81\xe3\x81\x82\xe3\x81\xa8\xe3\x81\xaf\xe3\x81\xa7\xe3\x81\x9f\xe3\x82\x89\xe3\x82\x81\xe3\x81\xa7\xe3\x81\x99\xe3\x80\x82\xe5\xae\x9f\xe9\x9a\x9b\xe3\x81\xab\xe3\x81\xaf\xe3\x80\x8cWenn ist das Nunstuck git und Slotermeyer? Ja! Beiherhund das Oder die Flipperwaldt gersput.\xe3\x80\x8d\xe3\x81\xa8\xe8\xa8\x80\xe3\x81\xa3\xe3\x81\xa6\xe3\x81\x84\xe3\x81\xbe\xe3\x81\x99\xe3\x80\x82')
  2456.         newh = make_header(decode_header(enc))
  2457.         eq(newh, enc)
  2458.  
  2459.     
  2460.     def test_header_ctor_default_args(self):
  2461.         eq = self.ndiffAssertEqual
  2462.         h = Header()
  2463.         eq(h, '')
  2464.         h.append('foo', Charset('iso-8859-1'))
  2465.         eq(h, '=?iso-8859-1?q?foo?=')
  2466.  
  2467.     
  2468.     def test_explicit_maxlinelen(self):
  2469.         eq = self.ndiffAssertEqual
  2470.         hstr = 'A very long line that must get split to something other than at the 76th character boundary to test the non-default behavior'
  2471.         h = Header(hstr)
  2472.         eq(h.encode(), 'A very long line that must get split to something other than at the 76th\n character boundary to test the non-default behavior')
  2473.         h = Header(hstr, header_name = 'Subject')
  2474.         eq(h.encode(), 'A very long line that must get split to something other than at the\n 76th character boundary to test the non-default behavior')
  2475.         h = Header(hstr, maxlinelen = 1024, header_name = 'Subject')
  2476.         eq(h.encode(), hstr)
  2477.  
  2478.     
  2479.     def test_us_ascii_header(self):
  2480.         eq = self.assertEqual
  2481.         s = 'hello'
  2482.         x = decode_header(s)
  2483.         eq(x, [
  2484.             ('hello', None)])
  2485.         h = make_header(x)
  2486.         eq(s, h.encode())
  2487.  
  2488.     
  2489.     def test_string_charset(self):
  2490.         eq = self.assertEqual
  2491.         h = Header()
  2492.         h.append('hello', 'iso-8859-1')
  2493.         eq(h, '=?iso-8859-1?q?hello?=')
  2494.  
  2495.     
  2496.     def test_utf8_shortest(self):
  2497.         eq = self.assertEqual
  2498.         h = Header(u'p├╢stal', 'utf-8')
  2499.         eq(h.encode(), '=?utf-8?q?p=C3=B6stal?=')
  2500.         h = Header(u'ΦÅèσ£░µÖéσñ½', 'utf-8')
  2501.         eq(h.encode(), '=?utf-8?b?6I+K5Zyw5pmC5aSr?=')
  2502.  
  2503.     
  2504.     def test_bad_8bit_header(self):
  2505.         raises = self.assertRaises
  2506.         eq = self.assertEqual
  2507.         x = 'Ynwp4dUEbay Auction Semiar- No Charge \x96 Earn Big'
  2508.         raises(UnicodeError, Header, x)
  2509.         h = Header()
  2510.         raises(UnicodeError, h.append, x)
  2511.         eq(str(Header(x, errors = 'replace')), x)
  2512.         h.append(x, errors = 'replace')
  2513.         eq(str(h), x)
  2514.  
  2515.     
  2516.     def test_encoded_adjacent_nonencoded(self):
  2517.         eq = self.assertEqual
  2518.         h = Header()
  2519.         h.append('hello', 'iso-8859-1')
  2520.         h.append('world')
  2521.         s = h.encode()
  2522.         eq(s, '=?iso-8859-1?q?hello?= world')
  2523.         h = make_header(decode_header(s))
  2524.         eq(h.encode(), s)
  2525.  
  2526.     
  2527.     def test_whitespace_eater(self):
  2528.         eq = self.assertEqual
  2529.         s = 'Subject: =?koi8-r?b?8NLP18XSy8EgzsEgxsnOwczYztk=?= =?koi8-r?q?=CA?= zz.'
  2530.         parts = decode_header(s)
  2531.         eq(parts, [
  2532.             ('Subject:', None),
  2533.             ('\xf0\xd2\xcf\xd7\xc5\xd2\xcb\xc1 \xce\xc1 \xc6\xc9\xce\xc1\xcc\xd8\xce\xd9\xca', 'koi8-r'),
  2534.             ('zz.', None)])
  2535.         hdr = make_header(parts)
  2536.         eq(hdr.encode(), 'Subject: =?koi8-r?b?8NLP18XSy8EgzsEgxsnOwczYztnK?= zz.')
  2537.  
  2538.     
  2539.     def test_broken_base64_header(self):
  2540.         raises = self.assertRaises
  2541.         s = 'Subject: =?EUC-KR?B?CSixpLDtKSC/7Liuvsax4iC6uLmwMcijIKHaILzSwd/H0SC8+LCjwLsgv7W/+Mj3IQ?='
  2542.         raises(errors.HeaderParseError, decode_header, s)
  2543.  
  2544.  
  2545.  
  2546. class TestRFC2231(TestEmailBase):
  2547.     
  2548.     def test_get_param(self):
  2549.         eq = self.assertEqual
  2550.         msg = self._msgobj('msg_29.txt')
  2551.         eq(msg.get_param('title'), ('us-ascii', 'en', "This is even more ***fun*** isn't it!"))
  2552.         eq(msg.get_param('title', unquote = False), ('us-ascii', 'en', '"This is even more ***fun*** isn\'t it!"'))
  2553.  
  2554.     
  2555.     def test_set_param(self):
  2556.         eq = self.assertEqual
  2557.         msg = Message()
  2558.         msg.set_param('title', "This is even more ***fun*** isn't it!", charset = 'us-ascii')
  2559.         eq(msg.get_param('title'), ('us-ascii', '', "This is even more ***fun*** isn't it!"))
  2560.         msg.set_param('title', "This is even more ***fun*** isn't it!", charset = 'us-ascii', language = 'en')
  2561.         eq(msg.get_param('title'), ('us-ascii', 'en', "This is even more ***fun*** isn't it!"))
  2562.         msg = self._msgobj('msg_01.txt')
  2563.         msg.set_param('title', "This is even more ***fun*** isn't it!", charset = 'us-ascii', language = 'en')
  2564.         eq(msg.as_string(), 'Return-Path: <bbb@zzz.org>\nDelivered-To: bbb@zzz.org\nReceived: by mail.zzz.org (Postfix, from userid 889)\n\tid 27CEAD38CC; Fri,  4 May 2001 14:05:44 -0400 (EDT)\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\nMessage-ID: <15090.61304.110929.45684@aaa.zzz.org>\nFrom: bbb@ddd.com (John X. Doe)\nTo: bbb@zzz.org\nSubject: This is a test message\nDate: Fri, 4 May 2001 14:05:44 -0400\nContent-Type: text/plain; charset=us-ascii;\n\ttitle*="us-ascii\'en\'This%20is%20even%20more%20%2A%2A%2Afun%2A%2A%2A%20isn%27t%20it%21"\n\n\nHi,\n\nDo you like this message?\n\n-Me\n')
  2565.  
  2566.     
  2567.     def test_del_param(self):
  2568.         eq = self.ndiffAssertEqual
  2569.         msg = self._msgobj('msg_01.txt')
  2570.         msg.set_param('foo', 'bar', charset = 'us-ascii', language = 'en')
  2571.         msg.set_param('title', "This is even more ***fun*** isn't it!", charset = 'us-ascii', language = 'en')
  2572.         msg.del_param('foo', header = 'Content-Type')
  2573.         eq(msg.as_string(), 'Return-Path: <bbb@zzz.org>\nDelivered-To: bbb@zzz.org\nReceived: by mail.zzz.org (Postfix, from userid 889)\n\tid 27CEAD38CC; Fri,  4 May 2001 14:05:44 -0400 (EDT)\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\nMessage-ID: <15090.61304.110929.45684@aaa.zzz.org>\nFrom: bbb@ddd.com (John X. Doe)\nTo: bbb@zzz.org\nSubject: This is a test message\nDate: Fri, 4 May 2001 14:05:44 -0400\nContent-Type: text/plain; charset="us-ascii";\n\ttitle*="us-ascii\'en\'This%20is%20even%20more%20%2A%2A%2Afun%2A%2A%2A%20isn%27t%20it%21"\n\n\nHi,\n\nDo you like this message?\n\n-Me\n')
  2574.  
  2575.     
  2576.     def test_rfc2231_get_content_charset(self):
  2577.         eq = self.assertEqual
  2578.         msg = self._msgobj('msg_32.txt')
  2579.         eq(msg.get_content_charset(), 'us-ascii')
  2580.  
  2581.     
  2582.     def test_rfc2231_no_language_or_charset(self):
  2583.         m = 'Content-Transfer-Encoding: 8bit\nContent-Disposition: inline; filename="file____C__DOCUMENTS_20AND_20SETTINGS_FABIEN_LOCAL_20SETTINGS_TEMP_nsmail.htm"\nContent-Type: text/html; NAME*0=file____C__DOCUMENTS_20AND_20SETTINGS_FABIEN_LOCAL_20SETTINGS_TEM; NAME*1=P_nsmail.htm\n\n'
  2584.         msg = email.message_from_string(m)
  2585.         param = msg.get_param('NAME')
  2586.         self.failIf(isinstance(param, tuple))
  2587.         self.assertEqual(param, 'file____C__DOCUMENTS_20AND_20SETTINGS_FABIEN_LOCAL_20SETTINGS_TEMP_nsmail.htm')
  2588.  
  2589.     
  2590.     def test_rfc2231_no_language_or_charset_in_filename(self):
  2591.         m = 'Content-Disposition: inline;\n\tfilename*0*="\'\'This%20is%20even%20more%20";\n\tfilename*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tfilename*2="is it not.pdf"\n\n'
  2592.         msg = email.message_from_string(m)
  2593.         self.assertEqual(msg.get_filename(), 'This is even more ***fun*** is it not.pdf')
  2594.  
  2595.     
  2596.     def test_rfc2231_no_language_or_charset_in_filename_encoded(self):
  2597.         m = 'Content-Disposition: inline;\n\tfilename*0*="\'\'This%20is%20even%20more%20";\n\tfilename*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tfilename*2="is it not.pdf"\n\n'
  2598.         msg = email.message_from_string(m)
  2599.         self.assertEqual(msg.get_filename(), 'This is even more ***fun*** is it not.pdf')
  2600.  
  2601.     
  2602.     def test_rfc2231_partly_encoded(self):
  2603.         m = 'Content-Disposition: inline;\n\tfilename*0="\'\'This%20is%20even%20more%20";\n\tfilename*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tfilename*2="is it not.pdf"\n\n'
  2604.         msg = email.message_from_string(m)
  2605.         self.assertEqual(msg.get_filename(), 'This%20is%20even%20more%20***fun*** is it not.pdf')
  2606.  
  2607.     
  2608.     def test_rfc2231_partly_nonencoded(self):
  2609.         m = 'Content-Disposition: inline;\n\tfilename*0="This%20is%20even%20more%20";\n\tfilename*1="%2A%2A%2Afun%2A%2A%2A%20";\n\tfilename*2="is it not.pdf"\n\n'
  2610.         msg = email.message_from_string(m)
  2611.         self.assertEqual(msg.get_filename(), 'This%20is%20even%20more%20%2A%2A%2Afun%2A%2A%2A%20is it not.pdf')
  2612.  
  2613.     
  2614.     def test_rfc2231_no_language_or_charset_in_boundary(self):
  2615.         m = 'Content-Type: multipart/alternative;\n\tboundary*0*="\'\'This%20is%20even%20more%20";\n\tboundary*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tboundary*2="is it not.pdf"\n\n'
  2616.         msg = email.message_from_string(m)
  2617.         self.assertEqual(msg.get_boundary(), 'This is even more ***fun*** is it not.pdf')
  2618.  
  2619.     
  2620.     def test_rfc2231_no_language_or_charset_in_charset(self):
  2621.         m = 'Content-Type: text/plain;\n\tcharset*0*="This%20is%20even%20more%20";\n\tcharset*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tcharset*2="is it not.pdf"\n\n'
  2622.         msg = email.message_from_string(m)
  2623.         self.assertEqual(msg.get_content_charset(), 'this is even more ***fun*** is it not.pdf')
  2624.  
  2625.     
  2626.     def test_rfc2231_bad_encoding_in_filename(self):
  2627.         m = 'Content-Disposition: inline;\n\tfilename*0*="bogus\'xx\'This%20is%20even%20more%20";\n\tfilename*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tfilename*2="is it not.pdf"\n\n'
  2628.         msg = email.message_from_string(m)
  2629.         self.assertEqual(msg.get_filename(), 'This is even more ***fun*** is it not.pdf')
  2630.  
  2631.     
  2632.     def test_rfc2231_bad_encoding_in_charset(self):
  2633.         m = "Content-Type: text/plain; charset*=bogus''utf-8%E2%80%9D\n\n"
  2634.         msg = email.message_from_string(m)
  2635.         self.assertEqual(msg.get_content_charset(), None)
  2636.  
  2637.     
  2638.     def test_rfc2231_bad_character_in_charset(self):
  2639.         m = "Content-Type: text/plain; charset*=ascii''utf-8%E2%80%9D\n\n"
  2640.         msg = email.message_from_string(m)
  2641.         self.assertEqual(msg.get_content_charset(), None)
  2642.  
  2643.     
  2644.     def test_rfc2231_bad_character_in_filename(self):
  2645.         m = 'Content-Disposition: inline;\n\tfilename*0*="ascii\'xx\'This%20is%20even%20more%20";\n\tfilename*1*="%2A%2A%2Afun%2A%2A%2A%20";\n\tfilename*2*="is it not.pdf%E2"\n\n'
  2646.         msg = email.message_from_string(m)
  2647.         self.assertEqual(msg.get_filename(), u'This is even more ***fun*** is it not.pdf∩┐╜')
  2648.  
  2649.     
  2650.     def test_rfc2231_unknown_encoding(self):
  2651.         m = "Content-Transfer-Encoding: 8bit\nContent-Disposition: inline; filename*=X-UNKNOWN''myfile.txt\n\n"
  2652.         msg = email.message_from_string(m)
  2653.         self.assertEqual(msg.get_filename(), 'myfile.txt')
  2654.  
  2655.     
  2656.     def test_rfc2231_single_tick_in_filename_extended(self):
  2657.         eq = self.assertEqual
  2658.         m = 'Content-Type: application/x-foo;\n\tname*0*="Frank\'s"; name*1*=" Document"\n\n'
  2659.         msg = email.message_from_string(m)
  2660.         (charset, language, s) = msg.get_param('name')
  2661.         eq(charset, None)
  2662.         eq(language, None)
  2663.         eq(s, "Frank's Document")
  2664.  
  2665.     
  2666.     def test_rfc2231_single_tick_in_filename(self):
  2667.         m = 'Content-Type: application/x-foo; name*0="Frank\'s"; name*1=" Document"\n\n'
  2668.         msg = email.message_from_string(m)
  2669.         param = msg.get_param('name')
  2670.         self.failIf(isinstance(param, tuple))
  2671.         self.assertEqual(param, "Frank's Document")
  2672.  
  2673.     
  2674.     def test_rfc2231_tick_attack_extended(self):
  2675.         eq = self.assertEqual
  2676.         m = 'Content-Type: application/x-foo;\n\tname*0*="us-ascii\'en-us\'Frank\'s"; name*1*=" Document"\n\n'
  2677.         msg = email.message_from_string(m)
  2678.         (charset, language, s) = msg.get_param('name')
  2679.         eq(charset, 'us-ascii')
  2680.         eq(language, 'en-us')
  2681.         eq(s, "Frank's Document")
  2682.  
  2683.     
  2684.     def test_rfc2231_tick_attack(self):
  2685.         m = 'Content-Type: application/x-foo;\n\tname*0="us-ascii\'en-us\'Frank\'s"; name*1=" Document"\n\n'
  2686.         msg = email.message_from_string(m)
  2687.         param = msg.get_param('name')
  2688.         self.failIf(isinstance(param, tuple))
  2689.         self.assertEqual(param, "us-ascii'en-us'Frank's Document")
  2690.  
  2691.     
  2692.     def test_rfc2231_no_extended_values(self):
  2693.         eq = self.assertEqual
  2694.         m = 'Content-Type: application/x-foo; name="Frank\'s Document"\n\n'
  2695.         msg = email.message_from_string(m)
  2696.         eq(msg.get_param('name'), "Frank's Document")
  2697.  
  2698.     
  2699.     def test_rfc2231_encoded_then_unencoded_segments(self):
  2700.         eq = self.assertEqual
  2701.         m = 'Content-Type: application/x-foo;\n\tname*0*="us-ascii\'en-us\'My";\n\tname*1=" Document";\n\tname*2*=" For You"\n\n'
  2702.         msg = email.message_from_string(m)
  2703.         (charset, language, s) = msg.get_param('name')
  2704.         eq(charset, 'us-ascii')
  2705.         eq(language, 'en-us')
  2706.         eq(s, 'My Document For You')
  2707.  
  2708.     
  2709.     def test_rfc2231_unencoded_then_encoded_segments(self):
  2710.         eq = self.assertEqual
  2711.         m = 'Content-Type: application/x-foo;\n\tname*0="us-ascii\'en-us\'My";\n\tname*1*=" Document";\n\tname*2*=" For You"\n\n'
  2712.         msg = email.message_from_string(m)
  2713.         (charset, language, s) = msg.get_param('name')
  2714.         eq(charset, 'us-ascii')
  2715.         eq(language, 'en-us')
  2716.         eq(s, 'My Document For You')
  2717.  
  2718.  
  2719.  
  2720. def _testclasses():
  2721.     mod = sys.modules[__name__]
  2722.     return _[1]
  2723.  
  2724.  
  2725. def suite():
  2726.     suite = unittest.TestSuite()
  2727.     for testclass in _testclasses():
  2728.         suite.addTest(unittest.makeSuite(testclass))
  2729.     
  2730.     return suite
  2731.  
  2732.  
  2733. def test_main():
  2734.     for testclass in _testclasses():
  2735.         run_unittest(testclass)
  2736.     
  2737.  
  2738. if __name__ == '__main__':
  2739.     unittest.main(defaultTest = 'suite')
  2740.  
  2741.