blob: 4ad735055d0de490e0bf2d7705de9cf92ad1197c [file] [log] [blame]
Neal Norwitze241ce82003-02-17 18:17:05 +00001"Test posix functions"
2
Benjamin Petersonee8712c2008-05-20 21:35:26 +00003from test import support
R. David Murrayeb3615d2009-04-22 02:24:39 +00004
5# Skip these tests if there is no posix module.
6posix = support.import_module('posix')
Neal Norwitze241ce82003-02-17 18:17:05 +00007
Antoine Pitroub7572f02009-12-02 20:46:48 +00008import errno
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00009import sys
Neal Norwitze241ce82003-02-17 18:17:05 +000010import time
11import os
Charles-François Natali1e045b12011-05-22 20:42:32 +020012import fcntl
Charles-François Nataliab2d58e2012-04-17 19:48:35 +020013import platform
Christian Heimesd5e2b6f2008-03-19 21:50:51 +000014import pwd
Benjamin Petersondcf97b92008-07-02 17:30:14 +000015import shutil
Benjamin Peterson052a02b2010-08-17 01:27:09 +000016import stat
Ned Deilyba2eab22011-07-26 13:53:55 -070017import tempfile
Neal Norwitze241ce82003-02-17 18:17:05 +000018import unittest
19import warnings
R. David Murraya21e4ca2009-03-31 23:16:50 +000020
Ned Deilyba2eab22011-07-26 13:53:55 -070021_DUMMY_SYMLINK = os.path.join(tempfile.gettempdir(),
22 support.TESTFN + '-dummy-symlink')
Neal Norwitze241ce82003-02-17 18:17:05 +000023
24class PosixTester(unittest.TestCase):
25
26 def setUp(self):
27 # create empty file
Benjamin Petersonee8712c2008-05-20 21:35:26 +000028 fp = open(support.TESTFN, 'w+')
Neal Norwitze241ce82003-02-17 18:17:05 +000029 fp.close()
Ned Deily3eb67d52011-06-28 00:00:28 -070030 self.teardown_files = [ support.TESTFN ]
Brett Cannonc8d502e2010-03-20 21:53:28 +000031 self._warnings_manager = support.check_warnings()
32 self._warnings_manager.__enter__()
33 warnings.filterwarnings('ignore', '.* potential security risk .*',
34 RuntimeWarning)
Neal Norwitze241ce82003-02-17 18:17:05 +000035
36 def tearDown(self):
Ned Deily3eb67d52011-06-28 00:00:28 -070037 for teardown_file in self.teardown_files:
38 support.unlink(teardown_file)
Brett Cannonc8d502e2010-03-20 21:53:28 +000039 self._warnings_manager.__exit__(None, None, None)
Neal Norwitze241ce82003-02-17 18:17:05 +000040
41 def testNoArgFunctions(self):
42 # test posix functions which take no arguments and have
43 # no side-effects which we need to cleanup (e.g., fork, wait, abort)
Guido van Rossumf0af3e32008-10-02 18:55:37 +000044 NO_ARG_FUNCTIONS = [ "ctermid", "getcwd", "getcwdb", "uname",
Guido van Rossum687b9c02007-10-25 23:18:51 +000045 "times", "getloadavg",
Neal Norwitze241ce82003-02-17 18:17:05 +000046 "getegid", "geteuid", "getgid", "getgroups",
Ross Lagerwall7807c352011-03-17 20:20:30 +020047 "getpid", "getpgrp", "getppid", "getuid", "sync",
Neal Norwitze241ce82003-02-17 18:17:05 +000048 ]
Neal Norwitz71b13e82003-02-23 22:12:24 +000049
Neal Norwitze241ce82003-02-17 18:17:05 +000050 for name in NO_ARG_FUNCTIONS:
51 posix_func = getattr(posix, name, None)
52 if posix_func is not None:
53 posix_func()
Neal Norwitz2ff51a82003-02-17 22:40:31 +000054 self.assertRaises(TypeError, posix_func, 1)
Neal Norwitze241ce82003-02-17 18:17:05 +000055
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000056 if hasattr(posix, 'getresuid'):
57 def test_getresuid(self):
58 user_ids = posix.getresuid()
59 self.assertEqual(len(user_ids), 3)
60 for val in user_ids:
61 self.assertGreaterEqual(val, 0)
62
63 if hasattr(posix, 'getresgid'):
64 def test_getresgid(self):
65 group_ids = posix.getresgid()
66 self.assertEqual(len(group_ids), 3)
67 for val in group_ids:
68 self.assertGreaterEqual(val, 0)
69
70 if hasattr(posix, 'setresuid'):
71 def test_setresuid(self):
72 current_user_ids = posix.getresuid()
73 self.assertIsNone(posix.setresuid(*current_user_ids))
74 # -1 means don't change that value.
75 self.assertIsNone(posix.setresuid(-1, -1, -1))
76
77 def test_setresuid_exception(self):
78 # Don't do this test if someone is silly enough to run us as root.
79 current_user_ids = posix.getresuid()
80 if 0 not in current_user_ids:
81 new_user_ids = (current_user_ids[0]+1, -1, -1)
82 self.assertRaises(OSError, posix.setresuid, *new_user_ids)
83
84 if hasattr(posix, 'setresgid'):
85 def test_setresgid(self):
86 current_group_ids = posix.getresgid()
87 self.assertIsNone(posix.setresgid(*current_group_ids))
88 # -1 means don't change that value.
89 self.assertIsNone(posix.setresgid(-1, -1, -1))
90
91 def test_setresgid_exception(self):
92 # Don't do this test if someone is silly enough to run us as root.
93 current_group_ids = posix.getresgid()
94 if 0 not in current_group_ids:
95 new_group_ids = (current_group_ids[0]+1, -1, -1)
96 self.assertRaises(OSError, posix.setresgid, *new_group_ids)
97
Antoine Pitroub7572f02009-12-02 20:46:48 +000098 @unittest.skipUnless(hasattr(posix, 'initgroups'),
99 "test needs os.initgroups()")
100 def test_initgroups(self):
101 # It takes a string and an integer; check that it raises a TypeError
102 # for other argument lists.
103 self.assertRaises(TypeError, posix.initgroups)
104 self.assertRaises(TypeError, posix.initgroups, None)
105 self.assertRaises(TypeError, posix.initgroups, 3, "foo")
106 self.assertRaises(TypeError, posix.initgroups, "foo", 3, object())
107
108 # If a non-privileged user invokes it, it should fail with OSError
109 # EPERM.
110 if os.getuid() != 0:
Charles-François Natalie8a255a2012-05-02 20:01:38 +0200111 try:
112 name = pwd.getpwuid(posix.getuid()).pw_name
113 except KeyError:
114 # the current UID may not have a pwd entry
115 raise unittest.SkipTest("need a pwd entry")
Antoine Pitroub7572f02009-12-02 20:46:48 +0000116 try:
117 posix.initgroups(name, 13)
118 except OSError as e:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000119 self.assertEqual(e.errno, errno.EPERM)
Antoine Pitroub7572f02009-12-02 20:46:48 +0000120 else:
121 self.fail("Expected OSError to be raised by initgroups")
122
Neal Norwitze241ce82003-02-17 18:17:05 +0000123 def test_statvfs(self):
124 if hasattr(posix, 'statvfs'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000125 self.assertTrue(posix.statvfs(os.curdir))
Neal Norwitze241ce82003-02-17 18:17:05 +0000126
127 def test_fstatvfs(self):
128 if hasattr(posix, 'fstatvfs'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000129 fp = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000130 try:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000131 self.assertTrue(posix.fstatvfs(fp.fileno()))
Larry Hastings9cf065c2012-06-22 16:30:09 -0700132 self.assertTrue(posix.statvfs(fp.fileno()))
Neal Norwitze241ce82003-02-17 18:17:05 +0000133 finally:
134 fp.close()
135
136 def test_ftruncate(self):
137 if hasattr(posix, 'ftruncate'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000138 fp = open(support.TESTFN, 'w+')
Neal Norwitze241ce82003-02-17 18:17:05 +0000139 try:
140 # we need to have some data to truncate
141 fp.write('test')
142 fp.flush()
143 posix.ftruncate(fp.fileno(), 0)
144 finally:
145 fp.close()
146
Ross Lagerwall7807c352011-03-17 20:20:30 +0200147 @unittest.skipUnless(hasattr(posix, 'truncate'), "test needs posix.truncate()")
148 def test_truncate(self):
149 with open(support.TESTFN, 'w') as fp:
150 fp.write('test')
151 fp.flush()
152 posix.truncate(support.TESTFN, 0)
153
Larry Hastings9cf065c2012-06-22 16:30:09 -0700154 @unittest.skipUnless(getattr(os, 'execve', None) in os.supports_fd, "test needs execve() to support the fd parameter")
Ross Lagerwall7807c352011-03-17 20:20:30 +0200155 @unittest.skipUnless(hasattr(os, 'fork'), "test needs os.fork()")
Ross Lagerwalldedf6cf2011-03-20 18:27:05 +0200156 @unittest.skipUnless(hasattr(os, 'waitpid'), "test needs os.waitpid()")
Ross Lagerwall7807c352011-03-17 20:20:30 +0200157 def test_fexecve(self):
158 fp = os.open(sys.executable, os.O_RDONLY)
159 try:
160 pid = os.fork()
161 if pid == 0:
162 os.chdir(os.path.split(sys.executable)[0])
Larry Hastings9cf065c2012-06-22 16:30:09 -0700163 posix.execve(fp, [sys.executable, '-c', 'pass'], os.environ)
Ross Lagerwall7807c352011-03-17 20:20:30 +0200164 else:
Ross Lagerwalldedf6cf2011-03-20 18:27:05 +0200165 self.assertEqual(os.waitpid(pid, 0), (pid, 0))
Ross Lagerwall7807c352011-03-17 20:20:30 +0200166 finally:
167 os.close(fp)
168
169 @unittest.skipUnless(hasattr(posix, 'waitid'), "test needs posix.waitid()")
170 @unittest.skipUnless(hasattr(os, 'fork'), "test needs os.fork()")
171 def test_waitid(self):
172 pid = os.fork()
173 if pid == 0:
174 os.chdir(os.path.split(sys.executable)[0])
175 posix.execve(sys.executable, [sys.executable, '-c', 'pass'], os.environ)
176 else:
177 res = posix.waitid(posix.P_PID, pid, posix.WEXITED)
178 self.assertEqual(pid, res.si_pid)
179
180 @unittest.skipUnless(hasattr(posix, 'lockf'), "test needs posix.lockf()")
181 def test_lockf(self):
182 fd = os.open(support.TESTFN, os.O_WRONLY | os.O_CREAT)
183 try:
184 os.write(fd, b'test')
185 os.lseek(fd, 0, os.SEEK_SET)
186 posix.lockf(fd, posix.F_LOCK, 4)
187 # section is locked
188 posix.lockf(fd, posix.F_ULOCK, 4)
189 finally:
190 os.close(fd)
191
192 @unittest.skipUnless(hasattr(posix, 'pread'), "test needs posix.pread()")
193 def test_pread(self):
194 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
195 try:
196 os.write(fd, b'test')
197 os.lseek(fd, 0, os.SEEK_SET)
198 self.assertEqual(b'es', posix.pread(fd, 2, 1))
Florent Xiclunae41f0de2011-11-11 19:39:25 +0100199 # the first pread() shouldn't disturb the file offset
Ross Lagerwall7807c352011-03-17 20:20:30 +0200200 self.assertEqual(b'te', posix.read(fd, 2))
201 finally:
202 os.close(fd)
203
204 @unittest.skipUnless(hasattr(posix, 'pwrite'), "test needs posix.pwrite()")
205 def test_pwrite(self):
206 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
207 try:
208 os.write(fd, b'test')
209 os.lseek(fd, 0, os.SEEK_SET)
210 posix.pwrite(fd, b'xx', 1)
211 self.assertEqual(b'txxt', posix.read(fd, 4))
212 finally:
213 os.close(fd)
214
215 @unittest.skipUnless(hasattr(posix, 'posix_fallocate'),
216 "test needs posix.posix_fallocate()")
217 def test_posix_fallocate(self):
218 fd = os.open(support.TESTFN, os.O_WRONLY | os.O_CREAT)
219 try:
220 posix.posix_fallocate(fd, 0, 10)
221 except OSError as inst:
222 # issue10812, ZFS doesn't appear to support posix_fallocate,
223 # so skip Solaris-based since they are likely to have ZFS.
224 if inst.errno != errno.EINVAL or not sys.platform.startswith("sunos"):
225 raise
226 finally:
227 os.close(fd)
228
229 @unittest.skipUnless(hasattr(posix, 'posix_fadvise'),
230 "test needs posix.posix_fadvise()")
231 def test_posix_fadvise(self):
232 fd = os.open(support.TESTFN, os.O_RDONLY)
233 try:
234 posix.posix_fadvise(fd, 0, 0, posix.POSIX_FADV_WILLNEED)
235 finally:
236 os.close(fd)
237
Larry Hastings9cf065c2012-06-22 16:30:09 -0700238 @unittest.skipUnless(os.utime in os.supports_fd, "test needs fd support in os.utime")
239 def test_utime_with_fd(self):
Ross Lagerwall7807c352011-03-17 20:20:30 +0200240 now = time.time()
241 fd = os.open(support.TESTFN, os.O_RDONLY)
242 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700243 posix.utime(fd)
244 posix.utime(fd, None)
245 self.assertRaises(TypeError, posix.utime, fd, (None, None))
246 self.assertRaises(TypeError, posix.utime, fd, (now, None))
247 self.assertRaises(TypeError, posix.utime, fd, (None, now))
248 posix.utime(fd, (int(now), int(now)))
249 posix.utime(fd, (now, now))
250 self.assertRaises(ValueError, posix.utime, fd, (now, now), ns=(now, now))
251 self.assertRaises(ValueError, posix.utime, fd, (now, 0), ns=(None, None))
252 self.assertRaises(ValueError, posix.utime, fd, (None, None), ns=(now, 0))
253 posix.utime(fd, (int(now), int((now - int(now)) * 1e9)))
254 posix.utime(fd, ns=(int(now), int((now - int(now)) * 1e9)))
255
Ross Lagerwall7807c352011-03-17 20:20:30 +0200256 finally:
257 os.close(fd)
258
Larry Hastings9cf065c2012-06-22 16:30:09 -0700259 @unittest.skipUnless(os.utime in os.supports_follow_symlinks, "test needs follow_symlinks support in os.utime")
260 def test_utime_nofollow_symlinks(self):
Ross Lagerwall7807c352011-03-17 20:20:30 +0200261 now = time.time()
Larry Hastings9cf065c2012-06-22 16:30:09 -0700262 posix.utime(support.TESTFN, None, follow_symlinks=False)
263 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, None), follow_symlinks=False)
264 self.assertRaises(TypeError, posix.utime, support.TESTFN, (now, None), follow_symlinks=False)
265 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, now), follow_symlinks=False)
266 posix.utime(support.TESTFN, (int(now), int(now)), follow_symlinks=False)
267 posix.utime(support.TESTFN, (now, now), follow_symlinks=False)
268 posix.utime(support.TESTFN, follow_symlinks=False)
Ross Lagerwall7807c352011-03-17 20:20:30 +0200269
270 @unittest.skipUnless(hasattr(posix, 'writev'), "test needs posix.writev()")
271 def test_writev(self):
272 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
273 try:
274 os.writev(fd, (b'test1', b'tt2', b't3'))
275 os.lseek(fd, 0, os.SEEK_SET)
276 self.assertEqual(b'test1tt2t3', posix.read(fd, 10))
277 finally:
278 os.close(fd)
279
280 @unittest.skipUnless(hasattr(posix, 'readv'), "test needs posix.readv()")
281 def test_readv(self):
282 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
283 try:
284 os.write(fd, b'test1tt2t3')
285 os.lseek(fd, 0, os.SEEK_SET)
286 buf = [bytearray(i) for i in [5, 3, 2]]
287 self.assertEqual(posix.readv(fd, buf), 10)
288 self.assertEqual([b'test1', b'tt2', b't3'], [bytes(i) for i in buf])
289 finally:
290 os.close(fd)
291
Neal Norwitze241ce82003-02-17 18:17:05 +0000292 def test_dup(self):
293 if hasattr(posix, 'dup'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000294 fp = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000295 try:
296 fd = posix.dup(fp.fileno())
Ezio Melottie9615932010-01-24 19:26:24 +0000297 self.assertIsInstance(fd, int)
Neal Norwitze241ce82003-02-17 18:17:05 +0000298 os.close(fd)
299 finally:
300 fp.close()
301
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000302 def test_confstr(self):
303 if hasattr(posix, 'confstr'):
304 self.assertRaises(ValueError, posix.confstr, "CS_garbage")
305 self.assertEqual(len(posix.confstr("CS_PATH")) > 0, True)
306
Neal Norwitze241ce82003-02-17 18:17:05 +0000307 def test_dup2(self):
308 if hasattr(posix, 'dup2'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000309 fp1 = open(support.TESTFN)
310 fp2 = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000311 try:
312 posix.dup2(fp1.fileno(), fp2.fileno())
313 finally:
314 fp1.close()
315 fp2.close()
316
Charles-François Natali1e045b12011-05-22 20:42:32 +0200317 @unittest.skipUnless(hasattr(os, 'O_CLOEXEC'), "needs os.O_CLOEXEC")
Charles-François Natali239bb962011-06-03 12:55:15 +0200318 @support.requires_linux_version(2, 6, 23)
Charles-François Natali1e045b12011-05-22 20:42:32 +0200319 def test_oscloexec(self):
320 fd = os.open(support.TESTFN, os.O_RDONLY|os.O_CLOEXEC)
321 self.addCleanup(os.close, fd)
Victor Stinnere36f3752011-05-24 00:29:43 +0200322 self.assertTrue(fcntl.fcntl(fd, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
Charles-François Natali1e045b12011-05-22 20:42:32 +0200323
Skip Montanaro98470002005-06-17 01:14:49 +0000324 def test_osexlock(self):
325 if hasattr(posix, "O_EXLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000326 fd = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000327 os.O_WRONLY|os.O_EXLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000328 self.assertRaises(OSError, os.open, support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000329 os.O_WRONLY|os.O_EXLOCK|os.O_NONBLOCK)
330 os.close(fd)
331
332 if hasattr(posix, "O_SHLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000333 fd = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000334 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000335 self.assertRaises(OSError, os.open, support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000336 os.O_WRONLY|os.O_EXLOCK|os.O_NONBLOCK)
337 os.close(fd)
338
339 def test_osshlock(self):
340 if hasattr(posix, "O_SHLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000341 fd1 = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000342 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000343 fd2 = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000344 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
345 os.close(fd2)
346 os.close(fd1)
347
348 if hasattr(posix, "O_EXLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000349 fd = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000350 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000351 self.assertRaises(OSError, os.open, support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000352 os.O_RDONLY|os.O_EXLOCK|os.O_NONBLOCK)
353 os.close(fd)
354
Neal Norwitze241ce82003-02-17 18:17:05 +0000355 def test_fstat(self):
356 if hasattr(posix, 'fstat'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000357 fp = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000358 try:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000359 self.assertTrue(posix.fstat(fp.fileno()))
Larry Hastings9cf065c2012-06-22 16:30:09 -0700360 self.assertTrue(posix.stat(fp.fileno()))
Neal Norwitze241ce82003-02-17 18:17:05 +0000361 finally:
362 fp.close()
363
364 def test_stat(self):
365 if hasattr(posix, 'stat'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000366 self.assertTrue(posix.stat(support.TESTFN))
Neal Norwitze241ce82003-02-17 18:17:05 +0000367
Benjamin Peterson052a02b2010-08-17 01:27:09 +0000368 @unittest.skipUnless(hasattr(posix, 'mkfifo'), "don't have mkfifo()")
369 def test_mkfifo(self):
370 support.unlink(support.TESTFN)
371 posix.mkfifo(support.TESTFN, stat.S_IRUSR | stat.S_IWUSR)
372 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
373
374 @unittest.skipUnless(hasattr(posix, 'mknod') and hasattr(stat, 'S_IFIFO'),
375 "don't have mknod()/S_IFIFO")
376 def test_mknod(self):
377 # Test using mknod() to create a FIFO (the only use specified
378 # by POSIX).
379 support.unlink(support.TESTFN)
380 mode = stat.S_IFIFO | stat.S_IRUSR | stat.S_IWUSR
381 try:
382 posix.mknod(support.TESTFN, mode, 0)
383 except OSError as e:
384 # Some old systems don't allow unprivileged users to use
385 # mknod(), or only support creating device nodes.
386 self.assertIn(e.errno, (errno.EPERM, errno.EINVAL))
387 else:
388 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
389
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000390 def _test_all_chown_common(self, chown_func, first_param):
391 """Common code for chown, fchown and lchown tests."""
Charles-François Nataliab2d58e2012-04-17 19:48:35 +0200392 # test a successful chown call
393 chown_func(first_param, os.getuid(), os.getgid())
394
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000395 if os.getuid() == 0:
396 try:
397 # Many linux distros have a nfsnobody user as MAX_UID-2
398 # that makes a good test case for signedness issues.
399 # http://bugs.python.org/issue1747858
400 # This part of the test only runs when run as root.
401 # Only scary people run their tests as root.
402 ent = pwd.getpwnam('nfsnobody')
403 chown_func(first_param, ent.pw_uid, ent.pw_gid)
404 except KeyError:
405 pass
Charles-François Nataliab2d58e2012-04-17 19:48:35 +0200406 elif platform.system() in ('HP-UX', 'SunOS'):
407 # HP-UX and Solaris can allow a non-root user to chown() to root
408 # (issue #5113)
409 raise unittest.SkipTest("Skipping because of non-standard chown() "
410 "behavior")
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000411 else:
412 # non-root cannot chown to root, raises OSError
413 self.assertRaises(OSError, chown_func,
414 first_param, 0, 0)
Christian Heimesd5e2b6f2008-03-19 21:50:51 +0000415
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000416 @unittest.skipUnless(hasattr(posix, 'chown'), "test needs os.chown()")
417 def test_chown(self):
418 # raise an OSError if the file does not exist
419 os.unlink(support.TESTFN)
420 self.assertRaises(OSError, posix.chown, support.TESTFN, -1, -1)
Christian Heimesd5e2b6f2008-03-19 21:50:51 +0000421
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000422 # re-create the file
Victor Stinnerbf816222011-06-30 23:25:47 +0200423 support.create_empty_file(support.TESTFN)
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000424 self._test_all_chown_common(posix.chown, support.TESTFN)
425
426 @unittest.skipUnless(hasattr(posix, 'fchown'), "test needs os.fchown()")
427 def test_fchown(self):
428 os.unlink(support.TESTFN)
429
430 # re-create the file
431 test_file = open(support.TESTFN, 'w')
432 try:
433 fd = test_file.fileno()
434 self._test_all_chown_common(posix.fchown, fd)
435 finally:
436 test_file.close()
437
438 @unittest.skipUnless(hasattr(posix, 'lchown'), "test needs os.lchown()")
439 def test_lchown(self):
440 os.unlink(support.TESTFN)
441 # create a symlink
Ned Deily3eb67d52011-06-28 00:00:28 -0700442 os.symlink(_DUMMY_SYMLINK, support.TESTFN)
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000443 self._test_all_chown_common(posix.lchown, support.TESTFN)
Christian Heimesd5e2b6f2008-03-19 21:50:51 +0000444
Neal Norwitze241ce82003-02-17 18:17:05 +0000445 def test_chdir(self):
446 if hasattr(posix, 'chdir'):
447 posix.chdir(os.curdir)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000448 self.assertRaises(OSError, posix.chdir, support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000449
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +0000450 def test_listdir(self):
Larry Hastingsfdaea062012-06-25 04:42:23 -0700451 self.assertTrue(support.TESTFN in posix.listdir(os.curdir))
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +0000452
453 def test_listdir_default(self):
Larry Hastingsfdaea062012-06-25 04:42:23 -0700454 # When listdir is called without argument,
455 # it's the same as listdir(os.curdir).
456 self.assertTrue(support.TESTFN in posix.listdir())
Neal Norwitze241ce82003-02-17 18:17:05 +0000457
Larry Hastingsfdaea062012-06-25 04:42:23 -0700458 def test_listdir_bytes(self):
459 # When listdir is called with a bytes object,
460 # the returned strings are of type bytes.
461 self.assertTrue(os.fsencode(support.TESTFN) in posix.listdir(b'.'))
462
463 @unittest.skipUnless(posix.listdir in os.supports_fd,
464 "test needs fd support for posix.listdir()")
465 def test_listdir_fd(self):
Antoine Pitrou8250e232011-02-25 23:41:16 +0000466 f = posix.open(posix.getcwd(), posix.O_RDONLY)
Charles-François Natali7546ad32012-01-08 18:34:06 +0100467 self.addCleanup(posix.close, f)
Antoine Pitrou8250e232011-02-25 23:41:16 +0000468 self.assertEqual(
469 sorted(posix.listdir('.')),
Larry Hastings9cf065c2012-06-22 16:30:09 -0700470 sorted(posix.listdir(f))
Antoine Pitrou8250e232011-02-25 23:41:16 +0000471 )
Charles-François Natali7546ad32012-01-08 18:34:06 +0100472 # Check that the fd offset was reset (issue #13739)
Charles-François Natali7546ad32012-01-08 18:34:06 +0100473 self.assertEqual(
474 sorted(posix.listdir('.')),
Larry Hastings9cf065c2012-06-22 16:30:09 -0700475 sorted(posix.listdir(f))
Charles-François Natali7546ad32012-01-08 18:34:06 +0100476 )
Antoine Pitrou8250e232011-02-25 23:41:16 +0000477
Neal Norwitze241ce82003-02-17 18:17:05 +0000478 def test_access(self):
479 if hasattr(posix, 'access'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000480 self.assertTrue(posix.access(support.TESTFN, os.R_OK))
Neal Norwitze241ce82003-02-17 18:17:05 +0000481
482 def test_umask(self):
483 if hasattr(posix, 'umask'):
484 old_mask = posix.umask(0)
Ezio Melottie9615932010-01-24 19:26:24 +0000485 self.assertIsInstance(old_mask, int)
Neal Norwitze241ce82003-02-17 18:17:05 +0000486 posix.umask(old_mask)
487
488 def test_strerror(self):
489 if hasattr(posix, 'strerror'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000490 self.assertTrue(posix.strerror(0))
Neal Norwitze241ce82003-02-17 18:17:05 +0000491
492 def test_pipe(self):
493 if hasattr(posix, 'pipe'):
494 reader, writer = posix.pipe()
495 os.close(reader)
496 os.close(writer)
497
Charles-François Natalidaafdd52011-05-29 20:07:40 +0200498 @unittest.skipUnless(hasattr(os, 'pipe2'), "test needs os.pipe2()")
Charles-François Natali239bb962011-06-03 12:55:15 +0200499 @support.requires_linux_version(2, 6, 27)
Charles-François Natalidaafdd52011-05-29 20:07:40 +0200500 def test_pipe2(self):
501 self.assertRaises(TypeError, os.pipe2, 'DEADBEEF')
502 self.assertRaises(TypeError, os.pipe2, 0, 0)
503
Charles-François Natali368f34b2011-06-06 19:49:47 +0200504 # try calling with flags = 0, like os.pipe()
505 r, w = os.pipe2(0)
Charles-François Natalidaafdd52011-05-29 20:07:40 +0200506 os.close(r)
507 os.close(w)
508
509 # test flags
510 r, w = os.pipe2(os.O_CLOEXEC|os.O_NONBLOCK)
511 self.addCleanup(os.close, r)
512 self.addCleanup(os.close, w)
513 self.assertTrue(fcntl.fcntl(r, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
514 self.assertTrue(fcntl.fcntl(w, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
515 # try reading from an empty pipe: this should fail, not block
516 self.assertRaises(OSError, os.read, r, 1)
517 # try a write big enough to fill-up the pipe: this should either
518 # fail or perform a partial write, not block
519 try:
520 os.write(w, b'x' * support.PIPE_MAX_SIZE)
521 except OSError:
522 pass
523
Neal Norwitze241ce82003-02-17 18:17:05 +0000524 def test_utime(self):
525 if hasattr(posix, 'utime'):
526 now = time.time()
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000527 posix.utime(support.TESTFN, None)
528 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, None))
529 self.assertRaises(TypeError, posix.utime, support.TESTFN, (now, None))
530 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, now))
531 posix.utime(support.TESTFN, (int(now), int(now)))
532 posix.utime(support.TESTFN, (now, now))
Neal Norwitze241ce82003-02-17 18:17:05 +0000533
Larry Hastings9cf065c2012-06-22 16:30:09 -0700534 def _test_chflags_regular_file(self, chflags_func, target_file, **kwargs):
Ned Deily3eb67d52011-06-28 00:00:28 -0700535 st = os.stat(target_file)
536 self.assertTrue(hasattr(st, 'st_flags'))
Trent Nelson75959cf2012-08-21 23:59:31 +0000537
538 # ZFS returns EOPNOTSUPP when attempting to set flag UF_IMMUTABLE.
539 flags = st.st_flags | stat.UF_IMMUTABLE
540 try:
541 chflags_func(target_file, flags, **kwargs)
542 except OSError as err:
543 if err.errno != errno.EOPNOTSUPP:
544 raise
545 msg = 'chflag UF_IMMUTABLE not supported by underlying fs'
546 self.skipTest(msg)
547
Ned Deily3eb67d52011-06-28 00:00:28 -0700548 try:
549 new_st = os.stat(target_file)
550 self.assertEqual(st.st_flags | stat.UF_IMMUTABLE, new_st.st_flags)
551 try:
552 fd = open(target_file, 'w+')
553 except IOError as e:
554 self.assertEqual(e.errno, errno.EPERM)
555 finally:
556 posix.chflags(target_file, st.st_flags)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000557
Ned Deily3eb67d52011-06-28 00:00:28 -0700558 @unittest.skipUnless(hasattr(posix, 'chflags'), 'test needs os.chflags()')
559 def test_chflags(self):
560 self._test_chflags_regular_file(posix.chflags, support.TESTFN)
561
562 @unittest.skipUnless(hasattr(posix, 'lchflags'), 'test needs os.lchflags()')
563 def test_lchflags_regular_file(self):
564 self._test_chflags_regular_file(posix.lchflags, support.TESTFN)
Larry Hastings9cf065c2012-06-22 16:30:09 -0700565 self._test_chflags_regular_file(posix.chflags, support.TESTFN, follow_symlinks=False)
Ned Deily3eb67d52011-06-28 00:00:28 -0700566
567 @unittest.skipUnless(hasattr(posix, 'lchflags'), 'test needs os.lchflags()')
568 def test_lchflags_symlink(self):
569 testfn_st = os.stat(support.TESTFN)
570
571 self.assertTrue(hasattr(testfn_st, 'st_flags'))
572
573 os.symlink(support.TESTFN, _DUMMY_SYMLINK)
574 self.teardown_files.append(_DUMMY_SYMLINK)
575 dummy_symlink_st = os.lstat(_DUMMY_SYMLINK)
576
Larry Hastings9cf065c2012-06-22 16:30:09 -0700577 def chflags_nofollow(path, flags):
578 return posix.chflags(path, flags, follow_symlinks=False)
Ned Deily3eb67d52011-06-28 00:00:28 -0700579
Larry Hastings9cf065c2012-06-22 16:30:09 -0700580 for fn in (posix.lchflags, chflags_nofollow):
Trent Nelson75959cf2012-08-21 23:59:31 +0000581 # ZFS returns EOPNOTSUPP when attempting to set flag UF_IMMUTABLE.
582 flags = dummy_symlink_st.st_flags | stat.UF_IMMUTABLE
583 try:
584 fn(_DUMMY_SYMLINK, flags)
585 except OSError as err:
586 if err.errno != errno.EOPNOTSUPP:
587 raise
588 msg = 'chflag UF_IMMUTABLE not supported by underlying fs'
589 self.skipTest(msg)
Larry Hastings9cf065c2012-06-22 16:30:09 -0700590 try:
591 new_testfn_st = os.stat(support.TESTFN)
592 new_dummy_symlink_st = os.lstat(_DUMMY_SYMLINK)
593
594 self.assertEqual(testfn_st.st_flags, new_testfn_st.st_flags)
595 self.assertEqual(dummy_symlink_st.st_flags | stat.UF_IMMUTABLE,
596 new_dummy_symlink_st.st_flags)
597 finally:
598 fn(_DUMMY_SYMLINK, dummy_symlink_st.st_flags)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000599
Guido van Rossum98297ee2007-11-06 21:34:58 +0000600 def test_environ(self):
Victor Stinner17b490d2010-05-06 22:19:30 +0000601 if os.name == "nt":
602 item_type = str
603 else:
604 item_type = bytes
Guido van Rossum98297ee2007-11-06 21:34:58 +0000605 for k, v in posix.environ.items():
Victor Stinner17b490d2010-05-06 22:19:30 +0000606 self.assertEqual(type(k), item_type)
607 self.assertEqual(type(v), item_type)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000608
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000609 def test_getcwd_long_pathnames(self):
610 if hasattr(posix, 'getcwd'):
611 dirname = 'getcwd-test-directory-0123456789abcdef-01234567890abcdef'
612 curdir = os.getcwd()
613 base_path = os.path.abspath(support.TESTFN) + '.getcwd'
614
615 try:
616 os.mkdir(base_path)
617 os.chdir(base_path)
618 except:
Benjamin Petersone549ead2009-03-28 21:42:05 +0000619# Just returning nothing instead of the SkipTest exception,
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000620# because the test results in Error in that case.
621# Is that ok?
Benjamin Petersone549ead2009-03-28 21:42:05 +0000622# raise unittest.SkipTest("cannot create directory for testing")
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000623 return
624
625 def _create_and_do_getcwd(dirname, current_path_length = 0):
626 try:
627 os.mkdir(dirname)
628 except:
Benjamin Petersone549ead2009-03-28 21:42:05 +0000629 raise unittest.SkipTest("mkdir cannot create directory sufficiently deep for getcwd test")
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000630
631 os.chdir(dirname)
632 try:
633 os.getcwd()
634 if current_path_length < 1027:
635 _create_and_do_getcwd(dirname, current_path_length + len(dirname) + 1)
636 finally:
637 os.chdir('..')
638 os.rmdir(dirname)
639
640 _create_and_do_getcwd(dirname)
641
642 finally:
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000643 os.chdir(curdir)
R. David Murray414c91f2009-07-09 20:12:31 +0000644 support.rmtree(base_path)
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000645
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +0200646 @unittest.skipUnless(hasattr(posix, 'getgrouplist'), "test needs posix.getgrouplist()")
647 @unittest.skipUnless(hasattr(pwd, 'getpwuid'), "test needs pwd.getpwuid()")
648 @unittest.skipUnless(hasattr(os, 'getuid'), "test needs os.getuid()")
649 def test_getgrouplist(self):
Ross Lagerwalla0b315f2012-12-13 15:20:26 +0000650 user = pwd.getpwuid(os.getuid())[0]
651 group = pwd.getpwuid(os.getuid())[3]
652 self.assertIn(group, posix.getgrouplist(user, group))
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +0200653
Ross Lagerwallb0ae53d2011-06-10 07:30:30 +0200654
Antoine Pitrou318b8f32011-01-12 18:45:27 +0000655 @unittest.skipUnless(hasattr(os, 'getegid'), "test needs os.getegid()")
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000656 def test_getgroups(self):
657 with os.popen('id -G') as idg:
658 groups = idg.read().strip()
Charles-François Natalie8a255a2012-05-02 20:01:38 +0200659 ret = idg.close()
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000660
Benjamin Petersonb29614e2012-10-09 11:16:03 -0400661 if ret is not None or not groups:
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000662 raise unittest.SkipTest("need working 'id -G'")
663
Ronald Oussoren7fb6f512010-08-01 19:18:13 +0000664 # 'id -G' and 'os.getgroups()' should return the same
665 # groups, ignoring order and duplicates.
Antoine Pitrou318b8f32011-01-12 18:45:27 +0000666 # #10822 - it is implementation defined whether posix.getgroups()
667 # includes the effective gid so we include it anyway, since id -G does
Ronald Oussorencb615e62010-07-24 14:15:19 +0000668 self.assertEqual(
Ronald Oussoren7fb6f512010-08-01 19:18:13 +0000669 set([int(x) for x in groups.split()]),
Antoine Pitrou318b8f32011-01-12 18:45:27 +0000670 set(posix.getgroups() + [posix.getegid()]))
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000671
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000672 # tests for the posix *at functions follow
673
Larry Hastings9cf065c2012-06-22 16:30:09 -0700674 @unittest.skipUnless(os.access in os.supports_dir_fd, "test needs dir_fd support for os.access()")
675 def test_access_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000676 f = posix.open(posix.getcwd(), posix.O_RDONLY)
677 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700678 self.assertTrue(posix.access(support.TESTFN, os.R_OK, dir_fd=f))
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000679 finally:
680 posix.close(f)
681
Larry Hastings9cf065c2012-06-22 16:30:09 -0700682 @unittest.skipUnless(os.chmod in os.supports_dir_fd, "test needs dir_fd support in os.chmod()")
683 def test_chmod_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000684 os.chmod(support.TESTFN, stat.S_IRUSR)
685
686 f = posix.open(posix.getcwd(), posix.O_RDONLY)
687 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700688 posix.chmod(support.TESTFN, stat.S_IRUSR | stat.S_IWUSR, dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000689
690 s = posix.stat(support.TESTFN)
691 self.assertEqual(s[0] & stat.S_IRWXU, stat.S_IRUSR | stat.S_IWUSR)
692 finally:
693 posix.close(f)
694
Larry Hastings9cf065c2012-06-22 16:30:09 -0700695 @unittest.skipUnless(os.chown in os.supports_dir_fd, "test needs dir_fd support in os.chown()")
696 def test_chown_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000697 support.unlink(support.TESTFN)
Victor Stinnerbf816222011-06-30 23:25:47 +0200698 support.create_empty_file(support.TESTFN)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000699
700 f = posix.open(posix.getcwd(), posix.O_RDONLY)
701 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700702 posix.chown(support.TESTFN, os.getuid(), os.getgid(), dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000703 finally:
704 posix.close(f)
705
Larry Hastings9cf065c2012-06-22 16:30:09 -0700706 @unittest.skipUnless(os.stat in os.supports_dir_fd, "test needs dir_fd support in os.stat()")
707 def test_stat_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000708 support.unlink(support.TESTFN)
709 with open(support.TESTFN, 'w') as outfile:
710 outfile.write("testline\n")
711
712 f = posix.open(posix.getcwd(), posix.O_RDONLY)
713 try:
714 s1 = posix.stat(support.TESTFN)
Larry Hastings9cf065c2012-06-22 16:30:09 -0700715 s2 = posix.stat(support.TESTFN, dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000716 self.assertEqual(s1, s2)
717 finally:
718 posix.close(f)
719
Larry Hastings9cf065c2012-06-22 16:30:09 -0700720 @unittest.skipUnless(os.utime in os.supports_dir_fd, "test needs dir_fd support in os.utime()")
721 def test_utime_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000722 f = posix.open(posix.getcwd(), posix.O_RDONLY)
723 try:
724 now = time.time()
Larry Hastings9cf065c2012-06-22 16:30:09 -0700725 posix.utime(support.TESTFN, None, dir_fd=f)
726 posix.utime(support.TESTFN, dir_fd=f)
727 self.assertRaises(TypeError, posix.utime, support.TESTFN, now, dir_fd=f)
728 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, None), dir_fd=f)
729 self.assertRaises(TypeError, posix.utime, support.TESTFN, (now, None), dir_fd=f)
730 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, now), dir_fd=f)
731 self.assertRaises(TypeError, posix.utime, support.TESTFN, (now, "x"), dir_fd=f)
732 posix.utime(support.TESTFN, (int(now), int(now)), dir_fd=f)
733 posix.utime(support.TESTFN, (now, now), dir_fd=f)
734 posix.utime(support.TESTFN,
735 (int(now), int((now - int(now)) * 1e9)), dir_fd=f)
736 posix.utime(support.TESTFN, dir_fd=f,
737 times=(int(now), int((now - int(now)) * 1e9)))
738
Larry Hastings90867a52012-06-22 17:01:41 -0700739 # try dir_fd and follow_symlinks together
Larry Hastings9cf065c2012-06-22 16:30:09 -0700740 if os.utime in os.supports_follow_symlinks:
Larry Hastings90867a52012-06-22 17:01:41 -0700741 try:
742 posix.utime(support.TESTFN, follow_symlinks=False, dir_fd=f)
Georg Brandl969288e2012-06-26 09:25:44 +0200743 except ValueError:
Larry Hastings90867a52012-06-22 17:01:41 -0700744 # whoops! using both together not supported on this platform.
745 pass
Larry Hastings9cf065c2012-06-22 16:30:09 -0700746
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000747 finally:
748 posix.close(f)
749
Larry Hastings9cf065c2012-06-22 16:30:09 -0700750 @unittest.skipUnless(os.link in os.supports_dir_fd, "test needs dir_fd support in os.link()")
751 def test_link_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000752 f = posix.open(posix.getcwd(), posix.O_RDONLY)
753 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700754 posix.link(support.TESTFN, support.TESTFN + 'link', src_dir_fd=f, dst_dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000755 # should have same inodes
756 self.assertEqual(posix.stat(support.TESTFN)[1],
757 posix.stat(support.TESTFN + 'link')[1])
758 finally:
759 posix.close(f)
760 support.unlink(support.TESTFN + 'link')
761
Larry Hastings9cf065c2012-06-22 16:30:09 -0700762 @unittest.skipUnless(os.mkdir in os.supports_dir_fd, "test needs dir_fd support in os.mkdir()")
763 def test_mkdir_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000764 f = posix.open(posix.getcwd(), posix.O_RDONLY)
765 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700766 posix.mkdir(support.TESTFN + 'dir', dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000767 posix.stat(support.TESTFN + 'dir') # should not raise exception
768 finally:
769 posix.close(f)
770 support.rmtree(support.TESTFN + 'dir')
771
Larry Hastings9cf065c2012-06-22 16:30:09 -0700772 @unittest.skipUnless((os.mknod in os.supports_dir_fd) and hasattr(stat, 'S_IFIFO'),
773 "test requires both stat.S_IFIFO and dir_fd support for os.mknod()")
774 def test_mknod_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000775 # Test using mknodat() to create a FIFO (the only use specified
776 # by POSIX).
777 support.unlink(support.TESTFN)
778 mode = stat.S_IFIFO | stat.S_IRUSR | stat.S_IWUSR
779 f = posix.open(posix.getcwd(), posix.O_RDONLY)
780 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700781 posix.mknod(support.TESTFN, mode, 0, dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000782 except OSError as e:
783 # Some old systems don't allow unprivileged users to use
784 # mknod(), or only support creating device nodes.
785 self.assertIn(e.errno, (errno.EPERM, errno.EINVAL))
786 else:
787 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
788 finally:
789 posix.close(f)
790
Larry Hastings9cf065c2012-06-22 16:30:09 -0700791 @unittest.skipUnless(os.open in os.supports_dir_fd, "test needs dir_fd support in os.open()")
792 def test_open_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000793 support.unlink(support.TESTFN)
794 with open(support.TESTFN, 'w') as outfile:
795 outfile.write("testline\n")
796 a = posix.open(posix.getcwd(), posix.O_RDONLY)
Larry Hastings9cf065c2012-06-22 16:30:09 -0700797 b = posix.open(support.TESTFN, posix.O_RDONLY, dir_fd=a)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000798 try:
799 res = posix.read(b, 9).decode(encoding="utf-8")
800 self.assertEqual("testline\n", res)
801 finally:
802 posix.close(a)
803 posix.close(b)
804
Larry Hastings9cf065c2012-06-22 16:30:09 -0700805 @unittest.skipUnless(os.readlink in os.supports_dir_fd, "test needs dir_fd support in os.readlink()")
806 def test_readlink_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000807 os.symlink(support.TESTFN, support.TESTFN + 'link')
808 f = posix.open(posix.getcwd(), posix.O_RDONLY)
809 try:
810 self.assertEqual(posix.readlink(support.TESTFN + 'link'),
Larry Hastings9cf065c2012-06-22 16:30:09 -0700811 posix.readlink(support.TESTFN + 'link', dir_fd=f))
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000812 finally:
813 support.unlink(support.TESTFN + 'link')
814 posix.close(f)
815
Larry Hastings9cf065c2012-06-22 16:30:09 -0700816 @unittest.skipUnless(os.rename in os.supports_dir_fd, "test needs dir_fd support in os.rename()")
817 def test_rename_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000818 support.unlink(support.TESTFN)
Victor Stinnerbf816222011-06-30 23:25:47 +0200819 support.create_empty_file(support.TESTFN + 'ren')
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000820 f = posix.open(posix.getcwd(), posix.O_RDONLY)
821 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700822 posix.rename(support.TESTFN + 'ren', support.TESTFN, src_dir_fd=f, dst_dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000823 except:
824 posix.rename(support.TESTFN + 'ren', support.TESTFN)
825 raise
826 else:
827 posix.stat(support.TESTFN) # should not throw exception
828 finally:
829 posix.close(f)
830
Larry Hastings9cf065c2012-06-22 16:30:09 -0700831 @unittest.skipUnless(os.symlink in os.supports_dir_fd, "test needs dir_fd support in os.symlink()")
832 def test_symlink_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000833 f = posix.open(posix.getcwd(), posix.O_RDONLY)
834 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700835 posix.symlink(support.TESTFN, support.TESTFN + 'link', dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000836 self.assertEqual(posix.readlink(support.TESTFN + 'link'), support.TESTFN)
837 finally:
838 posix.close(f)
839 support.unlink(support.TESTFN + 'link')
840
Larry Hastings9cf065c2012-06-22 16:30:09 -0700841 @unittest.skipUnless(os.unlink in os.supports_dir_fd, "test needs dir_fd support in os.unlink()")
842 def test_unlink_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000843 f = posix.open(posix.getcwd(), posix.O_RDONLY)
Victor Stinnerbf816222011-06-30 23:25:47 +0200844 support.create_empty_file(support.TESTFN + 'del')
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000845 posix.stat(support.TESTFN + 'del') # should not throw exception
846 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700847 posix.unlink(support.TESTFN + 'del', dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000848 except:
849 support.unlink(support.TESTFN + 'del')
850 raise
851 else:
852 self.assertRaises(OSError, posix.stat, support.TESTFN + 'link')
853 finally:
854 posix.close(f)
855
Larry Hastings9cf065c2012-06-22 16:30:09 -0700856 @unittest.skipUnless(os.mkfifo in os.supports_dir_fd, "test needs dir_fd support in os.mkfifo()")
857 def test_mkfifo_dir_fd(self):
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000858 support.unlink(support.TESTFN)
859 f = posix.open(posix.getcwd(), posix.O_RDONLY)
860 try:
Larry Hastings9cf065c2012-06-22 16:30:09 -0700861 posix.mkfifo(support.TESTFN, stat.S_IRUSR | stat.S_IWUSR, dir_fd=f)
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000862 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
863 finally:
864 posix.close(f)
865
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500866 requires_sched_h = unittest.skipUnless(hasattr(posix, 'sched_yield'),
867 "don't have scheduling support")
Antoine Pitrou84869872012-08-04 16:16:35 +0200868 requires_sched_affinity = unittest.skipUnless(hasattr(posix, 'sched_setaffinity'),
Benjamin Peterson50ba2712011-08-02 22:15:40 -0500869 "don't have sched affinity support")
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500870
871 @requires_sched_h
872 def test_sched_yield(self):
873 # This has no error conditions (at least on Linux).
874 posix.sched_yield()
875
876 @requires_sched_h
Charles-François Nataliea0d5fc2011-09-06 19:03:35 +0200877 @unittest.skipUnless(hasattr(posix, 'sched_get_priority_max'),
878 "requires sched_get_priority_max()")
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500879 def test_sched_priority(self):
880 # Round-robin usually has interesting priorities.
881 pol = posix.SCHED_RR
882 lo = posix.sched_get_priority_min(pol)
883 hi = posix.sched_get_priority_max(pol)
884 self.assertIsInstance(lo, int)
885 self.assertIsInstance(hi, int)
886 self.assertGreaterEqual(hi, lo)
Benjamin Peterson539b6c42011-08-02 22:09:37 -0500887 # OSX evidently just returns 15 without checking the argument.
888 if sys.platform != "darwin":
Benjamin Petersonc1581582011-08-02 22:10:55 -0500889 self.assertRaises(OSError, posix.sched_get_priority_min, -23)
890 self.assertRaises(OSError, posix.sched_get_priority_max, -23)
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500891
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -0500892 @unittest.skipUnless(hasattr(posix, 'sched_setscheduler'), "can't change scheduler")
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500893 def test_get_and_set_scheduler_and_param(self):
894 possible_schedulers = [sched for name, sched in posix.__dict__.items()
895 if name.startswith("SCHED_")]
896 mine = posix.sched_getscheduler(0)
897 self.assertIn(mine, possible_schedulers)
898 try:
Jesus Ceaceb5d162011-09-10 01:16:55 +0200899 parent = posix.sched_getscheduler(os.getppid())
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500900 except OSError as e:
Jesus Ceaceb5d162011-09-10 01:16:55 +0200901 if e.errno != errno.EPERM:
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500902 raise
903 else:
Jesus Ceaceb5d162011-09-10 01:16:55 +0200904 self.assertIn(parent, possible_schedulers)
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500905 self.assertRaises(OSError, posix.sched_getscheduler, -1)
906 self.assertRaises(OSError, posix.sched_getparam, -1)
907 param = posix.sched_getparam(0)
908 self.assertIsInstance(param.sched_priority, int)
Benjamin Peterson18592ca2011-08-02 18:48:59 -0500909 try:
910 posix.sched_setscheduler(0, mine, param)
911 except OSError as e:
912 if e.errno != errno.EPERM:
913 raise
Charles-François Natali7b911cb2011-08-21 12:41:43 +0200914
915 # POSIX states that calling sched_setparam() on a process with a
916 # scheduling policy other than SCHED_FIFO or SCHED_RR is
917 # implementation-defined: FreeBSD returns EINVAL.
918 if not sys.platform.startswith('freebsd'):
919 posix.sched_setparam(0, param)
920 self.assertRaises(OSError, posix.sched_setparam, -1, param)
921
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500922 self.assertRaises(OSError, posix.sched_setscheduler, -1, mine, param)
923 self.assertRaises(TypeError, posix.sched_setscheduler, 0, mine, None)
924 self.assertRaises(TypeError, posix.sched_setparam, 0, 43)
925 param = posix.sched_param(None)
926 self.assertRaises(TypeError, posix.sched_setparam, 0, param)
927 large = 214748364700
928 param = posix.sched_param(large)
929 self.assertRaises(OverflowError, posix.sched_setparam, 0, param)
930 param = posix.sched_param(sched_priority=-large)
931 self.assertRaises(OverflowError, posix.sched_setparam, 0, param)
932
Benjamin Petersonc5fce4d2011-08-02 18:07:32 -0500933 @unittest.skipUnless(hasattr(posix, "sched_rr_get_interval"), "no function")
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500934 def test_sched_rr_get_interval(self):
Benjamin Peterson43234ab2011-08-02 22:19:14 -0500935 try:
936 interval = posix.sched_rr_get_interval(0)
937 except OSError as e:
938 # This likely means that sched_rr_get_interval is only valid for
939 # processes with the SCHED_RR scheduler in effect.
940 if e.errno != errno.EINVAL:
941 raise
942 self.skipTest("only works on SCHED_RR processes")
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500943 self.assertIsInstance(interval, float)
944 # Reasonable constraints, I think.
945 self.assertGreaterEqual(interval, 0.)
946 self.assertLess(interval, 1.)
947
Benjamin Peterson2740af82011-08-02 17:41:34 -0500948 @requires_sched_affinity
Antoine Pitrou84869872012-08-04 16:16:35 +0200949 def test_sched_getaffinity(self):
950 mask = posix.sched_getaffinity(0)
951 self.assertIsInstance(mask, set)
952 self.assertGreaterEqual(len(mask), 1)
953 self.assertRaises(OSError, posix.sched_getaffinity, -1)
954 for cpu in mask:
955 self.assertIsInstance(cpu, int)
956 self.assertGreaterEqual(cpu, 0)
957 self.assertLess(cpu, 1 << 32)
958
959 @requires_sched_affinity
960 def test_sched_setaffinity(self):
961 mask = posix.sched_getaffinity(0)
962 if len(mask) > 1:
963 # Empty masks are forbidden
964 mask.pop()
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500965 posix.sched_setaffinity(0, mask)
Antoine Pitrou84869872012-08-04 16:16:35 +0200966 self.assertEqual(posix.sched_getaffinity(0), mask)
967 self.assertRaises(OSError, posix.sched_setaffinity, 0, [])
968 self.assertRaises(ValueError, posix.sched_setaffinity, 0, [-10])
969 self.assertRaises(OverflowError, posix.sched_setaffinity, 0, [1<<128])
Benjamin Peterson94b580d2011-08-02 17:30:04 -0500970 self.assertRaises(OSError, posix.sched_setaffinity, -1, mask)
971
Victor Stinner8b905bd2011-10-25 13:34:04 +0200972 def test_rtld_constants(self):
973 # check presence of major RTLD_* constants
974 posix.RTLD_LAZY
975 posix.RTLD_NOW
976 posix.RTLD_GLOBAL
977 posix.RTLD_LOCAL
978
Jesus Cea60c13dd2012-06-23 02:58:14 +0200979 @unittest.skipUnless(hasattr(os, 'SEEK_HOLE'),
980 "test needs an OS that reports file holes")
Hynek Schlawackf841e422012-06-24 09:51:46 +0200981 def test_fs_holes(self):
Jesus Cea94363612012-06-22 18:32:07 +0200982 # Even if the filesystem doesn't report holes,
983 # if the OS supports it the SEEK_* constants
984 # will be defined and will have a consistent
985 # behaviour:
986 # os.SEEK_DATA = current position
987 # os.SEEK_HOLE = end of file position
Hynek Schlawackf841e422012-06-24 09:51:46 +0200988 with open(support.TESTFN, 'r+b') as fp:
Jesus Cea94363612012-06-22 18:32:07 +0200989 fp.write(b"hello")
990 fp.flush()
991 size = fp.tell()
992 fno = fp.fileno()
Jesus Cead46f7d22012-07-07 14:56:04 +0200993 try :
994 for i in range(size):
995 self.assertEqual(i, os.lseek(fno, i, os.SEEK_DATA))
996 self.assertLessEqual(size, os.lseek(fno, i, os.SEEK_HOLE))
997 self.assertRaises(OSError, os.lseek, fno, size, os.SEEK_DATA)
998 self.assertRaises(OSError, os.lseek, fno, size, os.SEEK_HOLE)
999 except OSError :
1000 # Some OSs claim to support SEEK_HOLE/SEEK_DATA
1001 # but it is not true.
1002 # For instance:
1003 # http://lists.freebsd.org/pipermail/freebsd-amd64/2012-January/014332.html
1004 raise unittest.SkipTest("OSError raised!")
Jesus Cea94363612012-06-22 18:32:07 +02001005
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00001006class PosixGroupsTester(unittest.TestCase):
1007
1008 def setUp(self):
1009 if posix.getuid() != 0:
1010 raise unittest.SkipTest("not enough privileges")
1011 if not hasattr(posix, 'getgroups'):
1012 raise unittest.SkipTest("need posix.getgroups")
1013 if sys.platform == 'darwin':
1014 raise unittest.SkipTest("getgroups(2) is broken on OSX")
1015 self.saved_groups = posix.getgroups()
1016
1017 def tearDown(self):
1018 if hasattr(posix, 'setgroups'):
1019 posix.setgroups(self.saved_groups)
1020 elif hasattr(posix, 'initgroups'):
1021 name = pwd.getpwuid(posix.getuid()).pw_name
1022 posix.initgroups(name, self.saved_groups[0])
1023
1024 @unittest.skipUnless(hasattr(posix, 'initgroups'),
1025 "test needs posix.initgroups()")
1026 def test_initgroups(self):
1027 # find missing group
1028
Antoine Pitroue5a91012010-09-04 17:32:06 +00001029 g = max(self.saved_groups) + 1
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00001030 name = pwd.getpwuid(posix.getuid()).pw_name
1031 posix.initgroups(name, g)
1032 self.assertIn(g, posix.getgroups())
1033
1034 @unittest.skipUnless(hasattr(posix, 'setgroups'),
1035 "test needs posix.setgroups()")
1036 def test_setgroups(self):
Antoine Pitroue5a91012010-09-04 17:32:06 +00001037 for groups in [[0], list(range(16))]:
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +00001038 posix.setgroups(groups)
1039 self.assertListEqual(groups, posix.getgroups())
1040
Neal Norwitze241ce82003-02-17 18:17:05 +00001041def test_main():
Antoine Pitrou68c95922011-03-20 17:33:57 +01001042 try:
1043 support.run_unittest(PosixTester, PosixGroupsTester)
1044 finally:
1045 support.reap_children()
Neal Norwitze241ce82003-02-17 18:17:05 +00001046
1047if __name__ == '__main__':
1048 test_main()