blob: 70a47b02f425c1108f88ba182c8d33a32edeaa3f [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
Christian Heimesd5e2b6f2008-03-19 21:50:51 +000013import pwd
Benjamin Petersondcf97b92008-07-02 17:30:14 +000014import shutil
Benjamin Peterson052a02b2010-08-17 01:27:09 +000015import stat
Neal Norwitze241ce82003-02-17 18:17:05 +000016import unittest
17import warnings
R. David Murraya21e4ca2009-03-31 23:16:50 +000018
Neal Norwitze241ce82003-02-17 18:17:05 +000019
20class PosixTester(unittest.TestCase):
21
22 def setUp(self):
23 # create empty file
Benjamin Petersonee8712c2008-05-20 21:35:26 +000024 fp = open(support.TESTFN, 'w+')
Neal Norwitze241ce82003-02-17 18:17:05 +000025 fp.close()
Brett Cannonc8d502e2010-03-20 21:53:28 +000026 self._warnings_manager = support.check_warnings()
27 self._warnings_manager.__enter__()
28 warnings.filterwarnings('ignore', '.* potential security risk .*',
29 RuntimeWarning)
Neal Norwitze241ce82003-02-17 18:17:05 +000030
31 def tearDown(self):
Neal Norwitzc34177c2008-08-25 01:04:16 +000032 support.unlink(support.TESTFN)
Brett Cannonc8d502e2010-03-20 21:53:28 +000033 self._warnings_manager.__exit__(None, None, None)
Neal Norwitze241ce82003-02-17 18:17:05 +000034
35 def testNoArgFunctions(self):
36 # test posix functions which take no arguments and have
37 # no side-effects which we need to cleanup (e.g., fork, wait, abort)
Guido van Rossumf0af3e32008-10-02 18:55:37 +000038 NO_ARG_FUNCTIONS = [ "ctermid", "getcwd", "getcwdb", "uname",
Guido van Rossum687b9c02007-10-25 23:18:51 +000039 "times", "getloadavg",
Neal Norwitze241ce82003-02-17 18:17:05 +000040 "getegid", "geteuid", "getgid", "getgroups",
Ross Lagerwall7807c352011-03-17 20:20:30 +020041 "getpid", "getpgrp", "getppid", "getuid", "sync",
Neal Norwitze241ce82003-02-17 18:17:05 +000042 ]
Neal Norwitz71b13e82003-02-23 22:12:24 +000043
Neal Norwitze241ce82003-02-17 18:17:05 +000044 for name in NO_ARG_FUNCTIONS:
45 posix_func = getattr(posix, name, None)
46 if posix_func is not None:
47 posix_func()
Neal Norwitz2ff51a82003-02-17 22:40:31 +000048 self.assertRaises(TypeError, posix_func, 1)
Neal Norwitze241ce82003-02-17 18:17:05 +000049
Martin v. Löwis7aed61a2009-11-27 14:09:49 +000050 if hasattr(posix, 'getresuid'):
51 def test_getresuid(self):
52 user_ids = posix.getresuid()
53 self.assertEqual(len(user_ids), 3)
54 for val in user_ids:
55 self.assertGreaterEqual(val, 0)
56
57 if hasattr(posix, 'getresgid'):
58 def test_getresgid(self):
59 group_ids = posix.getresgid()
60 self.assertEqual(len(group_ids), 3)
61 for val in group_ids:
62 self.assertGreaterEqual(val, 0)
63
64 if hasattr(posix, 'setresuid'):
65 def test_setresuid(self):
66 current_user_ids = posix.getresuid()
67 self.assertIsNone(posix.setresuid(*current_user_ids))
68 # -1 means don't change that value.
69 self.assertIsNone(posix.setresuid(-1, -1, -1))
70
71 def test_setresuid_exception(self):
72 # Don't do this test if someone is silly enough to run us as root.
73 current_user_ids = posix.getresuid()
74 if 0 not in current_user_ids:
75 new_user_ids = (current_user_ids[0]+1, -1, -1)
76 self.assertRaises(OSError, posix.setresuid, *new_user_ids)
77
78 if hasattr(posix, 'setresgid'):
79 def test_setresgid(self):
80 current_group_ids = posix.getresgid()
81 self.assertIsNone(posix.setresgid(*current_group_ids))
82 # -1 means don't change that value.
83 self.assertIsNone(posix.setresgid(-1, -1, -1))
84
85 def test_setresgid_exception(self):
86 # Don't do this test if someone is silly enough to run us as root.
87 current_group_ids = posix.getresgid()
88 if 0 not in current_group_ids:
89 new_group_ids = (current_group_ids[0]+1, -1, -1)
90 self.assertRaises(OSError, posix.setresgid, *new_group_ids)
91
Antoine Pitroub7572f02009-12-02 20:46:48 +000092 @unittest.skipUnless(hasattr(posix, 'initgroups'),
93 "test needs os.initgroups()")
94 def test_initgroups(self):
95 # It takes a string and an integer; check that it raises a TypeError
96 # for other argument lists.
97 self.assertRaises(TypeError, posix.initgroups)
98 self.assertRaises(TypeError, posix.initgroups, None)
99 self.assertRaises(TypeError, posix.initgroups, 3, "foo")
100 self.assertRaises(TypeError, posix.initgroups, "foo", 3, object())
101
102 # If a non-privileged user invokes it, it should fail with OSError
103 # EPERM.
104 if os.getuid() != 0:
105 name = pwd.getpwuid(posix.getuid()).pw_name
106 try:
107 posix.initgroups(name, 13)
108 except OSError as e:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000109 self.assertEqual(e.errno, errno.EPERM)
Antoine Pitroub7572f02009-12-02 20:46:48 +0000110 else:
111 self.fail("Expected OSError to be raised by initgroups")
112
Neal Norwitze241ce82003-02-17 18:17:05 +0000113 def test_statvfs(self):
114 if hasattr(posix, 'statvfs'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000115 self.assertTrue(posix.statvfs(os.curdir))
Neal Norwitze241ce82003-02-17 18:17:05 +0000116
117 def test_fstatvfs(self):
118 if hasattr(posix, 'fstatvfs'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000119 fp = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000120 try:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000121 self.assertTrue(posix.fstatvfs(fp.fileno()))
Neal Norwitze241ce82003-02-17 18:17:05 +0000122 finally:
123 fp.close()
124
125 def test_ftruncate(self):
126 if hasattr(posix, 'ftruncate'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000127 fp = open(support.TESTFN, 'w+')
Neal Norwitze241ce82003-02-17 18:17:05 +0000128 try:
129 # we need to have some data to truncate
130 fp.write('test')
131 fp.flush()
132 posix.ftruncate(fp.fileno(), 0)
133 finally:
134 fp.close()
135
Ross Lagerwall7807c352011-03-17 20:20:30 +0200136 @unittest.skipUnless(hasattr(posix, 'truncate'), "test needs posix.truncate()")
137 def test_truncate(self):
138 with open(support.TESTFN, 'w') as fp:
139 fp.write('test')
140 fp.flush()
141 posix.truncate(support.TESTFN, 0)
142
143 @unittest.skipUnless(hasattr(posix, 'fexecve'), "test needs posix.fexecve()")
144 @unittest.skipUnless(hasattr(os, 'fork'), "test needs os.fork()")
Ross Lagerwalldedf6cf2011-03-20 18:27:05 +0200145 @unittest.skipUnless(hasattr(os, 'waitpid'), "test needs os.waitpid()")
Ross Lagerwall7807c352011-03-17 20:20:30 +0200146 def test_fexecve(self):
147 fp = os.open(sys.executable, os.O_RDONLY)
148 try:
149 pid = os.fork()
150 if pid == 0:
151 os.chdir(os.path.split(sys.executable)[0])
152 posix.fexecve(fp, [sys.executable, '-c', 'pass'], os.environ)
153 else:
Ross Lagerwalldedf6cf2011-03-20 18:27:05 +0200154 self.assertEqual(os.waitpid(pid, 0), (pid, 0))
Ross Lagerwall7807c352011-03-17 20:20:30 +0200155 finally:
156 os.close(fp)
157
158 @unittest.skipUnless(hasattr(posix, 'waitid'), "test needs posix.waitid()")
159 @unittest.skipUnless(hasattr(os, 'fork'), "test needs os.fork()")
160 def test_waitid(self):
161 pid = os.fork()
162 if pid == 0:
163 os.chdir(os.path.split(sys.executable)[0])
164 posix.execve(sys.executable, [sys.executable, '-c', 'pass'], os.environ)
165 else:
166 res = posix.waitid(posix.P_PID, pid, posix.WEXITED)
167 self.assertEqual(pid, res.si_pid)
168
169 @unittest.skipUnless(hasattr(posix, 'lockf'), "test needs posix.lockf()")
170 def test_lockf(self):
171 fd = os.open(support.TESTFN, os.O_WRONLY | os.O_CREAT)
172 try:
173 os.write(fd, b'test')
174 os.lseek(fd, 0, os.SEEK_SET)
175 posix.lockf(fd, posix.F_LOCK, 4)
176 # section is locked
177 posix.lockf(fd, posix.F_ULOCK, 4)
178 finally:
179 os.close(fd)
180
181 @unittest.skipUnless(hasattr(posix, 'pread'), "test needs posix.pread()")
182 def test_pread(self):
183 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
184 try:
185 os.write(fd, b'test')
186 os.lseek(fd, 0, os.SEEK_SET)
187 self.assertEqual(b'es', posix.pread(fd, 2, 1))
188 # the first pread() shoudn't disturb the file offset
189 self.assertEqual(b'te', posix.read(fd, 2))
190 finally:
191 os.close(fd)
192
193 @unittest.skipUnless(hasattr(posix, 'pwrite'), "test needs posix.pwrite()")
194 def test_pwrite(self):
195 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
196 try:
197 os.write(fd, b'test')
198 os.lseek(fd, 0, os.SEEK_SET)
199 posix.pwrite(fd, b'xx', 1)
200 self.assertEqual(b'txxt', posix.read(fd, 4))
201 finally:
202 os.close(fd)
203
204 @unittest.skipUnless(hasattr(posix, 'posix_fallocate'),
205 "test needs posix.posix_fallocate()")
206 def test_posix_fallocate(self):
207 fd = os.open(support.TESTFN, os.O_WRONLY | os.O_CREAT)
208 try:
209 posix.posix_fallocate(fd, 0, 10)
210 except OSError as inst:
211 # issue10812, ZFS doesn't appear to support posix_fallocate,
212 # so skip Solaris-based since they are likely to have ZFS.
213 if inst.errno != errno.EINVAL or not sys.platform.startswith("sunos"):
214 raise
215 finally:
216 os.close(fd)
217
218 @unittest.skipUnless(hasattr(posix, 'posix_fadvise'),
219 "test needs posix.posix_fadvise()")
220 def test_posix_fadvise(self):
221 fd = os.open(support.TESTFN, os.O_RDONLY)
222 try:
223 posix.posix_fadvise(fd, 0, 0, posix.POSIX_FADV_WILLNEED)
224 finally:
225 os.close(fd)
226
227 @unittest.skipUnless(hasattr(posix, 'futimes'), "test needs posix.futimes()")
228 def test_futimes(self):
229 now = time.time()
230 fd = os.open(support.TESTFN, os.O_RDONLY)
231 try:
232 posix.futimes(fd, None)
233 self.assertRaises(TypeError, posix.futimes, fd, (None, None))
234 self.assertRaises(TypeError, posix.futimes, fd, (now, None))
235 self.assertRaises(TypeError, posix.futimes, fd, (None, now))
236 posix.futimes(fd, (int(now), int(now)))
237 posix.futimes(fd, (now, now))
238 finally:
239 os.close(fd)
240
241 @unittest.skipUnless(hasattr(posix, 'lutimes'), "test needs posix.lutimes()")
242 def test_lutimes(self):
243 now = time.time()
244 posix.lutimes(support.TESTFN, None)
245 self.assertRaises(TypeError, posix.lutimes, support.TESTFN, (None, None))
246 self.assertRaises(TypeError, posix.lutimes, support.TESTFN, (now, None))
247 self.assertRaises(TypeError, posix.lutimes, support.TESTFN, (None, now))
248 posix.lutimes(support.TESTFN, (int(now), int(now)))
249 posix.lutimes(support.TESTFN, (now, now))
250
251 @unittest.skipUnless(hasattr(posix, 'futimens'), "test needs posix.futimens()")
252 def test_futimens(self):
253 now = time.time()
254 fd = os.open(support.TESTFN, os.O_RDONLY)
255 try:
256 self.assertRaises(TypeError, posix.futimens, fd, (None, None), (None, None))
257 self.assertRaises(TypeError, posix.futimens, fd, (now, 0), None)
258 self.assertRaises(TypeError, posix.futimens, fd, None, (now, 0))
259 posix.futimens(fd, (int(now), int((now - int(now)) * 1e9)),
260 (int(now), int((now - int(now)) * 1e9)))
261 finally:
262 os.close(fd)
263
264 @unittest.skipUnless(hasattr(posix, 'writev'), "test needs posix.writev()")
265 def test_writev(self):
266 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
267 try:
268 os.writev(fd, (b'test1', b'tt2', b't3'))
269 os.lseek(fd, 0, os.SEEK_SET)
270 self.assertEqual(b'test1tt2t3', posix.read(fd, 10))
271 finally:
272 os.close(fd)
273
274 @unittest.skipUnless(hasattr(posix, 'readv'), "test needs posix.readv()")
275 def test_readv(self):
276 fd = os.open(support.TESTFN, os.O_RDWR | os.O_CREAT)
277 try:
278 os.write(fd, b'test1tt2t3')
279 os.lseek(fd, 0, os.SEEK_SET)
280 buf = [bytearray(i) for i in [5, 3, 2]]
281 self.assertEqual(posix.readv(fd, buf), 10)
282 self.assertEqual([b'test1', b'tt2', b't3'], [bytes(i) for i in buf])
283 finally:
284 os.close(fd)
285
Neal Norwitze241ce82003-02-17 18:17:05 +0000286 def test_dup(self):
287 if hasattr(posix, 'dup'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000288 fp = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000289 try:
290 fd = posix.dup(fp.fileno())
Ezio Melottie9615932010-01-24 19:26:24 +0000291 self.assertIsInstance(fd, int)
Neal Norwitze241ce82003-02-17 18:17:05 +0000292 os.close(fd)
293 finally:
294 fp.close()
295
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000296 def test_confstr(self):
297 if hasattr(posix, 'confstr'):
298 self.assertRaises(ValueError, posix.confstr, "CS_garbage")
299 self.assertEqual(len(posix.confstr("CS_PATH")) > 0, True)
300
Neal Norwitze241ce82003-02-17 18:17:05 +0000301 def test_dup2(self):
302 if hasattr(posix, 'dup2'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000303 fp1 = open(support.TESTFN)
304 fp2 = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000305 try:
306 posix.dup2(fp1.fileno(), fp2.fileno())
307 finally:
308 fp1.close()
309 fp2.close()
310
Charles-François Natali1e045b12011-05-22 20:42:32 +0200311 @unittest.skipUnless(hasattr(os, 'O_CLOEXEC'), "needs os.O_CLOEXEC")
Charles-François Natali239bb962011-06-03 12:55:15 +0200312 @support.requires_linux_version(2, 6, 23)
Charles-François Natali1e045b12011-05-22 20:42:32 +0200313 def test_oscloexec(self):
314 fd = os.open(support.TESTFN, os.O_RDONLY|os.O_CLOEXEC)
315 self.addCleanup(os.close, fd)
Victor Stinnere36f3752011-05-24 00:29:43 +0200316 self.assertTrue(fcntl.fcntl(fd, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
Charles-François Natali1e045b12011-05-22 20:42:32 +0200317
Skip Montanaro98470002005-06-17 01:14:49 +0000318 def test_osexlock(self):
319 if hasattr(posix, "O_EXLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000320 fd = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000321 os.O_WRONLY|os.O_EXLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000322 self.assertRaises(OSError, os.open, support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000323 os.O_WRONLY|os.O_EXLOCK|os.O_NONBLOCK)
324 os.close(fd)
325
326 if hasattr(posix, "O_SHLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000327 fd = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000328 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000329 self.assertRaises(OSError, os.open, support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000330 os.O_WRONLY|os.O_EXLOCK|os.O_NONBLOCK)
331 os.close(fd)
332
333 def test_osshlock(self):
334 if hasattr(posix, "O_SHLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000335 fd1 = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000336 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000337 fd2 = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000338 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
339 os.close(fd2)
340 os.close(fd1)
341
342 if hasattr(posix, "O_EXLOCK"):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000343 fd = os.open(support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000344 os.O_WRONLY|os.O_SHLOCK|os.O_CREAT)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000345 self.assertRaises(OSError, os.open, support.TESTFN,
Skip Montanaro98470002005-06-17 01:14:49 +0000346 os.O_RDONLY|os.O_EXLOCK|os.O_NONBLOCK)
347 os.close(fd)
348
Neal Norwitze241ce82003-02-17 18:17:05 +0000349 def test_fstat(self):
350 if hasattr(posix, 'fstat'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000351 fp = open(support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000352 try:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000353 self.assertTrue(posix.fstat(fp.fileno()))
Neal Norwitze241ce82003-02-17 18:17:05 +0000354 finally:
355 fp.close()
356
357 def test_stat(self):
358 if hasattr(posix, 'stat'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000359 self.assertTrue(posix.stat(support.TESTFN))
Neal Norwitze241ce82003-02-17 18:17:05 +0000360
Benjamin Peterson052a02b2010-08-17 01:27:09 +0000361 @unittest.skipUnless(hasattr(posix, 'mkfifo'), "don't have mkfifo()")
362 def test_mkfifo(self):
363 support.unlink(support.TESTFN)
364 posix.mkfifo(support.TESTFN, stat.S_IRUSR | stat.S_IWUSR)
365 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
366
367 @unittest.skipUnless(hasattr(posix, 'mknod') and hasattr(stat, 'S_IFIFO'),
368 "don't have mknod()/S_IFIFO")
369 def test_mknod(self):
370 # Test using mknod() to create a FIFO (the only use specified
371 # by POSIX).
372 support.unlink(support.TESTFN)
373 mode = stat.S_IFIFO | stat.S_IRUSR | stat.S_IWUSR
374 try:
375 posix.mknod(support.TESTFN, mode, 0)
376 except OSError as e:
377 # Some old systems don't allow unprivileged users to use
378 # mknod(), or only support creating device nodes.
379 self.assertIn(e.errno, (errno.EPERM, errno.EINVAL))
380 else:
381 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
382
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000383 def _test_all_chown_common(self, chown_func, first_param):
384 """Common code for chown, fchown and lchown tests."""
385 if os.getuid() == 0:
386 try:
387 # Many linux distros have a nfsnobody user as MAX_UID-2
388 # that makes a good test case for signedness issues.
389 # http://bugs.python.org/issue1747858
390 # This part of the test only runs when run as root.
391 # Only scary people run their tests as root.
392 ent = pwd.getpwnam('nfsnobody')
393 chown_func(first_param, ent.pw_uid, ent.pw_gid)
394 except KeyError:
395 pass
396 else:
397 # non-root cannot chown to root, raises OSError
398 self.assertRaises(OSError, chown_func,
399 first_param, 0, 0)
400 # test a successful chown call
401 chown_func(first_param, os.getuid(), os.getgid())
Christian Heimesd5e2b6f2008-03-19 21:50:51 +0000402
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000403 @unittest.skipUnless(hasattr(posix, 'chown'), "test needs os.chown()")
404 def test_chown(self):
405 # raise an OSError if the file does not exist
406 os.unlink(support.TESTFN)
407 self.assertRaises(OSError, posix.chown, support.TESTFN, -1, -1)
Christian Heimesd5e2b6f2008-03-19 21:50:51 +0000408
Benjamin Peterson1baf4652009-12-31 03:11:23 +0000409 # re-create the file
410 open(support.TESTFN, 'w').close()
411 self._test_all_chown_common(posix.chown, support.TESTFN)
412
413 @unittest.skipUnless(hasattr(posix, 'fchown'), "test needs os.fchown()")
414 def test_fchown(self):
415 os.unlink(support.TESTFN)
416
417 # re-create the file
418 test_file = open(support.TESTFN, 'w')
419 try:
420 fd = test_file.fileno()
421 self._test_all_chown_common(posix.fchown, fd)
422 finally:
423 test_file.close()
424
425 @unittest.skipUnless(hasattr(posix, 'lchown'), "test needs os.lchown()")
426 def test_lchown(self):
427 os.unlink(support.TESTFN)
428 # create a symlink
429 os.symlink('/tmp/dummy-symlink-target', support.TESTFN)
430 self._test_all_chown_common(posix.lchown, support.TESTFN)
Christian Heimesd5e2b6f2008-03-19 21:50:51 +0000431
Neal Norwitze241ce82003-02-17 18:17:05 +0000432 def test_chdir(self):
433 if hasattr(posix, 'chdir'):
434 posix.chdir(os.curdir)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000435 self.assertRaises(OSError, posix.chdir, support.TESTFN)
Neal Norwitze241ce82003-02-17 18:17:05 +0000436
Martin v. Löwisc9e1c7d2010-07-23 12:16:41 +0000437 def test_listdir(self):
438 if hasattr(posix, 'listdir'):
439 self.assertTrue(support.TESTFN in posix.listdir(os.curdir))
440
441 def test_listdir_default(self):
442 # When listdir is called without argument, it's the same as listdir(os.curdir)
443 if hasattr(posix, 'listdir'):
444 self.assertTrue(support.TESTFN in posix.listdir())
Neal Norwitze241ce82003-02-17 18:17:05 +0000445
Antoine Pitrou8250e232011-02-25 23:41:16 +0000446 @unittest.skipUnless(hasattr(posix, 'fdlistdir'), "test needs posix.fdlistdir()")
447 def test_fdlistdir(self):
448 f = posix.open(posix.getcwd(), posix.O_RDONLY)
449 self.assertEqual(
450 sorted(posix.listdir('.')),
451 sorted(posix.fdlistdir(f))
452 )
453 # Check the fd was closed by fdlistdir
454 with self.assertRaises(OSError) as ctx:
455 posix.close(f)
456 self.assertEqual(ctx.exception.errno, errno.EBADF)
457
Neal Norwitze241ce82003-02-17 18:17:05 +0000458 def test_access(self):
459 if hasattr(posix, 'access'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000460 self.assertTrue(posix.access(support.TESTFN, os.R_OK))
Neal Norwitze241ce82003-02-17 18:17:05 +0000461
462 def test_umask(self):
463 if hasattr(posix, 'umask'):
464 old_mask = posix.umask(0)
Ezio Melottie9615932010-01-24 19:26:24 +0000465 self.assertIsInstance(old_mask, int)
Neal Norwitze241ce82003-02-17 18:17:05 +0000466 posix.umask(old_mask)
467
468 def test_strerror(self):
469 if hasattr(posix, 'strerror'):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000470 self.assertTrue(posix.strerror(0))
Neal Norwitze241ce82003-02-17 18:17:05 +0000471
472 def test_pipe(self):
473 if hasattr(posix, 'pipe'):
474 reader, writer = posix.pipe()
475 os.close(reader)
476 os.close(writer)
477
Charles-François Natalidaafdd52011-05-29 20:07:40 +0200478 @unittest.skipUnless(hasattr(os, 'pipe2'), "test needs os.pipe2()")
Charles-François Natali239bb962011-06-03 12:55:15 +0200479 @support.requires_linux_version(2, 6, 27)
Charles-François Natalidaafdd52011-05-29 20:07:40 +0200480 def test_pipe2(self):
481 self.assertRaises(TypeError, os.pipe2, 'DEADBEEF')
482 self.assertRaises(TypeError, os.pipe2, 0, 0)
483
Charles-François Natali368f34b2011-06-06 19:49:47 +0200484 # try calling with flags = 0, like os.pipe()
485 r, w = os.pipe2(0)
Charles-François Natalidaafdd52011-05-29 20:07:40 +0200486 os.close(r)
487 os.close(w)
488
489 # test flags
490 r, w = os.pipe2(os.O_CLOEXEC|os.O_NONBLOCK)
491 self.addCleanup(os.close, r)
492 self.addCleanup(os.close, w)
493 self.assertTrue(fcntl.fcntl(r, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
494 self.assertTrue(fcntl.fcntl(w, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
495 # try reading from an empty pipe: this should fail, not block
496 self.assertRaises(OSError, os.read, r, 1)
497 # try a write big enough to fill-up the pipe: this should either
498 # fail or perform a partial write, not block
499 try:
500 os.write(w, b'x' * support.PIPE_MAX_SIZE)
501 except OSError:
502 pass
503
Neal Norwitze241ce82003-02-17 18:17:05 +0000504 def test_utime(self):
505 if hasattr(posix, 'utime'):
506 now = time.time()
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000507 posix.utime(support.TESTFN, None)
508 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, None))
509 self.assertRaises(TypeError, posix.utime, support.TESTFN, (now, None))
510 self.assertRaises(TypeError, posix.utime, support.TESTFN, (None, now))
511 posix.utime(support.TESTFN, (int(now), int(now)))
512 posix.utime(support.TESTFN, (now, now))
Neal Norwitze241ce82003-02-17 18:17:05 +0000513
Thomas Wouterscf297e42007-02-23 15:07:44 +0000514 def test_chflags(self):
515 if hasattr(posix, 'chflags'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000516 st = os.stat(support.TESTFN)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000517 if hasattr(st, 'st_flags'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000518 posix.chflags(support.TESTFN, st.st_flags)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000519
520 def test_lchflags(self):
521 if hasattr(posix, 'lchflags'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000522 st = os.stat(support.TESTFN)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000523 if hasattr(st, 'st_flags'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000524 posix.lchflags(support.TESTFN, st.st_flags)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000525
Guido van Rossum98297ee2007-11-06 21:34:58 +0000526 def test_environ(self):
Victor Stinner17b490d2010-05-06 22:19:30 +0000527 if os.name == "nt":
528 item_type = str
529 else:
530 item_type = bytes
Guido van Rossum98297ee2007-11-06 21:34:58 +0000531 for k, v in posix.environ.items():
Victor Stinner17b490d2010-05-06 22:19:30 +0000532 self.assertEqual(type(k), item_type)
533 self.assertEqual(type(v), item_type)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000534
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000535 def test_getcwd_long_pathnames(self):
536 if hasattr(posix, 'getcwd'):
537 dirname = 'getcwd-test-directory-0123456789abcdef-01234567890abcdef'
538 curdir = os.getcwd()
539 base_path = os.path.abspath(support.TESTFN) + '.getcwd'
540
541 try:
542 os.mkdir(base_path)
543 os.chdir(base_path)
544 except:
Benjamin Petersone549ead2009-03-28 21:42:05 +0000545# Just returning nothing instead of the SkipTest exception,
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000546# because the test results in Error in that case.
547# Is that ok?
Benjamin Petersone549ead2009-03-28 21:42:05 +0000548# raise unittest.SkipTest("cannot create directory for testing")
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000549 return
550
551 def _create_and_do_getcwd(dirname, current_path_length = 0):
552 try:
553 os.mkdir(dirname)
554 except:
Benjamin Petersone549ead2009-03-28 21:42:05 +0000555 raise unittest.SkipTest("mkdir cannot create directory sufficiently deep for getcwd test")
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000556
557 os.chdir(dirname)
558 try:
559 os.getcwd()
560 if current_path_length < 1027:
561 _create_and_do_getcwd(dirname, current_path_length + len(dirname) + 1)
562 finally:
563 os.chdir('..')
564 os.rmdir(dirname)
565
566 _create_and_do_getcwd(dirname)
567
568 finally:
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000569 os.chdir(curdir)
R. David Murray414c91f2009-07-09 20:12:31 +0000570 support.rmtree(base_path)
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000571
Antoine Pitrou318b8f32011-01-12 18:45:27 +0000572 @unittest.skipUnless(hasattr(os, 'getegid'), "test needs os.getegid()")
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000573 def test_getgroups(self):
574 with os.popen('id -G') as idg:
575 groups = idg.read().strip()
576
577 if not groups:
578 raise unittest.SkipTest("need working 'id -G'")
579
Ronald Oussoren7fb6f512010-08-01 19:18:13 +0000580 # 'id -G' and 'os.getgroups()' should return the same
581 # groups, ignoring order and duplicates.
Antoine Pitrou318b8f32011-01-12 18:45:27 +0000582 # #10822 - it is implementation defined whether posix.getgroups()
583 # includes the effective gid so we include it anyway, since id -G does
Ronald Oussorencb615e62010-07-24 14:15:19 +0000584 self.assertEqual(
Ronald Oussoren7fb6f512010-08-01 19:18:13 +0000585 set([int(x) for x in groups.split()]),
Antoine Pitrou318b8f32011-01-12 18:45:27 +0000586 set(posix.getgroups() + [posix.getegid()]))
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000587
Antoine Pitrouf65132d2011-02-25 23:25:17 +0000588 # tests for the posix *at functions follow
589
590 @unittest.skipUnless(hasattr(posix, 'faccessat'), "test needs posix.faccessat()")
591 def test_faccessat(self):
592 f = posix.open(posix.getcwd(), posix.O_RDONLY)
593 try:
594 self.assertTrue(posix.faccessat(f, support.TESTFN, os.R_OK))
595 finally:
596 posix.close(f)
597
598 @unittest.skipUnless(hasattr(posix, 'fchmodat'), "test needs posix.fchmodat()")
599 def test_fchmodat(self):
600 os.chmod(support.TESTFN, stat.S_IRUSR)
601
602 f = posix.open(posix.getcwd(), posix.O_RDONLY)
603 try:
604 posix.fchmodat(f, support.TESTFN, stat.S_IRUSR | stat.S_IWUSR)
605
606 s = posix.stat(support.TESTFN)
607 self.assertEqual(s[0] & stat.S_IRWXU, stat.S_IRUSR | stat.S_IWUSR)
608 finally:
609 posix.close(f)
610
611 @unittest.skipUnless(hasattr(posix, 'fchownat'), "test needs posix.fchownat()")
612 def test_fchownat(self):
613 support.unlink(support.TESTFN)
614 open(support.TESTFN, 'w').close()
615
616 f = posix.open(posix.getcwd(), posix.O_RDONLY)
617 try:
618 posix.fchownat(f, support.TESTFN, os.getuid(), os.getgid())
619 finally:
620 posix.close(f)
621
622 @unittest.skipUnless(hasattr(posix, 'fstatat'), "test needs posix.fstatat()")
623 def test_fstatat(self):
624 support.unlink(support.TESTFN)
625 with open(support.TESTFN, 'w') as outfile:
626 outfile.write("testline\n")
627
628 f = posix.open(posix.getcwd(), posix.O_RDONLY)
629 try:
630 s1 = posix.stat(support.TESTFN)
631 s2 = posix.fstatat(f, support.TESTFN)
632 self.assertEqual(s1, s2)
633 finally:
634 posix.close(f)
635
636 @unittest.skipUnless(hasattr(posix, 'futimesat'), "test needs posix.futimesat()")
637 def test_futimesat(self):
638 f = posix.open(posix.getcwd(), posix.O_RDONLY)
639 try:
640 now = time.time()
641 posix.futimesat(f, support.TESTFN, None)
642 self.assertRaises(TypeError, posix.futimesat, f, support.TESTFN, (None, None))
643 self.assertRaises(TypeError, posix.futimesat, f, support.TESTFN, (now, None))
644 self.assertRaises(TypeError, posix.futimesat, f, support.TESTFN, (None, now))
645 posix.futimesat(f, support.TESTFN, (int(now), int(now)))
646 posix.futimesat(f, support.TESTFN, (now, now))
647 finally:
648 posix.close(f)
649
650 @unittest.skipUnless(hasattr(posix, 'linkat'), "test needs posix.linkat()")
651 def test_linkat(self):
652 f = posix.open(posix.getcwd(), posix.O_RDONLY)
653 try:
654 posix.linkat(f, support.TESTFN, f, support.TESTFN + 'link')
655 # should have same inodes
656 self.assertEqual(posix.stat(support.TESTFN)[1],
657 posix.stat(support.TESTFN + 'link')[1])
658 finally:
659 posix.close(f)
660 support.unlink(support.TESTFN + 'link')
661
662 @unittest.skipUnless(hasattr(posix, 'mkdirat'), "test needs posix.mkdirat()")
663 def test_mkdirat(self):
664 f = posix.open(posix.getcwd(), posix.O_RDONLY)
665 try:
666 posix.mkdirat(f, support.TESTFN + 'dir')
667 posix.stat(support.TESTFN + 'dir') # should not raise exception
668 finally:
669 posix.close(f)
670 support.rmtree(support.TESTFN + 'dir')
671
672 @unittest.skipUnless(hasattr(posix, 'mknodat') and hasattr(stat, 'S_IFIFO'),
673 "don't have mknodat()/S_IFIFO")
674 def test_mknodat(self):
675 # Test using mknodat() to create a FIFO (the only use specified
676 # by POSIX).
677 support.unlink(support.TESTFN)
678 mode = stat.S_IFIFO | stat.S_IRUSR | stat.S_IWUSR
679 f = posix.open(posix.getcwd(), posix.O_RDONLY)
680 try:
681 posix.mknodat(f, support.TESTFN, mode, 0)
682 except OSError as e:
683 # Some old systems don't allow unprivileged users to use
684 # mknod(), or only support creating device nodes.
685 self.assertIn(e.errno, (errno.EPERM, errno.EINVAL))
686 else:
687 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
688 finally:
689 posix.close(f)
690
691 @unittest.skipUnless(hasattr(posix, 'openat'), "test needs posix.openat()")
692 def test_openat(self):
693 support.unlink(support.TESTFN)
694 with open(support.TESTFN, 'w') as outfile:
695 outfile.write("testline\n")
696 a = posix.open(posix.getcwd(), posix.O_RDONLY)
697 b = posix.openat(a, support.TESTFN, posix.O_RDONLY)
698 try:
699 res = posix.read(b, 9).decode(encoding="utf-8")
700 self.assertEqual("testline\n", res)
701 finally:
702 posix.close(a)
703 posix.close(b)
704
705 @unittest.skipUnless(hasattr(posix, 'readlinkat'), "test needs posix.readlinkat()")
706 def test_readlinkat(self):
707 os.symlink(support.TESTFN, support.TESTFN + 'link')
708 f = posix.open(posix.getcwd(), posix.O_RDONLY)
709 try:
710 self.assertEqual(posix.readlink(support.TESTFN + 'link'),
711 posix.readlinkat(f, support.TESTFN + 'link'))
712 finally:
713 support.unlink(support.TESTFN + 'link')
714 posix.close(f)
715
716 @unittest.skipUnless(hasattr(posix, 'renameat'), "test needs posix.renameat()")
717 def test_renameat(self):
718 support.unlink(support.TESTFN)
719 open(support.TESTFN + 'ren', 'w').close()
720 f = posix.open(posix.getcwd(), posix.O_RDONLY)
721 try:
722 posix.renameat(f, support.TESTFN + 'ren', f, support.TESTFN)
723 except:
724 posix.rename(support.TESTFN + 'ren', support.TESTFN)
725 raise
726 else:
727 posix.stat(support.TESTFN) # should not throw exception
728 finally:
729 posix.close(f)
730
731 @unittest.skipUnless(hasattr(posix, 'symlinkat'), "test needs posix.symlinkat()")
732 def test_symlinkat(self):
733 f = posix.open(posix.getcwd(), posix.O_RDONLY)
734 try:
735 posix.symlinkat(support.TESTFN, f, support.TESTFN + 'link')
736 self.assertEqual(posix.readlink(support.TESTFN + 'link'), support.TESTFN)
737 finally:
738 posix.close(f)
739 support.unlink(support.TESTFN + 'link')
740
741 @unittest.skipUnless(hasattr(posix, 'unlinkat'), "test needs posix.unlinkat()")
742 def test_unlinkat(self):
743 f = posix.open(posix.getcwd(), posix.O_RDONLY)
744 open(support.TESTFN + 'del', 'w').close()
745 posix.stat(support.TESTFN + 'del') # should not throw exception
746 try:
747 posix.unlinkat(f, support.TESTFN + 'del')
748 except:
749 support.unlink(support.TESTFN + 'del')
750 raise
751 else:
752 self.assertRaises(OSError, posix.stat, support.TESTFN + 'link')
753 finally:
754 posix.close(f)
755
756 @unittest.skipUnless(hasattr(posix, 'utimensat'), "test needs posix.utimensat()")
757 def test_utimensat(self):
758 f = posix.open(posix.getcwd(), posix.O_RDONLY)
759 try:
760 now = time.time()
761 posix.utimensat(f, support.TESTFN, None, None)
762 self.assertRaises(TypeError, posix.utimensat, f, support.TESTFN, (None, None), (None, None))
763 self.assertRaises(TypeError, posix.utimensat, f, support.TESTFN, (now, 0), None)
764 self.assertRaises(TypeError, posix.utimensat, f, support.TESTFN, None, (now, 0))
765 posix.utimensat(f, support.TESTFN, (int(now), int((now - int(now)) * 1e9)),
766 (int(now), int((now - int(now)) * 1e9)))
767 finally:
768 posix.close(f)
769
770 @unittest.skipUnless(hasattr(posix, 'mkfifoat'), "don't have mkfifoat()")
771 def test_mkfifoat(self):
772 support.unlink(support.TESTFN)
773 f = posix.open(posix.getcwd(), posix.O_RDONLY)
774 try:
775 posix.mkfifoat(f, support.TESTFN, stat.S_IRUSR | stat.S_IWUSR)
776 self.assertTrue(stat.S_ISFIFO(posix.stat(support.TESTFN).st_mode))
777 finally:
778 posix.close(f)
779
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000780class PosixGroupsTester(unittest.TestCase):
781
782 def setUp(self):
783 if posix.getuid() != 0:
784 raise unittest.SkipTest("not enough privileges")
785 if not hasattr(posix, 'getgroups'):
786 raise unittest.SkipTest("need posix.getgroups")
787 if sys.platform == 'darwin':
788 raise unittest.SkipTest("getgroups(2) is broken on OSX")
789 self.saved_groups = posix.getgroups()
790
791 def tearDown(self):
792 if hasattr(posix, 'setgroups'):
793 posix.setgroups(self.saved_groups)
794 elif hasattr(posix, 'initgroups'):
795 name = pwd.getpwuid(posix.getuid()).pw_name
796 posix.initgroups(name, self.saved_groups[0])
797
798 @unittest.skipUnless(hasattr(posix, 'initgroups'),
799 "test needs posix.initgroups()")
800 def test_initgroups(self):
801 # find missing group
802
Antoine Pitroue5a91012010-09-04 17:32:06 +0000803 g = max(self.saved_groups) + 1
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000804 name = pwd.getpwuid(posix.getuid()).pw_name
805 posix.initgroups(name, g)
806 self.assertIn(g, posix.getgroups())
807
808 @unittest.skipUnless(hasattr(posix, 'setgroups'),
809 "test needs posix.setgroups()")
810 def test_setgroups(self):
Antoine Pitroue5a91012010-09-04 17:32:06 +0000811 for groups in [[0], list(range(16))]:
Ronald Oussorenb6ee4f52010-07-23 13:53:51 +0000812 posix.setgroups(groups)
813 self.assertListEqual(groups, posix.getgroups())
814
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000815
Neal Norwitze241ce82003-02-17 18:17:05 +0000816def test_main():
Antoine Pitrou68c95922011-03-20 17:33:57 +0100817 try:
818 support.run_unittest(PosixTester, PosixGroupsTester)
819 finally:
820 support.reap_children()
Neal Norwitze241ce82003-02-17 18:17:05 +0000821
822if __name__ == '__main__':
823 test_main()