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