Python code coverage for Lib/test/test_uu.py

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