Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1 | from test import support |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 2 | import unittest |
| 3 | |
Martin v. Löwis | 48f4cf9 | 2007-07-28 17:58:14 +0000 | [diff] [blame] | 4 | import sys, os, io, subprocess |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 5 | import quopri |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 6 | |
| 7 | |
Tim Peters | 87cc0c3 | 2001-07-21 01:41:30 +0000 | [diff] [blame] | 8 | |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 9 | ENCSAMPLE = b"""\ |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 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 |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 28 | DECSAMPLE = b"Here's a bunch of special \n" + \ |
| 29 | b"""\ |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 30 | |
Tim Peters | e861365 | 2001-08-03 20:40:18 +0000 | [diff] [blame] | 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 |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 42 | |
| 43 | characters... have fun! |
| 44 | """ |
| 45 | |
| 46 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 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 |
Tim Peters | 87cc0c3 | 2001-07-21 01:41:30 +0000 | [diff] [blame] | 64 | |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 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 |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 70 | (b'hello', b'hello'), |
| 71 | (b'''hello |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 72 | there |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 73 | world''', b'''hello |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 74 | there |
| 75 | world'''), |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 76 | (b'''hello |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 77 | there |
| 78 | world |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 79 | ''', b'''hello |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 80 | there |
| 81 | world |
| 82 | '''), |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 83 | (b'\201\202\203', b'=81=82=83'), |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 84 | # Add some trailing MUST QUOTE strings |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 85 | (b'hello ', b'hello=20'), |
| 86 | (b'hello\t', b'hello=09'), |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 87 | # Some long lines. First, a single line of 108 characters |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 88 | (b'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\xd8\xd9\xda\xdb\xdc\xdd\xde\xdfxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx', |
| 89 | b'''xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx=D8=D9=DA=DB=DC=DD=DE=DFx= |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 90 | xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'''), |
| 91 | # A line of exactly 76 characters, no soft line break should be needed |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 92 | (b'yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy', |
| 93 | b'yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy'), |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 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). |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 97 | (b'zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz', |
| 98 | b'''zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz= |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 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 = |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 102 | (b'zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz', |
| 103 | b'''zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz= |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 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. |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 109 | (b'''yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 110 | zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz''', |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 111 | b'''yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy= |
Barry Warsaw | 7599a3f | 2001-06-19 22:48:42 +0000 | [diff] [blame] | 112 | yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy |
| 113 | zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz'''), |
| 114 | # Now some really complex stuff ;) |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 115 | (DECSAMPLE, ENCSAMPLE), |
| 116 | ) |
| 117 | |
| 118 | # These are used in the "quotetabs=1" tests. |
| 119 | ESTRINGS = ( |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 120 | (b'hello world', b'hello=20world'), |
| 121 | (b'hello\tworld', b'hello=09world'), |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 122 | ) |
Tim Peters | 87cc0c3 | 2001-07-21 01:41:30 +0000 | [diff] [blame] | 123 | |
Martin v. Löwis | 16dc7f4 | 2001-09-30 20:32:11 +0000 | [diff] [blame] | 124 | # These are used in the "header=1" tests. |
| 125 | HSTRINGS = ( |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 126 | (b'hello world', b'hello_world'), |
| 127 | (b'hello_world', b'hello=5Fworld'), |
Martin v. Löwis | 16dc7f4 | 2001-09-30 20:32:11 +0000 | [diff] [blame] | 128 | ) |
| 129 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 130 | @withpythonimplementation |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 131 | def test_encodestring(self): |
| 132 | for p, e in self.STRINGS: |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 133 | self.assertEqual(quopri.encodestring(p), e) |
Tim Peters | 87cc0c3 | 2001-07-21 01:41:30 +0000 | [diff] [blame] | 134 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 135 | @withpythonimplementation |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 136 | def test_decodestring(self): |
| 137 | for p, e in self.STRINGS: |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 138 | self.assertEqual(quopri.decodestring(e), p) |
Tim Peters | 87cc0c3 | 2001-07-21 01:41:30 +0000 | [diff] [blame] | 139 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 140 | @withpythonimplementation |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 141 | def test_idempotent_string(self): |
| 142 | for p, e in self.STRINGS: |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 143 | self.assertEqual(quopri.decodestring(quopri.encodestring(e)), e) |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 144 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 145 | @withpythonimplementation |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 146 | def test_encode(self): |
| 147 | for p, e in self.STRINGS: |
Martin v. Löwis | 48f4cf9 | 2007-07-28 17:58:14 +0000 | [diff] [blame] | 148 | infp = io.BytesIO(p) |
| 149 | outfp = io.BytesIO() |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 150 | quopri.encode(infp, outfp, quotetabs=False) |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 151 | self.assertEqual(outfp.getvalue(), e) |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 152 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 153 | @withpythonimplementation |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 154 | def test_decode(self): |
| 155 | for p, e in self.STRINGS: |
Martin v. Löwis | 48f4cf9 | 2007-07-28 17:58:14 +0000 | [diff] [blame] | 156 | infp = io.BytesIO(e) |
| 157 | outfp = io.BytesIO() |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 158 | quopri.decode(infp, outfp) |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 159 | self.assertEqual(outfp.getvalue(), p) |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 160 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 161 | @withpythonimplementation |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 162 | def test_embedded_ws(self): |
| 163 | for p, e in self.ESTRINGS: |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 164 | self.assertEqual(quopri.encodestring(p, quotetabs=True), e) |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 165 | self.assertEqual(quopri.decodestring(e), p) |
Barry Warsaw | 7069763 | 2001-06-19 19:08:13 +0000 | [diff] [blame] | 166 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 167 | @withpythonimplementation |
Martin v. Löwis | 16dc7f4 | 2001-09-30 20:32:11 +0000 | [diff] [blame] | 168 | def test_encode_header(self): |
| 169 | for p, e in self.HSTRINGS: |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 170 | self.assertEqual(quopri.encodestring(p, header=True), e) |
Martin v. Löwis | 16dc7f4 | 2001-09-30 20:32:11 +0000 | [diff] [blame] | 171 | |
Walter Dörwald | b197543 | 2005-12-14 23:32:22 +0000 | [diff] [blame] | 172 | @withpythonimplementation |
Martin v. Löwis | 16dc7f4 | 2001-09-30 20:32:11 +0000 | [diff] [blame] | 173 | def test_decode_header(self): |
| 174 | for p, e in self.HSTRINGS: |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 175 | self.assertEqual(quopri.decodestring(e, header=True), p) |
Tim Peters | 87cc0c3 | 2001-07-21 01:41:30 +0000 | [diff] [blame] | 176 | |
Walter Dörwald | 3b28770 | 2005-12-15 20:17:20 +0000 | [diff] [blame] | 177 | def test_scriptencode(self): |
Tim Peters | 536cf99 | 2005-12-25 23:18:31 +0000 | [diff] [blame] | 178 | (p, e) = self.STRINGS[-1] |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 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. |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 185 | cout = cout.decode('latin-1').splitlines() |
| 186 | e = e.decode('latin-1').splitlines() |
| 187 | assert len(cout)==len(e) |
| 188 | for i in range(len(cout)): |
| 189 | self.assertEqual(cout[i], e[i]) |
| 190 | self.assertEqual(cout, e) |
Walter Dörwald | 3b28770 | 2005-12-15 20:17:20 +0000 | [diff] [blame] | 191 | |
| 192 | def test_scriptdecode(self): |
Tim Peters | 536cf99 | 2005-12-25 23:18:31 +0000 | [diff] [blame] | 193 | (p, e) = self.STRINGS[-1] |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 194 | process = subprocess.Popen([sys.executable, "-mquopri", "-d"], |
| 195 | stdin=subprocess.PIPE, stdout=subprocess.PIPE) |
| 196 | cout, cerr = process.communicate(e) |
Martin v. Löwis | c582bfc | 2007-07-28 17:52:25 +0000 | [diff] [blame] | 197 | cout = cout.decode('latin-1') |
| 198 | p = p.decode('latin-1') |
Guido van Rossum | e61fd5b | 2007-07-11 12:20:59 +0000 | [diff] [blame] | 199 | self.assertEqual(cout.splitlines(), p.splitlines()) |
Walter Dörwald | 3b28770 | 2005-12-15 20:17:20 +0000 | [diff] [blame] | 200 | |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 201 | def test_main(): |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 202 | support.run_unittest(QuopriTestCase) |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 203 | |
| 204 | |
| 205 | if __name__ == "__main__": |
| 206 | test_main() |