Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 1 | """ |
| 2 | Tests for uu module. |
| 3 | Nick Mathewson |
| 4 | """ |
| 5 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 6 | import unittest |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 7 | from test import support |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 8 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 9 | import sys, os |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 10 | import uu |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 11 | import io |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 12 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 13 | plaintext = b"The smooth-scaled python crept over the sleeping dog\n" |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 14 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 15 | encodedtext = b"""\ |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 16 | M5&AE('-M;V]T:\"US8V%L960@<'ET:&]N(&-R97!T(&]V97(@=&AE('-L965P |
| 17 | (:6YG(&1O9PH """ |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 18 | |
Alexandre Vassalotti | 794652d | 2008-06-11 22:58:36 +0000 | [diff] [blame] | 19 | # Stolen from io.py |
| 20 | class FakeIO(io.TextIOWrapper): |
| 21 | """Text I/O implementation using an in-memory buffer. |
| 22 | |
| 23 | Can be a used as a drop-in replacement for sys.stdin and sys.stdout. |
| 24 | """ |
| 25 | |
| 26 | # XXX This is really slow, but fully functional |
| 27 | |
| 28 | def __init__(self, initial_value="", encoding="utf-8", |
| 29 | errors="strict", newline="\n"): |
| 30 | super(FakeIO, self).__init__(io.BytesIO(), |
| 31 | encoding=encoding, |
| 32 | errors=errors, |
| 33 | newline=newline) |
Benjamin Peterson | 4fa88fa | 2009-03-04 00:14:51 +0000 | [diff] [blame] | 34 | self._encoding = encoding |
| 35 | self._errors = errors |
Alexandre Vassalotti | 794652d | 2008-06-11 22:58:36 +0000 | [diff] [blame] | 36 | if initial_value: |
| 37 | if not isinstance(initial_value, str): |
| 38 | initial_value = str(initial_value) |
| 39 | self.write(initial_value) |
| 40 | self.seek(0) |
| 41 | |
| 42 | def getvalue(self): |
| 43 | self.flush() |
| 44 | return self.buffer.getvalue().decode(self._encoding, self._errors) |
| 45 | |
| 46 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 47 | def encodedtextwrapped(mode, filename): |
| 48 | return (bytes("begin %03o %s\n" % (mode, filename), "ascii") + |
| 49 | encodedtext + b"\n \nend\n") |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 50 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 51 | class UUTest(unittest.TestCase): |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 52 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 53 | def test_encode(self): |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 54 | inp = io.BytesIO(plaintext) |
| 55 | out = io.BytesIO() |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 56 | uu.encode(inp, out, "t1") |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 57 | self.assertEqual(out.getvalue(), encodedtextwrapped(0o666, "t1")) |
| 58 | inp = io.BytesIO(plaintext) |
| 59 | out = io.BytesIO() |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 60 | uu.encode(inp, out, "t1", 0o644) |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 61 | self.assertEqual(out.getvalue(), encodedtextwrapped(0o644, "t1")) |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 62 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 63 | def test_decode(self): |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 64 | inp = io.BytesIO(encodedtextwrapped(0o666, "t1")) |
| 65 | out = io.BytesIO() |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 66 | uu.decode(inp, out) |
| 67 | self.assertEqual(out.getvalue(), plaintext) |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 68 | inp = io.BytesIO( |
| 69 | b"UUencoded files may contain many lines,\n" + |
| 70 | b"even some that have 'begin' in them.\n" + |
| 71 | encodedtextwrapped(0o666, "t1") |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 72 | ) |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 73 | out = io.BytesIO() |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 74 | uu.decode(inp, out) |
| 75 | self.assertEqual(out.getvalue(), plaintext) |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 76 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 77 | def test_truncatedinput(self): |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 78 | inp = io.BytesIO(b"begin 644 t1\n" + encodedtext) |
| 79 | out = io.BytesIO() |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 80 | try: |
| 81 | uu.decode(inp, out) |
Andrew Svetlov | 737fb89 | 2012-12-18 21:14:22 +0200 | [diff] [blame] | 82 | self.fail("No exception raised") |
Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 83 | except uu.Error as e: |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 84 | self.assertEqual(str(e), "Truncated input file") |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 85 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 86 | def test_missingbegin(self): |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 87 | inp = io.BytesIO(b"") |
| 88 | out = io.BytesIO() |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 89 | try: |
| 90 | uu.decode(inp, out) |
Andrew Svetlov | 737fb89 | 2012-12-18 21:14:22 +0200 | [diff] [blame] | 91 | self.fail("No exception raised") |
Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 92 | except uu.Error as e: |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 93 | self.assertEqual(str(e), "No valid begin line found in input file") |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 94 | |
Serhiy Storchaka | 519114d | 2014-11-07 14:04:37 +0200 | [diff] [blame] | 95 | def test_garbage_padding(self): |
| 96 | # Issue #22406 |
| 97 | encodedtext = ( |
| 98 | b"begin 644 file\n" |
| 99 | # length 1; bits 001100 111111 111111 111111 |
| 100 | b"\x21\x2C\x5F\x5F\x5F\n" |
| 101 | b"\x20\n" |
| 102 | b"end\n" |
| 103 | ) |
| 104 | plaintext = b"\x33" # 00110011 |
| 105 | |
| 106 | with self.subTest("uu.decode()"): |
| 107 | inp = io.BytesIO(encodedtext) |
| 108 | out = io.BytesIO() |
| 109 | uu.decode(inp, out, quiet=True) |
| 110 | self.assertEqual(out.getvalue(), plaintext) |
| 111 | |
| 112 | with self.subTest("uu_codec"): |
| 113 | import codecs |
| 114 | decoded = codecs.decode(encodedtext, "uu_codec") |
| 115 | self.assertEqual(decoded, plaintext) |
| 116 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 117 | class UUStdIOTest(unittest.TestCase): |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 118 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 119 | def setUp(self): |
| 120 | self.stdin = sys.stdin |
| 121 | self.stdout = sys.stdout |
Tim Peters | 48dacc6 | 2001-07-11 21:43:42 +0000 | [diff] [blame] | 122 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 123 | def tearDown(self): |
| 124 | sys.stdin = self.stdin |
| 125 | sys.stdout = self.stdout |
Barry Warsaw | d179570 | 2001-08-17 20:00:11 +0000 | [diff] [blame] | 126 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 127 | def test_encode(self): |
Alexandre Vassalotti | 794652d | 2008-06-11 22:58:36 +0000 | [diff] [blame] | 128 | sys.stdin = FakeIO(plaintext.decode("ascii")) |
| 129 | sys.stdout = FakeIO() |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 130 | uu.encode("-", "-", "t1", 0o666) |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 131 | self.assertEqual(sys.stdout.getvalue(), |
| 132 | encodedtextwrapped(0o666, "t1").decode("ascii")) |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 133 | |
| 134 | def test_decode(self): |
Alexandre Vassalotti | 794652d | 2008-06-11 22:58:36 +0000 | [diff] [blame] | 135 | sys.stdin = FakeIO(encodedtextwrapped(0o666, "t1").decode("ascii")) |
| 136 | sys.stdout = FakeIO() |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 137 | uu.decode("-", "-") |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 138 | stdout = sys.stdout |
| 139 | sys.stdout = self.stdout |
| 140 | sys.stdin = self.stdin |
| 141 | self.assertEqual(stdout.getvalue(), plaintext.decode("ascii")) |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 142 | |
| 143 | class UUFileTest(unittest.TestCase): |
| 144 | |
| 145 | def _kill(self, f): |
| 146 | # close and remove file |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 147 | if f is None: |
| 148 | return |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 149 | try: |
| 150 | f.close() |
| 151 | except (SystemExit, KeyboardInterrupt): |
| 152 | raise |
| 153 | except: |
| 154 | pass |
| 155 | try: |
| 156 | os.unlink(f.name) |
| 157 | except (SystemExit, KeyboardInterrupt): |
| 158 | raise |
| 159 | except: |
| 160 | pass |
| 161 | |
| 162 | def setUp(self): |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 163 | self.tmpin = support.TESTFN + "i" |
| 164 | self.tmpout = support.TESTFN + "o" |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 165 | |
| 166 | def tearDown(self): |
| 167 | del self.tmpin |
| 168 | del self.tmpout |
| 169 | |
| 170 | def test_encode(self): |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 171 | fin = fout = None |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 172 | try: |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 173 | support.unlink(self.tmpin) |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 174 | fin = open(self.tmpin, 'wb') |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 175 | fin.write(plaintext) |
| 176 | fin.close() |
| 177 | |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 178 | fin = open(self.tmpin, 'rb') |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 179 | fout = open(self.tmpout, 'wb') |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 180 | uu.encode(fin, fout, self.tmpin, mode=0o644) |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 181 | fin.close() |
| 182 | fout.close() |
| 183 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 184 | fout = open(self.tmpout, 'rb') |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 185 | s = fout.read() |
| 186 | fout.close() |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 187 | self.assertEqual(s, encodedtextwrapped(0o644, self.tmpin)) |
Walter Dörwald | 705fd47 | 2005-11-21 18:55:56 +0000 | [diff] [blame] | 188 | |
| 189 | # in_file and out_file as filenames |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 190 | uu.encode(self.tmpin, self.tmpout, self.tmpin, mode=0o644) |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 191 | fout = open(self.tmpout, 'rb') |
Walter Dörwald | 705fd47 | 2005-11-21 18:55:56 +0000 | [diff] [blame] | 192 | s = fout.read() |
| 193 | fout.close() |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 194 | self.assertEqual(s, encodedtextwrapped(0o644, self.tmpin)) |
Walter Dörwald | 705fd47 | 2005-11-21 18:55:56 +0000 | [diff] [blame] | 195 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 196 | finally: |
| 197 | self._kill(fin) |
| 198 | self._kill(fout) |
| 199 | |
| 200 | def test_decode(self): |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 201 | f = None |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 202 | try: |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 203 | support.unlink(self.tmpin) |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 204 | f = open(self.tmpin, 'wb') |
| 205 | f.write(encodedtextwrapped(0o644, self.tmpout)) |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 206 | f.close() |
| 207 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 208 | f = open(self.tmpin, 'rb') |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 209 | uu.decode(f) |
| 210 | f.close() |
| 211 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 212 | f = open(self.tmpout, 'rb') |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 213 | s = f.read() |
| 214 | f.close() |
| 215 | self.assertEqual(s, plaintext) |
| 216 | # XXX is there an xp way to verify the mode? |
| 217 | finally: |
| 218 | self._kill(f) |
| 219 | |
Antoine Pitrou | f569826 | 2010-10-31 16:04:14 +0000 | [diff] [blame] | 220 | def test_decode_filename(self): |
| 221 | f = None |
| 222 | try: |
| 223 | support.unlink(self.tmpin) |
| 224 | f = open(self.tmpin, 'wb') |
| 225 | f.write(encodedtextwrapped(0o644, self.tmpout)) |
| 226 | f.close() |
| 227 | |
| 228 | uu.decode(self.tmpin) |
| 229 | |
| 230 | f = open(self.tmpout, 'rb') |
| 231 | s = f.read() |
| 232 | f.close() |
| 233 | self.assertEqual(s, plaintext) |
| 234 | finally: |
| 235 | self._kill(f) |
| 236 | |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 237 | def test_decodetwice(self): |
| 238 | # Verify that decode() will refuse to overwrite an existing file |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 239 | f = None |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 240 | try: |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 241 | f = io.BytesIO(encodedtextwrapped(0o644, self.tmpout)) |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 242 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 243 | f = open(self.tmpin, 'rb') |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 244 | uu.decode(f) |
| 245 | f.close() |
| 246 | |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 247 | f = open(self.tmpin, 'rb') |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 248 | self.assertRaises(uu.Error, uu.decode, f) |
| 249 | f.close() |
| 250 | finally: |
| 251 | self._kill(f) |
| 252 | |
| 253 | def test_main(): |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 254 | support.run_unittest(UUTest, |
Guido van Rossum | 34d1928 | 2007-08-09 01:03:29 +0000 | [diff] [blame] | 255 | UUStdIOTest, |
| 256 | UUFileTest, |
| 257 | ) |
Walter Dörwald | 411e5a2 | 2003-05-06 08:57:41 +0000 | [diff] [blame] | 258 | |
| 259 | if __name__=="__main__": |
| 260 | test_main() |