blob: 14a9a7a0cf37aec1e4875815b0cb82a3fa28cb9b [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
Fred Drake38c2ef02001-07-17 20:52:51 +000016
Mark Dickinson7cf03892010-04-16 13:45:35 +000017# Detect whether we're on a Linux system that uses the (now outdated
18# and unmaintained) linuxthreads threading library. There's an issue
19# when combining linuxthreads with a failed execv call: see
20# http://bugs.python.org/issue4970.
Mark Dickinson89589c92010-04-16 13:51:27 +000021if (hasattr(os, "confstr_names") and
22 "CS_GNU_LIBPTHREAD_VERSION" in os.confstr_names):
Mark Dickinson7cf03892010-04-16 13:45:35 +000023 libpthread = os.confstr("CS_GNU_LIBPTHREAD_VERSION")
24 USING_LINUXTHREADS= libpthread.startswith("linuxthreads")
25else:
26 USING_LINUXTHREADS= False
Brian Curtineb24d742010-04-12 17:16:38 +000027
Thomas Wouters0e3f5912006-08-11 14:57:12 +000028# Tests creating TESTFN
29class FileTests(unittest.TestCase):
30 def setUp(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +000031 if os.path.exists(support.TESTFN):
32 os.unlink(support.TESTFN)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000033 tearDown = setUp
34
35 def test_access(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +000036 f = os.open(support.TESTFN, os.O_CREAT|os.O_RDWR)
Thomas Wouters0e3f5912006-08-11 14:57:12 +000037 os.close(f)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000038 self.assertTrue(os.access(support.TESTFN, os.W_OK))
Thomas Wouters0e3f5912006-08-11 14:57:12 +000039
Christian Heimesfdab48e2008-01-20 09:06:41 +000040 def test_closerange(self):
Antoine Pitroub9ee06c2008-08-16 22:03:17 +000041 first = os.open(support.TESTFN, os.O_CREAT|os.O_RDWR)
42 # We must allocate two consecutive file descriptors, otherwise
43 # it will mess up other file descriptors (perhaps even the three
44 # standard ones).
45 second = os.dup(first)
46 try:
47 retries = 0
48 while second != first + 1:
49 os.close(first)
50 retries += 1
51 if retries > 10:
52 # XXX test skipped
Benjamin Petersonfa0d7032009-06-01 22:42:33 +000053 self.skipTest("couldn't allocate two consecutive fds")
Antoine Pitroub9ee06c2008-08-16 22:03:17 +000054 first, second = second, os.dup(second)
55 finally:
56 os.close(second)
Christian Heimesfdab48e2008-01-20 09:06:41 +000057 # close a fd that is open, and one that isn't
Antoine Pitroub9ee06c2008-08-16 22:03:17 +000058 os.closerange(first, first + 2)
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +000059 self.assertRaises(OSError, os.write, first, b"a")
Thomas Wouters0e3f5912006-08-11 14:57:12 +000060
Benjamin Peterson1cc6df92010-06-30 17:39:45 +000061 @support.cpython_only
Hirokazu Yamamoto4c19e6e2008-09-08 23:41:21 +000062 def test_rename(self):
63 path = support.TESTFN
64 old = sys.getrefcount(path)
65 self.assertRaises(TypeError, os.rename, path, 0)
66 new = sys.getrefcount(path)
67 self.assertEqual(old, new)
68
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +000069 def test_read(self):
70 with open(support.TESTFN, "w+b") as fobj:
71 fobj.write(b"spam")
72 fobj.flush()
73 fd = fobj.fileno()
74 os.lseek(fd, 0, 0)
75 s = os.read(fd, 4)
76 self.assertEqual(type(s), bytes)
77 self.assertEqual(s, b"spam")
78
79 def test_write(self):
80 # os.write() accepts bytes- and buffer-like objects but not strings
81 fd = os.open(support.TESTFN, os.O_CREAT | os.O_WRONLY)
82 self.assertRaises(TypeError, os.write, fd, "beans")
83 os.write(fd, b"bacon\n")
84 os.write(fd, bytearray(b"eggs\n"))
85 os.write(fd, memoryview(b"spam\n"))
86 os.close(fd)
87 with open(support.TESTFN, "rb") as fobj:
Antoine Pitroud62269f2008-09-15 23:54:52 +000088 self.assertEqual(fobj.read().splitlines(),
89 [b"bacon", b"eggs", b"spam"])
Antoine Pitrou9cadb1b2008-09-15 23:02:56 +000090
91
Christian Heimesdd15f6c2008-03-16 00:07:10 +000092class TemporaryFileTests(unittest.TestCase):
93 def setUp(self):
94 self.files = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +000095 os.mkdir(support.TESTFN)
Christian Heimesdd15f6c2008-03-16 00:07:10 +000096
97 def tearDown(self):
98 for name in self.files:
99 os.unlink(name)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000100 os.rmdir(support.TESTFN)
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000101
102 def check_tempfile(self, name):
103 # make sure it doesn't already exist:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000104 self.assertFalse(os.path.exists(name),
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000105 "file already exists for temporary file")
106 # make sure we can create the file
107 open(name, "w")
108 self.files.append(name)
109
110 def test_tempnam(self):
111 if not hasattr(os, "tempnam"):
112 return
113 warnings.filterwarnings("ignore", "tempnam", RuntimeWarning,
114 r"test_os$")
115 self.check_tempfile(os.tempnam())
116
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000117 name = os.tempnam(support.TESTFN)
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000118 self.check_tempfile(name)
119
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000120 name = os.tempnam(support.TESTFN, "pfx")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000121 self.assertTrue(os.path.basename(name)[:3] == "pfx")
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000122 self.check_tempfile(name)
123
124 def test_tmpfile(self):
125 if not hasattr(os, "tmpfile"):
126 return
127 # As with test_tmpnam() below, the Windows implementation of tmpfile()
128 # attempts to create a file in the root directory of the current drive.
129 # On Vista and Server 2008, this test will always fail for normal users
130 # as writing to the root directory requires elevated privileges. With
131 # XP and below, the semantics of tmpfile() are the same, but the user
132 # running the test is more likely to have administrative privileges on
133 # their account already. If that's the case, then os.tmpfile() should
134 # work. In order to make this test as useful as possible, rather than
135 # trying to detect Windows versions or whether or not the user has the
136 # right permissions, just try and create a file in the root directory
137 # and see if it raises a 'Permission denied' OSError. If it does, then
138 # test that a subsequent call to os.tmpfile() raises the same error. If
139 # it doesn't, assume we're on XP or below and the user running the test
140 # has administrative privileges, and proceed with the test as normal.
141 if sys.platform == 'win32':
142 name = '\\python_test_os_test_tmpfile.txt'
143 if os.path.exists(name):
144 os.remove(name)
145 try:
146 fp = open(name, 'w')
147 except IOError as first:
148 # open() failed, assert tmpfile() fails in the same way.
149 # Although open() raises an IOError and os.tmpfile() raises an
150 # OSError(), 'args' will be (13, 'Permission denied') in both
151 # cases.
152 try:
153 fp = os.tmpfile()
154 except OSError as second:
155 self.assertEqual(first.args, second.args)
156 else:
157 self.fail("expected os.tmpfile() to raise OSError")
158 return
159 else:
160 # open() worked, therefore, tmpfile() should work. Close our
161 # dummy file and proceed with the test as normal.
162 fp.close()
163 os.remove(name)
164
165 fp = os.tmpfile()
166 fp.write("foobar")
167 fp.seek(0,0)
168 s = fp.read()
169 fp.close()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000170 self.assertTrue(s == "foobar")
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000171
172 def test_tmpnam(self):
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000173 if not hasattr(os, "tmpnam"):
174 return
175 warnings.filterwarnings("ignore", "tmpnam", RuntimeWarning,
176 r"test_os$")
177 name = os.tmpnam()
178 if sys.platform in ("win32",):
179 # The Windows tmpnam() seems useless. From the MS docs:
180 #
181 # The character string that tmpnam creates consists of
182 # the path prefix, defined by the entry P_tmpdir in the
183 # file STDIO.H, followed by a sequence consisting of the
184 # digit characters '0' through '9'; the numerical value
185 # of this string is in the range 1 - 65,535. Changing the
186 # definitions of L_tmpnam or P_tmpdir in STDIO.H does not
187 # change the operation of tmpnam.
188 #
189 # The really bizarre part is that, at least under MSVC6,
190 # P_tmpdir is "\\". That is, the path returned refers to
191 # the root of the current drive. That's a terrible place to
192 # put temp files, and, depending on privileges, the user
193 # may not even be able to open a file in the root directory.
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000194 self.assertFalse(os.path.exists(name),
Christian Heimesdd15f6c2008-03-16 00:07:10 +0000195 "file already exists for temporary file")
196 else:
197 self.check_tempfile(name)
198
Amaury Forgeot d'Arce2e36ba2008-08-01 00:14:22 +0000199 def fdopen_helper(self, *args):
200 fd = os.open(support.TESTFN, os.O_RDONLY)
201 fp2 = os.fdopen(fd, *args)
202 fp2.close()
203
204 def test_fdopen(self):
205 self.fdopen_helper()
206 self.fdopen_helper('r')
207 self.fdopen_helper('r', 100)
208
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000209# Test attributes on return values from os.*stat* family.
210class StatAttributeTests(unittest.TestCase):
211 def setUp(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000212 os.mkdir(support.TESTFN)
213 self.fname = os.path.join(support.TESTFN, "f1")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000214 f = open(self.fname, 'wb')
Guido van Rossum26d95c32007-08-27 23:18:54 +0000215 f.write(b"ABC")
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000216 f.close()
Tim Peterse0c446b2001-10-18 21:57:37 +0000217
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000218 def tearDown(self):
219 os.unlink(self.fname)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000220 os.rmdir(support.TESTFN)
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000221
222 def test_stat_attributes(self):
223 if not hasattr(os, "stat"):
224 return
225
226 import stat
227 result = os.stat(self.fname)
228
229 # Make sure direct access works
230 self.assertEquals(result[stat.ST_SIZE], 3)
231 self.assertEquals(result.st_size, 3)
232
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000233 # Make sure all the attributes are there
234 members = dir(result)
235 for name in dir(stat):
236 if name[:3] == 'ST_':
237 attr = name.lower()
Martin v. Löwis4d394df2005-01-23 09:19:22 +0000238 if name.endswith("TIME"):
239 def trunc(x): return int(x)
240 else:
241 def trunc(x): return x
242 self.assertEquals(trunc(getattr(result, attr)),
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000243 result[getattr(stat, name)])
Benjamin Peterson577473f2010-01-19 00:09:57 +0000244 self.assertIn(attr, members)
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000245
246 try:
247 result[200]
248 self.fail("No exception thrown")
249 except IndexError:
250 pass
251
252 # Make sure that assignment fails
253 try:
254 result.st_mode = 1
255 self.fail("No exception thrown")
Collin Winter42dae6a2007-03-28 21:44:53 +0000256 except AttributeError:
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000257 pass
258
259 try:
260 result.st_rdev = 1
261 self.fail("No exception thrown")
Guido van Rossum1fff8782001-10-18 21:19:31 +0000262 except (AttributeError, TypeError):
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000263 pass
264
265 try:
266 result.parrot = 1
267 self.fail("No exception thrown")
268 except AttributeError:
269 pass
270
271 # Use the stat_result constructor with a too-short tuple.
272 try:
273 result2 = os.stat_result((10,))
274 self.fail("No exception thrown")
275 except TypeError:
276 pass
277
278 # Use the constructr with a too-long tuple.
279 try:
280 result2 = os.stat_result((0,1,2,3,4,5,6,7,8,9,10,11,12,13,14))
281 except TypeError:
282 pass
283
Tim Peterse0c446b2001-10-18 21:57:37 +0000284
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000285 def test_statvfs_attributes(self):
286 if not hasattr(os, "statvfs"):
287 return
288
Martin v. Löwisf90ae202002-06-11 06:22:31 +0000289 try:
290 result = os.statvfs(self.fname)
Guido van Rossumb940e112007-01-10 16:19:56 +0000291 except OSError as e:
Martin v. Löwisf90ae202002-06-11 06:22:31 +0000292 # On AtheOS, glibc always returns ENOSYS
Martin v. Löwisf90ae202002-06-11 06:22:31 +0000293 if e.errno == errno.ENOSYS:
294 return
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000295
296 # Make sure direct access works
Brett Cannoncfaf10c2008-05-16 00:45:35 +0000297 self.assertEquals(result.f_bfree, result[3])
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000298
Brett Cannoncfaf10c2008-05-16 00:45:35 +0000299 # Make sure all the attributes are there.
300 members = ('bsize', 'frsize', 'blocks', 'bfree', 'bavail', 'files',
301 'ffree', 'favail', 'flag', 'namemax')
302 for value, member in enumerate(members):
303 self.assertEquals(getattr(result, 'f_' + member), result[value])
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000304
305 # Make sure that assignment really fails
306 try:
307 result.f_bfree = 1
308 self.fail("No exception thrown")
Collin Winter42dae6a2007-03-28 21:44:53 +0000309 except AttributeError:
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000310 pass
311
312 try:
313 result.parrot = 1
314 self.fail("No exception thrown")
315 except AttributeError:
316 pass
317
318 # Use the constructor with a too-short tuple.
319 try:
320 result2 = os.statvfs_result((10,))
321 self.fail("No exception thrown")
322 except TypeError:
323 pass
324
325 # Use the constructr with a too-long tuple.
326 try:
327 result2 = os.statvfs_result((0,1,2,3,4,5,6,7,8,9,10,11,12,13,14))
328 except TypeError:
329 pass
Fred Drake38c2ef02001-07-17 20:52:51 +0000330
Thomas Wouters89f507f2006-12-13 04:49:30 +0000331 def test_utime_dir(self):
332 delta = 1000000
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000333 st = os.stat(support.TESTFN)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000334 # round to int, because some systems may support sub-second
335 # time stamps in stat, but not in utime.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000336 os.utime(support.TESTFN, (st.st_atime, int(st.st_mtime-delta)))
337 st2 = os.stat(support.TESTFN)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000338 self.assertEquals(st2.st_mtime, int(st.st_mtime-delta))
339
340 # Restrict test to Win32, since there is no guarantee other
341 # systems support centiseconds
342 if sys.platform == 'win32':
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000343 def get_file_system(path):
Hirokazu Yamamoto5ef6d182008-08-20 04:17:24 +0000344 root = os.path.splitdrive(os.path.abspath(path))[0] + '\\'
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000345 import ctypes
Hirokazu Yamamotoca765d52008-08-20 16:18:19 +0000346 kernel32 = ctypes.windll.kernel32
Hirokazu Yamamoto5ef6d182008-08-20 04:17:24 +0000347 buf = ctypes.create_unicode_buffer("", 100)
Hirokazu Yamamotoca765d52008-08-20 16:18:19 +0000348 if kernel32.GetVolumeInformationW(root, None, 0, None, None, None, buf, len(buf)):
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000349 return buf.value
350
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000351 if get_file_system(support.TESTFN) == "NTFS":
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000352 def test_1565150(self):
353 t1 = 1159195039.25
354 os.utime(self.fname, (t1, t1))
355 self.assertEquals(os.stat(self.fname).st_mtime, t1)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000356
Guido van Rossumd8faa362007-04-27 19:54:29 +0000357 def test_1686475(self):
358 # Verify that an open file can be stat'ed
359 try:
360 os.stat(r"c:\pagefile.sys")
361 except WindowsError as e:
Benjamin Petersonc4fe6f32008-08-19 18:57:56 +0000362 if e.errno == 2: # file does not exist; cannot run test
Guido van Rossumd8faa362007-04-27 19:54:29 +0000363 return
364 self.fail("Could not stat pagefile.sys")
365
Walter Dörwald0a6d0ff2004-05-31 16:29:04 +0000366from test import mapping_tests
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000367
Walter Dörwald0a6d0ff2004-05-31 16:29:04 +0000368class EnvironTests(mapping_tests.BasicTestMappingProtocol):
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000369 """check that os.environ object conform to mapping protocol"""
Walter Dörwald118f9312004-06-02 18:42:25 +0000370 type2test = None
Christian Heimes90333392007-11-01 19:08:42 +0000371
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000372 def setUp(self):
373 self.__save = dict(os.environ)
Victor Stinnerb745a742010-05-18 17:17:23 +0000374 if os.supports_bytes_environ:
Victor Stinner208d28c2010-05-07 00:54:14 +0000375 self.__saveb = dict(os.environb)
Christian Heimes90333392007-11-01 19:08:42 +0000376 for key, value in self._reference().items():
377 os.environ[key] = value
378
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000379 def tearDown(self):
380 os.environ.clear()
381 os.environ.update(self.__save)
Victor Stinnerb745a742010-05-18 17:17:23 +0000382 if os.supports_bytes_environ:
Victor Stinner208d28c2010-05-07 00:54:14 +0000383 os.environb.clear()
384 os.environb.update(self.__saveb)
Raymond Hettinger2c2d3222003-03-09 07:05:43 +0000385
Christian Heimes90333392007-11-01 19:08:42 +0000386 def _reference(self):
387 return {"KEY1":"VALUE1", "KEY2":"VALUE2", "KEY3":"VALUE3"}
388
389 def _empty_mapping(self):
390 os.environ.clear()
391 return os.environ
392
Martin v. Löwis1d11de62005-01-29 13:29:23 +0000393 # Bug 1110478
Martin v. Löwis5510f652005-02-17 21:23:20 +0000394 def test_update2(self):
Christian Heimes90333392007-11-01 19:08:42 +0000395 os.environ.clear()
Martin v. Löwis1d11de62005-01-29 13:29:23 +0000396 if os.path.exists("/bin/sh"):
397 os.environ.update(HELLO="World")
398 value = os.popen("/bin/sh -c 'echo $HELLO'").read().strip()
399 self.assertEquals(value, "World")
400
Christian Heimes1a13d592007-11-08 14:16:55 +0000401 def test_os_popen_iter(self):
402 if os.path.exists("/bin/sh"):
403 popen = os.popen("/bin/sh -c 'echo \"line1\nline2\nline3\"'")
404 it = iter(popen)
405 self.assertEquals(next(it), "line1\n")
406 self.assertEquals(next(it), "line2\n")
407 self.assertEquals(next(it), "line3\n")
408 self.assertRaises(StopIteration, next, it)
409
Guido van Rossum67aca9e2007-06-13 21:51:27 +0000410 # Verify environ keys and values from the OS are of the
411 # correct str type.
412 def test_keyvalue_types(self):
413 for key, val in os.environ.items():
414 self.assertEquals(type(key), str)
415 self.assertEquals(type(val), str)
416
Christian Heimes90333392007-11-01 19:08:42 +0000417 def test_items(self):
418 for key, value in self._reference().items():
419 self.assertEqual(os.environ.get(key), value)
420
Ezio Melotti19e4acf2010-02-22 15:59:01 +0000421 # Issue 7310
422 def test___repr__(self):
423 """Check that the repr() of os.environ looks like environ({...})."""
424 env = os.environ
425 self.assertTrue(isinstance(env.data, dict))
426 self.assertEqual(repr(env), 'environ({!r})'.format(env.data))
427
Gregory P. Smithb6e8c7e2010-02-27 07:22:22 +0000428 def test_get_exec_path(self):
429 defpath_list = os.defpath.split(os.pathsep)
430 test_path = ['/monty', '/python', '', '/flying/circus']
431 test_env = {'PATH': os.pathsep.join(test_path)}
432
433 saved_environ = os.environ
434 try:
435 os.environ = dict(test_env)
436 # Test that defaulting to os.environ works.
437 self.assertSequenceEqual(test_path, os.get_exec_path())
438 self.assertSequenceEqual(test_path, os.get_exec_path(env=None))
439 finally:
440 os.environ = saved_environ
441
442 # No PATH environment variable
443 self.assertSequenceEqual(defpath_list, os.get_exec_path({}))
444 # Empty PATH environment variable
445 self.assertSequenceEqual(('',), os.get_exec_path({'PATH':''}))
446 # Supplied PATH environment variable
447 self.assertSequenceEqual(test_path, os.get_exec_path(test_env))
448
Victor Stinnerb745a742010-05-18 17:17:23 +0000449 if os.supports_bytes_environ:
450 # env cannot contain 'PATH' and b'PATH' keys
451 self.assertRaises(ValueError,
452 os.get_exec_path, {'PATH': '1', b'PATH': b'2'})
453
454 # bytes key and/or value
455 self.assertSequenceEqual(os.get_exec_path({b'PATH': b'abc'}),
456 ['abc'])
457 self.assertSequenceEqual(os.get_exec_path({b'PATH': 'abc'}),
458 ['abc'])
459 self.assertSequenceEqual(os.get_exec_path({'PATH': b'abc'}),
460 ['abc'])
461
462 @unittest.skipUnless(os.supports_bytes_environ,
463 "os.environb required for this test.")
Victor Stinner84ae1182010-05-06 22:05:07 +0000464 def test_environb(self):
465 # os.environ -> os.environb
466 value = 'euro\u20ac'
467 try:
Benjamin Peterson180799d2010-05-06 22:25:42 +0000468 value_bytes = value.encode(sys.getfilesystemencoding(),
469 'surrogateescape')
Victor Stinner84ae1182010-05-06 22:05:07 +0000470 except UnicodeEncodeError:
Benjamin Peterson180799d2010-05-06 22:25:42 +0000471 msg = "U+20AC character is not encodable to %s" % (
472 sys.getfilesystemencoding(),)
Benjamin Peterson932d3f42010-05-06 22:26:31 +0000473 self.skipTest(msg)
Victor Stinner84ae1182010-05-06 22:05:07 +0000474 os.environ['unicode'] = value
475 self.assertEquals(os.environ['unicode'], value)
476 self.assertEquals(os.environb[b'unicode'], value_bytes)
477
478 # os.environb -> os.environ
479 value = b'\xff'
480 os.environb[b'bytes'] = value
481 self.assertEquals(os.environb[b'bytes'], value)
482 value_str = value.decode(sys.getfilesystemencoding(), 'surrogateescape')
483 self.assertEquals(os.environ['bytes'], value_str)
Ezio Melotti19e4acf2010-02-22 15:59:01 +0000484
Tim Petersc4e09402003-04-25 07:11:48 +0000485class WalkTests(unittest.TestCase):
486 """Tests for os.walk()."""
487
488 def test_traversal(self):
489 import os
490 from os.path import join
491
492 # Build:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000493 # TESTFN/
494 # TEST1/ a file kid and two directory kids
Tim Petersc4e09402003-04-25 07:11:48 +0000495 # tmp1
496 # SUB1/ a file kid and a directory kid
Guido van Rossumd8faa362007-04-27 19:54:29 +0000497 # tmp2
498 # SUB11/ no kids
499 # SUB2/ a file kid and a dirsymlink kid
500 # tmp3
501 # link/ a symlink to TESTFN.2
502 # TEST2/
503 # tmp4 a lone file
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000504 walk_path = join(support.TESTFN, "TEST1")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000505 sub1_path = join(walk_path, "SUB1")
Tim Petersc4e09402003-04-25 07:11:48 +0000506 sub11_path = join(sub1_path, "SUB11")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000507 sub2_path = join(walk_path, "SUB2")
508 tmp1_path = join(walk_path, "tmp1")
Tim Petersc4e09402003-04-25 07:11:48 +0000509 tmp2_path = join(sub1_path, "tmp2")
510 tmp3_path = join(sub2_path, "tmp3")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000511 link_path = join(sub2_path, "link")
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000512 t2_path = join(support.TESTFN, "TEST2")
513 tmp4_path = join(support.TESTFN, "TEST2", "tmp4")
Tim Petersc4e09402003-04-25 07:11:48 +0000514
515 # Create stuff.
516 os.makedirs(sub11_path)
517 os.makedirs(sub2_path)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000518 os.makedirs(t2_path)
519 for path in tmp1_path, tmp2_path, tmp3_path, tmp4_path:
Alex Martelli01c77c62006-08-24 02:58:11 +0000520 f = open(path, "w")
Tim Petersc4e09402003-04-25 07:11:48 +0000521 f.write("I'm " + path + " and proud of it. Blame test_os.\n")
522 f.close()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000523 if hasattr(os, "symlink"):
524 os.symlink(os.path.abspath(t2_path), link_path)
525 sub2_tree = (sub2_path, ["link"], ["tmp3"])
526 else:
527 sub2_tree = (sub2_path, [], ["tmp3"])
Tim Petersc4e09402003-04-25 07:11:48 +0000528
529 # Walk top-down.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000530 all = list(os.walk(walk_path))
Tim Petersc4e09402003-04-25 07:11:48 +0000531 self.assertEqual(len(all), 4)
532 # We can't know which order SUB1 and SUB2 will appear in.
533 # Not flipped: TESTFN, SUB1, SUB11, SUB2
534 # flipped: TESTFN, SUB2, SUB1, SUB11
535 flipped = all[0][1][0] != "SUB1"
536 all[0][1].sort()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000537 self.assertEqual(all[0], (walk_path, ["SUB1", "SUB2"], ["tmp1"]))
Tim Petersc4e09402003-04-25 07:11:48 +0000538 self.assertEqual(all[1 + flipped], (sub1_path, ["SUB11"], ["tmp2"]))
539 self.assertEqual(all[2 + flipped], (sub11_path, [], []))
Guido van Rossumd8faa362007-04-27 19:54:29 +0000540 self.assertEqual(all[3 - 2 * flipped], sub2_tree)
Tim Petersc4e09402003-04-25 07:11:48 +0000541
542 # Prune the search.
543 all = []
Guido van Rossumd8faa362007-04-27 19:54:29 +0000544 for root, dirs, files in os.walk(walk_path):
Tim Petersc4e09402003-04-25 07:11:48 +0000545 all.append((root, dirs, files))
546 # Don't descend into SUB1.
547 if 'SUB1' in dirs:
548 # Note that this also mutates the dirs we appended to all!
549 dirs.remove('SUB1')
550 self.assertEqual(len(all), 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000551 self.assertEqual(all[0], (walk_path, ["SUB2"], ["tmp1"]))
552 self.assertEqual(all[1], sub2_tree)
Tim Petersc4e09402003-04-25 07:11:48 +0000553
554 # Walk bottom-up.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000555 all = list(os.walk(walk_path, topdown=False))
Tim Petersc4e09402003-04-25 07:11:48 +0000556 self.assertEqual(len(all), 4)
557 # We can't know which order SUB1 and SUB2 will appear in.
558 # Not flipped: SUB11, SUB1, SUB2, TESTFN
559 # flipped: SUB2, SUB11, SUB1, TESTFN
560 flipped = all[3][1][0] != "SUB1"
561 all[3][1].sort()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000562 self.assertEqual(all[3], (walk_path, ["SUB1", "SUB2"], ["tmp1"]))
Tim Petersc4e09402003-04-25 07:11:48 +0000563 self.assertEqual(all[flipped], (sub11_path, [], []))
564 self.assertEqual(all[flipped + 1], (sub1_path, ["SUB11"], ["tmp2"]))
Guido van Rossumd8faa362007-04-27 19:54:29 +0000565 self.assertEqual(all[2 - 2 * flipped], sub2_tree)
Tim Petersc4e09402003-04-25 07:11:48 +0000566
Guido van Rossumd8faa362007-04-27 19:54:29 +0000567 if hasattr(os, "symlink"):
568 # Walk, following symlinks.
569 for root, dirs, files in os.walk(walk_path, followlinks=True):
570 if root == link_path:
571 self.assertEqual(dirs, [])
572 self.assertEqual(files, ["tmp4"])
573 break
574 else:
575 self.fail("Didn't follow symlink with followlinks=True")
576
577 def tearDown(self):
Tim Petersc4e09402003-04-25 07:11:48 +0000578 # Tear everything down. This is a decent use for bottom-up on
579 # Windows, which doesn't have a recursive delete command. The
580 # (not so) subtlety is that rmdir will fail unless the dir's
581 # kids are removed first, so bottom up is essential.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000582 for root, dirs, files in os.walk(support.TESTFN, topdown=False):
Tim Petersc4e09402003-04-25 07:11:48 +0000583 for name in files:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000584 os.remove(os.path.join(root, name))
Tim Petersc4e09402003-04-25 07:11:48 +0000585 for name in dirs:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000586 dirname = os.path.join(root, name)
587 if not os.path.islink(dirname):
588 os.rmdir(dirname)
589 else:
590 os.remove(dirname)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000591 os.rmdir(support.TESTFN)
Tim Petersc4e09402003-04-25 07:11:48 +0000592
Guido van Rossume7ba4952007-06-06 23:52:48 +0000593class MakedirTests(unittest.TestCase):
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000594 def setUp(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000595 os.mkdir(support.TESTFN)
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000596
597 def test_makedir(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000598 base = support.TESTFN
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000599 path = os.path.join(base, 'dir1', 'dir2', 'dir3')
600 os.makedirs(path) # Should work
601 path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4')
602 os.makedirs(path)
603
604 # Try paths with a '.' in them
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000605 self.assertRaises(OSError, os.makedirs, os.curdir)
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000606 path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4', 'dir5', os.curdir)
607 os.makedirs(path)
608 path = os.path.join(base, 'dir1', os.curdir, 'dir2', 'dir3', 'dir4',
609 'dir5', 'dir6')
610 os.makedirs(path)
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000611
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000612 def tearDown(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000613 path = os.path.join(support.TESTFN, 'dir1', 'dir2', 'dir3',
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000614 'dir4', 'dir5', 'dir6')
615 # If the tests failed, the bottom-most directory ('../dir6')
616 # may not have been created, so we look for the outermost directory
617 # that exists.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000618 while not os.path.exists(path) and path != support.TESTFN:
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000619 path = os.path.dirname(path)
620
621 os.removedirs(path)
622
Guido van Rossume7ba4952007-06-06 23:52:48 +0000623class DevNullTests(unittest.TestCase):
Martin v. Löwisbdec50f2004-06-08 08:29:33 +0000624 def test_devnull(self):
Alex Martelli01c77c62006-08-24 02:58:11 +0000625 f = open(os.devnull, 'w')
Martin v. Löwisbdec50f2004-06-08 08:29:33 +0000626 f.write('hello')
627 f.close()
Alex Martelli01c77c62006-08-24 02:58:11 +0000628 f = open(os.devnull, 'r')
Tim Peters4182cfd2004-06-08 20:34:34 +0000629 self.assertEqual(f.read(), '')
Martin v. Löwisbdec50f2004-06-08 08:29:33 +0000630 f.close()
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +0000631
Guido van Rossume7ba4952007-06-06 23:52:48 +0000632class URandomTests(unittest.TestCase):
Martin v. Löwisdc3883f2004-08-29 15:46:35 +0000633 def test_urandom(self):
634 try:
635 self.assertEqual(len(os.urandom(1)), 1)
636 self.assertEqual(len(os.urandom(10)), 10)
637 self.assertEqual(len(os.urandom(100)), 100)
638 self.assertEqual(len(os.urandom(1000)), 1000)
639 except NotImplementedError:
640 pass
641
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000642@contextlib.contextmanager
643def _execvpe_mockup(defpath=None):
644 """
645 Stubs out execv and execve functions when used as context manager.
646 Records exec calls. The mock execv and execve functions always raise an
647 exception as they would normally never return.
648 """
649 # A list of tuples containing (function name, first arg, args)
650 # of calls to execv or execve that have been made.
651 calls = []
652
653 def mock_execv(name, *args):
654 calls.append(('execv', name, args))
655 raise RuntimeError("execv called")
656
657 def mock_execve(name, *args):
658 calls.append(('execve', name, args))
659 raise OSError(errno.ENOTDIR, "execve called")
660
661 try:
662 orig_execv = os.execv
663 orig_execve = os.execve
664 orig_defpath = os.defpath
665 os.execv = mock_execv
666 os.execve = mock_execve
667 if defpath is not None:
668 os.defpath = defpath
669 yield calls
670 finally:
671 os.execv = orig_execv
672 os.execve = orig_execve
673 os.defpath = orig_defpath
674
Guido van Rossume7ba4952007-06-06 23:52:48 +0000675class ExecTests(unittest.TestCase):
Mark Dickinson7cf03892010-04-16 13:45:35 +0000676 @unittest.skipIf(USING_LINUXTHREADS,
677 "avoid triggering a linuxthreads bug: see issue #4970")
Guido van Rossume7ba4952007-06-06 23:52:48 +0000678 def test_execvpe_with_bad_program(self):
Mark Dickinson7cf03892010-04-16 13:45:35 +0000679 self.assertRaises(OSError, os.execvpe, 'no such app-',
680 ['no such app-'], None)
Guido van Rossume7ba4952007-06-06 23:52:48 +0000681
Thomas Heller6790d602007-08-30 17:15:14 +0000682 def test_execvpe_with_bad_arglist(self):
683 self.assertRaises(ValueError, os.execvpe, 'notepad', [], None)
684
Gregory P. Smith4ae37772010-05-08 18:05:46 +0000685 @unittest.skipUnless(hasattr(os, '_execvpe'),
686 "No internal os._execvpe function to test.")
Victor Stinnerb745a742010-05-18 17:17:23 +0000687 def _test_internal_execvpe(self, test_type):
688 program_path = os.sep + 'absolutepath'
689 if test_type is bytes:
690 program = b'executable'
691 fullpath = os.path.join(os.fsencode(program_path), program)
692 native_fullpath = fullpath
693 arguments = [b'progname', 'arg1', 'arg2']
694 else:
695 program = 'executable'
696 arguments = ['progname', 'arg1', 'arg2']
697 fullpath = os.path.join(program_path, program)
698 if os.name != "nt":
699 native_fullpath = os.fsencode(fullpath)
700 else:
701 native_fullpath = fullpath
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000702 env = {'spam': 'beans'}
703
Victor Stinnerb745a742010-05-18 17:17:23 +0000704 # test os._execvpe() with an absolute path
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000705 with _execvpe_mockup() as calls:
Victor Stinnerb745a742010-05-18 17:17:23 +0000706 self.assertRaises(RuntimeError,
707 os._execvpe, fullpath, arguments)
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000708 self.assertEqual(len(calls), 1)
709 self.assertEqual(calls[0], ('execv', fullpath, (arguments,)))
710
Victor Stinnerb745a742010-05-18 17:17:23 +0000711 # test os._execvpe() with a relative path:
712 # os.get_exec_path() returns defpath
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000713 with _execvpe_mockup(defpath=program_path) as calls:
Victor Stinnerb745a742010-05-18 17:17:23 +0000714 self.assertRaises(OSError,
715 os._execvpe, program, arguments, env=env)
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000716 self.assertEqual(len(calls), 1)
Victor Stinnerb745a742010-05-18 17:17:23 +0000717 self.assertSequenceEqual(calls[0],
718 ('execve', native_fullpath, (arguments, env)))
719
720 # test os._execvpe() with a relative path:
721 # os.get_exec_path() reads the 'PATH' variable
722 with _execvpe_mockup() as calls:
723 env_path = env.copy()
724 env_path['PATH'] = program_path
725 self.assertRaises(OSError,
726 os._execvpe, program, arguments, env=env_path)
727 self.assertEqual(len(calls), 1)
728 self.assertSequenceEqual(calls[0],
729 ('execve', native_fullpath, (arguments, env_path)))
730
731 def test_internal_execvpe_str(self):
732 self._test_internal_execvpe(str)
733 if os.name != "nt":
734 self._test_internal_execvpe(bytes)
Victor Stinnerc2d095f2010-05-17 00:14:53 +0000735
Gregory P. Smith4ae37772010-05-08 18:05:46 +0000736
Thomas Wouters477c8d52006-05-27 19:21:47 +0000737class Win32ErrorTests(unittest.TestCase):
738 def test_rename(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000739 self.assertRaises(WindowsError, os.rename, support.TESTFN, support.TESTFN+".bak")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000740
741 def test_remove(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000742 self.assertRaises(WindowsError, os.remove, support.TESTFN)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000743
744 def test_chdir(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000745 self.assertRaises(WindowsError, os.chdir, support.TESTFN)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000746
747 def test_mkdir(self):
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000748 f = open(support.TESTFN, "w")
Benjamin Petersonf91df042009-02-13 02:50:59 +0000749 try:
750 self.assertRaises(WindowsError, os.mkdir, support.TESTFN)
751 finally:
752 f.close()
Amaury Forgeot d'Arc2fc224f2009-02-19 23:23:47 +0000753 os.unlink(support.TESTFN)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000754
755 def test_utime(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000756 self.assertRaises(WindowsError, os.utime, support.TESTFN, None)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000757
Thomas Wouters477c8d52006-05-27 19:21:47 +0000758 def test_chmod(self):
Benjamin Petersonf91df042009-02-13 02:50:59 +0000759 self.assertRaises(WindowsError, os.chmod, support.TESTFN, 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000760
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000761class TestInvalidFD(unittest.TestCase):
Benjamin Peterson05e782f2009-01-19 15:15:02 +0000762 singles = ["fchdir", "dup", "fdopen", "fdatasync", "fstat",
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000763 "fstatvfs", "fsync", "tcgetpgrp", "ttyname"]
764 #singles.append("close")
765 #We omit close because it doesn'r raise an exception on some platforms
766 def get_single(f):
767 def helper(self):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000768 if hasattr(os, f):
769 self.check(getattr(os, f))
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000770 return helper
771 for f in singles:
772 locals()["test_"+f] = get_single(f)
773
Benjamin Peterson7522c742009-01-19 21:00:09 +0000774 def check(self, f, *args):
Benjamin Peterson5c6d7872009-02-06 02:40:07 +0000775 try:
776 f(support.make_bad_fd(), *args)
777 except OSError as e:
778 self.assertEqual(e.errno, errno.EBADF)
779 else:
780 self.fail("%r didn't raise a OSError with a bad file descriptor"
781 % f)
Benjamin Peterson7522c742009-01-19 21:00:09 +0000782
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000783 def test_isatty(self):
784 if hasattr(os, "isatty"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000785 self.assertEqual(os.isatty(support.make_bad_fd()), False)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000786
787 def test_closerange(self):
788 if hasattr(os, "closerange"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000789 fd = support.make_bad_fd()
R. David Murray630cc482009-07-22 15:20:27 +0000790 # Make sure none of the descriptors we are about to close are
791 # currently valid (issue 6542).
792 for i in range(10):
793 try: os.fstat(fd+i)
794 except OSError:
795 pass
796 else:
797 break
798 if i < 2:
799 raise unittest.SkipTest(
800 "Unable to acquire a range of invalid file descriptors")
801 self.assertEqual(os.closerange(fd, fd + i-1), None)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000802
803 def test_dup2(self):
804 if hasattr(os, "dup2"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000805 self.check(os.dup2, 20)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000806
807 def test_fchmod(self):
808 if hasattr(os, "fchmod"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000809 self.check(os.fchmod, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000810
811 def test_fchown(self):
812 if hasattr(os, "fchown"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000813 self.check(os.fchown, -1, -1)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000814
815 def test_fpathconf(self):
816 if hasattr(os, "fpathconf"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000817 self.check(os.fpathconf, "PC_NAME_MAX")
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000818
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000819 def test_ftruncate(self):
820 if hasattr(os, "ftruncate"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000821 self.check(os.ftruncate, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000822
823 def test_lseek(self):
824 if hasattr(os, "lseek"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000825 self.check(os.lseek, 0, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000826
827 def test_read(self):
828 if hasattr(os, "read"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000829 self.check(os.read, 1)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000830
831 def test_tcsetpgrpt(self):
832 if hasattr(os, "tcsetpgrp"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000833 self.check(os.tcsetpgrp, 0)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000834
835 def test_write(self):
836 if hasattr(os, "write"):
Benjamin Peterson7522c742009-01-19 21:00:09 +0000837 self.check(os.write, b" ")
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000838
Thomas Wouters477c8d52006-05-27 19:21:47 +0000839if sys.platform != 'win32':
840 class Win32ErrorTests(unittest.TestCase):
841 pass
842
Benjamin Petersonef3e4c22009-04-11 19:48:14 +0000843 class PosixUidGidTests(unittest.TestCase):
844 if hasattr(os, 'setuid'):
845 def test_setuid(self):
846 if os.getuid() != 0:
847 self.assertRaises(os.error, os.setuid, 0)
848 self.assertRaises(OverflowError, os.setuid, 1<<32)
849
850 if hasattr(os, 'setgid'):
851 def test_setgid(self):
852 if os.getuid() != 0:
853 self.assertRaises(os.error, os.setgid, 0)
854 self.assertRaises(OverflowError, os.setgid, 1<<32)
855
856 if hasattr(os, 'seteuid'):
857 def test_seteuid(self):
858 if os.getuid() != 0:
859 self.assertRaises(os.error, os.seteuid, 0)
860 self.assertRaises(OverflowError, os.seteuid, 1<<32)
861
862 if hasattr(os, 'setegid'):
863 def test_setegid(self):
864 if os.getuid() != 0:
865 self.assertRaises(os.error, os.setegid, 0)
866 self.assertRaises(OverflowError, os.setegid, 1<<32)
867
868 if hasattr(os, 'setreuid'):
869 def test_setreuid(self):
870 if os.getuid() != 0:
871 self.assertRaises(os.error, os.setreuid, 0, 0)
872 self.assertRaises(OverflowError, os.setreuid, 1<<32, 0)
873 self.assertRaises(OverflowError, os.setreuid, 0, 1<<32)
Benjamin Petersonebe87ba2010-03-06 20:34:24 +0000874
875 def test_setreuid_neg1(self):
876 # Needs to accept -1. We run this in a subprocess to avoid
877 # altering the test runner's process state (issue8045).
Benjamin Petersonebe87ba2010-03-06 20:34:24 +0000878 subprocess.check_call([
879 sys.executable, '-c',
880 'import os,sys;os.setreuid(-1,-1);sys.exit(0)'])
Benjamin Petersonef3e4c22009-04-11 19:48:14 +0000881
882 if hasattr(os, 'setregid'):
883 def test_setregid(self):
884 if os.getuid() != 0:
885 self.assertRaises(os.error, os.setregid, 0, 0)
886 self.assertRaises(OverflowError, os.setregid, 1<<32, 0)
887 self.assertRaises(OverflowError, os.setregid, 0, 1<<32)
Benjamin Petersonebe87ba2010-03-06 20:34:24 +0000888
889 def test_setregid_neg1(self):
890 # Needs to accept -1. We run this in a subprocess to avoid
891 # altering the test runner's process state (issue8045).
Benjamin Petersonebe87ba2010-03-06 20:34:24 +0000892 subprocess.check_call([
893 sys.executable, '-c',
894 'import os,sys;os.setregid(-1,-1);sys.exit(0)'])
Martin v. Löwis011e8422009-05-05 04:43:17 +0000895
Mark Dickinson70613682009-05-05 21:34:59 +0000896 @unittest.skipIf(sys.platform == 'darwin', "tests don't apply to OS X")
Martin v. Löwis011e8422009-05-05 04:43:17 +0000897 class Pep383Tests(unittest.TestCase):
898 filenames = [b'foo\xf6bar', 'foo\xf6bar'.encode("utf-8")]
899
900 def setUp(self):
901 self.fsencoding = sys.getfilesystemencoding()
902 sys.setfilesystemencoding("utf-8")
903 self.dir = support.TESTFN
Martin v. Löwis43c57782009-05-10 08:15:24 +0000904 self.bdir = self.dir.encode("utf-8", "surrogateescape")
Martin v. Löwis011e8422009-05-05 04:43:17 +0000905 os.mkdir(self.dir)
906 self.unicodefn = []
907 for fn in self.filenames:
908 f = open(os.path.join(self.bdir, fn), "w")
909 f.close()
Martin v. Löwis43c57782009-05-10 08:15:24 +0000910 self.unicodefn.append(fn.decode("utf-8", "surrogateescape"))
Martin v. Löwis011e8422009-05-05 04:43:17 +0000911
912 def tearDown(self):
913 shutil.rmtree(self.dir)
914 sys.setfilesystemencoding(self.fsencoding)
915
916 def test_listdir(self):
917 expected = set(self.unicodefn)
918 found = set(os.listdir(support.TESTFN))
919 self.assertEquals(found, expected)
920
921 def test_open(self):
922 for fn in self.unicodefn:
923 f = open(os.path.join(self.dir, fn))
924 f.close()
925
926 def test_stat(self):
927 for fn in self.unicodefn:
928 os.stat(os.path.join(self.dir, fn))
Benjamin Petersonef3e4c22009-04-11 19:48:14 +0000929else:
930 class PosixUidGidTests(unittest.TestCase):
931 pass
Martin v. Löwis011e8422009-05-05 04:43:17 +0000932 class Pep383Tests(unittest.TestCase):
933 pass
Benjamin Petersonef3e4c22009-04-11 19:48:14 +0000934
Brian Curtineb24d742010-04-12 17:16:38 +0000935@unittest.skipUnless(sys.platform == "win32", "Win32 specific tests")
936class Win32KillTests(unittest.TestCase):
Brian Curtinc3acbc32010-05-28 16:08:40 +0000937 def _kill(self, sig):
938 # Start sys.executable as a subprocess and communicate from the
939 # subprocess to the parent that the interpreter is ready. When it
940 # becomes ready, send *sig* via os.kill to the subprocess and check
941 # that the return code is equal to *sig*.
942 import ctypes
943 from ctypes import wintypes
944 import msvcrt
945
946 # Since we can't access the contents of the process' stdout until the
947 # process has exited, use PeekNamedPipe to see what's inside stdout
948 # without waiting. This is done so we can tell that the interpreter
949 # is started and running at a point where it could handle a signal.
950 PeekNamedPipe = ctypes.windll.kernel32.PeekNamedPipe
951 PeekNamedPipe.restype = wintypes.BOOL
952 PeekNamedPipe.argtypes = (wintypes.HANDLE, # Pipe handle
953 ctypes.POINTER(ctypes.c_char), # stdout buf
954 wintypes.DWORD, # Buffer size
955 ctypes.POINTER(wintypes.DWORD), # bytes read
956 ctypes.POINTER(wintypes.DWORD), # bytes avail
957 ctypes.POINTER(wintypes.DWORD)) # bytes left
958 msg = "running"
959 proc = subprocess.Popen([sys.executable, "-c",
960 "import sys;"
961 "sys.stdout.write('{}');"
962 "sys.stdout.flush();"
963 "input()".format(msg)],
964 stdout=subprocess.PIPE,
965 stderr=subprocess.PIPE,
966 stdin=subprocess.PIPE)
967
968 count, max = 0, 100
969 while count < max and proc.poll() is None:
970 # Create a string buffer to store the result of stdout from the pipe
971 buf = ctypes.create_string_buffer(len(msg))
972 # Obtain the text currently in proc.stdout
973 # Bytes read/avail/left are left as NULL and unused
974 rslt = PeekNamedPipe(msvcrt.get_osfhandle(proc.stdout.fileno()),
975 buf, ctypes.sizeof(buf), None, None, None)
976 self.assertNotEqual(rslt, 0, "PeekNamedPipe failed")
977 if buf.value:
978 self.assertEqual(msg, buf.value.decode())
979 break
980 time.sleep(0.1)
981 count += 1
982 else:
983 self.fail("Did not receive communication from the subprocess")
984
Brian Curtineb24d742010-04-12 17:16:38 +0000985 os.kill(proc.pid, sig)
986 self.assertEqual(proc.wait(), sig)
987
988 def test_kill_sigterm(self):
989 # SIGTERM doesn't mean anything special, but make sure it works
Brian Curtinc3acbc32010-05-28 16:08:40 +0000990 self._kill(signal.SIGTERM)
Brian Curtineb24d742010-04-12 17:16:38 +0000991
992 def test_kill_int(self):
993 # os.kill on Windows can take an int which gets set as the exit code
Brian Curtinc3acbc32010-05-28 16:08:40 +0000994 self._kill(100)
Brian Curtineb24d742010-04-12 17:16:38 +0000995
996 def _kill_with_event(self, event, name):
997 # Run a script which has console control handling enabled.
998 proc = subprocess.Popen([sys.executable,
999 os.path.join(os.path.dirname(__file__),
1000 "win_console_handler.py")],
1001 creationflags=subprocess.CREATE_NEW_PROCESS_GROUP)
1002 # Let the interpreter startup before we send signals. See #3137.
1003 time.sleep(0.5)
1004 os.kill(proc.pid, event)
1005 # proc.send_signal(event) could also be done here.
1006 # Allow time for the signal to be passed and the process to exit.
1007 time.sleep(0.5)
1008 if not proc.poll():
1009 # Forcefully kill the process if we weren't able to signal it.
1010 os.kill(proc.pid, signal.SIGINT)
1011 self.fail("subprocess did not stop on {}".format(name))
1012
1013 @unittest.skip("subprocesses aren't inheriting CTRL+C property")
1014 def test_CTRL_C_EVENT(self):
1015 from ctypes import wintypes
1016 import ctypes
1017
1018 # Make a NULL value by creating a pointer with no argument.
1019 NULL = ctypes.POINTER(ctypes.c_int)()
1020 SetConsoleCtrlHandler = ctypes.windll.kernel32.SetConsoleCtrlHandler
1021 SetConsoleCtrlHandler.argtypes = (ctypes.POINTER(ctypes.c_int),
1022 wintypes.BOOL)
1023 SetConsoleCtrlHandler.restype = wintypes.BOOL
1024
1025 # Calling this with NULL and FALSE causes the calling process to
1026 # handle CTRL+C, rather than ignore it. This property is inherited
1027 # by subprocesses.
1028 SetConsoleCtrlHandler(NULL, 0)
1029
1030 self._kill_with_event(signal.CTRL_C_EVENT, "CTRL_C_EVENT")
1031
1032 def test_CTRL_BREAK_EVENT(self):
1033 self._kill_with_event(signal.CTRL_BREAK_EVENT, "CTRL_BREAK_EVENT")
1034
1035
Victor Stinnerbf9bcab2010-05-09 03:15:33 +00001036class MiscTests(unittest.TestCase):
Benjamin Peterson31191a92010-05-09 03:22:58 +00001037
1038 @unittest.skipIf(os.name == "nt", "POSIX specific test")
Victor Stinnerbf9bcab2010-05-09 03:15:33 +00001039 def test_fsencode(self):
1040 self.assertEquals(os.fsencode(b'ab\xff'), b'ab\xff')
1041 self.assertEquals(os.fsencode('ab\uDCFF'), b'ab\xff')
1042
1043
Fred Drake2e2be372001-09-20 21:33:42 +00001044def test_main():
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001045 support.run_unittest(
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001046 FileTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001047 StatAttributeTests,
1048 EnvironTests,
Andrew M. Kuchlingb386f6a2003-12-23 16:36:11 +00001049 WalkTests,
1050 MakedirTests,
Martin v. Löwisbdec50f2004-06-08 08:29:33 +00001051 DevNullTests,
Thomas Wouters477c8d52006-05-27 19:21:47 +00001052 URandomTests,
Guido van Rossume7ba4952007-06-06 23:52:48 +00001053 ExecTests,
Benjamin Petersone1cdfd72009-01-18 21:02:37 +00001054 Win32ErrorTests,
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00001055 TestInvalidFD,
Martin v. Löwis011e8422009-05-05 04:43:17 +00001056 PosixUidGidTests,
Brian Curtineb24d742010-04-12 17:16:38 +00001057 Pep383Tests,
Victor Stinnerbf9bcab2010-05-09 03:15:33 +00001058 Win32KillTests,
1059 MiscTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001060 )
Fred Drake2e2be372001-09-20 21:33:42 +00001061
1062if __name__ == "__main__":
1063 test_main()