blob: ec7b3efb06b681ddae574469231efccc98ca4118 [file] [log] [blame]
Fred Drake38c2ef02001-07-17 20:52:51 +00001# As a test suite for the os module, this is woefully inadequate, but this
2# does add tests for a few functions which have been determined to be more
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00003# portable than they had been thought to be.
Fred Drake38c2ef02001-07-17 20:52:51 +00004
5import os
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00006import errno
Fred Drake38c2ef02001-07-17 20:52:51 +00007import unittest
Jeremy Hyltona7fc21b2001-08-20 20:10:01 +00008import warnings
Thomas Wouters477c8d52006-05-27 19:21:47 +00009import sys
Brian Curtineb24d742010-04-12 17:16:38 +000010import signal
11import subprocess
12import time
Martin v. Löwis011e8422009-05-05 04:43:17 +000013import shutil
Benjamin Petersonee8712c2008-05-20 21:35:26 +000014from test import support
Victor Stinnerc2d095f2010-05-17 00:14:53 +000015import contextlib
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +000016import mmap
Benjamin Peterson799bd802011-08-31 22:15:17 -040017import platform
18import re
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +000019import uuid
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000020import asyncore
21import asynchat
22import socket
Charles-François Natali7372b062012-02-05 15:15:38 +010023import itertools
24import stat
Brett Cannonefb00c02012-02-29 18:31:31 -050025import locale
26import codecs
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +000027try:
28 import threading
29except ImportError:
30 threading = None
Georg Brandl2daf6ae2012-02-20 19:54:16 +010031from test.script_helper import assert_python_ok
Fred Drake38c2ef02001-07-17 20:52:51 +000032
Victor Stinner034d0aa2012-06-05 01:22:15 +020033with warnings.catch_warnings():
34 warnings.simplefilter("ignore", DeprecationWarning)
35 os.stat_float_times(True)
Victor Stinner1aa54a42012-02-08 04:09:37 +010036st = os.stat(__file__)
37stat_supports_subsecond = (
38 # check if float and int timestamps are different
39 (st.st_atime != st[7])
40 or (st.st_mtime != st[8])
41 or (st.st_ctime != st[9]))
42
Mark Dickinson7cf03892010-04-16 13:45:35 +000043# Detect whether we're on a Linux system that uses the (now outdated
44# and unmaintained) linuxthreads threading library. There's an issue
45# when combining linuxthreads with a failed execv call: see
46# http://bugs.python.org/issue4970.
Victor Stinnerd5c355c2011-04-30 14:53:09 +020047if hasattr(sys, 'thread_info') and sys.thread_info.version:
48 USING_LINUXTHREADS = sys.thread_info.version.startswith("linuxthreads")
49else:
50 USING_LINUXTHREADS = False
Brian Curtineb24d742010-04-12 17:16:38 +000051
Thomas Wouters0e3f5912006-08-11 14:57:12 +000052# Tests creating TESTFN
53class FileTests(unittest.TestCase):
54 def setUp(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +000055 if os.path.exists(support.TESTFN):
56 os.unlink(support.TESTFN)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000057 tearDown = setUp
58
59 def test_access(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +000060 f = os.open(support.TESTFN, os.O_CREAT|os.O_RDWR)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000061 os.close(f)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000062 self.assertTrue(os.access(support.TESTFN, os.W_OK))
Thomas Wouters0e3f5912006-08-11 14:57:12 +000063
Christian Heimesfdab48e2008-01-20 09:06:41 +000064 def test_closerange(self):
Antoine Pitroub9ee06c2008-08-16 22:03:17 +000065 first = os.open(support.TESTFN, os.O_CREAT|os.O_RDWR)
66 # We must allocate two consecutive file descriptors, otherwise
67 # it will mess up other file descriptors (perhaps even the three
68 # standard ones).
69 second = os.dup(first)
70 try:
71 retries = 0
72 while second != first + 1:
73 os.close(first)
74 retries += 1
75 if retries > 10:
76 # XXX test skipped
Benjamin Petersonfa0d7032009-06-01 22:42:33 +000077 self.skipTest("couldn't allocate two consecutive fds")
Antoine Pitroub9ee06c2008-08-16 22:03:17 +000078 first, second = second, os.dup(second)
79 finally:
80 os.close(second)
Christian Heimesfdab48e2008-01-20 09:06:41 +000081 # close a fd that is open, and one that isn't
Antoine Pitroub9ee06c2008-08-16 22:03:17 +000082 os.closerange(first, first + 2)
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +000083 self.assertRaises(OSError, os.write, first, b"a")
Thomas Wouters0e3f5912006-08-11 14:57:12 +000084
Benjamin Peterson1cc6df92010-06-30 17:39:45 +000085 @support.cpython_only
Hirokazu Yamamoto4c19e6e2008-09-08 23:41:21 +000086 def test_rename(self):
87 path = support.TESTFN
88 old = sys.getrefcount(path)
89 self.assertRaises(TypeError, os.rename, path, 0)
90 new = sys.getrefcount(path)
91 self.assertEqual(old, new)
92
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +000093 def test_read(self):
94 with open(support.TESTFN, "w+b") as fobj:
95 fobj.write(b"spam")
96 fobj.flush()
97 fd = fobj.fileno()
98 os.lseek(fd, 0, 0)
99 s = os.read(fd, 4)
100 self.assertEqual(type(s), bytes)
101 self.assertEqual(s, b"spam")
102
103 def test_write(self):
104 # os.write() accepts bytes- and buffer-like objects but not strings
105 fd = os.open(support.TESTFN, os.O_CREAT | os.O_WRONLY)
106 self.assertRaises(TypeError, os.write, fd, "beans")
107 os.write(fd, b"bacon\n")
108 os.write(fd, bytearray(b"eggs\n"))
109 os.write(fd, memoryview(b"spam\n"))
110 os.close(fd)
111 with open(support.TESTFN, "rb") as fobj:
Antoine Pitroud62269f2008-09-15 23:54:52 +0000112 self.assertEqual(fobj.read().splitlines(),
113 [b"bacon", b"eggs", b"spam"])
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +0000114
Victor Stinnere0daff12011-03-20 23:36:35 +0100115 def write_windows_console(self, *args):
116 retcode = subprocess.call(args,
117 # use a new console to not flood the test output
118 creationflags=subprocess.CREATE_NEW_CONSOLE,
119 # use a shell to hide the console window (SW_HIDE)
120 shell=True)
121 self.assertEqual(retcode, 0)
122
123 @unittest.skipUnless(sys.platform == 'win32',
124 'test specific to the Windows console')
125 def test_write_windows_console(self):
126 # Issue #11395: the Windows console returns an error (12: not enough
127 # space error) on writing into stdout if stdout mode is binary and the
128 # length is greater than 66,000 bytes (or less, depending on heap
129 # usage).
130 code = "print('x' * 100000)"
131 self.write_windows_console(sys.executable, "-c", code)
132 self.write_windows_console(sys.executable, "-u", "-c", code)
133
Amaury Forgeot d'Arce2e36ba2008-08-01 00:14:22 +0000134 def fdopen_helper(self, *args):
135 fd = os.open(support.TESTFN, os.O_RDONLY)
Victor Stinnerbef7fdf2011-07-01 13:45:30 +0200136 f = os.fdopen(fd, *args)
137 f.close()
Amaury Forgeot d'Arce2e36ba2008-08-01 00:14:22 +0000138
139 def test_fdopen(self):
Victor Stinnerbef7fdf2011-07-01 13:45:30 +0200140 fd = os.open(support.TESTFN, os.O_CREAT|os.O_RDWR)
141 os.close(fd)
142
Amaury Forgeot d'Arce2e36ba2008-08-01 00:14:22 +0000143 self.fdopen_helper()
144 self.fdopen_helper('r')
145 self.fdopen_helper('r', 100)
146
Antoine Pitrouf3b2d882012-01-30 22:08:52 +0100147 def test_replace(self):
148 TESTFN2 = support.TESTFN + ".2"
149 with open(support.TESTFN, 'w') as f:
150 f.write("1")
151 with open(TESTFN2, 'w') as f:
152 f.write("2")
153 self.addCleanup(os.unlink, TESTFN2)
154 os.replace(support.TESTFN, TESTFN2)
155 self.assertRaises(FileNotFoundError, os.stat, support.TESTFN)
156 with open(TESTFN2, 'r') as f:
157 self.assertEqual(f.read(), "1")
158
Victor Stinnerbef7fdf2011-07-01 13:45:30 +0200159
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000160# Test attributes on return values from os.*stat* family.
161class StatAttributeTests(unittest.TestCase):
162 def setUp(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000163 os.mkdir(support.TESTFN)
164 self.fname = os.path.join(support.TESTFN, "f1")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000165 f = open(self.fname, 'wb')
Guido van Rossum26d95c32007-08-27 23:18:54 +0000166 f.write(b"ABC")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000167 f.close()
Tim Peterse0c446b2001-10-18 21:57:37 +0000168
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000169 def tearDown(self):
170 os.unlink(self.fname)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000171 os.rmdir(support.TESTFN)
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000172
Antoine Pitrou38425292010-09-21 18:19:07 +0000173 def check_stat_attributes(self, fname):
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000174 if not hasattr(os, "stat"):
175 return
176
Antoine Pitrou38425292010-09-21 18:19:07 +0000177 result = os.stat(fname)
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000178
179 # Make sure direct access works
Ezio Melottib3aedd42010-11-20 19:04:17 +0000180 self.assertEqual(result[stat.ST_SIZE], 3)
181 self.assertEqual(result.st_size, 3)
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000182
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000183 # Make sure all the attributes are there
184 members = dir(result)
185 for name in dir(stat):
186 if name[:3] == 'ST_':
187 attr = name.lower()
Martin v. Löwis4d394df2005-01-23 09:19:22 +0000188 if name.endswith("TIME"):
189 def trunc(x): return int(x)
190 else:
191 def trunc(x): return x
Ezio Melottib3aedd42010-11-20 19:04:17 +0000192 self.assertEqual(trunc(getattr(result, attr)),
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000193 result[getattr(stat, name)])
Benjamin Peterson577473f2010-01-19 00:09:57 +0000194 self.assertIn(attr, members)
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000195
Larry Hastings6fe20b32012-04-19 15:07:49 -0700196 # Make sure that the st_?time and st_?time_ns fields roughly agree
Larry Hastings76ad59b2012-05-03 00:30:07 -0700197 # (they should always agree up to around tens-of-microseconds)
Larry Hastings6fe20b32012-04-19 15:07:49 -0700198 for name in 'st_atime st_mtime st_ctime'.split():
199 floaty = int(getattr(result, name) * 100000)
200 nanosecondy = getattr(result, name + "_ns") // 10000
Larry Hastings76ad59b2012-05-03 00:30:07 -0700201 self.assertAlmostEqual(floaty, nanosecondy, delta=2)
Larry Hastings6fe20b32012-04-19 15:07:49 -0700202
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000203 try:
204 result[200]
Andrew Svetlov737fb892012-12-18 21:14:22 +0200205 self.fail("No exception raised")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000206 except IndexError:
207 pass
208
209 # Make sure that assignment fails
210 try:
211 result.st_mode = 1
Andrew Svetlov737fb892012-12-18 21:14:22 +0200212 self.fail("No exception raised")
Collin Winter42dae6a2007-03-28 21:44:53 +0000213 except AttributeError:
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000214 pass
215
216 try:
217 result.st_rdev = 1
Andrew Svetlov737fb892012-12-18 21:14:22 +0200218 self.fail("No exception raised")
Guido van Rossum1fff8782001-10-18 21:19:31 +0000219 except (AttributeError, TypeError):
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000220 pass
221
222 try:
223 result.parrot = 1
Andrew Svetlov737fb892012-12-18 21:14:22 +0200224 self.fail("No exception raised")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000225 except AttributeError:
226 pass
227
228 # Use the stat_result constructor with a too-short tuple.
229 try:
230 result2 = os.stat_result((10,))
Andrew Svetlov737fb892012-12-18 21:14:22 +0200231 self.fail("No exception raised")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000232 except TypeError:
233 pass
234
Ezio Melotti42da6632011-03-15 05:18:48 +0200235 # Use the constructor with a too-long tuple.
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000236 try:
237 result2 = os.stat_result((0,1,2,3,4,5,6,7,8,9,10,11,12,13,14))
238 except TypeError:
239 pass
240
Antoine Pitrou38425292010-09-21 18:19:07 +0000241 def test_stat_attributes(self):
242 self.check_stat_attributes(self.fname)
243
244 def test_stat_attributes_bytes(self):
245 try:
246 fname = self.fname.encode(sys.getfilesystemencoding())
247 except UnicodeEncodeError:
248 self.skipTest("cannot encode %a for the filesystem" % self.fname)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +0100249 with warnings.catch_warnings():
250 warnings.simplefilter("ignore", DeprecationWarning)
251 self.check_stat_attributes(fname)
Tim Peterse0c446b2001-10-18 21:57:37 +0000252
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000253 def test_statvfs_attributes(self):
254 if not hasattr(os, "statvfs"):
255 return
256
Martin v. Löwisf90ae202002-06-11 06:22:31 +0000257 try:
258 result = os.statvfs(self.fname)
Guido van Rossumb940e112007-01-10 16:19:56 +0000259 except OSError as e:
Martin v. Löwisf90ae202002-06-11 06:22:31 +0000260 # On AtheOS, glibc always returns ENOSYS
Martin v. Löwisf90ae202002-06-11 06:22:31 +0000261 if e.errno == errno.ENOSYS:
262 return
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000263
264 # Make sure direct access works
Ezio Melottib3aedd42010-11-20 19:04:17 +0000265 self.assertEqual(result.f_bfree, result[3])
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000266
Brett Cannoncfaf10c2008-05-16 00:45:35 +0000267 # Make sure all the attributes are there.
268 members = ('bsize', 'frsize', 'blocks', 'bfree', 'bavail', 'files',
269 'ffree', 'favail', 'flag', 'namemax')
270 for value, member in enumerate(members):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000271 self.assertEqual(getattr(result, 'f_' + member), result[value])
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000272
273 # Make sure that assignment really fails
274 try:
275 result.f_bfree = 1
Andrew Svetlov737fb892012-12-18 21:14:22 +0200276 self.fail("No exception raised")
Collin Winter42dae6a2007-03-28 21:44:53 +0000277 except AttributeError:
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000278 pass
279
280 try:
281 result.parrot = 1
Andrew Svetlov737fb892012-12-18 21:14:22 +0200282 self.fail("No exception raised")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000283 except AttributeError:
284 pass
285
286 # Use the constructor with a too-short tuple.
287 try:
288 result2 = os.statvfs_result((10,))
Andrew Svetlov737fb892012-12-18 21:14:22 +0200289 self.fail("No exception raised")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000290 except TypeError:
291 pass
292
Ezio Melotti42da6632011-03-15 05:18:48 +0200293 # Use the constructor with a too-long tuple.
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000294 try:
295 result2 = os.statvfs_result((0,1,2,3,4,5,6,7,8,9,10,11,12,13,14))
296 except TypeError:
297 pass
Fred Drake38c2ef02001-07-17 20:52:51 +0000298
Thomas Wouters89f507f2006-12-13 04:49:30 +0000299 def test_utime_dir(self):
300 delta = 1000000
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000301 st = os.stat(support.TESTFN)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000302 # round to int, because some systems may support sub-second
303 # time stamps in stat, but not in utime.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000304 os.utime(support.TESTFN, (st.st_atime, int(st.st_mtime-delta)))
305 st2 = os.stat(support.TESTFN)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000306 self.assertEqual(st2.st_mtime, int(st.st_mtime-delta))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000307
Larry Hastings76ad59b2012-05-03 00:30:07 -0700308 def _test_utime(self, filename, attr, utime, delta):
Brian Curtin0277aa32011-11-06 13:50:15 -0600309 # Issue #13327 removed the requirement to pass None as the
Brian Curtin52fbea12011-11-06 13:41:17 -0600310 # second argument. Check that the previous methods of passing
311 # a time tuple or None work in addition to no argument.
Larry Hastings76ad59b2012-05-03 00:30:07 -0700312 st0 = os.stat(filename)
Brian Curtin52fbea12011-11-06 13:41:17 -0600313 # Doesn't set anything new, but sets the time tuple way
Larry Hastings76ad59b2012-05-03 00:30:07 -0700314 utime(filename, (attr(st0, "st_atime"), attr(st0, "st_mtime")))
315 # Setting the time to the time you just read, then reading again,
316 # should always return exactly the same times.
317 st1 = os.stat(filename)
318 self.assertEqual(attr(st0, "st_mtime"), attr(st1, "st_mtime"))
319 self.assertEqual(attr(st0, "st_atime"), attr(st1, "st_atime"))
Brian Curtin52fbea12011-11-06 13:41:17 -0600320 # Set to the current time in the old explicit way.
Larry Hastings76ad59b2012-05-03 00:30:07 -0700321 os.utime(filename, None)
Brian Curtin52fbea12011-11-06 13:41:17 -0600322 st2 = os.stat(support.TESTFN)
Larry Hastings76ad59b2012-05-03 00:30:07 -0700323 # Set to the current time in the new way
324 os.utime(filename)
325 st3 = os.stat(filename)
326 self.assertAlmostEqual(attr(st2, "st_mtime"), attr(st3, "st_mtime"), delta=delta)
327
328 def test_utime(self):
329 def utime(file, times):
330 return os.utime(file, times)
331 self._test_utime(self.fname, getattr, utime, 10)
332 self._test_utime(support.TESTFN, getattr, utime, 10)
333
334
335 def _test_utime_ns(self, set_times_ns, test_dir=True):
336 def getattr_ns(o, attr):
337 return getattr(o, attr + "_ns")
338 ten_s = 10 * 1000 * 1000 * 1000
339 self._test_utime(self.fname, getattr_ns, set_times_ns, ten_s)
340 if test_dir:
341 self._test_utime(support.TESTFN, getattr_ns, set_times_ns, ten_s)
342
343 def test_utime_ns(self):
344 def utime_ns(file, times):
345 return os.utime(file, ns=times)
346 self._test_utime_ns(utime_ns)
347
Larry Hastings9cf065c2012-06-22 16:30:09 -0700348 requires_utime_dir_fd = unittest.skipUnless(
349 os.utime in os.supports_dir_fd,
350 "dir_fd support for utime required for this test.")
351 requires_utime_fd = unittest.skipUnless(
352 os.utime in os.supports_fd,
353 "fd support for utime required for this test.")
354 requires_utime_nofollow_symlinks = unittest.skipUnless(
355 os.utime in os.supports_follow_symlinks,
356 "follow_symlinks support for utime required for this test.")
Larry Hastings76ad59b2012-05-03 00:30:07 -0700357
Larry Hastings9cf065c2012-06-22 16:30:09 -0700358 @requires_utime_nofollow_symlinks
Larry Hastings76ad59b2012-05-03 00:30:07 -0700359 def test_lutimes_ns(self):
360 def lutimes_ns(file, times):
Larry Hastings9cf065c2012-06-22 16:30:09 -0700361 return os.utime(file, ns=times, follow_symlinks=False)
Larry Hastings76ad59b2012-05-03 00:30:07 -0700362 self._test_utime_ns(lutimes_ns)
363
Larry Hastings9cf065c2012-06-22 16:30:09 -0700364 @requires_utime_fd
Larry Hastings76ad59b2012-05-03 00:30:07 -0700365 def test_futimes_ns(self):
366 def futimes_ns(file, times):
367 with open(file, "wb") as f:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700368 os.utime(f.fileno(), ns=times)
Larry Hastings76ad59b2012-05-03 00:30:07 -0700369 self._test_utime_ns(futimes_ns, test_dir=False)
370
371 def _utime_invalid_arguments(self, name, arg):
Larry Hastings9cf065c2012-06-22 16:30:09 -0700372 with self.assertRaises(ValueError):
Larry Hastings76ad59b2012-05-03 00:30:07 -0700373 getattr(os, name)(arg, (5, 5), ns=(5, 5))
374
375 def test_utime_invalid_arguments(self):
376 self._utime_invalid_arguments('utime', self.fname)
377
Brian Curtin52fbea12011-11-06 13:41:17 -0600378
Victor Stinner1aa54a42012-02-08 04:09:37 +0100379 @unittest.skipUnless(stat_supports_subsecond,
380 "os.stat() doesn't has a subsecond resolution")
Victor Stinnera2f7c002012-02-08 03:36:25 +0100381 def _test_utime_subsecond(self, set_time_func):
Victor Stinnerbe557de2012-02-08 03:01:11 +0100382 asec, amsec = 1, 901
383 atime = asec + amsec * 1e-3
Victor Stinnera2f7c002012-02-08 03:36:25 +0100384 msec, mmsec = 2, 901
Victor Stinnerbe557de2012-02-08 03:01:11 +0100385 mtime = msec + mmsec * 1e-3
386 filename = self.fname
Victor Stinnera2f7c002012-02-08 03:36:25 +0100387 os.utime(filename, (0, 0))
388 set_time_func(filename, atime, mtime)
Victor Stinner034d0aa2012-06-05 01:22:15 +0200389 with warnings.catch_warnings():
390 warnings.simplefilter("ignore", DeprecationWarning)
391 os.stat_float_times(True)
Victor Stinnera2f7c002012-02-08 03:36:25 +0100392 st = os.stat(filename)
393 self.assertAlmostEqual(st.st_atime, atime, places=3)
394 self.assertAlmostEqual(st.st_mtime, mtime, places=3)
Victor Stinnerbe557de2012-02-08 03:01:11 +0100395
Victor Stinnera2f7c002012-02-08 03:36:25 +0100396 def test_utime_subsecond(self):
397 def set_time(filename, atime, mtime):
398 os.utime(filename, (atime, mtime))
399 self._test_utime_subsecond(set_time)
400
Larry Hastings9cf065c2012-06-22 16:30:09 -0700401 @requires_utime_fd
Victor Stinnera2f7c002012-02-08 03:36:25 +0100402 def test_futimes_subsecond(self):
403 def set_time(filename, atime, mtime):
404 with open(filename, "wb") as f:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700405 os.utime(f.fileno(), times=(atime, mtime))
Victor Stinnera2f7c002012-02-08 03:36:25 +0100406 self._test_utime_subsecond(set_time)
407
Larry Hastings9cf065c2012-06-22 16:30:09 -0700408 @requires_utime_fd
Victor Stinnera2f7c002012-02-08 03:36:25 +0100409 def test_futimens_subsecond(self):
410 def set_time(filename, atime, mtime):
411 with open(filename, "wb") as f:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700412 os.utime(f.fileno(), times=(atime, mtime))
Victor Stinnera2f7c002012-02-08 03:36:25 +0100413 self._test_utime_subsecond(set_time)
414
Larry Hastings9cf065c2012-06-22 16:30:09 -0700415 @requires_utime_dir_fd
Victor Stinnera2f7c002012-02-08 03:36:25 +0100416 def test_futimesat_subsecond(self):
417 def set_time(filename, atime, mtime):
418 dirname = os.path.dirname(filename)
419 dirfd = os.open(dirname, os.O_RDONLY)
420 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700421 os.utime(os.path.basename(filename), dir_fd=dirfd,
422 times=(atime, mtime))
Victor Stinnera2f7c002012-02-08 03:36:25 +0100423 finally:
424 os.close(dirfd)
425 self._test_utime_subsecond(set_time)
426
Larry Hastings9cf065c2012-06-22 16:30:09 -0700427 @requires_utime_nofollow_symlinks
Victor Stinnera2f7c002012-02-08 03:36:25 +0100428 def test_lutimes_subsecond(self):
429 def set_time(filename, atime, mtime):
Larry Hastings9cf065c2012-06-22 16:30:09 -0700430 os.utime(filename, (atime, mtime), follow_symlinks=False)
Victor Stinnera2f7c002012-02-08 03:36:25 +0100431 self._test_utime_subsecond(set_time)
432
Larry Hastings9cf065c2012-06-22 16:30:09 -0700433 @requires_utime_dir_fd
Victor Stinnera2f7c002012-02-08 03:36:25 +0100434 def test_utimensat_subsecond(self):
435 def set_time(filename, atime, mtime):
436 dirname = os.path.dirname(filename)
437 dirfd = os.open(dirname, os.O_RDONLY)
438 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700439 os.utime(os.path.basename(filename), dir_fd=dirfd,
440 times=(atime, mtime))
Victor Stinnera2f7c002012-02-08 03:36:25 +0100441 finally:
442 os.close(dirfd)
443 self._test_utime_subsecond(set_time)
Victor Stinnerbe557de2012-02-08 03:01:11 +0100444
Thomas Wouters89f507f2006-12-13 04:49:30 +0000445 # Restrict test to Win32, since there is no guarantee other
446 # systems support centiseconds
447 if sys.platform == 'win32':
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000448 def get_file_system(path):
Hirokazu Yamamoto5ef6d182008-08-20 04:17:24 +0000449 root = os.path.splitdrive(os.path.abspath(path))[0] + '\\'
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000450 import ctypes
Hirokazu Yamamotoca765d52008-08-20 16:18:19 +0000451 kernel32 = ctypes.windll.kernel32
Hirokazu Yamamoto5ef6d182008-08-20 04:17:24 +0000452 buf = ctypes.create_unicode_buffer("", 100)
Hirokazu Yamamotoca765d52008-08-20 16:18:19 +0000453 if kernel32.GetVolumeInformationW(root, None, 0, None, None, None, buf, len(buf)):
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000454 return buf.value
455
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000456 if get_file_system(support.TESTFN) == "NTFS":
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000457 def test_1565150(self):
458 t1 = 1159195039.25
459 os.utime(self.fname, (t1, t1))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000460 self.assertEqual(os.stat(self.fname).st_mtime, t1)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000461
Amaury Forgeot d'Arca251a852011-01-03 00:19:11 +0000462 def test_large_time(self):
463 t1 = 5000000000 # some day in 2128
464 os.utime(self.fname, (t1, t1))
465 self.assertEqual(os.stat(self.fname).st_mtime, t1)
466
Guido van Rossumd8faa362007-04-27 19:54:29 +0000467 def test_1686475(self):
468 # Verify that an open file can be stat'ed
469 try:
470 os.stat(r"c:\pagefile.sys")
471 except WindowsError as e:
Benjamin Petersonc4fe6f32008-08-19 18:57:56 +0000472 if e.errno == 2: # file does not exist; cannot run test
Guido van Rossumd8faa362007-04-27 19:54:29 +0000473 return
474 self.fail("Could not stat pagefile.sys")
475
Richard Oudkerk2240ac12012-07-06 12:05:32 +0100476 @unittest.skipUnless(hasattr(os, "pipe"), "requires os.pipe()")
477 def test_15261(self):
478 # Verify that stat'ing a closed fd does not cause crash
479 r, w = os.pipe()
480 try:
481 os.stat(r) # should not raise error
482 finally:
483 os.close(r)
484 os.close(w)
485 with self.assertRaises(OSError) as ctx:
486 os.stat(r)
487 self.assertEqual(ctx.exception.errno, errno.EBADF)
488
Walter Dörwald0a6d0ff2004-05-31 16:29:04 +0000489from test import mapping_tests
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000490
Walter Dörwald0a6d0ff2004-05-31 16:29:04 +0000491class EnvironTests(mapping_tests.BasicTestMappingProtocol):
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000492 """check that os.environ object conform to mapping protocol"""
Walter Dörwald118f9312004-06-02 18:42:25 +0000493 type2test = None
Christian Heimes90333392007-11-01 19:08:42 +0000494
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000495 def setUp(self):
496 self.__save = dict(os.environ)
Victor Stinnerb745a742010-05-18 17:17:23 +0000497 if os.supports_bytes_environ:
Victor Stinner208d28c2010-05-07 00:54:14 +0000498 self.__saveb = dict(os.environb)
Christian Heimes90333392007-11-01 19:08:42 +0000499 for key, value in self._reference().items():
500 os.environ[key] = value
501
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000502 def tearDown(self):
503 os.environ.clear()
504 os.environ.update(self.__save)
Victor Stinnerb745a742010-05-18 17:17:23 +0000505 if os.supports_bytes_environ:
Victor Stinner208d28c2010-05-07 00:54:14 +0000506 os.environb.clear()
507 os.environb.update(self.__saveb)
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000508
Christian Heimes90333392007-11-01 19:08:42 +0000509 def _reference(self):
510 return {"KEY1":"VALUE1", "KEY2":"VALUE2", "KEY3":"VALUE3"}
511
512 def _empty_mapping(self):
513 os.environ.clear()
514 return os.environ
515
Martin v. Löwis1d11de62005-01-29 13:29:23 +0000516 # Bug 1110478
Ezio Melottic7e139b2012-09-26 20:01:34 +0300517 @unittest.skipUnless(os.path.exists('/bin/sh'), 'requires /bin/sh')
Martin v. Löwis5510f652005-02-17 21:23:20 +0000518 def test_update2(self):
Christian Heimes90333392007-11-01 19:08:42 +0000519 os.environ.clear()
Ezio Melottic7e139b2012-09-26 20:01:34 +0300520 os.environ.update(HELLO="World")
521 with os.popen("/bin/sh -c 'echo $HELLO'") as popen:
522 value = popen.read().strip()
523 self.assertEqual(value, "World")
Martin v. Löwis1d11de62005-01-29 13:29:23 +0000524
Ezio Melottic7e139b2012-09-26 20:01:34 +0300525 @unittest.skipUnless(os.path.exists('/bin/sh'), 'requires /bin/sh')
Christian Heimes1a13d592007-11-08 14:16:55 +0000526 def test_os_popen_iter(self):
Ezio Melottic7e139b2012-09-26 20:01:34 +0300527 with os.popen(
528 "/bin/sh -c 'echo \"line1\nline2\nline3\"'") as popen:
529 it = iter(popen)
530 self.assertEqual(next(it), "line1\n")
531 self.assertEqual(next(it), "line2\n")
532 self.assertEqual(next(it), "line3\n")
533 self.assertRaises(StopIteration, next, it)
Christian Heimes1a13d592007-11-08 14:16:55 +0000534
Guido van Rossum67aca9e2007-06-13 21:51:27 +0000535 # Verify environ keys and values from the OS are of the
536 # correct str type.
537 def test_keyvalue_types(self):
538 for key, val in os.environ.items():
Ezio Melottib3aedd42010-11-20 19:04:17 +0000539 self.assertEqual(type(key), str)
540 self.assertEqual(type(val), str)
Guido van Rossum67aca9e2007-06-13 21:51:27 +0000541
Christian Heimes90333392007-11-01 19:08:42 +0000542 def test_items(self):
543 for key, value in self._reference().items():
544 self.assertEqual(os.environ.get(key), value)
545
Ezio Melotti19e4acf2010-02-22 15:59:01 +0000546 # Issue 7310
547 def test___repr__(self):
548 """Check that the repr() of os.environ looks like environ({...})."""
549 env = os.environ
Victor Stinner96f0de92010-07-29 00:29:00 +0000550 self.assertEqual(repr(env), 'environ({{{}}})'.format(', '.join(
551 '{!r}: {!r}'.format(key, value)
552 for key, value in env.items())))
Ezio Melotti19e4acf2010-02-22 15:59:01 +0000553
Gregory P. Smithb6e8c7e2010-02-27 07:22:22 +0000554 def test_get_exec_path(self):
555 defpath_list = os.defpath.split(os.pathsep)
556 test_path = ['/monty', '/python', '', '/flying/circus']
557 test_env = {'PATH': os.pathsep.join(test_path)}
558
559 saved_environ = os.environ
560 try:
561 os.environ = dict(test_env)
562 # Test that defaulting to os.environ works.
563 self.assertSequenceEqual(test_path, os.get_exec_path())
564 self.assertSequenceEqual(test_path, os.get_exec_path(env=None))
565 finally:
566 os.environ = saved_environ
567
568 # No PATH environment variable
569 self.assertSequenceEqual(defpath_list, os.get_exec_path({}))
570 # Empty PATH environment variable
571 self.assertSequenceEqual(('',), os.get_exec_path({'PATH':''}))
572 # Supplied PATH environment variable
573 self.assertSequenceEqual(test_path, os.get_exec_path(test_env))
574
Victor Stinnerb745a742010-05-18 17:17:23 +0000575 if os.supports_bytes_environ:
576 # env cannot contain 'PATH' and b'PATH' keys
Victor Stinner38430e22010-08-19 17:10:18 +0000577 try:
Victor Stinner6f35eda2010-10-29 00:38:58 +0000578 # ignore BytesWarning warning
579 with warnings.catch_warnings(record=True):
580 mixed_env = {'PATH': '1', b'PATH': b'2'}
Victor Stinner38430e22010-08-19 17:10:18 +0000581 except BytesWarning:
Victor Stinner6f35eda2010-10-29 00:38:58 +0000582 # mixed_env cannot be created with python -bb
Victor Stinner38430e22010-08-19 17:10:18 +0000583 pass
584 else:
585 self.assertRaises(ValueError, os.get_exec_path, mixed_env)
Victor Stinnerb745a742010-05-18 17:17:23 +0000586
587 # bytes key and/or value
588 self.assertSequenceEqual(os.get_exec_path({b'PATH': b'abc'}),
589 ['abc'])
590 self.assertSequenceEqual(os.get_exec_path({b'PATH': 'abc'}),
591 ['abc'])
592 self.assertSequenceEqual(os.get_exec_path({'PATH': b'abc'}),
593 ['abc'])
594
595 @unittest.skipUnless(os.supports_bytes_environ,
596 "os.environb required for this test.")
Victor Stinner84ae1182010-05-06 22:05:07 +0000597 def test_environb(self):
598 # os.environ -> os.environb
599 value = 'euro\u20ac'
600 try:
Benjamin Peterson180799d2010-05-06 22:25:42 +0000601 value_bytes = value.encode(sys.getfilesystemencoding(),
602 'surrogateescape')
Victor Stinner84ae1182010-05-06 22:05:07 +0000603 except UnicodeEncodeError:
Benjamin Peterson180799d2010-05-06 22:25:42 +0000604 msg = "U+20AC character is not encodable to %s" % (
605 sys.getfilesystemencoding(),)
Benjamin Peterson932d3f42010-05-06 22:26:31 +0000606 self.skipTest(msg)
Victor Stinner84ae1182010-05-06 22:05:07 +0000607 os.environ['unicode'] = value
Ezio Melottib3aedd42010-11-20 19:04:17 +0000608 self.assertEqual(os.environ['unicode'], value)
609 self.assertEqual(os.environb[b'unicode'], value_bytes)
Victor Stinner84ae1182010-05-06 22:05:07 +0000610
611 # os.environb -> os.environ
612 value = b'\xff'
613 os.environb[b'bytes'] = value
Ezio Melottib3aedd42010-11-20 19:04:17 +0000614 self.assertEqual(os.environb[b'bytes'], value)
Victor Stinner84ae1182010-05-06 22:05:07 +0000615 value_str = value.decode(sys.getfilesystemencoding(), 'surrogateescape')
Ezio Melottib3aedd42010-11-20 19:04:17 +0000616 self.assertEqual(os.environ['bytes'], value_str)
Ezio Melotti19e4acf2010-02-22 15:59:01 +0000617
Charles-François Natali2966f102011-11-26 11:32:46 +0100618 # On FreeBSD < 7 and OS X < 10.6, unsetenv() doesn't return a value (issue
619 # #13415).
620 @support.requires_freebsd_version(7)
621 @support.requires_mac_ver(10, 6)
Victor Stinner60b385e2011-11-22 22:01:28 +0100622 def test_unset_error(self):
623 if sys.platform == "win32":
624 # an environment variable is limited to 32,767 characters
625 key = 'x' * 50000
Victor Stinnerb3f82682011-11-22 22:30:19 +0100626 self.assertRaises(ValueError, os.environ.__delitem__, key)
Victor Stinner60b385e2011-11-22 22:01:28 +0100627 else:
628 # "=" is not allowed in a variable name
629 key = 'key='
Victor Stinnerb3f82682011-11-22 22:30:19 +0100630 self.assertRaises(OSError, os.environ.__delitem__, key)
Victor Stinner60b385e2011-11-22 22:01:28 +0100631
Tim Petersc4e09402003-04-25 07:11:48 +0000632class WalkTests(unittest.TestCase):
633 """Tests for os.walk()."""
634
Charles-François Natali7372b062012-02-05 15:15:38 +0100635 def setUp(self):
Tim Petersc4e09402003-04-25 07:11:48 +0000636 import os
637 from os.path import join
638
639 # Build:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000640 # TESTFN/
641 # TEST1/ a file kid and two directory kids
Tim Petersc4e09402003-04-25 07:11:48 +0000642 # tmp1
643 # SUB1/ a file kid and a directory kid
Guido van Rossumd8faa362007-04-27 19:54:29 +0000644 # tmp2
645 # SUB11/ no kids
646 # SUB2/ a file kid and a dirsymlink kid
647 # tmp3
648 # link/ a symlink to TESTFN.2
Hynek Schlawack66bfcc12012-05-15 16:32:21 +0200649 # broken_link
Guido van Rossumd8faa362007-04-27 19:54:29 +0000650 # TEST2/
651 # tmp4 a lone file
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000652 walk_path = join(support.TESTFN, "TEST1")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000653 sub1_path = join(walk_path, "SUB1")
Tim Petersc4e09402003-04-25 07:11:48 +0000654 sub11_path = join(sub1_path, "SUB11")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000655 sub2_path = join(walk_path, "SUB2")
656 tmp1_path = join(walk_path, "tmp1")
Tim Petersc4e09402003-04-25 07:11:48 +0000657 tmp2_path = join(sub1_path, "tmp2")
658 tmp3_path = join(sub2_path, "tmp3")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000659 link_path = join(sub2_path, "link")
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000660 t2_path = join(support.TESTFN, "TEST2")
661 tmp4_path = join(support.TESTFN, "TEST2", "tmp4")
Hynek Schlawack66bfcc12012-05-15 16:32:21 +0200662 link_path = join(sub2_path, "link")
663 broken_link_path = join(sub2_path, "broken_link")
Tim Petersc4e09402003-04-25 07:11:48 +0000664
665 # Create stuff.
666 os.makedirs(sub11_path)
667 os.makedirs(sub2_path)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000668 os.makedirs(t2_path)
669 for path in tmp1_path, tmp2_path, tmp3_path, tmp4_path:
Alex Martelli01c77c62006-08-24 02:58:11 +0000670 f = open(path, "w")
Tim Petersc4e09402003-04-25 07:11:48 +0000671 f.write("I'm " + path + " and proud of it. Blame test_os.\n")
672 f.close()
Brian Curtin3b4499c2010-12-28 14:31:47 +0000673 if support.can_symlink():
Antoine Pitrou5311c1d2012-01-24 08:59:28 +0100674 if os.name == 'nt':
675 def symlink_to_dir(src, dest):
676 os.symlink(src, dest, True)
677 else:
678 symlink_to_dir = os.symlink
679 symlink_to_dir(os.path.abspath(t2_path), link_path)
Hynek Schlawack66bfcc12012-05-15 16:32:21 +0200680 symlink_to_dir('broken', broken_link_path)
681 sub2_tree = (sub2_path, ["link"], ["broken_link", "tmp3"])
Guido van Rossumd8faa362007-04-27 19:54:29 +0000682 else:
683 sub2_tree = (sub2_path, [], ["tmp3"])
Tim Petersc4e09402003-04-25 07:11:48 +0000684
685 # Walk top-down.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000686 all = list(os.walk(walk_path))
Tim Petersc4e09402003-04-25 07:11:48 +0000687 self.assertEqual(len(all), 4)
688 # We can't know which order SUB1 and SUB2 will appear in.
689 # Not flipped: TESTFN, SUB1, SUB11, SUB2
690 # flipped: TESTFN, SUB2, SUB1, SUB11
691 flipped = all[0][1][0] != "SUB1"
692 all[0][1].sort()
Hynek Schlawackc96f5a02012-05-15 17:55:38 +0200693 all[3 - 2 * flipped][-1].sort()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000694 self.assertEqual(all[0], (walk_path, ["SUB1", "SUB2"], ["tmp1"]))
Tim Petersc4e09402003-04-25 07:11:48 +0000695 self.assertEqual(all[1 + flipped], (sub1_path, ["SUB11"], ["tmp2"]))
696 self.assertEqual(all[2 + flipped], (sub11_path, [], []))
Guido van Rossumd8faa362007-04-27 19:54:29 +0000697 self.assertEqual(all[3 - 2 * flipped], sub2_tree)
Tim Petersc4e09402003-04-25 07:11:48 +0000698
699 # Prune the search.
700 all = []
Guido van Rossumd8faa362007-04-27 19:54:29 +0000701 for root, dirs, files in os.walk(walk_path):
Tim Petersc4e09402003-04-25 07:11:48 +0000702 all.append((root, dirs, files))
703 # Don't descend into SUB1.
704 if 'SUB1' in dirs:
705 # Note that this also mutates the dirs we appended to all!
706 dirs.remove('SUB1')
707 self.assertEqual(len(all), 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000708 self.assertEqual(all[0], (walk_path, ["SUB2"], ["tmp1"]))
Hynek Schlawack39bf90d2012-05-15 18:40:17 +0200709 all[1][-1].sort()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000710 self.assertEqual(all[1], sub2_tree)
Tim Petersc4e09402003-04-25 07:11:48 +0000711
712 # Walk bottom-up.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000713 all = list(os.walk(walk_path, topdown=False))
Tim Petersc4e09402003-04-25 07:11:48 +0000714 self.assertEqual(len(all), 4)
715 # We can't know which order SUB1 and SUB2 will appear in.
716 # Not flipped: SUB11, SUB1, SUB2, TESTFN
717 # flipped: SUB2, SUB11, SUB1, TESTFN
718 flipped = all[3][1][0] != "SUB1"
719 all[3][1].sort()
Hynek Schlawack39bf90d2012-05-15 18:40:17 +0200720 all[2 - 2 * flipped][-1].sort()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000721 self.assertEqual(all[3], (walk_path, ["SUB1", "SUB2"], ["tmp1"]))
Tim Petersc4e09402003-04-25 07:11:48 +0000722 self.assertEqual(all[flipped], (sub11_path, [], []))
723 self.assertEqual(all[flipped + 1], (sub1_path, ["SUB11"], ["tmp2"]))
Guido van Rossumd8faa362007-04-27 19:54:29 +0000724 self.assertEqual(all[2 - 2 * flipped], sub2_tree)
Tim Petersc4e09402003-04-25 07:11:48 +0000725
Brian Curtin3b4499c2010-12-28 14:31:47 +0000726 if support.can_symlink():
Guido van Rossumd8faa362007-04-27 19:54:29 +0000727 # Walk, following symlinks.
728 for root, dirs, files in os.walk(walk_path, followlinks=True):
729 if root == link_path:
730 self.assertEqual(dirs, [])
731 self.assertEqual(files, ["tmp4"])
732 break
733 else:
734 self.fail("Didn't follow symlink with followlinks=True")
735
736 def tearDown(self):
Tim Petersc4e09402003-04-25 07:11:48 +0000737 # Tear everything down. This is a decent use for bottom-up on
738 # Windows, which doesn't have a recursive delete command. The
739 # (not so) subtlety is that rmdir will fail unless the dir's
740 # kids are removed first, so bottom up is essential.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000741 for root, dirs, files in os.walk(support.TESTFN, topdown=False):
Tim Petersc4e09402003-04-25 07:11:48 +0000742 for name in files:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000743 os.remove(os.path.join(root, name))
Tim Petersc4e09402003-04-25 07:11:48 +0000744 for name in dirs:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000745 dirname = os.path.join(root, name)
746 if not os.path.islink(dirname):
747 os.rmdir(dirname)
748 else:
749 os.remove(dirname)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000750 os.rmdir(support.TESTFN)
Tim Petersc4e09402003-04-25 07:11:48 +0000751
Charles-François Natali7372b062012-02-05 15:15:38 +0100752
753@unittest.skipUnless(hasattr(os, 'fwalk'), "Test needs os.fwalk()")
754class FwalkTests(WalkTests):
755 """Tests for os.fwalk()."""
756
Larry Hastingsc48fe982012-06-25 04:49:05 -0700757 def _compare_to_walk(self, walk_kwargs, fwalk_kwargs):
758 """
759 compare with walk() results.
760 """
Larry Hastingsb4038062012-07-15 10:57:38 -0700761 walk_kwargs = walk_kwargs.copy()
762 fwalk_kwargs = fwalk_kwargs.copy()
763 for topdown, follow_symlinks in itertools.product((True, False), repeat=2):
764 walk_kwargs.update(topdown=topdown, followlinks=follow_symlinks)
765 fwalk_kwargs.update(topdown=topdown, follow_symlinks=follow_symlinks)
Larry Hastingsc48fe982012-06-25 04:49:05 -0700766
Charles-François Natali7372b062012-02-05 15:15:38 +0100767 expected = {}
Larry Hastingsc48fe982012-06-25 04:49:05 -0700768 for root, dirs, files in os.walk(**walk_kwargs):
Charles-François Natali7372b062012-02-05 15:15:38 +0100769 expected[root] = (set(dirs), set(files))
770
Larry Hastingsc48fe982012-06-25 04:49:05 -0700771 for root, dirs, files, rootfd in os.fwalk(**fwalk_kwargs):
Charles-François Natali7372b062012-02-05 15:15:38 +0100772 self.assertIn(root, expected)
773 self.assertEqual(expected[root], (set(dirs), set(files)))
774
Larry Hastingsc48fe982012-06-25 04:49:05 -0700775 def test_compare_to_walk(self):
776 kwargs = {'top': support.TESTFN}
777 self._compare_to_walk(kwargs, kwargs)
778
Charles-François Natali7372b062012-02-05 15:15:38 +0100779 def test_dir_fd(self):
Larry Hastingsc48fe982012-06-25 04:49:05 -0700780 try:
781 fd = os.open(".", os.O_RDONLY)
782 walk_kwargs = {'top': support.TESTFN}
783 fwalk_kwargs = walk_kwargs.copy()
784 fwalk_kwargs['dir_fd'] = fd
785 self._compare_to_walk(walk_kwargs, fwalk_kwargs)
786 finally:
787 os.close(fd)
788
789 def test_yields_correct_dir_fd(self):
Charles-François Natali7372b062012-02-05 15:15:38 +0100790 # check returned file descriptors
Larry Hastingsb4038062012-07-15 10:57:38 -0700791 for topdown, follow_symlinks in itertools.product((True, False), repeat=2):
792 args = support.TESTFN, topdown, None
793 for root, dirs, files, rootfd in os.fwalk(*args, follow_symlinks=follow_symlinks):
Charles-François Natali7372b062012-02-05 15:15:38 +0100794 # check that the FD is valid
795 os.fstat(rootfd)
Larry Hastings9cf065c2012-06-22 16:30:09 -0700796 # redundant check
797 os.stat(rootfd)
798 # check that listdir() returns consistent information
799 self.assertEqual(set(os.listdir(rootfd)), set(dirs) | set(files))
Charles-François Natali7372b062012-02-05 15:15:38 +0100800
801 def test_fd_leak(self):
802 # Since we're opening a lot of FDs, we must be careful to avoid leaks:
803 # we both check that calling fwalk() a large number of times doesn't
804 # yield EMFILE, and that the minimum allocated FD hasn't changed.
805 minfd = os.dup(1)
806 os.close(minfd)
807 for i in range(256):
808 for x in os.fwalk(support.TESTFN):
809 pass
810 newfd = os.dup(1)
811 self.addCleanup(os.close, newfd)
812 self.assertEqual(newfd, minfd)
813
814 def tearDown(self):
815 # cleanup
816 for root, dirs, files, rootfd in os.fwalk(support.TESTFN, topdown=False):
817 for name in files:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700818 os.unlink(name, dir_fd=rootfd)
Charles-François Natali7372b062012-02-05 15:15:38 +0100819 for name in dirs:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700820 st = os.stat(name, dir_fd=rootfd, follow_symlinks=False)
Larry Hastingsb698d8e2012-06-23 16:55:07 -0700821 if stat.S_ISDIR(st.st_mode):
822 os.rmdir(name, dir_fd=rootfd)
823 else:
824 os.unlink(name, dir_fd=rootfd)
Charles-François Natali7372b062012-02-05 15:15:38 +0100825 os.rmdir(support.TESTFN)
826
827
Guido van Rossume7ba4952007-06-06 23:52:48 +0000828class MakedirTests(unittest.TestCase):
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000829 def setUp(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000830 os.mkdir(support.TESTFN)
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000831
832 def test_makedir(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000833 base = support.TESTFN
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000834 path = os.path.join(base, 'dir1', 'dir2', 'dir3')
835 os.makedirs(path) # Should work
836 path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4')
837 os.makedirs(path)
838
839 # Try paths with a '.' in them
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000840 self.assertRaises(OSError, os.makedirs, os.curdir)
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000841 path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4', 'dir5', os.curdir)
842 os.makedirs(path)
843 path = os.path.join(base, 'dir1', os.curdir, 'dir2', 'dir3', 'dir4',
844 'dir5', 'dir6')
845 os.makedirs(path)
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000846
Terry Reedy5a22b652010-12-02 07:05:56 +0000847 def test_exist_ok_existing_directory(self):
848 path = os.path.join(support.TESTFN, 'dir1')
849 mode = 0o777
850 old_mask = os.umask(0o022)
851 os.makedirs(path, mode)
852 self.assertRaises(OSError, os.makedirs, path, mode)
853 self.assertRaises(OSError, os.makedirs, path, mode, exist_ok=False)
854 self.assertRaises(OSError, os.makedirs, path, 0o776, exist_ok=True)
855 os.makedirs(path, mode=mode, exist_ok=True)
856 os.umask(old_mask)
857
Gregory P. Smitha81c8562012-06-03 14:30:44 -0700858 def test_exist_ok_s_isgid_directory(self):
859 path = os.path.join(support.TESTFN, 'dir1')
860 S_ISGID = stat.S_ISGID
861 mode = 0o777
862 old_mask = os.umask(0o022)
863 try:
864 existing_testfn_mode = stat.S_IMODE(
865 os.lstat(support.TESTFN).st_mode)
Ned Deilyc622f422012-08-08 20:57:24 -0700866 try:
867 os.chmod(support.TESTFN, existing_testfn_mode | S_ISGID)
Ned Deily3a2b97e2012-08-08 21:03:02 -0700868 except PermissionError:
Ned Deilyc622f422012-08-08 20:57:24 -0700869 raise unittest.SkipTest('Cannot set S_ISGID for dir.')
Gregory P. Smitha81c8562012-06-03 14:30:44 -0700870 if (os.lstat(support.TESTFN).st_mode & S_ISGID != S_ISGID):
871 raise unittest.SkipTest('No support for S_ISGID dir mode.')
872 # The os should apply S_ISGID from the parent dir for us, but
873 # this test need not depend on that behavior. Be explicit.
874 os.makedirs(path, mode | S_ISGID)
875 # http://bugs.python.org/issue14992
876 # Should not fail when the bit is already set.
877 os.makedirs(path, mode, exist_ok=True)
878 # remove the bit.
879 os.chmod(path, stat.S_IMODE(os.lstat(path).st_mode) & ~S_ISGID)
880 with self.assertRaises(OSError):
881 # Should fail when the bit is not already set when demanded.
882 os.makedirs(path, mode | S_ISGID, exist_ok=True)
883 finally:
884 os.umask(old_mask)
Terry Reedy5a22b652010-12-02 07:05:56 +0000885
886 def test_exist_ok_existing_regular_file(self):
887 base = support.TESTFN
888 path = os.path.join(support.TESTFN, 'dir1')
889 f = open(path, 'w')
890 f.write('abc')
891 f.close()
892 self.assertRaises(OSError, os.makedirs, path)
893 self.assertRaises(OSError, os.makedirs, path, exist_ok=False)
894 self.assertRaises(OSError, os.makedirs, path, exist_ok=True)
895 os.remove(path)
896
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000897 def tearDown(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000898 path = os.path.join(support.TESTFN, 'dir1', 'dir2', 'dir3',
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000899 'dir4', 'dir5', 'dir6')
900 # If the tests failed, the bottom-most directory ('../dir6')
901 # may not have been created, so we look for the outermost directory
902 # that exists.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000903 while not os.path.exists(path) and path != support.TESTFN:
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000904 path = os.path.dirname(path)
905
906 os.removedirs(path)
907
Andrew Svetlov405faed2012-12-25 12:18:09 +0200908
909class RemoveDirsTests(unittest.TestCase):
910 def setUp(self):
911 os.makedirs(support.TESTFN)
912
913 def tearDown(self):
914 support.rmtree(support.TESTFN)
915
916 def test_remove_all(self):
917 dira = os.path.join(support.TESTFN, 'dira')
918 os.mkdir(dira)
919 dirb = os.path.join(dira, 'dirb')
920 os.mkdir(dirb)
921 os.removedirs(dirb)
922 self.assertFalse(os.path.exists(dirb))
923 self.assertFalse(os.path.exists(dira))
924 self.assertFalse(os.path.exists(support.TESTFN))
925
926 def test_remove_partial(self):
927 dira = os.path.join(support.TESTFN, 'dira')
928 os.mkdir(dira)
929 dirb = os.path.join(dira, 'dirb')
930 os.mkdir(dirb)
931 with open(os.path.join(dira, 'file.txt'), 'w') as f:
932 f.write('text')
933 os.removedirs(dirb)
934 self.assertFalse(os.path.exists(dirb))
935 self.assertTrue(os.path.exists(dira))
936 self.assertTrue(os.path.exists(support.TESTFN))
937
938 def test_remove_nothing(self):
939 dira = os.path.join(support.TESTFN, 'dira')
940 os.mkdir(dira)
941 dirb = os.path.join(dira, 'dirb')
942 os.mkdir(dirb)
943 with open(os.path.join(dirb, 'file.txt'), 'w') as f:
944 f.write('text')
945 with self.assertRaises(OSError):
946 os.removedirs(dirb)
947 self.assertTrue(os.path.exists(dirb))
948 self.assertTrue(os.path.exists(dira))
949 self.assertTrue(os.path.exists(support.TESTFN))
950
951
Guido van Rossume7ba4952007-06-06 23:52:48 +0000952class DevNullTests(unittest.TestCase):
Martin v. Löwisbdec50f2004-06-08 08:29:33 +0000953 def test_devnull(self):
Victor Stinnera6d2c762011-06-30 18:20:11 +0200954 with open(os.devnull, 'wb') as f:
955 f.write(b'hello')
956 f.close()
957 with open(os.devnull, 'rb') as f:
958 self.assertEqual(f.read(), b'')
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000959
Andrew Svetlov405faed2012-12-25 12:18:09 +0200960
Guido van Rossume7ba4952007-06-06 23:52:48 +0000961class URandomTests(unittest.TestCase):
Georg Brandl2daf6ae2012-02-20 19:54:16 +0100962 def test_urandom_length(self):
963 self.assertEqual(len(os.urandom(0)), 0)
964 self.assertEqual(len(os.urandom(1)), 1)
965 self.assertEqual(len(os.urandom(10)), 10)
966 self.assertEqual(len(os.urandom(100)), 100)
967 self.assertEqual(len(os.urandom(1000)), 1000)
968
969 def test_urandom_value(self):
970 data1 = os.urandom(16)
971 data2 = os.urandom(16)
972 self.assertNotEqual(data1, data2)
973
974 def get_urandom_subprocess(self, count):
975 code = '\n'.join((
976 'import os, sys',
977 'data = os.urandom(%s)' % count,
978 'sys.stdout.buffer.write(data)',
979 'sys.stdout.buffer.flush()'))
980 out = assert_python_ok('-c', code)
981 stdout = out[1]
982 self.assertEqual(len(stdout), 16)
983 return stdout
984
985 def test_urandom_subprocess(self):
986 data1 = self.get_urandom_subprocess(16)
987 data2 = self.get_urandom_subprocess(16)
988 self.assertNotEqual(data1, data2)
Martin v. Löwisdc3883f2004-08-29 15:46:35 +0000989
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000990@contextlib.contextmanager
991def _execvpe_mockup(defpath=None):
992 """
993 Stubs out execv and execve functions when used as context manager.
994 Records exec calls. The mock execv and execve functions always raise an
995 exception as they would normally never return.
996 """
997 # A list of tuples containing (function name, first arg, args)
998 # of calls to execv or execve that have been made.
999 calls = []
1000
1001 def mock_execv(name, *args):
1002 calls.append(('execv', name, args))
1003 raise RuntimeError("execv called")
1004
1005 def mock_execve(name, *args):
1006 calls.append(('execve', name, args))
1007 raise OSError(errno.ENOTDIR, "execve called")
1008
1009 try:
1010 orig_execv = os.execv
1011 orig_execve = os.execve
1012 orig_defpath = os.defpath
1013 os.execv = mock_execv
1014 os.execve = mock_execve
1015 if defpath is not None:
1016 os.defpath = defpath
1017 yield calls
1018 finally:
1019 os.execv = orig_execv
1020 os.execve = orig_execve
1021 os.defpath = orig_defpath
1022
Guido van Rossume7ba4952007-06-06 23:52:48 +00001023class ExecTests(unittest.TestCase):
Mark Dickinson7cf03892010-04-16 13:45:35 +00001024 @unittest.skipIf(USING_LINUXTHREADS,
1025 "avoid triggering a linuxthreads bug: see issue #4970")
Guido van Rossume7ba4952007-06-06 23:52:48 +00001026 def test_execvpe_with_bad_program(self):
Mark Dickinson7cf03892010-04-16 13:45:35 +00001027 self.assertRaises(OSError, os.execvpe, 'no such app-',
1028 ['no such app-'], None)
Guido van Rossume7ba4952007-06-06 23:52:48 +00001029
Thomas Heller6790d602007-08-30 17:15:14 +00001030 def test_execvpe_with_bad_arglist(self):
1031 self.assertRaises(ValueError, os.execvpe, 'notepad', [], None)
1032
Gregory P. Smith4ae37772010-05-08 18:05:46 +00001033 @unittest.skipUnless(hasattr(os, '_execvpe'),
1034 "No internal os._execvpe function to test.")
Victor Stinnerb745a742010-05-18 17:17:23 +00001035 def _test_internal_execvpe(self, test_type):
1036 program_path = os.sep + 'absolutepath'
1037 if test_type is bytes:
1038 program = b'executable'
1039 fullpath = os.path.join(os.fsencode(program_path), program)
1040 native_fullpath = fullpath
1041 arguments = [b'progname', 'arg1', 'arg2']
1042 else:
1043 program = 'executable'
1044 arguments = ['progname', 'arg1', 'arg2']
1045 fullpath = os.path.join(program_path, program)
1046 if os.name != "nt":
1047 native_fullpath = os.fsencode(fullpath)
1048 else:
1049 native_fullpath = fullpath
Victor Stinnerc2d095f2010-05-17 00:14:53 +00001050 env = {'spam': 'beans'}
1051
Victor Stinnerb745a742010-05-18 17:17:23 +00001052 # test os._execvpe() with an absolute path
Victor Stinnerc2d095f2010-05-17 00:14:53 +00001053 with _execvpe_mockup() as calls:
Victor Stinnerb745a742010-05-18 17:17:23 +00001054 self.assertRaises(RuntimeError,
1055 os._execvpe, fullpath, arguments)
Victor Stinnerc2d095f2010-05-17 00:14:53 +00001056 self.assertEqual(len(calls), 1)
1057 self.assertEqual(calls[0], ('execv', fullpath, (arguments,)))
1058
Victor Stinnerb745a742010-05-18 17:17:23 +00001059 # test os._execvpe() with a relative path:
1060 # os.get_exec_path() returns defpath
Victor Stinnerc2d095f2010-05-17 00:14:53 +00001061 with _execvpe_mockup(defpath=program_path) as calls:
Victor Stinnerb745a742010-05-18 17:17:23 +00001062 self.assertRaises(OSError,
1063 os._execvpe, program, arguments, env=env)
Victor Stinnerc2d095f2010-05-17 00:14:53 +00001064 self.assertEqual(len(calls), 1)
Victor Stinnerb745a742010-05-18 17:17:23 +00001065 self.assertSequenceEqual(calls[0],
1066 ('execve', native_fullpath, (arguments, env)))
1067
1068 # test os._execvpe() with a relative path:
1069 # os.get_exec_path() reads the 'PATH' variable
1070 with _execvpe_mockup() as calls:
1071 env_path = env.copy()
Victor Stinner38430e22010-08-19 17:10:18 +00001072 if test_type is bytes:
1073 env_path[b'PATH'] = program_path
1074 else:
1075 env_path['PATH'] = program_path
Victor Stinnerb745a742010-05-18 17:17:23 +00001076 self.assertRaises(OSError,
1077 os._execvpe, program, arguments, env=env_path)
1078 self.assertEqual(len(calls), 1)
1079 self.assertSequenceEqual(calls[0],
1080 ('execve', native_fullpath, (arguments, env_path)))
1081
1082 def test_internal_execvpe_str(self):
1083 self._test_internal_execvpe(str)
1084 if os.name != "nt":
1085 self._test_internal_execvpe(bytes)
Victor Stinnerc2d095f2010-05-17 00:14:53 +00001086
Gregory P. Smith4ae37772010-05-08 18:05:46 +00001087
Thomas Wouters477c8d52006-05-27 19:21:47 +00001088class Win32ErrorTests(unittest.TestCase):
1089 def test_rename(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001090 self.assertRaises(WindowsError, os.rename, support.TESTFN, support.TESTFN+".bak")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001091
1092 def test_remove(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001093 self.assertRaises(WindowsError, os.remove, support.TESTFN)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001094
1095 def test_chdir(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001096 self.assertRaises(WindowsError, os.chdir, support.TESTFN)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001097
1098 def test_mkdir(self):
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00001099 f = open(support.TESTFN, "w")
Benjamin Petersonf91df042009-02-13 02:50:59 +00001100 try:
1101 self.assertRaises(WindowsError, os.mkdir, support.TESTFN)
1102 finally:
1103 f.close()
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +00001104 os.unlink(support.TESTFN)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001105
1106 def test_utime(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001107 self.assertRaises(WindowsError, os.utime, support.TESTFN, None)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001108
Thomas Wouters477c8d52006-05-27 19:21:47 +00001109 def test_chmod(self):
Benjamin Petersonf91df042009-02-13 02:50:59 +00001110 self.assertRaises(WindowsError, os.chmod, support.TESTFN, 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001111
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001112class TestInvalidFD(unittest.TestCase):
Benjamin Peterson05e782f2009-01-19 15:15:02 +00001113 singles = ["fchdir", "dup", "fdopen", "fdatasync", "fstat",
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001114 "fstatvfs", "fsync", "tcgetpgrp", "ttyname"]
1115 #singles.append("close")
1116 #We omit close because it doesn'r raise an exception on some platforms
1117 def get_single(f):
1118 def helper(self):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001119 if hasattr(os, f):
1120 self.check(getattr(os, f))
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001121 return helper
1122 for f in singles:
1123 locals()["test_"+f] = get_single(f)
1124
Benjamin Peterson7522c742009-01-19 21:00:09 +00001125 def check(self, f, *args):
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001126 try:
1127 f(support.make_bad_fd(), *args)
1128 except OSError as e:
1129 self.assertEqual(e.errno, errno.EBADF)
1130 else:
1131 self.fail("%r didn't raise a OSError with a bad file descriptor"
1132 % f)
Benjamin Peterson7522c742009-01-19 21:00:09 +00001133
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001134 def test_isatty(self):
1135 if hasattr(os, "isatty"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001136 self.assertEqual(os.isatty(support.make_bad_fd()), False)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001137
1138 def test_closerange(self):
1139 if hasattr(os, "closerange"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001140 fd = support.make_bad_fd()
R. David Murray630cc482009-07-22 15:20:27 +00001141 # Make sure none of the descriptors we are about to close are
1142 # currently valid (issue 6542).
1143 for i in range(10):
1144 try: os.fstat(fd+i)
1145 except OSError:
1146 pass
1147 else:
1148 break
1149 if i < 2:
1150 raise unittest.SkipTest(
1151 "Unable to acquire a range of invalid file descriptors")
1152 self.assertEqual(os.closerange(fd, fd + i-1), None)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001153
1154 def test_dup2(self):
1155 if hasattr(os, "dup2"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001156 self.check(os.dup2, 20)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001157
1158 def test_fchmod(self):
1159 if hasattr(os, "fchmod"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001160 self.check(os.fchmod, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001161
1162 def test_fchown(self):
1163 if hasattr(os, "fchown"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001164 self.check(os.fchown, -1, -1)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001165
1166 def test_fpathconf(self):
1167 if hasattr(os, "fpathconf"):
Georg Brandl306336b2012-06-24 12:55:33 +02001168 self.check(os.pathconf, "PC_NAME_MAX")
Benjamin Peterson7522c742009-01-19 21:00:09 +00001169 self.check(os.fpathconf, "PC_NAME_MAX")
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001170
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001171 def test_ftruncate(self):
1172 if hasattr(os, "ftruncate"):
Georg Brandl306336b2012-06-24 12:55:33 +02001173 self.check(os.truncate, 0)
Benjamin Peterson7522c742009-01-19 21:00:09 +00001174 self.check(os.ftruncate, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001175
1176 def test_lseek(self):
1177 if hasattr(os, "lseek"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001178 self.check(os.lseek, 0, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001179
1180 def test_read(self):
1181 if hasattr(os, "read"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001182 self.check(os.read, 1)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001183
1184 def test_tcsetpgrpt(self):
1185 if hasattr(os, "tcsetpgrp"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001186 self.check(os.tcsetpgrp, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001187
1188 def test_write(self):
1189 if hasattr(os, "write"):
Benjamin Peterson7522c742009-01-19 21:00:09 +00001190 self.check(os.write, b" ")
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001191
Brian Curtin1b9df392010-11-24 20:24:31 +00001192
1193class LinkTests(unittest.TestCase):
1194 def setUp(self):
1195 self.file1 = support.TESTFN
1196 self.file2 = os.path.join(support.TESTFN + "2")
1197
Brian Curtinc0abc4e2010-11-30 23:46:54 +00001198 def tearDown(self):
Brian Curtin1b9df392010-11-24 20:24:31 +00001199 for file in (self.file1, self.file2):
1200 if os.path.exists(file):
1201 os.unlink(file)
1202
Brian Curtin1b9df392010-11-24 20:24:31 +00001203 def _test_link(self, file1, file2):
1204 with open(file1, "w") as f1:
1205 f1.write("test")
1206
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001207 with warnings.catch_warnings():
1208 warnings.simplefilter("ignore", DeprecationWarning)
1209 os.link(file1, file2)
Brian Curtin1b9df392010-11-24 20:24:31 +00001210 with open(file1, "r") as f1, open(file2, "r") as f2:
1211 self.assertTrue(os.path.sameopenfile(f1.fileno(), f2.fileno()))
1212
1213 def test_link(self):
1214 self._test_link(self.file1, self.file2)
1215
1216 def test_link_bytes(self):
1217 self._test_link(bytes(self.file1, sys.getfilesystemencoding()),
1218 bytes(self.file2, sys.getfilesystemencoding()))
1219
Brian Curtinf498b752010-11-30 15:54:04 +00001220 def test_unicode_name(self):
Brian Curtin43f0c272010-11-30 15:40:04 +00001221 try:
Brian Curtinf498b752010-11-30 15:54:04 +00001222 os.fsencode("\xf1")
Brian Curtin43f0c272010-11-30 15:40:04 +00001223 except UnicodeError:
1224 raise unittest.SkipTest("Unable to encode for this platform.")
1225
Brian Curtinf498b752010-11-30 15:54:04 +00001226 self.file1 += "\xf1"
Brian Curtinfc889c42010-11-28 23:59:46 +00001227 self.file2 = self.file1 + "2"
1228 self._test_link(self.file1, self.file2)
1229
Thomas Wouters477c8d52006-05-27 19:21:47 +00001230if sys.platform != 'win32':
1231 class Win32ErrorTests(unittest.TestCase):
1232 pass
1233
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00001234 class PosixUidGidTests(unittest.TestCase):
1235 if hasattr(os, 'setuid'):
1236 def test_setuid(self):
1237 if os.getuid() != 0:
1238 self.assertRaises(os.error, os.setuid, 0)
1239 self.assertRaises(OverflowError, os.setuid, 1<<32)
1240
1241 if hasattr(os, 'setgid'):
1242 def test_setgid(self):
1243 if os.getuid() != 0:
1244 self.assertRaises(os.error, os.setgid, 0)
1245 self.assertRaises(OverflowError, os.setgid, 1<<32)
1246
1247 if hasattr(os, 'seteuid'):
1248 def test_seteuid(self):
1249 if os.getuid() != 0:
1250 self.assertRaises(os.error, os.seteuid, 0)
1251 self.assertRaises(OverflowError, os.seteuid, 1<<32)
1252
1253 if hasattr(os, 'setegid'):
1254 def test_setegid(self):
1255 if os.getuid() != 0:
1256 self.assertRaises(os.error, os.setegid, 0)
1257 self.assertRaises(OverflowError, os.setegid, 1<<32)
1258
1259 if hasattr(os, 'setreuid'):
1260 def test_setreuid(self):
1261 if os.getuid() != 0:
1262 self.assertRaises(os.error, os.setreuid, 0, 0)
1263 self.assertRaises(OverflowError, os.setreuid, 1<<32, 0)
1264 self.assertRaises(OverflowError, os.setreuid, 0, 1<<32)
Benjamin Petersonebe87ba2010-03-06 20:34:24 +00001265
1266 def test_setreuid_neg1(self):
1267 # Needs to accept -1. We run this in a subprocess to avoid
1268 # altering the test runner's process state (issue8045).
Benjamin Petersonebe87ba2010-03-06 20:34:24 +00001269 subprocess.check_call([
1270 sys.executable, '-c',
1271 'import os,sys;os.setreuid(-1,-1);sys.exit(0)'])
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00001272
1273 if hasattr(os, 'setregid'):
1274 def test_setregid(self):
1275 if os.getuid() != 0:
1276 self.assertRaises(os.error, os.setregid, 0, 0)
1277 self.assertRaises(OverflowError, os.setregid, 1<<32, 0)
1278 self.assertRaises(OverflowError, os.setregid, 0, 1<<32)
Benjamin Petersonebe87ba2010-03-06 20:34:24 +00001279
1280 def test_setregid_neg1(self):
1281 # Needs to accept -1. We run this in a subprocess to avoid
1282 # altering the test runner's process state (issue8045).
Benjamin Petersonebe87ba2010-03-06 20:34:24 +00001283 subprocess.check_call([
1284 sys.executable, '-c',
1285 'import os,sys;os.setregid(-1,-1);sys.exit(0)'])
Martin v. Löwis011e8422009-05-05 04:43:17 +00001286
1287 class Pep383Tests(unittest.TestCase):
Martin v. Löwis011e8422009-05-05 04:43:17 +00001288 def setUp(self):
Victor Stinnerd91df1a2010-08-18 10:56:19 +00001289 if support.TESTFN_UNENCODABLE:
1290 self.dir = support.TESTFN_UNENCODABLE
Victor Stinnere667e982012-11-12 01:23:15 +01001291 elif support.TESTFN_NONASCII:
1292 self.dir = support.TESTFN_NONASCII
Victor Stinnerd91df1a2010-08-18 10:56:19 +00001293 else:
1294 self.dir = support.TESTFN
1295 self.bdir = os.fsencode(self.dir)
1296
1297 bytesfn = []
1298 def add_filename(fn):
1299 try:
1300 fn = os.fsencode(fn)
1301 except UnicodeEncodeError:
1302 return
1303 bytesfn.append(fn)
1304 add_filename(support.TESTFN_UNICODE)
1305 if support.TESTFN_UNENCODABLE:
1306 add_filename(support.TESTFN_UNENCODABLE)
Victor Stinnere667e982012-11-12 01:23:15 +01001307 if support.TESTFN_NONASCII:
1308 add_filename(support.TESTFN_NONASCII)
Victor Stinnerd91df1a2010-08-18 10:56:19 +00001309 if not bytesfn:
1310 self.skipTest("couldn't create any non-ascii filename")
1311
1312 self.unicodefn = set()
Martin v. Löwis011e8422009-05-05 04:43:17 +00001313 os.mkdir(self.dir)
Victor Stinnerd91df1a2010-08-18 10:56:19 +00001314 try:
1315 for fn in bytesfn:
Victor Stinnerbf816222011-06-30 23:25:47 +02001316 support.create_empty_file(os.path.join(self.bdir, fn))
Victor Stinnere8d51452010-08-19 01:05:19 +00001317 fn = os.fsdecode(fn)
Victor Stinnerd91df1a2010-08-18 10:56:19 +00001318 if fn in self.unicodefn:
1319 raise ValueError("duplicate filename")
1320 self.unicodefn.add(fn)
1321 except:
1322 shutil.rmtree(self.dir)
1323 raise
Martin v. Löwis011e8422009-05-05 04:43:17 +00001324
1325 def tearDown(self):
1326 shutil.rmtree(self.dir)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001327
1328 def test_listdir(self):
Victor Stinnerd91df1a2010-08-18 10:56:19 +00001329 expected = self.unicodefn
1330 found = set(os.listdir(self.dir))
Ezio Melottib3aedd42010-11-20 19:04:17 +00001331 self.assertEqual(found, expected)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001332 # test listdir without arguments
1333 current_directory = os.getcwd()
1334 try:
1335 os.chdir(os.sep)
1336 self.assertEqual(set(os.listdir()), set(os.listdir(os.sep)))
1337 finally:
1338 os.chdir(current_directory)
Martin v. Löwis011e8422009-05-05 04:43:17 +00001339
1340 def test_open(self):
1341 for fn in self.unicodefn:
Victor Stinnera6d2c762011-06-30 18:20:11 +02001342 f = open(os.path.join(self.dir, fn), 'rb')
Martin v. Löwis011e8422009-05-05 04:43:17 +00001343 f.close()
1344
1345 def test_stat(self):
1346 for fn in self.unicodefn:
1347 os.stat(os.path.join(self.dir, fn))
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00001348else:
1349 class PosixUidGidTests(unittest.TestCase):
1350 pass
Martin v. Löwis011e8422009-05-05 04:43:17 +00001351 class Pep383Tests(unittest.TestCase):
1352 pass
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00001353
Brian Curtineb24d742010-04-12 17:16:38 +00001354@unittest.skipUnless(sys.platform == "win32", "Win32 specific tests")
1355class Win32KillTests(unittest.TestCase):
Brian Curtinc3acbc32010-05-28 16:08:40 +00001356 def _kill(self, sig):
1357 # Start sys.executable as a subprocess and communicate from the
1358 # subprocess to the parent that the interpreter is ready. When it
1359 # becomes ready, send *sig* via os.kill to the subprocess and check
1360 # that the return code is equal to *sig*.
1361 import ctypes
1362 from ctypes import wintypes
1363 import msvcrt
1364
1365 # Since we can't access the contents of the process' stdout until the
1366 # process has exited, use PeekNamedPipe to see what's inside stdout
1367 # without waiting. This is done so we can tell that the interpreter
1368 # is started and running at a point where it could handle a signal.
1369 PeekNamedPipe = ctypes.windll.kernel32.PeekNamedPipe
1370 PeekNamedPipe.restype = wintypes.BOOL
1371 PeekNamedPipe.argtypes = (wintypes.HANDLE, # Pipe handle
1372 ctypes.POINTER(ctypes.c_char), # stdout buf
1373 wintypes.DWORD, # Buffer size
1374 ctypes.POINTER(wintypes.DWORD), # bytes read
1375 ctypes.POINTER(wintypes.DWORD), # bytes avail
1376 ctypes.POINTER(wintypes.DWORD)) # bytes left
1377 msg = "running"
1378 proc = subprocess.Popen([sys.executable, "-c",
1379 "import sys;"
1380 "sys.stdout.write('{}');"
1381 "sys.stdout.flush();"
1382 "input()".format(msg)],
1383 stdout=subprocess.PIPE,
1384 stderr=subprocess.PIPE,
1385 stdin=subprocess.PIPE)
Brian Curtin43ec5772010-11-05 15:17:11 +00001386 self.addCleanup(proc.stdout.close)
1387 self.addCleanup(proc.stderr.close)
1388 self.addCleanup(proc.stdin.close)
Brian Curtinc3acbc32010-05-28 16:08:40 +00001389
1390 count, max = 0, 100
1391 while count < max and proc.poll() is None:
1392 # Create a string buffer to store the result of stdout from the pipe
1393 buf = ctypes.create_string_buffer(len(msg))
1394 # Obtain the text currently in proc.stdout
1395 # Bytes read/avail/left are left as NULL and unused
1396 rslt = PeekNamedPipe(msvcrt.get_osfhandle(proc.stdout.fileno()),
1397 buf, ctypes.sizeof(buf), None, None, None)
1398 self.assertNotEqual(rslt, 0, "PeekNamedPipe failed")
1399 if buf.value:
1400 self.assertEqual(msg, buf.value.decode())
1401 break
1402 time.sleep(0.1)
1403 count += 1
1404 else:
1405 self.fail("Did not receive communication from the subprocess")
1406
Brian Curtineb24d742010-04-12 17:16:38 +00001407 os.kill(proc.pid, sig)
1408 self.assertEqual(proc.wait(), sig)
1409
1410 def test_kill_sigterm(self):
1411 # SIGTERM doesn't mean anything special, but make sure it works
Brian Curtinc3acbc32010-05-28 16:08:40 +00001412 self._kill(signal.SIGTERM)
Brian Curtineb24d742010-04-12 17:16:38 +00001413
1414 def test_kill_int(self):
1415 # os.kill on Windows can take an int which gets set as the exit code
Brian Curtinc3acbc32010-05-28 16:08:40 +00001416 self._kill(100)
Brian Curtineb24d742010-04-12 17:16:38 +00001417
1418 def _kill_with_event(self, event, name):
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +00001419 tagname = "test_os_%s" % uuid.uuid1()
1420 m = mmap.mmap(-1, 1, tagname)
1421 m[0] = 0
Brian Curtineb24d742010-04-12 17:16:38 +00001422 # Run a script which has console control handling enabled.
1423 proc = subprocess.Popen([sys.executable,
1424 os.path.join(os.path.dirname(__file__),
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +00001425 "win_console_handler.py"), tagname],
Brian Curtineb24d742010-04-12 17:16:38 +00001426 creationflags=subprocess.CREATE_NEW_PROCESS_GROUP)
1427 # Let the interpreter startup before we send signals. See #3137.
Hirokazu Yamamoto8e9fe9f2010-12-05 02:41:46 +00001428 count, max = 0, 100
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +00001429 while count < max and proc.poll() is None:
Brian Curtinf668df52010-10-15 14:21:06 +00001430 if m[0] == 1:
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +00001431 break
Hirokazu Yamamoto8e9fe9f2010-12-05 02:41:46 +00001432 time.sleep(0.1)
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +00001433 count += 1
1434 else:
Hirokazu Yamamoto8e9fe9f2010-12-05 02:41:46 +00001435 # Forcefully kill the process if we weren't able to signal it.
1436 os.kill(proc.pid, signal.SIGINT)
Hirokazu Yamamoto54c950f2010-10-08 08:38:15 +00001437 self.fail("Subprocess didn't finish initialization")
Brian Curtineb24d742010-04-12 17:16:38 +00001438 os.kill(proc.pid, event)
1439 # proc.send_signal(event) could also be done here.
1440 # Allow time for the signal to be passed and the process to exit.
1441 time.sleep(0.5)
1442 if not proc.poll():
1443 # Forcefully kill the process if we weren't able to signal it.
1444 os.kill(proc.pid, signal.SIGINT)
1445 self.fail("subprocess did not stop on {}".format(name))
1446
1447 @unittest.skip("subprocesses aren't inheriting CTRL+C property")
1448 def test_CTRL_C_EVENT(self):
1449 from ctypes import wintypes
1450 import ctypes
1451
1452 # Make a NULL value by creating a pointer with no argument.
1453 NULL = ctypes.POINTER(ctypes.c_int)()
1454 SetConsoleCtrlHandler = ctypes.windll.kernel32.SetConsoleCtrlHandler
1455 SetConsoleCtrlHandler.argtypes = (ctypes.POINTER(ctypes.c_int),
1456 wintypes.BOOL)
1457 SetConsoleCtrlHandler.restype = wintypes.BOOL
1458
1459 # Calling this with NULL and FALSE causes the calling process to
1460 # handle CTRL+C, rather than ignore it. This property is inherited
1461 # by subprocesses.
1462 SetConsoleCtrlHandler(NULL, 0)
1463
1464 self._kill_with_event(signal.CTRL_C_EVENT, "CTRL_C_EVENT")
1465
1466 def test_CTRL_BREAK_EVENT(self):
1467 self._kill_with_event(signal.CTRL_BREAK_EVENT, "CTRL_BREAK_EVENT")
1468
1469
Brian Curtind40e6f72010-07-08 21:39:08 +00001470@unittest.skipUnless(sys.platform == "win32", "Win32 specific tests")
Brian Curtin3b4499c2010-12-28 14:31:47 +00001471@support.skip_unless_symlink
Brian Curtind40e6f72010-07-08 21:39:08 +00001472class Win32SymlinkTests(unittest.TestCase):
1473 filelink = 'filelinktest'
1474 filelink_target = os.path.abspath(__file__)
1475 dirlink = 'dirlinktest'
1476 dirlink_target = os.path.dirname(filelink_target)
1477 missing_link = 'missing link'
1478
1479 def setUp(self):
1480 assert os.path.exists(self.dirlink_target)
1481 assert os.path.exists(self.filelink_target)
1482 assert not os.path.exists(self.dirlink)
1483 assert not os.path.exists(self.filelink)
1484 assert not os.path.exists(self.missing_link)
1485
1486 def tearDown(self):
1487 if os.path.exists(self.filelink):
1488 os.remove(self.filelink)
1489 if os.path.exists(self.dirlink):
1490 os.rmdir(self.dirlink)
1491 if os.path.lexists(self.missing_link):
1492 os.remove(self.missing_link)
1493
1494 def test_directory_link(self):
Antoine Pitrou5311c1d2012-01-24 08:59:28 +01001495 os.symlink(self.dirlink_target, self.dirlink, True)
Brian Curtind40e6f72010-07-08 21:39:08 +00001496 self.assertTrue(os.path.exists(self.dirlink))
1497 self.assertTrue(os.path.isdir(self.dirlink))
1498 self.assertTrue(os.path.islink(self.dirlink))
1499 self.check_stat(self.dirlink, self.dirlink_target)
1500
1501 def test_file_link(self):
1502 os.symlink(self.filelink_target, self.filelink)
1503 self.assertTrue(os.path.exists(self.filelink))
1504 self.assertTrue(os.path.isfile(self.filelink))
1505 self.assertTrue(os.path.islink(self.filelink))
1506 self.check_stat(self.filelink, self.filelink_target)
1507
1508 def _create_missing_dir_link(self):
1509 'Create a "directory" link to a non-existent target'
1510 linkname = self.missing_link
1511 if os.path.lexists(linkname):
1512 os.remove(linkname)
1513 target = r'c:\\target does not exist.29r3c740'
1514 assert not os.path.exists(target)
1515 target_is_dir = True
1516 os.symlink(target, linkname, target_is_dir)
1517
1518 def test_remove_directory_link_to_missing_target(self):
1519 self._create_missing_dir_link()
1520 # For compatibility with Unix, os.remove will check the
1521 # directory status and call RemoveDirectory if the symlink
1522 # was created with target_is_dir==True.
1523 os.remove(self.missing_link)
1524
1525 @unittest.skip("currently fails; consider for improvement")
1526 def test_isdir_on_directory_link_to_missing_target(self):
1527 self._create_missing_dir_link()
1528 # consider having isdir return true for directory links
1529 self.assertTrue(os.path.isdir(self.missing_link))
1530
1531 @unittest.skip("currently fails; consider for improvement")
1532 def test_rmdir_on_directory_link_to_missing_target(self):
1533 self._create_missing_dir_link()
1534 # consider allowing rmdir to remove directory links
1535 os.rmdir(self.missing_link)
1536
1537 def check_stat(self, link, target):
1538 self.assertEqual(os.stat(link), os.stat(target))
1539 self.assertNotEqual(os.lstat(link), os.stat(link))
1540
Brian Curtind25aef52011-06-13 15:16:04 -05001541 bytes_link = os.fsencode(link)
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01001542 with warnings.catch_warnings():
1543 warnings.simplefilter("ignore", DeprecationWarning)
1544 self.assertEqual(os.stat(bytes_link), os.stat(target))
1545 self.assertNotEqual(os.lstat(bytes_link), os.stat(bytes_link))
Brian Curtind25aef52011-06-13 15:16:04 -05001546
1547 def test_12084(self):
1548 level1 = os.path.abspath(support.TESTFN)
1549 level2 = os.path.join(level1, "level2")
1550 level3 = os.path.join(level2, "level3")
1551 try:
1552 os.mkdir(level1)
1553 os.mkdir(level2)
1554 os.mkdir(level3)
1555
1556 file1 = os.path.abspath(os.path.join(level1, "file1"))
1557
1558 with open(file1, "w") as f:
1559 f.write("file1")
1560
1561 orig_dir = os.getcwd()
1562 try:
1563 os.chdir(level2)
1564 link = os.path.join(level2, "link")
1565 os.symlink(os.path.relpath(file1), "link")
1566 self.assertIn("link", os.listdir(os.getcwd()))
1567
1568 # Check os.stat calls from the same dir as the link
1569 self.assertEqual(os.stat(file1), os.stat("link"))
1570
1571 # Check os.stat calls from a dir below the link
1572 os.chdir(level1)
1573 self.assertEqual(os.stat(file1),
1574 os.stat(os.path.relpath(link)))
1575
1576 # Check os.stat calls from a dir above the link
1577 os.chdir(level3)
1578 self.assertEqual(os.stat(file1),
1579 os.stat(os.path.relpath(link)))
1580 finally:
1581 os.chdir(orig_dir)
1582 except OSError as err:
1583 self.fail(err)
1584 finally:
1585 os.remove(file1)
1586 shutil.rmtree(level1)
1587
Brian Curtind40e6f72010-07-08 21:39:08 +00001588
Victor Stinnere8d51452010-08-19 01:05:19 +00001589class FSEncodingTests(unittest.TestCase):
1590 def test_nop(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +00001591 self.assertEqual(os.fsencode(b'abc\xff'), b'abc\xff')
1592 self.assertEqual(os.fsdecode('abc\u0141'), 'abc\u0141')
Benjamin Peterson31191a92010-05-09 03:22:58 +00001593
Victor Stinnere8d51452010-08-19 01:05:19 +00001594 def test_identity(self):
1595 # assert fsdecode(fsencode(x)) == x
1596 for fn in ('unicode\u0141', 'latin\xe9', 'ascii'):
1597 try:
1598 bytesfn = os.fsencode(fn)
1599 except UnicodeEncodeError:
1600 continue
Ezio Melottib3aedd42010-11-20 19:04:17 +00001601 self.assertEqual(os.fsdecode(bytesfn), fn)
Victor Stinnere8d51452010-08-19 01:05:19 +00001602
Victor Stinnerbf9bcab2010-05-09 03:15:33 +00001603
Brett Cannonefb00c02012-02-29 18:31:31 -05001604
1605class DeviceEncodingTests(unittest.TestCase):
1606
1607 def test_bad_fd(self):
1608 # Return None when an fd doesn't actually exist.
1609 self.assertIsNone(os.device_encoding(123456))
1610
Philip Jenveye308b7c2012-02-29 16:16:15 -08001611 @unittest.skipUnless(os.isatty(0) and (sys.platform.startswith('win') or
1612 (hasattr(locale, 'nl_langinfo') and hasattr(locale, 'CODESET'))),
Philip Jenveyd7aff2d2012-02-29 16:21:25 -08001613 'test requires a tty and either Windows or nl_langinfo(CODESET)')
Brett Cannonefb00c02012-02-29 18:31:31 -05001614 def test_device_encoding(self):
1615 encoding = os.device_encoding(0)
1616 self.assertIsNotNone(encoding)
1617 self.assertTrue(codecs.lookup(encoding))
1618
1619
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00001620class PidTests(unittest.TestCase):
1621 @unittest.skipUnless(hasattr(os, 'getppid'), "test needs os.getppid")
1622 def test_getppid(self):
1623 p = subprocess.Popen([sys.executable, '-c',
1624 'import os; print(os.getppid())'],
1625 stdout=subprocess.PIPE)
1626 stdout, _ = p.communicate()
1627 # We are the parent of our subprocess
1628 self.assertEqual(int(stdout), os.getpid())
1629
1630
Brian Curtin0151b8e2010-09-24 13:43:43 +00001631# The introduction of this TestCase caused at least two different errors on
1632# *nix buildbots. Temporarily skip this to let the buildbots move along.
1633@unittest.skip("Skip due to platform/environment differences on *NIX buildbots")
Brian Curtine8e4b3b2010-09-23 20:04:14 +00001634@unittest.skipUnless(hasattr(os, 'getlogin'), "test needs os.getlogin")
1635class LoginTests(unittest.TestCase):
1636 def test_getlogin(self):
1637 user_name = os.getlogin()
1638 self.assertNotEqual(len(user_name), 0)
1639
1640
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00001641@unittest.skipUnless(hasattr(os, 'getpriority') and hasattr(os, 'setpriority'),
1642 "needs os.getpriority and os.setpriority")
1643class ProgramPriorityTests(unittest.TestCase):
1644 """Tests for os.getpriority() and os.setpriority()."""
1645
1646 def test_set_get_priority(self):
Giampaolo Rodolàcfbcec32011-02-28 19:27:16 +00001647
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00001648 base = os.getpriority(os.PRIO_PROCESS, os.getpid())
1649 os.setpriority(os.PRIO_PROCESS, os.getpid(), base + 1)
1650 try:
Giampaolo Rodolàcfbcec32011-02-28 19:27:16 +00001651 new_prio = os.getpriority(os.PRIO_PROCESS, os.getpid())
1652 if base >= 19 and new_prio <= 19:
1653 raise unittest.SkipTest(
1654 "unable to reliably test setpriority at current nice level of %s" % base)
1655 else:
1656 self.assertEqual(new_prio, base + 1)
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00001657 finally:
1658 try:
1659 os.setpriority(os.PRIO_PROCESS, os.getpid(), base)
1660 except OSError as err:
Antoine Pitrou692f0382011-02-26 00:22:25 +00001661 if err.errno != errno.EACCES:
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00001662 raise
1663
1664
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001665if threading is not None:
1666 class SendfileTestServer(asyncore.dispatcher, threading.Thread):
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001667
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001668 class Handler(asynchat.async_chat):
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001669
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001670 def __init__(self, conn):
1671 asynchat.async_chat.__init__(self, conn)
1672 self.in_buffer = []
1673 self.closed = False
1674 self.push(b"220 ready\r\n")
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001675
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001676 def handle_read(self):
1677 data = self.recv(4096)
1678 self.in_buffer.append(data)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001679
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001680 def get_data(self):
1681 return b''.join(self.in_buffer)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001682
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001683 def handle_close(self):
1684 self.close()
1685 self.closed = True
1686
1687 def handle_error(self):
1688 raise
1689
1690 def __init__(self, address):
1691 threading.Thread.__init__(self)
1692 asyncore.dispatcher.__init__(self)
1693 self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
1694 self.bind(address)
1695 self.listen(5)
1696 self.host, self.port = self.socket.getsockname()[:2]
1697 self.handler_instance = None
1698 self._active = False
1699 self._active_lock = threading.Lock()
1700
1701 # --- public API
1702
1703 @property
1704 def running(self):
1705 return self._active
1706
1707 def start(self):
1708 assert not self.running
1709 self.__flag = threading.Event()
1710 threading.Thread.start(self)
1711 self.__flag.wait()
1712
1713 def stop(self):
1714 assert self.running
1715 self._active = False
1716 self.join()
1717
1718 def wait(self):
1719 # wait for handler connection to be closed, then stop the server
1720 while not getattr(self.handler_instance, "closed", False):
1721 time.sleep(0.001)
1722 self.stop()
1723
1724 # --- internals
1725
1726 def run(self):
1727 self._active = True
1728 self.__flag.set()
1729 while self._active and asyncore.socket_map:
1730 self._active_lock.acquire()
1731 asyncore.loop(timeout=0.001, count=1)
1732 self._active_lock.release()
1733 asyncore.close_all()
1734
1735 def handle_accept(self):
1736 conn, addr = self.accept()
1737 self.handler_instance = self.Handler(conn)
1738
1739 def handle_connect(self):
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001740 self.close()
Giampaolo Rodola'566f8a62011-05-18 21:28:39 +02001741 handle_read = handle_connect
1742
1743 def writable(self):
1744 return 0
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001745
1746 def handle_error(self):
1747 raise
1748
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001749
Giampaolo Rodolà46134642011-02-25 20:01:05 +00001750@unittest.skipUnless(threading is not None, "test needs threading module")
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001751@unittest.skipUnless(hasattr(os, 'sendfile'), "test needs os.sendfile()")
1752class TestSendfile(unittest.TestCase):
1753
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001754 DATA = b"12345abcde" * 16 * 1024 # 160 KB
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001755 SUPPORT_HEADERS_TRAILERS = not sys.platform.startswith("linux") and \
Giampaolo Rodolà4bc68572011-02-25 21:46:01 +00001756 not sys.platform.startswith("solaris") and \
1757 not sys.platform.startswith("sunos")
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001758
1759 @classmethod
1760 def setUpClass(cls):
1761 with open(support.TESTFN, "wb") as f:
1762 f.write(cls.DATA)
1763
1764 @classmethod
1765 def tearDownClass(cls):
1766 support.unlink(support.TESTFN)
1767
1768 def setUp(self):
1769 self.server = SendfileTestServer((support.HOST, 0))
1770 self.server.start()
1771 self.client = socket.socket()
1772 self.client.connect((self.server.host, self.server.port))
1773 self.client.settimeout(1)
1774 # synchronize by waiting for "220 ready" response
1775 self.client.recv(1024)
1776 self.sockno = self.client.fileno()
1777 self.file = open(support.TESTFN, 'rb')
1778 self.fileno = self.file.fileno()
1779
1780 def tearDown(self):
1781 self.file.close()
1782 self.client.close()
1783 if self.server.running:
1784 self.server.stop()
1785
1786 def sendfile_wrapper(self, sock, file, offset, nbytes, headers=[], trailers=[]):
1787 """A higher level wrapper representing how an application is
1788 supposed to use sendfile().
1789 """
1790 while 1:
1791 try:
1792 if self.SUPPORT_HEADERS_TRAILERS:
1793 return os.sendfile(sock, file, offset, nbytes, headers,
1794 trailers)
1795 else:
1796 return os.sendfile(sock, file, offset, nbytes)
1797 except OSError as err:
1798 if err.errno == errno.ECONNRESET:
1799 # disconnected
1800 raise
1801 elif err.errno in (errno.EAGAIN, errno.EBUSY):
1802 # we have to retry send data
1803 continue
1804 else:
1805 raise
1806
1807 def test_send_whole_file(self):
1808 # normal send
1809 total_sent = 0
1810 offset = 0
1811 nbytes = 4096
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001812 while total_sent < len(self.DATA):
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001813 sent = self.sendfile_wrapper(self.sockno, self.fileno, offset, nbytes)
1814 if sent == 0:
1815 break
1816 offset += sent
1817 total_sent += sent
1818 self.assertTrue(sent <= nbytes)
1819 self.assertEqual(offset, total_sent)
1820
1821 self.assertEqual(total_sent, len(self.DATA))
Antoine Pitrou2de51ff2011-02-26 17:52:50 +00001822 self.client.shutdown(socket.SHUT_RDWR)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001823 self.client.close()
1824 self.server.wait()
1825 data = self.server.handler_instance.get_data()
Antoine Pitrou2de51ff2011-02-26 17:52:50 +00001826 self.assertEqual(len(data), len(self.DATA))
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001827 self.assertEqual(data, self.DATA)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001828
1829 def test_send_at_certain_offset(self):
1830 # start sending a file at a certain offset
1831 total_sent = 0
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001832 offset = len(self.DATA) // 2
1833 must_send = len(self.DATA) - offset
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001834 nbytes = 4096
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001835 while total_sent < must_send:
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001836 sent = self.sendfile_wrapper(self.sockno, self.fileno, offset, nbytes)
1837 if sent == 0:
1838 break
1839 offset += sent
1840 total_sent += sent
1841 self.assertTrue(sent <= nbytes)
1842
Antoine Pitrou2de51ff2011-02-26 17:52:50 +00001843 self.client.shutdown(socket.SHUT_RDWR)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001844 self.client.close()
1845 self.server.wait()
1846 data = self.server.handler_instance.get_data()
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001847 expected = self.DATA[len(self.DATA) // 2:]
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001848 self.assertEqual(total_sent, len(expected))
Antoine Pitrou2de51ff2011-02-26 17:52:50 +00001849 self.assertEqual(len(data), len(expected))
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001850 self.assertEqual(data, expected)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001851
1852 def test_offset_overflow(self):
1853 # specify an offset > file size
1854 offset = len(self.DATA) + 4096
Antoine Pitrou18dd0df2011-02-26 14:29:24 +00001855 try:
1856 sent = os.sendfile(self.sockno, self.fileno, offset, 4096)
1857 except OSError as e:
1858 # Solaris can raise EINVAL if offset >= file length, ignore.
1859 if e.errno != errno.EINVAL:
1860 raise
1861 else:
1862 self.assertEqual(sent, 0)
Antoine Pitrou2de51ff2011-02-26 17:52:50 +00001863 self.client.shutdown(socket.SHUT_RDWR)
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001864 self.client.close()
1865 self.server.wait()
1866 data = self.server.handler_instance.get_data()
1867 self.assertEqual(data, b'')
1868
1869 def test_invalid_offset(self):
1870 with self.assertRaises(OSError) as cm:
1871 os.sendfile(self.sockno, self.fileno, -1, 4096)
1872 self.assertEqual(cm.exception.errno, errno.EINVAL)
1873
1874 # --- headers / trailers tests
1875
1876 if SUPPORT_HEADERS_TRAILERS:
1877
1878 def test_headers(self):
1879 total_sent = 0
1880 sent = os.sendfile(self.sockno, self.fileno, 0, 4096,
1881 headers=[b"x" * 512])
1882 total_sent += sent
1883 offset = 4096
1884 nbytes = 4096
1885 while 1:
1886 sent = self.sendfile_wrapper(self.sockno, self.fileno,
1887 offset, nbytes)
1888 if sent == 0:
1889 break
1890 total_sent += sent
1891 offset += sent
1892
1893 expected_data = b"x" * 512 + self.DATA
1894 self.assertEqual(total_sent, len(expected_data))
1895 self.client.close()
1896 self.server.wait()
1897 data = self.server.handler_instance.get_data()
1898 self.assertEqual(hash(data), hash(expected_data))
1899
1900 def test_trailers(self):
1901 TESTFN2 = support.TESTFN + "2"
Brett Cannonb6376802011-03-15 17:38:22 -04001902 with open(TESTFN2, 'wb') as f:
1903 f.write(b"abcde")
1904 with open(TESTFN2, 'rb')as f:
1905 self.addCleanup(os.remove, TESTFN2)
1906 os.sendfile(self.sockno, f.fileno(), 0, 4096,
1907 trailers=[b"12345"])
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001908 self.client.close()
1909 self.server.wait()
1910 data = self.server.handler_instance.get_data()
1911 self.assertEqual(data, b"abcde12345")
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00001912
1913 if hasattr(os, "SF_NODISKIO"):
1914 def test_flags(self):
1915 try:
1916 os.sendfile(self.sockno, self.fileno, 0, 4096,
1917 flags=os.SF_NODISKIO)
1918 except OSError as err:
1919 if err.errno not in (errno.EBUSY, errno.EAGAIN):
1920 raise
1921
1922
Larry Hastings9cf065c2012-06-22 16:30:09 -07001923def supports_extended_attributes():
1924 if not hasattr(os, "setxattr"):
1925 return False
1926 try:
1927 with open(support.TESTFN, "wb") as fp:
1928 try:
1929 os.setxattr(fp.fileno(), b"user.test", b"")
1930 except OSError:
1931 return False
1932 finally:
1933 support.unlink(support.TESTFN)
1934 # Kernels < 2.6.39 don't respect setxattr flags.
1935 kernel_version = platform.release()
1936 m = re.match("2.6.(\d{1,2})", kernel_version)
1937 return m is None or int(m.group(1)) >= 39
1938
1939
1940@unittest.skipUnless(supports_extended_attributes(),
1941 "no non-broken extended attribute support")
Benjamin Peterson799bd802011-08-31 22:15:17 -04001942class ExtendedAttributeTests(unittest.TestCase):
1943
1944 def tearDown(self):
1945 support.unlink(support.TESTFN)
1946
Larry Hastings9cf065c2012-06-22 16:30:09 -07001947 def _check_xattrs_str(self, s, getxattr, setxattr, removexattr, listxattr, **kwargs):
Benjamin Peterson799bd802011-08-31 22:15:17 -04001948 fn = support.TESTFN
1949 open(fn, "wb").close()
1950 with self.assertRaises(OSError) as cm:
Larry Hastings9cf065c2012-06-22 16:30:09 -07001951 getxattr(fn, s("user.test"), **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001952 self.assertEqual(cm.exception.errno, errno.ENODATA)
Victor Stinnerf12e5062011-10-16 22:12:03 +02001953 init_xattr = listxattr(fn)
1954 self.assertIsInstance(init_xattr, list)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001955 setxattr(fn, s("user.test"), b"", **kwargs)
Victor Stinnerf12e5062011-10-16 22:12:03 +02001956 xattr = set(init_xattr)
1957 xattr.add("user.test")
1958 self.assertEqual(set(listxattr(fn)), xattr)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001959 self.assertEqual(getxattr(fn, b"user.test", **kwargs), b"")
1960 setxattr(fn, s("user.test"), b"hello", os.XATTR_REPLACE, **kwargs)
1961 self.assertEqual(getxattr(fn, b"user.test", **kwargs), b"hello")
Benjamin Peterson799bd802011-08-31 22:15:17 -04001962 with self.assertRaises(OSError) as cm:
Larry Hastings9cf065c2012-06-22 16:30:09 -07001963 setxattr(fn, s("user.test"), b"bye", os.XATTR_CREATE, **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001964 self.assertEqual(cm.exception.errno, errno.EEXIST)
1965 with self.assertRaises(OSError) as cm:
Larry Hastings9cf065c2012-06-22 16:30:09 -07001966 setxattr(fn, s("user.test2"), b"bye", os.XATTR_REPLACE, **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001967 self.assertEqual(cm.exception.errno, errno.ENODATA)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001968 setxattr(fn, s("user.test2"), b"foo", os.XATTR_CREATE, **kwargs)
Victor Stinnerf12e5062011-10-16 22:12:03 +02001969 xattr.add("user.test2")
1970 self.assertEqual(set(listxattr(fn)), xattr)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001971 removexattr(fn, s("user.test"), **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001972 with self.assertRaises(OSError) as cm:
Larry Hastings9cf065c2012-06-22 16:30:09 -07001973 getxattr(fn, s("user.test"), **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001974 self.assertEqual(cm.exception.errno, errno.ENODATA)
Victor Stinnerf12e5062011-10-16 22:12:03 +02001975 xattr.remove("user.test")
1976 self.assertEqual(set(listxattr(fn)), xattr)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001977 self.assertEqual(getxattr(fn, s("user.test2"), **kwargs), b"foo")
1978 setxattr(fn, s("user.test"), b"a"*1024, **kwargs)
1979 self.assertEqual(getxattr(fn, s("user.test"), **kwargs), b"a"*1024)
1980 removexattr(fn, s("user.test"), **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001981 many = sorted("user.test{}".format(i) for i in range(100))
1982 for thing in many:
Larry Hastings9cf065c2012-06-22 16:30:09 -07001983 setxattr(fn, thing, b"x", **kwargs)
Victor Stinnerf12e5062011-10-16 22:12:03 +02001984 self.assertEqual(set(listxattr(fn)), set(init_xattr) | set(many))
Benjamin Peterson799bd802011-08-31 22:15:17 -04001985
Larry Hastings9cf065c2012-06-22 16:30:09 -07001986 def _check_xattrs(self, *args, **kwargs):
Benjamin Peterson799bd802011-08-31 22:15:17 -04001987 def make_bytes(s):
1988 return bytes(s, "ascii")
Larry Hastings9cf065c2012-06-22 16:30:09 -07001989 self._check_xattrs_str(str, *args, **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001990 support.unlink(support.TESTFN)
Larry Hastings9cf065c2012-06-22 16:30:09 -07001991 self._check_xattrs_str(make_bytes, *args, **kwargs)
Benjamin Peterson799bd802011-08-31 22:15:17 -04001992
1993 def test_simple(self):
1994 self._check_xattrs(os.getxattr, os.setxattr, os.removexattr,
1995 os.listxattr)
1996
1997 def test_lpath(self):
Larry Hastings9cf065c2012-06-22 16:30:09 -07001998 self._check_xattrs(os.getxattr, os.setxattr, os.removexattr,
1999 os.listxattr, follow_symlinks=False)
Benjamin Peterson799bd802011-08-31 22:15:17 -04002000
2001 def test_fds(self):
2002 def getxattr(path, *args):
2003 with open(path, "rb") as fp:
Larry Hastings9cf065c2012-06-22 16:30:09 -07002004 return os.getxattr(fp.fileno(), *args)
Benjamin Peterson799bd802011-08-31 22:15:17 -04002005 def setxattr(path, *args):
2006 with open(path, "wb") as fp:
Larry Hastings9cf065c2012-06-22 16:30:09 -07002007 os.setxattr(fp.fileno(), *args)
Benjamin Peterson799bd802011-08-31 22:15:17 -04002008 def removexattr(path, *args):
2009 with open(path, "wb") as fp:
Larry Hastings9cf065c2012-06-22 16:30:09 -07002010 os.removexattr(fp.fileno(), *args)
Benjamin Peterson799bd802011-08-31 22:15:17 -04002011 def listxattr(path, *args):
2012 with open(path, "rb") as fp:
Larry Hastings9cf065c2012-06-22 16:30:09 -07002013 return os.listxattr(fp.fileno(), *args)
Benjamin Peterson799bd802011-08-31 22:15:17 -04002014 self._check_xattrs(getxattr, setxattr, removexattr, listxattr)
2015
2016
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002017@unittest.skipUnless(sys.platform == "win32", "Win32 specific tests")
2018class Win32DeprecatedBytesAPI(unittest.TestCase):
2019 def test_deprecated(self):
2020 import nt
2021 filename = os.fsencode(support.TESTFN)
2022 with warnings.catch_warnings():
2023 warnings.simplefilter("error", DeprecationWarning)
2024 for func, *args in (
2025 (nt._getfullpathname, filename),
2026 (nt._isdir, filename),
2027 (os.access, filename, os.R_OK),
2028 (os.chdir, filename),
2029 (os.chmod, filename, 0o777),
Victor Stinnerf7c5ae22011-11-16 23:43:07 +01002030 (os.getcwdb,),
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002031 (os.link, filename, filename),
2032 (os.listdir, filename),
2033 (os.lstat, filename),
2034 (os.mkdir, filename),
2035 (os.open, filename, os.O_RDONLY),
2036 (os.rename, filename, filename),
2037 (os.rmdir, filename),
2038 (os.startfile, filename),
2039 (os.stat, filename),
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002040 (os.unlink, filename),
2041 (os.utime, filename),
2042 ):
2043 self.assertRaises(DeprecationWarning, func, *args)
2044
Victor Stinner28216442011-11-16 00:34:44 +01002045 @support.skip_unless_symlink
2046 def test_symlink(self):
2047 filename = os.fsencode(support.TESTFN)
2048 with warnings.catch_warnings():
2049 warnings.simplefilter("error", DeprecationWarning)
2050 self.assertRaises(DeprecationWarning,
2051 os.symlink, filename, filename)
2052
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002053
Antoine Pitroubcf2b592012-02-08 23:28:36 +01002054@unittest.skipUnless(hasattr(os, 'get_terminal_size'), "requires os.get_terminal_size")
2055class TermsizeTests(unittest.TestCase):
2056 def test_does_not_crash(self):
2057 """Check if get_terminal_size() returns a meaningful value.
2058
2059 There's no easy portable way to actually check the size of the
2060 terminal, so let's check if it returns something sensible instead.
2061 """
2062 try:
2063 size = os.get_terminal_size()
2064 except OSError as e:
Antoine Pitrou81a1fa52012-02-09 00:11:00 +01002065 if sys.platform == "win32" or e.errno in (errno.EINVAL, errno.ENOTTY):
Antoine Pitroubcf2b592012-02-08 23:28:36 +01002066 # Under win32 a generic OSError can be thrown if the
2067 # handle cannot be retrieved
2068 self.skipTest("failed to query terminal size")
2069 raise
2070
Antoine Pitroucfade362012-02-08 23:48:59 +01002071 self.assertGreaterEqual(size.columns, 0)
2072 self.assertGreaterEqual(size.lines, 0)
Antoine Pitroubcf2b592012-02-08 23:28:36 +01002073
2074 def test_stty_match(self):
2075 """Check if stty returns the same results
2076
2077 stty actually tests stdin, so get_terminal_size is invoked on
2078 stdin explicitly. If stty succeeded, then get_terminal_size()
2079 should work too.
2080 """
2081 try:
2082 size = subprocess.check_output(['stty', 'size']).decode().split()
2083 except (FileNotFoundError, subprocess.CalledProcessError):
2084 self.skipTest("stty invocation failed")
2085 expected = (int(size[1]), int(size[0])) # reversed order
2086
Antoine Pitrou81a1fa52012-02-09 00:11:00 +01002087 try:
2088 actual = os.get_terminal_size(sys.__stdin__.fileno())
2089 except OSError as e:
2090 if sys.platform == "win32" or e.errno in (errno.EINVAL, errno.ENOTTY):
2091 # Under win32 a generic OSError can be thrown if the
2092 # handle cannot be retrieved
2093 self.skipTest("failed to query terminal size")
2094 raise
Antoine Pitroubcf2b592012-02-08 23:28:36 +01002095 self.assertEqual(expected, actual)
2096
2097
Antoine Pitrouf26ad712011-07-15 23:00:56 +02002098@support.reap_threads
Fred Drake2e2be372001-09-20 21:33:42 +00002099def test_main():
Benjamin Petersonee8712c2008-05-20 21:35:26 +00002100 support.run_unittest(
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002101 FileTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00002102 StatAttributeTests,
2103 EnvironTests,
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +00002104 WalkTests,
Charles-François Natali7372b062012-02-05 15:15:38 +01002105 FwalkTests,
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +00002106 MakedirTests,
Martin v. Löwisbdec50f2004-06-08 08:29:33 +00002107 DevNullTests,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002108 URandomTests,
Guido van Rossume7ba4952007-06-06 23:52:48 +00002109 ExecTests,
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00002110 Win32ErrorTests,
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00002111 TestInvalidFD,
Martin v. Löwis011e8422009-05-05 04:43:17 +00002112 PosixUidGidTests,
Brian Curtineb24d742010-04-12 17:16:38 +00002113 Pep383Tests,
Victor Stinnerbf9bcab2010-05-09 03:15:33 +00002114 Win32KillTests,
Brian Curtind40e6f72010-07-08 21:39:08 +00002115 Win32SymlinkTests,
Victor Stinnere8d51452010-08-19 01:05:19 +00002116 FSEncodingTests,
Brett Cannonefb00c02012-02-29 18:31:31 -05002117 DeviceEncodingTests,
Amaury Forgeot d'Arc4b6fdf32010-09-07 21:31:17 +00002118 PidTests,
Brian Curtine8e4b3b2010-09-23 20:04:14 +00002119 LoginTests,
Brian Curtin1b9df392010-11-24 20:24:31 +00002120 LinkTests,
Giampaolo Rodolàc9c2c8b2011-02-25 14:39:16 +00002121 TestSendfile,
Giampaolo Rodolà18e8bcb2011-02-25 20:57:54 +00002122 ProgramPriorityTests,
Benjamin Peterson799bd802011-08-31 22:15:17 -04002123 ExtendedAttributeTests,
Victor Stinner1ab6c2d2011-11-15 22:27:41 +01002124 Win32DeprecatedBytesAPI,
Antoine Pitroubcf2b592012-02-08 23:28:36 +01002125 TermsizeTests,
Andrew Svetlov405faed2012-12-25 12:18:09 +02002126 RemoveDirsTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00002127 )
Fred Drake2e2be372001-09-20 21:33:42 +00002128
2129if __name__ == "__main__":
2130 test_main()