home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2011 June / maximum-cd-2011-06.iso / DiscContents / LibO_3.3.1_Win_x86_install_multi.exe / libreoffice1.cab / test_quopri.py < prev    next >
Encoding:
Python Source  |  2011-02-15  |  7.2 KB  |  200 lines
  1. from test import test_support
  2. import unittest
  3.  
  4. import sys, cStringIO, subprocess
  5. import quopri
  6.  
  7.  
  8.  
  9. ENCSAMPLE = """\
  10. Here's a bunch of special=20
  11.  
  12. =A1=A2=A3=A4=A5=A6=A7=A8=A9
  13. =AA=AB=AC=AD=AE=AF=B0=B1=B2=B3
  14. =B4=B5=B6=B7=B8=B9=BA=BB=BC=BD=BE
  15. =BF=C0=C1=C2=C3=C4=C5=C6
  16. =C7=C8=C9=CA=CB=CC=CD=CE=CF
  17. =D0=D1=D2=D3=D4=D5=D6=D7
  18. =D8=D9=DA=DB=DC=DD=DE=DF
  19. =E0=E1=E2=E3=E4=E5=E6=E7
  20. =E8=E9=EA=EB=EC=ED=EE=EF
  21. =F0=F1=F2=F3=F4=F5=F6=F7
  22. =F8=F9=FA=FB=FC=FD=FE=FF
  23.  
  24. characters... have fun!
  25. """
  26.  
  27. # First line ends with a space
  28. DECSAMPLE = "Here's a bunch of special \n" + \
  29. """\
  30.  
  31. \xa1\xa2\xa3\xa4\xa5\xa6\xa7\xa8\xa9
  32. \xaa\xab\xac\xad\xae\xaf\xb0\xb1\xb2\xb3
  33. \xb4\xb5\xb6\xb7\xb8\xb9\xba\xbb\xbc\xbd\xbe
  34. \xbf\xc0\xc1\xc2\xc3\xc4\xc5\xc6
  35. \xc7\xc8\xc9\xca\xcb\xcc\xcd\xce\xcf
  36. \xd0\xd1\xd2\xd3\xd4\xd5\xd6\xd7
  37. \xd8\xd9\xda\xdb\xdc\xdd\xde\xdf
  38. \xe0\xe1\xe2\xe3\xe4\xe5\xe6\xe7
  39. \xe8\xe9\xea\xeb\xec\xed\xee\xef
  40. \xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7
  41. \xf8\xf9\xfa\xfb\xfc\xfd\xfe\xff
  42.  
  43. characters... have fun!
  44. """
  45.  
  46.  
  47. def withpythonimplementation(testfunc):
  48.     def newtest(self):
  49.         # Test default implementation
  50.         testfunc(self)
  51.         # Test Python implementation
  52.         if quopri.b2a_qp is not None or quopri.a2b_qp is not None:
  53.             oldencode = quopri.b2a_qp
  54.             olddecode = quopri.a2b_qp
  55.             try:
  56.                 quopri.b2a_qp = None
  57.                 quopri.a2b_qp = None
  58.                 testfunc(self)
  59.             finally:
  60.                 quopri.b2a_qp = oldencode
  61.                 quopri.a2b_qp = olddecode
  62.     newtest.__name__ = testfunc.__name__
  63.     return newtest
  64.  
  65. class QuopriTestCase(unittest.TestCase):
  66.     # Each entry is a tuple of (plaintext, encoded string).  These strings are
  67.     # used in the "quotetabs=0" tests.
  68.     STRINGS = (
  69.         # Some normal strings
  70.         ('hello', 'hello'),
  71.         ('''hello
  72.         there
  73.         world''', '''hello
  74.         there
  75.         world'''),
  76.         ('''hello
  77.         there
  78.         world
  79. ''', '''hello
  80.         there
  81.         world
  82. '''),
  83.         ('\201\202\203', '=81=82=83'),
  84.         # Add some trailing MUST QUOTE strings
  85.         ('hello ', 'hello=20'),
  86.         ('hello\t', 'hello=09'),
  87.         # Some long lines.  First, a single line of 108 characters
  88.         ('xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\xd8\xd9\xda\xdb\xdc\xdd\xde\xdfxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx',
  89.          '''xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx=D8=D9=DA=DB=DC=DD=DE=DFx=
  90. xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'''),
  91.         # A line of exactly 76 characters, no soft line break should be needed
  92.         ('yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy',
  93.         'yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy'),
  94.         # A line of 77 characters, forcing a soft line break at position 75,
  95.         # and a second line of exactly 2 characters (because the soft line
  96.         # break `=' sign counts against the line length limit).
  97.         ('zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz',
  98.          '''zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz=
  99. zz'''),
  100.         # A line of 151 characters, forcing a soft line break at position 75,
  101.         # with a second line of exactly 76 characters and no trailing =
  102.         ('zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz',
  103.          '''zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz=
  104. zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz'''),
  105.         # A string containing a hard line break, but which the first line is
  106.         # 151 characters and the second line is exactly 76 characters.  This
  107.         # should leave us with three lines, the first which has a soft line
  108.         # break, and which the second and third do not.
  109.         ('''yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy
  110. zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz''',
  111.          '''yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy=
  112. yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy
  113. zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz'''),
  114.         # Now some really complex stuff ;)
  115.         (DECSAMPLE, ENCSAMPLE),
  116.         )
  117.  
  118.     # These are used in the "quotetabs=1" tests.
  119.     ESTRINGS = (
  120.         ('hello world', 'hello=20world'),
  121.         ('hello\tworld', 'hello=09world'),
  122.         )
  123.  
  124.     # These are used in the "header=1" tests.
  125.     HSTRINGS = (
  126.         ('hello world', 'hello_world'),
  127.         ('hello_world', 'hello=5Fworld'),
  128.         )
  129.  
  130.     @withpythonimplementation
  131.     def test_encodestring(self):
  132.         for p, e in self.STRINGS:
  133.             self.assert_(quopri.encodestring(p) == e)
  134.  
  135.     @withpythonimplementation
  136.     def test_decodestring(self):
  137.         for p, e in self.STRINGS:
  138.             self.assert_(quopri.decodestring(e) == p)
  139.  
  140.     @withpythonimplementation
  141.     def test_idempotent_string(self):
  142.         for p, e in self.STRINGS:
  143.             self.assert_(quopri.decodestring(quopri.encodestring(e)) == e)
  144.  
  145.     @withpythonimplementation
  146.     def test_encode(self):
  147.         for p, e in self.STRINGS:
  148.             infp = cStringIO.StringIO(p)
  149.             outfp = cStringIO.StringIO()
  150.             quopri.encode(infp, outfp, quotetabs=False)
  151.             self.assert_(outfp.getvalue() == e)
  152.  
  153.     @withpythonimplementation
  154.     def test_decode(self):
  155.         for p, e in self.STRINGS:
  156.             infp = cStringIO.StringIO(e)
  157.             outfp = cStringIO.StringIO()
  158.             quopri.decode(infp, outfp)
  159.             self.assert_(outfp.getvalue() == p)
  160.  
  161.     @withpythonimplementation
  162.     def test_embedded_ws(self):
  163.         for p, e in self.ESTRINGS:
  164.             self.assert_(quopri.encodestring(p, quotetabs=True) == e)
  165.             self.assert_(quopri.decodestring(e) == p)
  166.  
  167.     @withpythonimplementation
  168.     def test_encode_header(self):
  169.         for p, e in self.HSTRINGS:
  170.             self.assert_(quopri.encodestring(p, header=True) == e)
  171.  
  172.     @withpythonimplementation
  173.     def test_decode_header(self):
  174.         for p, e in self.HSTRINGS:
  175.             self.assert_(quopri.decodestring(e, header=True) == p)
  176.  
  177.     def test_scriptencode(self):
  178.         (p, e) = self.STRINGS[-1]
  179.         process = subprocess.Popen([sys.executable, "-mquopri"],
  180.                                    stdin=subprocess.PIPE, stdout=subprocess.PIPE)
  181.         cout, cerr = process.communicate(p)
  182.         # On Windows, Python will output the result to stdout using
  183.         # CRLF, as the mode of stdout is text mode. To compare this
  184.         # with the expected result, we need to do a line-by-line comparison.
  185.         self.assertEqual(cout.splitlines(), e.splitlines())
  186.  
  187.     def test_scriptdecode(self):
  188.         (p, e) = self.STRINGS[-1]
  189.         process = subprocess.Popen([sys.executable, "-mquopri", "-d"],
  190.                                    stdin=subprocess.PIPE, stdout=subprocess.PIPE)
  191.         cout, cerr = process.communicate(e)
  192.         self.assertEqual(cout.splitlines(), p.splitlines())
  193.  
  194. def test_main():
  195.     test_support.run_unittest(QuopriTestCase)
  196.  
  197.  
  198. if __name__ == "__main__":
  199.     test_main()
  200.