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