blob: 962adde38da541d914af7f95472095872257d6df [file] [log] [blame]
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001import collections.abc
Antoine Pitrou31119e42013-11-22 17:38:12 +01002import io
3import os
4import errno
5import pathlib
6import pickle
Antoine Pitrou31119e42013-11-22 17:38:12 +01007import socket
8import stat
Antoine Pitrou31119e42013-11-22 17:38:12 +01009import tempfile
10import unittest
Armin Rigo22a594a2017-04-13 20:08:15 +020011from unittest import mock
Antoine Pitrou31119e42013-11-22 17:38:12 +010012
13from test import support
Xavier de Gaye3a4e9892016-12-13 10:00:01 +010014android_not_root = support.android_not_root
Antoine Pitrou31119e42013-11-22 17:38:12 +010015TESTFN = support.TESTFN
16
17try:
18 import grp, pwd
19except ImportError:
20 grp = pwd = None
21
22
23class _BaseFlavourTest(object):
24
25 def _check_parse_parts(self, arg, expected):
26 f = self.flavour.parse_parts
27 sep = self.flavour.sep
28 altsep = self.flavour.altsep
29 actual = f([x.replace('/', sep) for x in arg])
30 self.assertEqual(actual, expected)
31 if altsep:
32 actual = f([x.replace('/', altsep) for x in arg])
33 self.assertEqual(actual, expected)
34
35 def test_parse_parts_common(self):
36 check = self._check_parse_parts
37 sep = self.flavour.sep
38 # Unanchored parts
39 check([], ('', '', []))
40 check(['a'], ('', '', ['a']))
41 check(['a/'], ('', '', ['a']))
42 check(['a', 'b'], ('', '', ['a', 'b']))
43 # Expansion
44 check(['a/b'], ('', '', ['a', 'b']))
45 check(['a/b/'], ('', '', ['a', 'b']))
46 check(['a', 'b/c', 'd'], ('', '', ['a', 'b', 'c', 'd']))
47 # Collapsing and stripping excess slashes
48 check(['a', 'b//c', 'd'], ('', '', ['a', 'b', 'c', 'd']))
49 check(['a', 'b/c/', 'd'], ('', '', ['a', 'b', 'c', 'd']))
50 # Eliminating standalone dots
51 check(['.'], ('', '', []))
52 check(['.', '.', 'b'], ('', '', ['b']))
53 check(['a', '.', 'b'], ('', '', ['a', 'b']))
54 check(['a', '.', '.'], ('', '', ['a']))
55 # The first part is anchored
56 check(['/a/b'], ('', sep, [sep, 'a', 'b']))
57 check(['/a', 'b'], ('', sep, [sep, 'a', 'b']))
58 check(['/a/', 'b'], ('', sep, [sep, 'a', 'b']))
59 # Ignoring parts before an anchored part
60 check(['a', '/b', 'c'], ('', sep, [sep, 'b', 'c']))
61 check(['a', '/b', '/c'], ('', sep, [sep, 'c']))
62
63
64class PosixFlavourTest(_BaseFlavourTest, unittest.TestCase):
65 flavour = pathlib._posix_flavour
66
67 def test_parse_parts(self):
68 check = self._check_parse_parts
69 # Collapsing of excess leading slashes, except for the double-slash
70 # special case.
71 check(['//a', 'b'], ('', '//', ['//', 'a', 'b']))
72 check(['///a', 'b'], ('', '/', ['/', 'a', 'b']))
73 check(['////a', 'b'], ('', '/', ['/', 'a', 'b']))
74 # Paths which look like NT paths aren't treated specially
75 check(['c:a'], ('', '', ['c:a']))
76 check(['c:\\a'], ('', '', ['c:\\a']))
77 check(['\\a'], ('', '', ['\\a']))
78
79 def test_splitroot(self):
80 f = self.flavour.splitroot
81 self.assertEqual(f(''), ('', '', ''))
82 self.assertEqual(f('a'), ('', '', 'a'))
83 self.assertEqual(f('a/b'), ('', '', 'a/b'))
84 self.assertEqual(f('a/b/'), ('', '', 'a/b/'))
85 self.assertEqual(f('/a'), ('', '/', 'a'))
86 self.assertEqual(f('/a/b'), ('', '/', 'a/b'))
87 self.assertEqual(f('/a/b/'), ('', '/', 'a/b/'))
88 # The root is collapsed when there are redundant slashes
89 # except when there are exactly two leading slashes, which
90 # is a special case in POSIX.
91 self.assertEqual(f('//a'), ('', '//', 'a'))
92 self.assertEqual(f('///a'), ('', '/', 'a'))
93 self.assertEqual(f('///a/b'), ('', '/', 'a/b'))
94 # Paths which look like NT paths aren't treated specially
95 self.assertEqual(f('c:/a/b'), ('', '', 'c:/a/b'))
96 self.assertEqual(f('\\/a/b'), ('', '', '\\/a/b'))
97 self.assertEqual(f('\\a\\b'), ('', '', '\\a\\b'))
98
99
100class NTFlavourTest(_BaseFlavourTest, unittest.TestCase):
101 flavour = pathlib._windows_flavour
102
103 def test_parse_parts(self):
104 check = self._check_parse_parts
105 # First part is anchored
106 check(['c:'], ('c:', '', ['c:']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100107 check(['c:/'], ('c:', '\\', ['c:\\']))
108 check(['/'], ('', '\\', ['\\']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100109 check(['c:a'], ('c:', '', ['c:', 'a']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100110 check(['c:/a'], ('c:', '\\', ['c:\\', 'a']))
111 check(['/a'], ('', '\\', ['\\', 'a']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100112 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100113 check(['//a/b'], ('\\\\a\\b', '\\', ['\\\\a\\b\\']))
114 check(['//a/b/'], ('\\\\a\\b', '\\', ['\\\\a\\b\\']))
115 check(['//a/b/c'], ('\\\\a\\b', '\\', ['\\\\a\\b\\', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100116 # Second part is anchored, so that the first part is ignored
117 check(['a', 'Z:b', 'c'], ('Z:', '', ['Z:', 'b', 'c']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100118 check(['a', 'Z:/b', 'c'], ('Z:', '\\', ['Z:\\', 'b', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100119 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100120 check(['a', '//b/c', 'd'], ('\\\\b\\c', '\\', ['\\\\b\\c\\', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100121 # Collapsing and stripping excess slashes
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100122 check(['a', 'Z://b//c/', 'd/'], ('Z:', '\\', ['Z:\\', 'b', 'c', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100123 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100124 check(['a', '//b/c//', 'd'], ('\\\\b\\c', '\\', ['\\\\b\\c\\', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100125 # Extended paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100126 check(['//?/c:/'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\']))
127 check(['//?/c:/a'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\', 'a']))
128 check(['//?/c:/a', '/b'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\', 'b']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100129 # Extended UNC paths (format is "\\?\UNC\server\share")
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100130 check(['//?/UNC/b/c'], ('\\\\?\\UNC\\b\\c', '\\', ['\\\\?\\UNC\\b\\c\\']))
131 check(['//?/UNC/b/c/d'], ('\\\\?\\UNC\\b\\c', '\\', ['\\\\?\\UNC\\b\\c\\', 'd']))
132 # Second part has a root but not drive
133 check(['a', '/b', 'c'], ('', '\\', ['\\', 'b', 'c']))
134 check(['Z:/a', '/b', 'c'], ('Z:', '\\', ['Z:\\', 'b', 'c']))
135 check(['//?/Z:/a', '/b', 'c'], ('\\\\?\\Z:', '\\', ['\\\\?\\Z:\\', 'b', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100136
137 def test_splitroot(self):
138 f = self.flavour.splitroot
139 self.assertEqual(f(''), ('', '', ''))
140 self.assertEqual(f('a'), ('', '', 'a'))
141 self.assertEqual(f('a\\b'), ('', '', 'a\\b'))
142 self.assertEqual(f('\\a'), ('', '\\', 'a'))
143 self.assertEqual(f('\\a\\b'), ('', '\\', 'a\\b'))
144 self.assertEqual(f('c:a\\b'), ('c:', '', 'a\\b'))
145 self.assertEqual(f('c:\\a\\b'), ('c:', '\\', 'a\\b'))
146 # Redundant slashes in the root are collapsed
147 self.assertEqual(f('\\\\a'), ('', '\\', 'a'))
148 self.assertEqual(f('\\\\\\a/b'), ('', '\\', 'a/b'))
149 self.assertEqual(f('c:\\\\a'), ('c:', '\\', 'a'))
150 self.assertEqual(f('c:\\\\\\a/b'), ('c:', '\\', 'a/b'))
151 # Valid UNC paths
152 self.assertEqual(f('\\\\a\\b'), ('\\\\a\\b', '\\', ''))
153 self.assertEqual(f('\\\\a\\b\\'), ('\\\\a\\b', '\\', ''))
154 self.assertEqual(f('\\\\a\\b\\c\\d'), ('\\\\a\\b', '\\', 'c\\d'))
155 # These are non-UNC paths (according to ntpath.py and test_ntpath)
156 # However, command.com says such paths are invalid, so it's
157 # difficult to know what the right semantics are
158 self.assertEqual(f('\\\\\\a\\b'), ('', '\\', 'a\\b'))
159 self.assertEqual(f('\\\\a'), ('', '\\', 'a'))
160
161
162#
163# Tests for the pure classes
164#
165
166class _BasePurePathTest(object):
167
168 # keys are canonical paths, values are list of tuples of arguments
169 # supposed to produce equal paths
170 equivalences = {
171 'a/b': [
172 ('a', 'b'), ('a/', 'b'), ('a', 'b/'), ('a/', 'b/'),
173 ('a/b/',), ('a//b',), ('a//b//',),
174 # empty components get removed
175 ('', 'a', 'b'), ('a', '', 'b'), ('a', 'b', ''),
176 ],
177 '/b/c/d': [
178 ('a', '/b/c', 'd'), ('a', '///b//c', 'd/'),
179 ('/a', '/b/c', 'd'),
180 # empty components get removed
181 ('/', 'b', '', 'c/d'), ('/', '', 'b/c/d'), ('', '/b/c/d'),
182 ],
183 }
184
185 def setUp(self):
186 p = self.cls('a')
187 self.flavour = p._flavour
188 self.sep = self.flavour.sep
189 self.altsep = self.flavour.altsep
190
191 def test_constructor_common(self):
192 P = self.cls
193 p = P('a')
194 self.assertIsInstance(p, P)
Brett Cannon568be632016-06-10 12:20:49 -0700195 class PathLike:
196 def __fspath__(self):
197 return "a/b/c"
Antoine Pitrou31119e42013-11-22 17:38:12 +0100198 P('a', 'b', 'c')
199 P('/a', 'b', 'c')
200 P('a/b/c')
201 P('/a/b/c')
Brett Cannon568be632016-06-10 12:20:49 -0700202 P(PathLike())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100203 self.assertEqual(P(P('a')), P('a'))
204 self.assertEqual(P(P('a'), 'b'), P('a/b'))
205 self.assertEqual(P(P('a'), P('b')), P('a/b'))
Brett Cannon568be632016-06-10 12:20:49 -0700206 self.assertEqual(P(P('a'), P('b'), P('c')), P(PathLike()))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100207
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200208 def _check_str_subclass(self, *args):
209 # Issue #21127: it should be possible to construct a PurePath object
Serhiy Storchaka6a7b3a72016-04-17 08:32:47 +0300210 # from a str subclass instance, and it then gets converted to
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200211 # a pure str object.
212 class StrSubclass(str):
213 pass
214 P = self.cls
215 p = P(*(StrSubclass(x) for x in args))
216 self.assertEqual(p, P(*args))
217 for part in p.parts:
218 self.assertIs(type(part), str)
219
220 def test_str_subclass_common(self):
221 self._check_str_subclass('')
222 self._check_str_subclass('.')
223 self._check_str_subclass('a')
224 self._check_str_subclass('a/b.txt')
225 self._check_str_subclass('/a/b.txt')
226
Antoine Pitrou31119e42013-11-22 17:38:12 +0100227 def test_join_common(self):
228 P = self.cls
229 p = P('a/b')
230 pp = p.joinpath('c')
231 self.assertEqual(pp, P('a/b/c'))
232 self.assertIs(type(pp), type(p))
233 pp = p.joinpath('c', 'd')
234 self.assertEqual(pp, P('a/b/c/d'))
235 pp = p.joinpath(P('c'))
236 self.assertEqual(pp, P('a/b/c'))
237 pp = p.joinpath('/c')
238 self.assertEqual(pp, P('/c'))
239
240 def test_div_common(self):
241 # Basically the same as joinpath()
242 P = self.cls
243 p = P('a/b')
244 pp = p / 'c'
245 self.assertEqual(pp, P('a/b/c'))
246 self.assertIs(type(pp), type(p))
247 pp = p / 'c/d'
248 self.assertEqual(pp, P('a/b/c/d'))
249 pp = p / 'c' / 'd'
250 self.assertEqual(pp, P('a/b/c/d'))
251 pp = 'c' / p / 'd'
252 self.assertEqual(pp, P('c/a/b/d'))
253 pp = p / P('c')
254 self.assertEqual(pp, P('a/b/c'))
255 pp = p/ '/c'
256 self.assertEqual(pp, P('/c'))
257
258 def _check_str(self, expected, args):
259 p = self.cls(*args)
260 self.assertEqual(str(p), expected.replace('/', self.sep))
261
262 def test_str_common(self):
263 # Canonicalized paths roundtrip
264 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
265 self._check_str(pathstr, (pathstr,))
266 # Special case for the empty path
267 self._check_str('.', ('',))
268 # Other tests for str() are in test_equivalences()
269
270 def test_as_posix_common(self):
271 P = self.cls
272 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
273 self.assertEqual(P(pathstr).as_posix(), pathstr)
274 # Other tests for as_posix() are in test_equivalences()
275
276 def test_as_bytes_common(self):
277 sep = os.fsencode(self.sep)
278 P = self.cls
279 self.assertEqual(bytes(P('a/b')), b'a' + sep + b'b')
280
281 def test_as_uri_common(self):
282 P = self.cls
283 with self.assertRaises(ValueError):
284 P('a').as_uri()
285 with self.assertRaises(ValueError):
286 P().as_uri()
287
288 def test_repr_common(self):
289 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
290 p = self.cls(pathstr)
291 clsname = p.__class__.__name__
292 r = repr(p)
293 # The repr() is in the form ClassName("forward-slashes path")
294 self.assertTrue(r.startswith(clsname + '('), r)
295 self.assertTrue(r.endswith(')'), r)
296 inner = r[len(clsname) + 1 : -1]
297 self.assertEqual(eval(inner), p.as_posix())
298 # The repr() roundtrips
299 q = eval(r, pathlib.__dict__)
300 self.assertIs(q.__class__, p.__class__)
301 self.assertEqual(q, p)
302 self.assertEqual(repr(q), r)
303
304 def test_eq_common(self):
305 P = self.cls
306 self.assertEqual(P('a/b'), P('a/b'))
307 self.assertEqual(P('a/b'), P('a', 'b'))
308 self.assertNotEqual(P('a/b'), P('a'))
309 self.assertNotEqual(P('a/b'), P('/a/b'))
310 self.assertNotEqual(P('a/b'), P())
311 self.assertNotEqual(P('/a/b'), P('/'))
312 self.assertNotEqual(P(), P('/'))
313 self.assertNotEqual(P(), "")
314 self.assertNotEqual(P(), {})
315 self.assertNotEqual(P(), int)
316
317 def test_match_common(self):
318 P = self.cls
319 self.assertRaises(ValueError, P('a').match, '')
320 self.assertRaises(ValueError, P('a').match, '.')
321 # Simple relative pattern
322 self.assertTrue(P('b.py').match('b.py'))
323 self.assertTrue(P('a/b.py').match('b.py'))
324 self.assertTrue(P('/a/b.py').match('b.py'))
325 self.assertFalse(P('a.py').match('b.py'))
326 self.assertFalse(P('b/py').match('b.py'))
327 self.assertFalse(P('/a.py').match('b.py'))
328 self.assertFalse(P('b.py/c').match('b.py'))
329 # Wilcard relative pattern
330 self.assertTrue(P('b.py').match('*.py'))
331 self.assertTrue(P('a/b.py').match('*.py'))
332 self.assertTrue(P('/a/b.py').match('*.py'))
333 self.assertFalse(P('b.pyc').match('*.py'))
334 self.assertFalse(P('b./py').match('*.py'))
335 self.assertFalse(P('b.py/c').match('*.py'))
336 # Multi-part relative pattern
337 self.assertTrue(P('ab/c.py').match('a*/*.py'))
338 self.assertTrue(P('/d/ab/c.py').match('a*/*.py'))
339 self.assertFalse(P('a.py').match('a*/*.py'))
340 self.assertFalse(P('/dab/c.py').match('a*/*.py'))
341 self.assertFalse(P('ab/c.py/d').match('a*/*.py'))
342 # Absolute pattern
343 self.assertTrue(P('/b.py').match('/*.py'))
344 self.assertFalse(P('b.py').match('/*.py'))
345 self.assertFalse(P('a/b.py').match('/*.py'))
346 self.assertFalse(P('/a/b.py').match('/*.py'))
347 # Multi-part absolute pattern
348 self.assertTrue(P('/a/b.py').match('/a/*.py'))
349 self.assertFalse(P('/ab.py').match('/a/*.py'))
350 self.assertFalse(P('/a/b/c.py').match('/a/*.py'))
351
352 def test_ordering_common(self):
353 # Ordering is tuple-alike
354 def assertLess(a, b):
355 self.assertLess(a, b)
356 self.assertGreater(b, a)
357 P = self.cls
358 a = P('a')
359 b = P('a/b')
360 c = P('abc')
361 d = P('b')
362 assertLess(a, b)
363 assertLess(a, c)
364 assertLess(a, d)
365 assertLess(b, c)
366 assertLess(c, d)
367 P = self.cls
368 a = P('/a')
369 b = P('/a/b')
370 c = P('/abc')
371 d = P('/b')
372 assertLess(a, b)
373 assertLess(a, c)
374 assertLess(a, d)
375 assertLess(b, c)
376 assertLess(c, d)
377 with self.assertRaises(TypeError):
378 P() < {}
379
380 def test_parts_common(self):
381 # `parts` returns a tuple
382 sep = self.sep
383 P = self.cls
384 p = P('a/b')
385 parts = p.parts
386 self.assertEqual(parts, ('a', 'b'))
387 # The object gets reused
388 self.assertIs(parts, p.parts)
389 # When the path is absolute, the anchor is a separate part
390 p = P('/a/b')
391 parts = p.parts
392 self.assertEqual(parts, (sep, 'a', 'b'))
393
Brett Cannon568be632016-06-10 12:20:49 -0700394 def test_fspath_common(self):
395 P = self.cls
396 p = P('a/b')
397 self._check_str(p.__fspath__(), ('a/b',))
398 self._check_str(os.fspath(p), ('a/b',))
399
Antoine Pitrou31119e42013-11-22 17:38:12 +0100400 def test_equivalences(self):
401 for k, tuples in self.equivalences.items():
402 canon = k.replace('/', self.sep)
403 posix = k.replace(self.sep, '/')
404 if canon != posix:
405 tuples = tuples + [
406 tuple(part.replace('/', self.sep) for part in t)
407 for t in tuples
408 ]
409 tuples.append((posix, ))
410 pcanon = self.cls(canon)
411 for t in tuples:
412 p = self.cls(*t)
413 self.assertEqual(p, pcanon, "failed with args {}".format(t))
414 self.assertEqual(hash(p), hash(pcanon))
415 self.assertEqual(str(p), canon)
416 self.assertEqual(p.as_posix(), posix)
417
418 def test_parent_common(self):
419 # Relative
420 P = self.cls
421 p = P('a/b/c')
422 self.assertEqual(p.parent, P('a/b'))
423 self.assertEqual(p.parent.parent, P('a'))
424 self.assertEqual(p.parent.parent.parent, P())
425 self.assertEqual(p.parent.parent.parent.parent, P())
426 # Anchored
427 p = P('/a/b/c')
428 self.assertEqual(p.parent, P('/a/b'))
429 self.assertEqual(p.parent.parent, P('/a'))
430 self.assertEqual(p.parent.parent.parent, P('/'))
431 self.assertEqual(p.parent.parent.parent.parent, P('/'))
432
433 def test_parents_common(self):
434 # Relative
435 P = self.cls
436 p = P('a/b/c')
437 par = p.parents
438 self.assertEqual(len(par), 3)
439 self.assertEqual(par[0], P('a/b'))
440 self.assertEqual(par[1], P('a'))
441 self.assertEqual(par[2], P('.'))
442 self.assertEqual(list(par), [P('a/b'), P('a'), P('.')])
443 with self.assertRaises(IndexError):
444 par[-1]
445 with self.assertRaises(IndexError):
446 par[3]
447 with self.assertRaises(TypeError):
448 par[0] = p
449 # Anchored
450 p = P('/a/b/c')
451 par = p.parents
452 self.assertEqual(len(par), 3)
453 self.assertEqual(par[0], P('/a/b'))
454 self.assertEqual(par[1], P('/a'))
455 self.assertEqual(par[2], P('/'))
456 self.assertEqual(list(par), [P('/a/b'), P('/a'), P('/')])
457 with self.assertRaises(IndexError):
458 par[3]
459
460 def test_drive_common(self):
461 P = self.cls
462 self.assertEqual(P('a/b').drive, '')
463 self.assertEqual(P('/a/b').drive, '')
464 self.assertEqual(P('').drive, '')
465
466 def test_root_common(self):
467 P = self.cls
468 sep = self.sep
469 self.assertEqual(P('').root, '')
470 self.assertEqual(P('a/b').root, '')
471 self.assertEqual(P('/').root, sep)
472 self.assertEqual(P('/a/b').root, sep)
473
474 def test_anchor_common(self):
475 P = self.cls
476 sep = self.sep
477 self.assertEqual(P('').anchor, '')
478 self.assertEqual(P('a/b').anchor, '')
479 self.assertEqual(P('/').anchor, sep)
480 self.assertEqual(P('/a/b').anchor, sep)
481
482 def test_name_common(self):
483 P = self.cls
484 self.assertEqual(P('').name, '')
485 self.assertEqual(P('.').name, '')
486 self.assertEqual(P('/').name, '')
487 self.assertEqual(P('a/b').name, 'b')
488 self.assertEqual(P('/a/b').name, 'b')
489 self.assertEqual(P('/a/b/.').name, 'b')
490 self.assertEqual(P('a/b.py').name, 'b.py')
491 self.assertEqual(P('/a/b.py').name, 'b.py')
492
493 def test_suffix_common(self):
494 P = self.cls
495 self.assertEqual(P('').suffix, '')
496 self.assertEqual(P('.').suffix, '')
497 self.assertEqual(P('..').suffix, '')
498 self.assertEqual(P('/').suffix, '')
499 self.assertEqual(P('a/b').suffix, '')
500 self.assertEqual(P('/a/b').suffix, '')
501 self.assertEqual(P('/a/b/.').suffix, '')
502 self.assertEqual(P('a/b.py').suffix, '.py')
503 self.assertEqual(P('/a/b.py').suffix, '.py')
504 self.assertEqual(P('a/.hgrc').suffix, '')
505 self.assertEqual(P('/a/.hgrc').suffix, '')
506 self.assertEqual(P('a/.hg.rc').suffix, '.rc')
507 self.assertEqual(P('/a/.hg.rc').suffix, '.rc')
508 self.assertEqual(P('a/b.tar.gz').suffix, '.gz')
509 self.assertEqual(P('/a/b.tar.gz').suffix, '.gz')
510 self.assertEqual(P('a/Some name. Ending with a dot.').suffix, '')
511 self.assertEqual(P('/a/Some name. Ending with a dot.').suffix, '')
512
513 def test_suffixes_common(self):
514 P = self.cls
515 self.assertEqual(P('').suffixes, [])
516 self.assertEqual(P('.').suffixes, [])
517 self.assertEqual(P('/').suffixes, [])
518 self.assertEqual(P('a/b').suffixes, [])
519 self.assertEqual(P('/a/b').suffixes, [])
520 self.assertEqual(P('/a/b/.').suffixes, [])
521 self.assertEqual(P('a/b.py').suffixes, ['.py'])
522 self.assertEqual(P('/a/b.py').suffixes, ['.py'])
523 self.assertEqual(P('a/.hgrc').suffixes, [])
524 self.assertEqual(P('/a/.hgrc').suffixes, [])
525 self.assertEqual(P('a/.hg.rc').suffixes, ['.rc'])
526 self.assertEqual(P('/a/.hg.rc').suffixes, ['.rc'])
527 self.assertEqual(P('a/b.tar.gz').suffixes, ['.tar', '.gz'])
528 self.assertEqual(P('/a/b.tar.gz').suffixes, ['.tar', '.gz'])
529 self.assertEqual(P('a/Some name. Ending with a dot.').suffixes, [])
530 self.assertEqual(P('/a/Some name. Ending with a dot.').suffixes, [])
531
532 def test_stem_common(self):
533 P = self.cls
534 self.assertEqual(P('').stem, '')
535 self.assertEqual(P('.').stem, '')
536 self.assertEqual(P('..').stem, '..')
537 self.assertEqual(P('/').stem, '')
538 self.assertEqual(P('a/b').stem, 'b')
539 self.assertEqual(P('a/b.py').stem, 'b')
540 self.assertEqual(P('a/.hgrc').stem, '.hgrc')
541 self.assertEqual(P('a/.hg.rc').stem, '.hg')
542 self.assertEqual(P('a/b.tar.gz').stem, 'b.tar')
543 self.assertEqual(P('a/Some name. Ending with a dot.').stem,
544 'Some name. Ending with a dot.')
545
546 def test_with_name_common(self):
547 P = self.cls
548 self.assertEqual(P('a/b').with_name('d.xml'), P('a/d.xml'))
549 self.assertEqual(P('/a/b').with_name('d.xml'), P('/a/d.xml'))
550 self.assertEqual(P('a/b.py').with_name('d.xml'), P('a/d.xml'))
551 self.assertEqual(P('/a/b.py').with_name('d.xml'), P('/a/d.xml'))
552 self.assertEqual(P('a/Dot ending.').with_name('d.xml'), P('a/d.xml'))
553 self.assertEqual(P('/a/Dot ending.').with_name('d.xml'), P('/a/d.xml'))
554 self.assertRaises(ValueError, P('').with_name, 'd.xml')
555 self.assertRaises(ValueError, P('.').with_name, 'd.xml')
556 self.assertRaises(ValueError, P('/').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400557 self.assertRaises(ValueError, P('a/b').with_name, '')
558 self.assertRaises(ValueError, P('a/b').with_name, '/c')
559 self.assertRaises(ValueError, P('a/b').with_name, 'c/')
560 self.assertRaises(ValueError, P('a/b').with_name, 'c/d')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100561
562 def test_with_suffix_common(self):
563 P = self.cls
564 self.assertEqual(P('a/b').with_suffix('.gz'), P('a/b.gz'))
565 self.assertEqual(P('/a/b').with_suffix('.gz'), P('/a/b.gz'))
566 self.assertEqual(P('a/b.py').with_suffix('.gz'), P('a/b.gz'))
567 self.assertEqual(P('/a/b.py').with_suffix('.gz'), P('/a/b.gz'))
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400568 # Stripping suffix
569 self.assertEqual(P('a/b.py').with_suffix(''), P('a/b'))
570 self.assertEqual(P('/a/b').with_suffix(''), P('/a/b'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100571 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100572 self.assertRaises(ValueError, P('').with_suffix, '.gz')
573 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
574 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100575 # Invalid suffix
576 self.assertRaises(ValueError, P('a/b').with_suffix, 'gz')
577 self.assertRaises(ValueError, P('a/b').with_suffix, '/')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400578 self.assertRaises(ValueError, P('a/b').with_suffix, '.')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100579 self.assertRaises(ValueError, P('a/b').with_suffix, '/.gz')
580 self.assertRaises(ValueError, P('a/b').with_suffix, 'c/d')
581 self.assertRaises(ValueError, P('a/b').with_suffix, '.c/.d')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400582 self.assertRaises(ValueError, P('a/b').with_suffix, './.d')
583 self.assertRaises(ValueError, P('a/b').with_suffix, '.d/.')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100584
585 def test_relative_to_common(self):
586 P = self.cls
587 p = P('a/b')
588 self.assertRaises(TypeError, p.relative_to)
Antoine Pitrou156b3612013-12-28 19:49:04 +0100589 self.assertRaises(TypeError, p.relative_to, b'a')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100590 self.assertEqual(p.relative_to(P()), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100591 self.assertEqual(p.relative_to(''), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100592 self.assertEqual(p.relative_to(P('a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100593 self.assertEqual(p.relative_to('a'), P('b'))
594 self.assertEqual(p.relative_to('a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100595 self.assertEqual(p.relative_to(P('a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100596 self.assertEqual(p.relative_to('a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100597 # With several args
598 self.assertEqual(p.relative_to('a', 'b'), P())
599 # Unrelated paths
600 self.assertRaises(ValueError, p.relative_to, P('c'))
601 self.assertRaises(ValueError, p.relative_to, P('a/b/c'))
602 self.assertRaises(ValueError, p.relative_to, P('a/c'))
603 self.assertRaises(ValueError, p.relative_to, P('/a'))
604 p = P('/a/b')
605 self.assertEqual(p.relative_to(P('/')), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100606 self.assertEqual(p.relative_to('/'), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100607 self.assertEqual(p.relative_to(P('/a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100608 self.assertEqual(p.relative_to('/a'), P('b'))
609 self.assertEqual(p.relative_to('/a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100610 self.assertEqual(p.relative_to(P('/a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100611 self.assertEqual(p.relative_to('/a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100612 # Unrelated paths
613 self.assertRaises(ValueError, p.relative_to, P('/c'))
614 self.assertRaises(ValueError, p.relative_to, P('/a/b/c'))
615 self.assertRaises(ValueError, p.relative_to, P('/a/c'))
616 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100617 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100618 self.assertRaises(ValueError, p.relative_to, P('a'))
619
620 def test_pickling_common(self):
621 P = self.cls
622 p = P('/a/b')
623 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
624 dumped = pickle.dumps(p, proto)
625 pp = pickle.loads(dumped)
626 self.assertIs(pp.__class__, p.__class__)
627 self.assertEqual(pp, p)
628 self.assertEqual(hash(pp), hash(p))
629 self.assertEqual(str(pp), str(p))
630
631
632class PurePosixPathTest(_BasePurePathTest, unittest.TestCase):
633 cls = pathlib.PurePosixPath
634
635 def test_root(self):
636 P = self.cls
637 self.assertEqual(P('/a/b').root, '/')
638 self.assertEqual(P('///a/b').root, '/')
639 # POSIX special case for two leading slashes
640 self.assertEqual(P('//a/b').root, '//')
641
642 def test_eq(self):
643 P = self.cls
644 self.assertNotEqual(P('a/b'), P('A/b'))
645 self.assertEqual(P('/a'), P('///a'))
646 self.assertNotEqual(P('/a'), P('//a'))
647
648 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100649 P = self.cls
650 self.assertEqual(P('/').as_uri(), 'file:///')
651 self.assertEqual(P('/a/b.c').as_uri(), 'file:///a/b.c')
652 self.assertEqual(P('/a/b%#c').as_uri(), 'file:///a/b%25%23c')
Antoine Pitrou29eac422013-11-22 17:57:03 +0100653
654 def test_as_uri_non_ascii(self):
655 from urllib.parse import quote_from_bytes
656 P = self.cls
657 try:
658 os.fsencode('\xe9')
659 except UnicodeEncodeError:
660 self.skipTest("\\xe9 cannot be encoded to the filesystem encoding")
Antoine Pitrou31119e42013-11-22 17:38:12 +0100661 self.assertEqual(P('/a/b\xe9').as_uri(),
662 'file:///a/b' + quote_from_bytes(os.fsencode('\xe9')))
663
664 def test_match(self):
665 P = self.cls
666 self.assertFalse(P('A.py').match('a.PY'))
667
668 def test_is_absolute(self):
669 P = self.cls
670 self.assertFalse(P().is_absolute())
671 self.assertFalse(P('a').is_absolute())
672 self.assertFalse(P('a/b/').is_absolute())
673 self.assertTrue(P('/').is_absolute())
674 self.assertTrue(P('/a').is_absolute())
675 self.assertTrue(P('/a/b/').is_absolute())
676 self.assertTrue(P('//a').is_absolute())
677 self.assertTrue(P('//a/b').is_absolute())
678
679 def test_is_reserved(self):
680 P = self.cls
681 self.assertIs(False, P('').is_reserved())
682 self.assertIs(False, P('/').is_reserved())
683 self.assertIs(False, P('/foo/bar').is_reserved())
684 self.assertIs(False, P('/dev/con/PRN/NUL').is_reserved())
685
686 def test_join(self):
687 P = self.cls
688 p = P('//a')
689 pp = p.joinpath('b')
690 self.assertEqual(pp, P('//a/b'))
691 pp = P('/a').joinpath('//c')
692 self.assertEqual(pp, P('//c'))
693 pp = P('//a').joinpath('/c')
694 self.assertEqual(pp, P('/c'))
695
696 def test_div(self):
697 # Basically the same as joinpath()
698 P = self.cls
699 p = P('//a')
700 pp = p / 'b'
701 self.assertEqual(pp, P('//a/b'))
702 pp = P('/a') / '//c'
703 self.assertEqual(pp, P('//c'))
704 pp = P('//a') / '/c'
705 self.assertEqual(pp, P('/c'))
706
707
708class PureWindowsPathTest(_BasePurePathTest, unittest.TestCase):
709 cls = pathlib.PureWindowsPath
710
711 equivalences = _BasePurePathTest.equivalences.copy()
712 equivalences.update({
713 'c:a': [ ('c:', 'a'), ('c:', 'a/'), ('/', 'c:', 'a') ],
714 'c:/a': [
715 ('c:/', 'a'), ('c:', '/', 'a'), ('c:', '/a'),
716 ('/z', 'c:/', 'a'), ('//x/y', 'c:/', 'a'),
717 ],
718 '//a/b/': [ ('//a/b',) ],
719 '//a/b/c': [
720 ('//a/b', 'c'), ('//a/b/', 'c'),
721 ],
722 })
723
724 def test_str(self):
725 p = self.cls('a/b/c')
726 self.assertEqual(str(p), 'a\\b\\c')
727 p = self.cls('c:/a/b/c')
728 self.assertEqual(str(p), 'c:\\a\\b\\c')
729 p = self.cls('//a/b')
730 self.assertEqual(str(p), '\\\\a\\b\\')
731 p = self.cls('//a/b/c')
732 self.assertEqual(str(p), '\\\\a\\b\\c')
733 p = self.cls('//a/b/c/d')
734 self.assertEqual(str(p), '\\\\a\\b\\c\\d')
735
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200736 def test_str_subclass(self):
737 self._check_str_subclass('c:')
738 self._check_str_subclass('c:a')
739 self._check_str_subclass('c:a\\b.txt')
740 self._check_str_subclass('c:\\')
741 self._check_str_subclass('c:\\a')
742 self._check_str_subclass('c:\\a\\b.txt')
743 self._check_str_subclass('\\\\some\\share')
744 self._check_str_subclass('\\\\some\\share\\a')
745 self._check_str_subclass('\\\\some\\share\\a\\b.txt')
746
Antoine Pitrou31119e42013-11-22 17:38:12 +0100747 def test_eq(self):
748 P = self.cls
749 self.assertEqual(P('c:a/b'), P('c:a/b'))
750 self.assertEqual(P('c:a/b'), P('c:', 'a', 'b'))
751 self.assertNotEqual(P('c:a/b'), P('d:a/b'))
752 self.assertNotEqual(P('c:a/b'), P('c:/a/b'))
753 self.assertNotEqual(P('/a/b'), P('c:/a/b'))
754 # Case-insensitivity
755 self.assertEqual(P('a/B'), P('A/b'))
756 self.assertEqual(P('C:a/B'), P('c:A/b'))
757 self.assertEqual(P('//Some/SHARE/a/B'), P('//somE/share/A/b'))
758
759 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100760 P = self.cls
761 with self.assertRaises(ValueError):
762 P('/a/b').as_uri()
763 with self.assertRaises(ValueError):
764 P('c:a/b').as_uri()
765 self.assertEqual(P('c:/').as_uri(), 'file:///c:/')
766 self.assertEqual(P('c:/a/b.c').as_uri(), 'file:///c:/a/b.c')
767 self.assertEqual(P('c:/a/b%#c').as_uri(), 'file:///c:/a/b%25%23c')
768 self.assertEqual(P('c:/a/b\xe9').as_uri(), 'file:///c:/a/b%C3%A9')
769 self.assertEqual(P('//some/share/').as_uri(), 'file://some/share/')
770 self.assertEqual(P('//some/share/a/b.c').as_uri(),
771 'file://some/share/a/b.c')
772 self.assertEqual(P('//some/share/a/b%#c\xe9').as_uri(),
773 'file://some/share/a/b%25%23c%C3%A9')
774
775 def test_match_common(self):
776 P = self.cls
777 # Absolute patterns
778 self.assertTrue(P('c:/b.py').match('/*.py'))
779 self.assertTrue(P('c:/b.py').match('c:*.py'))
780 self.assertTrue(P('c:/b.py').match('c:/*.py'))
781 self.assertFalse(P('d:/b.py').match('c:/*.py')) # wrong drive
782 self.assertFalse(P('b.py').match('/*.py'))
783 self.assertFalse(P('b.py').match('c:*.py'))
784 self.assertFalse(P('b.py').match('c:/*.py'))
785 self.assertFalse(P('c:b.py').match('/*.py'))
786 self.assertFalse(P('c:b.py').match('c:/*.py'))
787 self.assertFalse(P('/b.py').match('c:*.py'))
788 self.assertFalse(P('/b.py').match('c:/*.py'))
789 # UNC patterns
790 self.assertTrue(P('//some/share/a.py').match('/*.py'))
791 self.assertTrue(P('//some/share/a.py').match('//some/share/*.py'))
792 self.assertFalse(P('//other/share/a.py').match('//some/share/*.py'))
793 self.assertFalse(P('//some/share/a/b.py').match('//some/share/*.py'))
794 # Case-insensitivity
795 self.assertTrue(P('B.py').match('b.PY'))
796 self.assertTrue(P('c:/a/B.Py').match('C:/A/*.pY'))
797 self.assertTrue(P('//Some/Share/B.Py').match('//somE/sharE/*.pY'))
798
799 def test_ordering_common(self):
800 # Case-insensitivity
801 def assertOrderedEqual(a, b):
802 self.assertLessEqual(a, b)
803 self.assertGreaterEqual(b, a)
804 P = self.cls
805 p = P('c:A/b')
806 q = P('C:a/B')
807 assertOrderedEqual(p, q)
808 self.assertFalse(p < q)
809 self.assertFalse(p > q)
810 p = P('//some/Share/A/b')
811 q = P('//Some/SHARE/a/B')
812 assertOrderedEqual(p, q)
813 self.assertFalse(p < q)
814 self.assertFalse(p > q)
815
816 def test_parts(self):
817 P = self.cls
818 p = P('c:a/b')
819 parts = p.parts
820 self.assertEqual(parts, ('c:', 'a', 'b'))
821 p = P('c:/a/b')
822 parts = p.parts
823 self.assertEqual(parts, ('c:\\', 'a', 'b'))
824 p = P('//a/b/c/d')
825 parts = p.parts
826 self.assertEqual(parts, ('\\\\a\\b\\', 'c', 'd'))
827
828 def test_parent(self):
829 # Anchored
830 P = self.cls
831 p = P('z:a/b/c')
832 self.assertEqual(p.parent, P('z:a/b'))
833 self.assertEqual(p.parent.parent, P('z:a'))
834 self.assertEqual(p.parent.parent.parent, P('z:'))
835 self.assertEqual(p.parent.parent.parent.parent, P('z:'))
836 p = P('z:/a/b/c')
837 self.assertEqual(p.parent, P('z:/a/b'))
838 self.assertEqual(p.parent.parent, P('z:/a'))
839 self.assertEqual(p.parent.parent.parent, P('z:/'))
840 self.assertEqual(p.parent.parent.parent.parent, P('z:/'))
841 p = P('//a/b/c/d')
842 self.assertEqual(p.parent, P('//a/b/c'))
843 self.assertEqual(p.parent.parent, P('//a/b'))
844 self.assertEqual(p.parent.parent.parent, P('//a/b'))
845
846 def test_parents(self):
847 # Anchored
848 P = self.cls
849 p = P('z:a/b/')
850 par = p.parents
851 self.assertEqual(len(par), 2)
852 self.assertEqual(par[0], P('z:a'))
853 self.assertEqual(par[1], P('z:'))
854 self.assertEqual(list(par), [P('z:a'), P('z:')])
855 with self.assertRaises(IndexError):
856 par[2]
857 p = P('z:/a/b/')
858 par = p.parents
859 self.assertEqual(len(par), 2)
860 self.assertEqual(par[0], P('z:/a'))
861 self.assertEqual(par[1], P('z:/'))
862 self.assertEqual(list(par), [P('z:/a'), P('z:/')])
863 with self.assertRaises(IndexError):
864 par[2]
865 p = P('//a/b/c/d')
866 par = p.parents
867 self.assertEqual(len(par), 2)
868 self.assertEqual(par[0], P('//a/b/c'))
869 self.assertEqual(par[1], P('//a/b'))
870 self.assertEqual(list(par), [P('//a/b/c'), P('//a/b')])
871 with self.assertRaises(IndexError):
872 par[2]
873
874 def test_drive(self):
875 P = self.cls
876 self.assertEqual(P('c:').drive, 'c:')
877 self.assertEqual(P('c:a/b').drive, 'c:')
878 self.assertEqual(P('c:/').drive, 'c:')
879 self.assertEqual(P('c:/a/b/').drive, 'c:')
880 self.assertEqual(P('//a/b').drive, '\\\\a\\b')
881 self.assertEqual(P('//a/b/').drive, '\\\\a\\b')
882 self.assertEqual(P('//a/b/c/d').drive, '\\\\a\\b')
883
884 def test_root(self):
885 P = self.cls
886 self.assertEqual(P('c:').root, '')
887 self.assertEqual(P('c:a/b').root, '')
888 self.assertEqual(P('c:/').root, '\\')
889 self.assertEqual(P('c:/a/b/').root, '\\')
890 self.assertEqual(P('//a/b').root, '\\')
891 self.assertEqual(P('//a/b/').root, '\\')
892 self.assertEqual(P('//a/b/c/d').root, '\\')
893
894 def test_anchor(self):
895 P = self.cls
896 self.assertEqual(P('c:').anchor, 'c:')
897 self.assertEqual(P('c:a/b').anchor, 'c:')
898 self.assertEqual(P('c:/').anchor, 'c:\\')
899 self.assertEqual(P('c:/a/b/').anchor, 'c:\\')
900 self.assertEqual(P('//a/b').anchor, '\\\\a\\b\\')
901 self.assertEqual(P('//a/b/').anchor, '\\\\a\\b\\')
902 self.assertEqual(P('//a/b/c/d').anchor, '\\\\a\\b\\')
903
904 def test_name(self):
905 P = self.cls
906 self.assertEqual(P('c:').name, '')
907 self.assertEqual(P('c:/').name, '')
908 self.assertEqual(P('c:a/b').name, 'b')
909 self.assertEqual(P('c:/a/b').name, 'b')
910 self.assertEqual(P('c:a/b.py').name, 'b.py')
911 self.assertEqual(P('c:/a/b.py').name, 'b.py')
912 self.assertEqual(P('//My.py/Share.php').name, '')
913 self.assertEqual(P('//My.py/Share.php/a/b').name, 'b')
914
915 def test_suffix(self):
916 P = self.cls
917 self.assertEqual(P('c:').suffix, '')
918 self.assertEqual(P('c:/').suffix, '')
919 self.assertEqual(P('c:a/b').suffix, '')
920 self.assertEqual(P('c:/a/b').suffix, '')
921 self.assertEqual(P('c:a/b.py').suffix, '.py')
922 self.assertEqual(P('c:/a/b.py').suffix, '.py')
923 self.assertEqual(P('c:a/.hgrc').suffix, '')
924 self.assertEqual(P('c:/a/.hgrc').suffix, '')
925 self.assertEqual(P('c:a/.hg.rc').suffix, '.rc')
926 self.assertEqual(P('c:/a/.hg.rc').suffix, '.rc')
927 self.assertEqual(P('c:a/b.tar.gz').suffix, '.gz')
928 self.assertEqual(P('c:/a/b.tar.gz').suffix, '.gz')
929 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffix, '')
930 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffix, '')
931 self.assertEqual(P('//My.py/Share.php').suffix, '')
932 self.assertEqual(P('//My.py/Share.php/a/b').suffix, '')
933
934 def test_suffixes(self):
935 P = self.cls
936 self.assertEqual(P('c:').suffixes, [])
937 self.assertEqual(P('c:/').suffixes, [])
938 self.assertEqual(P('c:a/b').suffixes, [])
939 self.assertEqual(P('c:/a/b').suffixes, [])
940 self.assertEqual(P('c:a/b.py').suffixes, ['.py'])
941 self.assertEqual(P('c:/a/b.py').suffixes, ['.py'])
942 self.assertEqual(P('c:a/.hgrc').suffixes, [])
943 self.assertEqual(P('c:/a/.hgrc').suffixes, [])
944 self.assertEqual(P('c:a/.hg.rc').suffixes, ['.rc'])
945 self.assertEqual(P('c:/a/.hg.rc').suffixes, ['.rc'])
946 self.assertEqual(P('c:a/b.tar.gz').suffixes, ['.tar', '.gz'])
947 self.assertEqual(P('c:/a/b.tar.gz').suffixes, ['.tar', '.gz'])
948 self.assertEqual(P('//My.py/Share.php').suffixes, [])
949 self.assertEqual(P('//My.py/Share.php/a/b').suffixes, [])
950 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffixes, [])
951 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffixes, [])
952
953 def test_stem(self):
954 P = self.cls
955 self.assertEqual(P('c:').stem, '')
956 self.assertEqual(P('c:.').stem, '')
957 self.assertEqual(P('c:..').stem, '..')
958 self.assertEqual(P('c:/').stem, '')
959 self.assertEqual(P('c:a/b').stem, 'b')
960 self.assertEqual(P('c:a/b.py').stem, 'b')
961 self.assertEqual(P('c:a/.hgrc').stem, '.hgrc')
962 self.assertEqual(P('c:a/.hg.rc').stem, '.hg')
963 self.assertEqual(P('c:a/b.tar.gz').stem, 'b.tar')
964 self.assertEqual(P('c:a/Some name. Ending with a dot.').stem,
965 'Some name. Ending with a dot.')
966
967 def test_with_name(self):
968 P = self.cls
969 self.assertEqual(P('c:a/b').with_name('d.xml'), P('c:a/d.xml'))
970 self.assertEqual(P('c:/a/b').with_name('d.xml'), P('c:/a/d.xml'))
971 self.assertEqual(P('c:a/Dot ending.').with_name('d.xml'), P('c:a/d.xml'))
972 self.assertEqual(P('c:/a/Dot ending.').with_name('d.xml'), P('c:/a/d.xml'))
973 self.assertRaises(ValueError, P('c:').with_name, 'd.xml')
974 self.assertRaises(ValueError, P('c:/').with_name, 'd.xml')
975 self.assertRaises(ValueError, P('//My/Share').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400976 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:')
977 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:e')
978 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:/e')
979 self.assertRaises(ValueError, P('c:a/b').with_name, '//My/Share')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100980
981 def test_with_suffix(self):
982 P = self.cls
983 self.assertEqual(P('c:a/b').with_suffix('.gz'), P('c:a/b.gz'))
984 self.assertEqual(P('c:/a/b').with_suffix('.gz'), P('c:/a/b.gz'))
985 self.assertEqual(P('c:a/b.py').with_suffix('.gz'), P('c:a/b.gz'))
986 self.assertEqual(P('c:/a/b.py').with_suffix('.gz'), P('c:/a/b.gz'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100987 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100988 self.assertRaises(ValueError, P('').with_suffix, '.gz')
989 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
990 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
991 self.assertRaises(ValueError, P('//My/Share').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100992 # Invalid suffix
993 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'gz')
994 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/')
995 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\')
996 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:')
997 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/.gz')
998 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\.gz')
999 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:.gz')
1000 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c/d')
1001 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c\\d')
1002 self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c/d')
1003 self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c\\d')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001004
1005 def test_relative_to(self):
1006 P = self.cls
Antoine Pitrou156b3612013-12-28 19:49:04 +01001007 p = P('C:Foo/Bar')
1008 self.assertEqual(p.relative_to(P('c:')), P('Foo/Bar'))
1009 self.assertEqual(p.relative_to('c:'), P('Foo/Bar'))
1010 self.assertEqual(p.relative_to(P('c:foO')), P('Bar'))
1011 self.assertEqual(p.relative_to('c:foO'), P('Bar'))
1012 self.assertEqual(p.relative_to('c:foO/'), P('Bar'))
1013 self.assertEqual(p.relative_to(P('c:foO/baR')), P())
1014 self.assertEqual(p.relative_to('c:foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001015 # Unrelated paths
1016 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +01001017 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001018 self.assertRaises(ValueError, p.relative_to, P('d:'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001019 self.assertRaises(ValueError, p.relative_to, P('/'))
1020 self.assertRaises(ValueError, p.relative_to, P('Foo'))
1021 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1022 self.assertRaises(ValueError, p.relative_to, P('C:/Foo'))
1023 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Bar/Baz'))
1024 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Baz'))
1025 p = P('C:/Foo/Bar')
1026 self.assertEqual(p.relative_to(P('c:')), P('/Foo/Bar'))
1027 self.assertEqual(p.relative_to('c:'), P('/Foo/Bar'))
1028 self.assertEqual(str(p.relative_to(P('c:'))), '\\Foo\\Bar')
1029 self.assertEqual(str(p.relative_to('c:')), '\\Foo\\Bar')
1030 self.assertEqual(p.relative_to(P('c:/')), P('Foo/Bar'))
1031 self.assertEqual(p.relative_to('c:/'), P('Foo/Bar'))
1032 self.assertEqual(p.relative_to(P('c:/foO')), P('Bar'))
1033 self.assertEqual(p.relative_to('c:/foO'), P('Bar'))
1034 self.assertEqual(p.relative_to('c:/foO/'), P('Bar'))
1035 self.assertEqual(p.relative_to(P('c:/foO/baR')), P())
1036 self.assertEqual(p.relative_to('c:/foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001037 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001038 self.assertRaises(ValueError, p.relative_to, P('C:/Baz'))
1039 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Bar/Baz'))
1040 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Baz'))
1041 self.assertRaises(ValueError, p.relative_to, P('C:Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001042 self.assertRaises(ValueError, p.relative_to, P('d:'))
1043 self.assertRaises(ValueError, p.relative_to, P('d:/'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001044 self.assertRaises(ValueError, p.relative_to, P('/'))
1045 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1046 self.assertRaises(ValueError, p.relative_to, P('//C/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001047 # UNC paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001048 p = P('//Server/Share/Foo/Bar')
1049 self.assertEqual(p.relative_to(P('//sErver/sHare')), P('Foo/Bar'))
1050 self.assertEqual(p.relative_to('//sErver/sHare'), P('Foo/Bar'))
1051 self.assertEqual(p.relative_to('//sErver/sHare/'), P('Foo/Bar'))
1052 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo')), P('Bar'))
1053 self.assertEqual(p.relative_to('//sErver/sHare/Foo'), P('Bar'))
1054 self.assertEqual(p.relative_to('//sErver/sHare/Foo/'), P('Bar'))
1055 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo/Bar')), P())
1056 self.assertEqual(p.relative_to('//sErver/sHare/Foo/Bar'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001057 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001058 self.assertRaises(ValueError, p.relative_to, P('/Server/Share/Foo'))
1059 self.assertRaises(ValueError, p.relative_to, P('c:/Server/Share/Foo'))
1060 self.assertRaises(ValueError, p.relative_to, P('//z/Share/Foo'))
1061 self.assertRaises(ValueError, p.relative_to, P('//Server/z/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001062
1063 def test_is_absolute(self):
1064 P = self.cls
1065 # Under NT, only paths with both a drive and a root are absolute
1066 self.assertFalse(P().is_absolute())
1067 self.assertFalse(P('a').is_absolute())
1068 self.assertFalse(P('a/b/').is_absolute())
1069 self.assertFalse(P('/').is_absolute())
1070 self.assertFalse(P('/a').is_absolute())
1071 self.assertFalse(P('/a/b/').is_absolute())
1072 self.assertFalse(P('c:').is_absolute())
1073 self.assertFalse(P('c:a').is_absolute())
1074 self.assertFalse(P('c:a/b/').is_absolute())
1075 self.assertTrue(P('c:/').is_absolute())
1076 self.assertTrue(P('c:/a').is_absolute())
1077 self.assertTrue(P('c:/a/b/').is_absolute())
1078 # UNC paths are absolute by definition
1079 self.assertTrue(P('//a/b').is_absolute())
1080 self.assertTrue(P('//a/b/').is_absolute())
1081 self.assertTrue(P('//a/b/c').is_absolute())
1082 self.assertTrue(P('//a/b/c/d').is_absolute())
1083
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001084 def test_join(self):
1085 P = self.cls
1086 p = P('C:/a/b')
1087 pp = p.joinpath('x/y')
1088 self.assertEqual(pp, P('C:/a/b/x/y'))
1089 pp = p.joinpath('/x/y')
1090 self.assertEqual(pp, P('C:/x/y'))
1091 # Joining with a different drive => the first path is ignored, even
1092 # if the second path is relative.
1093 pp = p.joinpath('D:x/y')
1094 self.assertEqual(pp, P('D:x/y'))
1095 pp = p.joinpath('D:/x/y')
1096 self.assertEqual(pp, P('D:/x/y'))
1097 pp = p.joinpath('//host/share/x/y')
1098 self.assertEqual(pp, P('//host/share/x/y'))
1099 # Joining with the same drive => the first path is appended to if
1100 # the second path is relative.
1101 pp = p.joinpath('c:x/y')
1102 self.assertEqual(pp, P('C:/a/b/x/y'))
1103 pp = p.joinpath('c:/x/y')
1104 self.assertEqual(pp, P('C:/x/y'))
1105
1106 def test_div(self):
1107 # Basically the same as joinpath()
1108 P = self.cls
1109 p = P('C:/a/b')
1110 self.assertEqual(p / 'x/y', P('C:/a/b/x/y'))
1111 self.assertEqual(p / 'x' / 'y', P('C:/a/b/x/y'))
1112 self.assertEqual(p / '/x/y', P('C:/x/y'))
1113 self.assertEqual(p / '/x' / 'y', P('C:/x/y'))
1114 # Joining with a different drive => the first path is ignored, even
1115 # if the second path is relative.
1116 self.assertEqual(p / 'D:x/y', P('D:x/y'))
1117 self.assertEqual(p / 'D:' / 'x/y', P('D:x/y'))
1118 self.assertEqual(p / 'D:/x/y', P('D:/x/y'))
1119 self.assertEqual(p / 'D:' / '/x/y', P('D:/x/y'))
1120 self.assertEqual(p / '//host/share/x/y', P('//host/share/x/y'))
1121 # Joining with the same drive => the first path is appended to if
1122 # the second path is relative.
Serhiy Storchaka010ff582013-12-06 17:25:51 +02001123 self.assertEqual(p / 'c:x/y', P('C:/a/b/x/y'))
1124 self.assertEqual(p / 'c:/x/y', P('C:/x/y'))
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001125
Antoine Pitrou31119e42013-11-22 17:38:12 +01001126 def test_is_reserved(self):
1127 P = self.cls
1128 self.assertIs(False, P('').is_reserved())
1129 self.assertIs(False, P('/').is_reserved())
1130 self.assertIs(False, P('/foo/bar').is_reserved())
1131 self.assertIs(True, P('con').is_reserved())
1132 self.assertIs(True, P('NUL').is_reserved())
1133 self.assertIs(True, P('NUL.txt').is_reserved())
1134 self.assertIs(True, P('com1').is_reserved())
1135 self.assertIs(True, P('com9.bar').is_reserved())
1136 self.assertIs(False, P('bar.com9').is_reserved())
1137 self.assertIs(True, P('lpt1').is_reserved())
1138 self.assertIs(True, P('lpt9.bar').is_reserved())
1139 self.assertIs(False, P('bar.lpt9').is_reserved())
1140 # Only the last component matters
1141 self.assertIs(False, P('c:/NUL/con/baz').is_reserved())
1142 # UNC paths are never reserved
1143 self.assertIs(False, P('//my/share/nul/con/aux').is_reserved())
1144
Antoine Pitrou31119e42013-11-22 17:38:12 +01001145class PurePathTest(_BasePurePathTest, unittest.TestCase):
1146 cls = pathlib.PurePath
1147
1148 def test_concrete_class(self):
1149 p = self.cls('a')
1150 self.assertIs(type(p),
1151 pathlib.PureWindowsPath if os.name == 'nt' else pathlib.PurePosixPath)
1152
1153 def test_different_flavours_unequal(self):
1154 p = pathlib.PurePosixPath('a')
1155 q = pathlib.PureWindowsPath('a')
1156 self.assertNotEqual(p, q)
1157
1158 def test_different_flavours_unordered(self):
1159 p = pathlib.PurePosixPath('a')
1160 q = pathlib.PureWindowsPath('a')
1161 with self.assertRaises(TypeError):
1162 p < q
1163 with self.assertRaises(TypeError):
1164 p <= q
1165 with self.assertRaises(TypeError):
1166 p > q
1167 with self.assertRaises(TypeError):
1168 p >= q
1169
1170
1171#
1172# Tests for the concrete classes
1173#
1174
1175# Make sure any symbolic links in the base test path are resolved
1176BASE = os.path.realpath(TESTFN)
1177join = lambda *x: os.path.join(BASE, *x)
1178rel_join = lambda *x: os.path.join(TESTFN, *x)
1179
Antoine Pitrou31119e42013-11-22 17:38:12 +01001180only_nt = unittest.skipIf(os.name != 'nt',
1181 'test requires a Windows-compatible system')
1182only_posix = unittest.skipIf(os.name == 'nt',
1183 'test requires a POSIX-compatible system')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001184
1185@only_posix
1186class PosixPathAsPureTest(PurePosixPathTest):
1187 cls = pathlib.PosixPath
1188
1189@only_nt
1190class WindowsPathAsPureTest(PureWindowsPathTest):
1191 cls = pathlib.WindowsPath
1192
Victor Stinnerd7569632016-03-11 22:53:00 +01001193 def test_owner(self):
1194 P = self.cls
1195 with self.assertRaises(NotImplementedError):
1196 P('c:/').owner()
1197
1198 def test_group(self):
1199 P = self.cls
1200 with self.assertRaises(NotImplementedError):
1201 P('c:/').group()
1202
Antoine Pitrou31119e42013-11-22 17:38:12 +01001203
1204class _BasePathTest(object):
1205 """Tests for the FS-accessing functionalities of the Path classes."""
1206
1207 # (BASE)
1208 # |
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001209 # |-- brokenLink -> non-existing
1210 # |-- dirA
1211 # | `-- linkC -> ../dirB
1212 # |-- dirB
1213 # | |-- fileB
1214 # | `-- linkD -> ../dirB
1215 # |-- dirC
1216 # | |-- dirD
1217 # | | `-- fileD
1218 # | `-- fileC
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001219 # |-- dirE # No permissions
Antoine Pitrou31119e42013-11-22 17:38:12 +01001220 # |-- fileA
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001221 # |-- linkA -> fileA
1222 # `-- linkB -> dirB
Antoine Pitrou31119e42013-11-22 17:38:12 +01001223 #
1224
1225 def setUp(self):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001226 def cleanup():
1227 os.chmod(join('dirE'), 0o777)
1228 support.rmtree(BASE)
1229 self.addCleanup(cleanup)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001230 os.mkdir(BASE)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001231 os.mkdir(join('dirA'))
1232 os.mkdir(join('dirB'))
1233 os.mkdir(join('dirC'))
1234 os.mkdir(join('dirC', 'dirD'))
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001235 os.mkdir(join('dirE'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001236 with open(join('fileA'), 'wb') as f:
1237 f.write(b"this is file A\n")
1238 with open(join('dirB', 'fileB'), 'wb') as f:
1239 f.write(b"this is file B\n")
1240 with open(join('dirC', 'fileC'), 'wb') as f:
1241 f.write(b"this is file C\n")
1242 with open(join('dirC', 'dirD', 'fileD'), 'wb') as f:
1243 f.write(b"this is file D\n")
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001244 os.chmod(join('dirE'), 0)
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001245 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001246 # Relative symlinks
1247 os.symlink('fileA', join('linkA'))
1248 os.symlink('non-existing', join('brokenLink'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001249 self.dirlink('dirB', join('linkB'))
1250 self.dirlink(os.path.join('..', 'dirB'), join('dirA', 'linkC'))
Guido van Rossum69bfb152016-01-06 10:31:33 -08001251 # This one goes upwards, creating a loop
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001252 self.dirlink(os.path.join('..', 'dirB'), join('dirB', 'linkD'))
1253
1254 if os.name == 'nt':
1255 # Workaround for http://bugs.python.org/issue13772
1256 def dirlink(self, src, dest):
1257 os.symlink(src, dest, target_is_directory=True)
1258 else:
1259 def dirlink(self, src, dest):
1260 os.symlink(src, dest)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001261
1262 def assertSame(self, path_a, path_b):
1263 self.assertTrue(os.path.samefile(str(path_a), str(path_b)),
1264 "%r and %r don't point to the same file" %
1265 (path_a, path_b))
1266
1267 def assertFileNotFound(self, func, *args, **kwargs):
1268 with self.assertRaises(FileNotFoundError) as cm:
1269 func(*args, **kwargs)
1270 self.assertEqual(cm.exception.errno, errno.ENOENT)
1271
1272 def _test_cwd(self, p):
1273 q = self.cls(os.getcwd())
1274 self.assertEqual(p, q)
1275 self.assertEqual(str(p), str(q))
1276 self.assertIs(type(p), type(q))
1277 self.assertTrue(p.is_absolute())
1278
1279 def test_cwd(self):
1280 p = self.cls.cwd()
1281 self._test_cwd(p)
1282
Antoine Pitrou17cba7d2015-01-12 21:03:41 +01001283 def _test_home(self, p):
1284 q = self.cls(os.path.expanduser('~'))
1285 self.assertEqual(p, q)
1286 self.assertEqual(str(p), str(q))
1287 self.assertIs(type(p), type(q))
1288 self.assertTrue(p.is_absolute())
1289
1290 def test_home(self):
1291 p = self.cls.home()
1292 self._test_home(p)
1293
Antoine Pitrou43e3d942014-05-13 10:50:15 +02001294 def test_samefile(self):
1295 fileA_path = os.path.join(BASE, 'fileA')
1296 fileB_path = os.path.join(BASE, 'dirB', 'fileB')
1297 p = self.cls(fileA_path)
1298 pp = self.cls(fileA_path)
1299 q = self.cls(fileB_path)
1300 self.assertTrue(p.samefile(fileA_path))
1301 self.assertTrue(p.samefile(pp))
1302 self.assertFalse(p.samefile(fileB_path))
1303 self.assertFalse(p.samefile(q))
1304 # Test the non-existent file case
1305 non_existent = os.path.join(BASE, 'foo')
1306 r = self.cls(non_existent)
1307 self.assertRaises(FileNotFoundError, p.samefile, r)
1308 self.assertRaises(FileNotFoundError, p.samefile, non_existent)
1309 self.assertRaises(FileNotFoundError, r.samefile, p)
1310 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1311 self.assertRaises(FileNotFoundError, r.samefile, r)
1312 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1313
Antoine Pitrou31119e42013-11-22 17:38:12 +01001314 def test_empty_path(self):
1315 # The empty path points to '.'
1316 p = self.cls('')
1317 self.assertEqual(p.stat(), os.stat('.'))
1318
Antoine Pitrou8477ed62014-12-30 20:54:45 +01001319 def test_expanduser_common(self):
1320 P = self.cls
1321 p = P('~')
1322 self.assertEqual(p.expanduser(), P(os.path.expanduser('~')))
1323 p = P('foo')
1324 self.assertEqual(p.expanduser(), p)
1325 p = P('/~')
1326 self.assertEqual(p.expanduser(), p)
1327 p = P('../~')
1328 self.assertEqual(p.expanduser(), p)
1329 p = P(P('').absolute().anchor) / '~'
1330 self.assertEqual(p.expanduser(), p)
1331
Antoine Pitrou31119e42013-11-22 17:38:12 +01001332 def test_exists(self):
1333 P = self.cls
1334 p = P(BASE)
1335 self.assertIs(True, p.exists())
1336 self.assertIs(True, (p / 'dirA').exists())
1337 self.assertIs(True, (p / 'fileA').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001338 self.assertIs(False, (p / 'fileA' / 'bah').exists())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001339 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001340 self.assertIs(True, (p / 'linkA').exists())
1341 self.assertIs(True, (p / 'linkB').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001342 self.assertIs(True, (p / 'linkB' / 'fileB').exists())
1343 self.assertIs(False, (p / 'linkA' / 'bah').exists())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001344 self.assertIs(False, (p / 'foo').exists())
1345 self.assertIs(False, P('/xyzzy').exists())
1346
1347 def test_open_common(self):
1348 p = self.cls(BASE)
1349 with (p / 'fileA').open('r') as f:
1350 self.assertIsInstance(f, io.TextIOBase)
1351 self.assertEqual(f.read(), "this is file A\n")
1352 with (p / 'fileA').open('rb') as f:
1353 self.assertIsInstance(f, io.BufferedIOBase)
1354 self.assertEqual(f.read().strip(), b"this is file A")
1355 with (p / 'fileA').open('rb', buffering=0) as f:
1356 self.assertIsInstance(f, io.RawIOBase)
1357 self.assertEqual(f.read().strip(), b"this is file A")
1358
Georg Brandlea683982014-10-01 19:12:33 +02001359 def test_read_write_bytes(self):
1360 p = self.cls(BASE)
1361 (p / 'fileA').write_bytes(b'abcdefg')
1362 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1363 # check that trying to write str does not truncate the file
1364 self.assertRaises(TypeError, (p / 'fileA').write_bytes, 'somestr')
1365 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1366
1367 def test_read_write_text(self):
1368 p = self.cls(BASE)
1369 (p / 'fileA').write_text('äbcdefg', encoding='latin-1')
1370 self.assertEqual((p / 'fileA').read_text(
1371 encoding='utf-8', errors='ignore'), 'bcdefg')
1372 # check that trying to write bytes does not truncate the file
1373 self.assertRaises(TypeError, (p / 'fileA').write_text, b'somebytes')
1374 self.assertEqual((p / 'fileA').read_text(encoding='latin-1'), 'äbcdefg')
1375
Antoine Pitrou31119e42013-11-22 17:38:12 +01001376 def test_iterdir(self):
1377 P = self.cls
1378 p = P(BASE)
1379 it = p.iterdir()
1380 paths = set(it)
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001381 expected = ['dirA', 'dirB', 'dirC', 'dirE', 'fileA']
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001382 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001383 expected += ['linkA', 'linkB', 'brokenLink']
1384 self.assertEqual(paths, { P(BASE, q) for q in expected })
1385
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001386 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001387 def test_iterdir_symlink(self):
1388 # __iter__ on a symlink to a directory
1389 P = self.cls
1390 p = P(BASE, 'linkB')
1391 paths = set(p.iterdir())
1392 expected = { P(BASE, 'linkB', q) for q in ['fileB', 'linkD'] }
1393 self.assertEqual(paths, expected)
1394
1395 def test_iterdir_nodir(self):
1396 # __iter__ on something that is not a directory
1397 p = self.cls(BASE, 'fileA')
1398 with self.assertRaises(OSError) as cm:
1399 next(p.iterdir())
1400 # ENOENT or EINVAL under Windows, ENOTDIR otherwise
1401 # (see issue #12802)
1402 self.assertIn(cm.exception.errno, (errno.ENOTDIR,
1403 errno.ENOENT, errno.EINVAL))
1404
1405 def test_glob_common(self):
1406 def _check(glob, expected):
1407 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1408 P = self.cls
1409 p = P(BASE)
1410 it = p.glob("fileA")
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001411 self.assertIsInstance(it, collections.abc.Iterator)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001412 _check(it, ["fileA"])
1413 _check(p.glob("fileB"), [])
1414 _check(p.glob("dir*/file*"), ["dirB/fileB", "dirC/fileC"])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001415 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001416 _check(p.glob("*A"), ['dirA', 'fileA'])
1417 else:
1418 _check(p.glob("*A"), ['dirA', 'fileA', 'linkA'])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001419 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001420 _check(p.glob("*B/*"), ['dirB/fileB'])
1421 else:
1422 _check(p.glob("*B/*"), ['dirB/fileB', 'dirB/linkD',
1423 'linkB/fileB', 'linkB/linkD'])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001424 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001425 _check(p.glob("*/fileB"), ['dirB/fileB'])
1426 else:
1427 _check(p.glob("*/fileB"), ['dirB/fileB', 'linkB/fileB'])
1428
1429 def test_rglob_common(self):
1430 def _check(glob, expected):
1431 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1432 P = self.cls
1433 p = P(BASE)
1434 it = p.rglob("fileA")
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001435 self.assertIsInstance(it, collections.abc.Iterator)
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001436 _check(it, ["fileA"])
1437 _check(p.rglob("fileB"), ["dirB/fileB"])
1438 _check(p.rglob("*/fileA"), [])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001439 if not support.can_symlink():
Guido van Rossum9c39b672016-01-07 13:12:34 -08001440 _check(p.rglob("*/fileB"), ["dirB/fileB"])
1441 else:
1442 _check(p.rglob("*/fileB"), ["dirB/fileB", "dirB/linkD/fileB",
1443 "linkB/fileB", "dirA/linkC/fileB"])
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001444 _check(p.rglob("file*"), ["fileA", "dirB/fileB",
1445 "dirC/fileC", "dirC/dirD/fileD"])
Antoine Pitrou31119e42013-11-22 17:38:12 +01001446 p = P(BASE, "dirC")
1447 _check(p.rglob("file*"), ["dirC/fileC", "dirC/dirD/fileD"])
1448 _check(p.rglob("*/*"), ["dirC/dirD/fileD"])
1449
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001450 @support.skip_unless_symlink
Guido van Rossum69bfb152016-01-06 10:31:33 -08001451 def test_rglob_symlink_loop(self):
1452 # Don't get fooled by symlink loops (Issue #26012)
1453 P = self.cls
1454 p = P(BASE)
1455 given = set(p.rglob('*'))
1456 expect = {'brokenLink',
1457 'dirA', 'dirA/linkC',
1458 'dirB', 'dirB/fileB', 'dirB/linkD',
1459 'dirC', 'dirC/dirD', 'dirC/dirD/fileD', 'dirC/fileC',
1460 'dirE',
1461 'fileA',
1462 'linkA',
1463 'linkB',
1464 }
1465 self.assertEqual(given, {p / x for x in expect})
1466
Antoine Pitrou31119e42013-11-22 17:38:12 +01001467 def test_glob_dotdot(self):
1468 # ".." is not special in globs
1469 P = self.cls
1470 p = P(BASE)
1471 self.assertEqual(set(p.glob("..")), { P(BASE, "..") })
1472 self.assertEqual(set(p.glob("dirA/../file*")), { P(BASE, "dirA/../fileA") })
1473 self.assertEqual(set(p.glob("../xyzzy")), set())
1474
Ezio Melottiac1e7f62015-12-28 23:50:19 +02001475
Steve Dower98eb3602016-11-09 12:58:17 -08001476 def _check_resolve(self, p, expected, strict=True):
1477 q = p.resolve(strict)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001478 self.assertEqual(q, expected)
1479
Ezio Melottiac1e7f62015-12-28 23:50:19 +02001480 # this can be used to check both relative and absolute resolutions
1481 _check_resolve_relative = _check_resolve_absolute = _check_resolve
Antoine Pitrou31119e42013-11-22 17:38:12 +01001482
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001483 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001484 def test_resolve_common(self):
1485 P = self.cls
1486 p = P(BASE, 'foo')
1487 with self.assertRaises(OSError) as cm:
Steve Dower98eb3602016-11-09 12:58:17 -08001488 p.resolve(strict=True)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001489 self.assertEqual(cm.exception.errno, errno.ENOENT)
Steve Dower98eb3602016-11-09 12:58:17 -08001490 # Non-strict
1491 self.assertEqual(str(p.resolve(strict=False)),
1492 os.path.join(BASE, 'foo'))
1493 p = P(BASE, 'foo', 'in', 'spam')
1494 self.assertEqual(str(p.resolve(strict=False)),
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001495 os.path.join(BASE, 'foo', 'in', 'spam'))
Steve Dower98eb3602016-11-09 12:58:17 -08001496 p = P(BASE, '..', 'foo', 'in', 'spam')
1497 self.assertEqual(str(p.resolve(strict=False)),
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001498 os.path.abspath(os.path.join('foo', 'in', 'spam')))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001499 # These are all relative symlinks
1500 p = P(BASE, 'dirB', 'fileB')
1501 self._check_resolve_relative(p, p)
1502 p = P(BASE, 'linkA')
1503 self._check_resolve_relative(p, P(BASE, 'fileA'))
1504 p = P(BASE, 'dirA', 'linkC', 'fileB')
1505 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
1506 p = P(BASE, 'dirB', 'linkD', 'fileB')
1507 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
Steve Dower98eb3602016-11-09 12:58:17 -08001508 # Non-strict
1509 p = P(BASE, 'dirA', 'linkC', 'fileB', 'foo', 'in', 'spam')
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001510 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB', 'foo', 'in',
1511 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001512 p = P(BASE, 'dirA', 'linkC', '..', 'foo', 'in', 'spam')
1513 if os.name == 'nt':
1514 # In Windows, if linkY points to dirB, 'dirA\linkY\..'
1515 # resolves to 'dirA' without resolving linkY first.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001516 self._check_resolve_relative(p, P(BASE, 'dirA', 'foo', 'in',
1517 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001518 else:
1519 # In Posix, if linkY points to dirB, 'dirA/linkY/..'
1520 # resolves to 'dirB/..' first before resolving to parent of dirB.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001521 self._check_resolve_relative(p, P(BASE, 'foo', 'in', 'spam'), False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001522 # Now create absolute symlinks
1523 d = tempfile.mkdtemp(suffix='-dirD')
Victor Stinnerec864692014-07-21 19:19:05 +02001524 self.addCleanup(support.rmtree, d)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001525 os.symlink(os.path.join(d), join('dirA', 'linkX'))
1526 os.symlink(join('dirB'), os.path.join(d, 'linkY'))
1527 p = P(BASE, 'dirA', 'linkX', 'linkY', 'fileB')
1528 self._check_resolve_absolute(p, P(BASE, 'dirB', 'fileB'))
Steve Dower98eb3602016-11-09 12:58:17 -08001529 # Non-strict
1530 p = P(BASE, 'dirA', 'linkX', 'linkY', 'foo', 'in', 'spam')
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001531 self._check_resolve_relative(p, P(BASE, 'dirB', 'foo', 'in', 'spam'),
1532 False)
Steve Dower98eb3602016-11-09 12:58:17 -08001533 p = P(BASE, 'dirA', 'linkX', 'linkY', '..', 'foo', 'in', 'spam')
1534 if os.name == 'nt':
1535 # In Windows, if linkY points to dirB, 'dirA\linkY\..'
1536 # resolves to 'dirA' without resolving linkY first.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001537 self._check_resolve_relative(p, P(d, 'foo', 'in', 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001538 else:
1539 # In Posix, if linkY points to dirB, 'dirA/linkY/..'
1540 # resolves to 'dirB/..' first before resolving to parent of dirB.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001541 self._check_resolve_relative(p, P(BASE, 'foo', 'in', 'spam'), False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001542
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001543 @support.skip_unless_symlink
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001544 def test_resolve_dot(self):
1545 # See https://bitbucket.org/pitrou/pathlib/issue/9/pathresolve-fails-on-complex-symlinks
1546 p = self.cls(BASE)
1547 self.dirlink('.', join('0'))
Antoine Pitroucc157512013-12-03 17:13:13 +01001548 self.dirlink(os.path.join('0', '0'), join('1'))
1549 self.dirlink(os.path.join('1', '1'), join('2'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001550 q = p / '2'
Steve Dower98eb3602016-11-09 12:58:17 -08001551 self.assertEqual(q.resolve(strict=True), p)
1552 r = q / '3' / '4'
1553 self.assertRaises(FileNotFoundError, r.resolve, strict=True)
1554 # Non-strict
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001555 self.assertEqual(r.resolve(strict=False), p / '3' / '4')
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001556
Antoine Pitrou31119e42013-11-22 17:38:12 +01001557 def test_with(self):
1558 p = self.cls(BASE)
1559 it = p.iterdir()
1560 it2 = p.iterdir()
1561 next(it2)
1562 with p:
1563 pass
1564 # I/O operation on closed path
1565 self.assertRaises(ValueError, next, it)
1566 self.assertRaises(ValueError, next, it2)
1567 self.assertRaises(ValueError, p.open)
1568 self.assertRaises(ValueError, p.resolve)
1569 self.assertRaises(ValueError, p.absolute)
1570 self.assertRaises(ValueError, p.__enter__)
1571
1572 def test_chmod(self):
1573 p = self.cls(BASE) / 'fileA'
1574 mode = p.stat().st_mode
1575 # Clear writable bit
1576 new_mode = mode & ~0o222
1577 p.chmod(new_mode)
1578 self.assertEqual(p.stat().st_mode, new_mode)
1579 # Set writable bit
1580 new_mode = mode | 0o222
1581 p.chmod(new_mode)
1582 self.assertEqual(p.stat().st_mode, new_mode)
1583
1584 # XXX also need a test for lchmod
1585
1586 def test_stat(self):
1587 p = self.cls(BASE) / 'fileA'
1588 st = p.stat()
1589 self.assertEqual(p.stat(), st)
1590 # Change file mode by flipping write bit
1591 p.chmod(st.st_mode ^ 0o222)
1592 self.addCleanup(p.chmod, st.st_mode)
1593 self.assertNotEqual(p.stat(), st)
1594
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001595 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001596 def test_lstat(self):
1597 p = self.cls(BASE)/ 'linkA'
1598 st = p.stat()
1599 self.assertNotEqual(st, p.lstat())
1600
1601 def test_lstat_nosymlink(self):
1602 p = self.cls(BASE) / 'fileA'
1603 st = p.stat()
1604 self.assertEqual(st, p.lstat())
1605
1606 @unittest.skipUnless(pwd, "the pwd module is needed for this test")
1607 def test_owner(self):
1608 p = self.cls(BASE) / 'fileA'
1609 uid = p.stat().st_uid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001610 try:
1611 name = pwd.getpwuid(uid).pw_name
1612 except KeyError:
1613 self.skipTest(
1614 "user %d doesn't have an entry in the system database" % uid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001615 self.assertEqual(name, p.owner())
1616
1617 @unittest.skipUnless(grp, "the grp module is needed for this test")
1618 def test_group(self):
1619 p = self.cls(BASE) / 'fileA'
1620 gid = p.stat().st_gid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001621 try:
1622 name = grp.getgrgid(gid).gr_name
1623 except KeyError:
1624 self.skipTest(
1625 "group %d doesn't have an entry in the system database" % gid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001626 self.assertEqual(name, p.group())
1627
1628 def test_unlink(self):
1629 p = self.cls(BASE) / 'fileA'
1630 p.unlink()
1631 self.assertFileNotFound(p.stat)
1632 self.assertFileNotFound(p.unlink)
1633
1634 def test_rmdir(self):
1635 p = self.cls(BASE) / 'dirA'
1636 for q in p.iterdir():
1637 q.unlink()
1638 p.rmdir()
1639 self.assertFileNotFound(p.stat)
1640 self.assertFileNotFound(p.unlink)
1641
1642 def test_rename(self):
1643 P = self.cls(BASE)
1644 p = P / 'fileA'
1645 size = p.stat().st_size
1646 # Renaming to another path
1647 q = P / 'dirA' / 'fileAA'
1648 p.rename(q)
1649 self.assertEqual(q.stat().st_size, size)
1650 self.assertFileNotFound(p.stat)
1651 # Renaming to a str of a relative path
1652 r = rel_join('fileAAA')
1653 q.rename(r)
1654 self.assertEqual(os.stat(r).st_size, size)
1655 self.assertFileNotFound(q.stat)
1656
1657 def test_replace(self):
1658 P = self.cls(BASE)
1659 p = P / 'fileA'
1660 size = p.stat().st_size
1661 # Replacing a non-existing path
1662 q = P / 'dirA' / 'fileAA'
1663 p.replace(q)
1664 self.assertEqual(q.stat().st_size, size)
1665 self.assertFileNotFound(p.stat)
1666 # Replacing another (existing) path
1667 r = rel_join('dirB', 'fileB')
1668 q.replace(r)
1669 self.assertEqual(os.stat(r).st_size, size)
1670 self.assertFileNotFound(q.stat)
1671
1672 def test_touch_common(self):
1673 P = self.cls(BASE)
1674 p = P / 'newfileA'
1675 self.assertFalse(p.exists())
1676 p.touch()
1677 self.assertTrue(p.exists())
Antoine Pitrou0f575642013-11-22 23:20:08 +01001678 st = p.stat()
1679 old_mtime = st.st_mtime
1680 old_mtime_ns = st.st_mtime_ns
Antoine Pitrou31119e42013-11-22 17:38:12 +01001681 # Rewind the mtime sufficiently far in the past to work around
1682 # filesystem-specific timestamp granularity.
1683 os.utime(str(p), (old_mtime - 10, old_mtime - 10))
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001684 # The file mtime should be refreshed by calling touch() again
Antoine Pitrou31119e42013-11-22 17:38:12 +01001685 p.touch()
Antoine Pitrou0f575642013-11-22 23:20:08 +01001686 st = p.stat()
Antoine Pitrou2cf39172013-11-23 15:25:59 +01001687 self.assertGreaterEqual(st.st_mtime_ns, old_mtime_ns)
1688 self.assertGreaterEqual(st.st_mtime, old_mtime)
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001689 # Now with exist_ok=False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001690 p = P / 'newfileB'
1691 self.assertFalse(p.exists())
1692 p.touch(mode=0o700, exist_ok=False)
1693 self.assertTrue(p.exists())
1694 self.assertRaises(OSError, p.touch, exist_ok=False)
1695
Antoine Pitrou8b784932013-11-23 14:52:39 +01001696 def test_touch_nochange(self):
1697 P = self.cls(BASE)
1698 p = P / 'fileA'
1699 p.touch()
1700 with p.open('rb') as f:
1701 self.assertEqual(f.read().strip(), b"this is file A")
1702
Antoine Pitrou31119e42013-11-22 17:38:12 +01001703 def test_mkdir(self):
1704 P = self.cls(BASE)
1705 p = P / 'newdirA'
1706 self.assertFalse(p.exists())
1707 p.mkdir()
1708 self.assertTrue(p.exists())
1709 self.assertTrue(p.is_dir())
1710 with self.assertRaises(OSError) as cm:
1711 p.mkdir()
1712 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001713
1714 def test_mkdir_parents(self):
1715 # Creating a chain of directories
1716 p = self.cls(BASE, 'newdirB', 'newdirC')
1717 self.assertFalse(p.exists())
1718 with self.assertRaises(OSError) as cm:
1719 p.mkdir()
1720 self.assertEqual(cm.exception.errno, errno.ENOENT)
1721 p.mkdir(parents=True)
1722 self.assertTrue(p.exists())
1723 self.assertTrue(p.is_dir())
1724 with self.assertRaises(OSError) as cm:
1725 p.mkdir(parents=True)
1726 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001727 # test `mode` arg
1728 mode = stat.S_IMODE(p.stat().st_mode) # default mode
1729 p = self.cls(BASE, 'newdirD', 'newdirE')
1730 p.mkdir(0o555, parents=True)
1731 self.assertTrue(p.exists())
1732 self.assertTrue(p.is_dir())
1733 if os.name != 'nt':
1734 # the directory's permissions follow the mode argument
Gregory P. Smithb599c612014-01-20 01:10:33 -08001735 self.assertEqual(stat.S_IMODE(p.stat().st_mode), 0o7555 & mode)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001736 # the parent's permissions follow the default process settings
1737 self.assertEqual(stat.S_IMODE(p.parent.stat().st_mode), mode)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001738
Barry Warsaw7c549c42014-08-05 11:28:12 -04001739 def test_mkdir_exist_ok(self):
1740 p = self.cls(BASE, 'dirB')
1741 st_ctime_first = p.stat().st_ctime
1742 self.assertTrue(p.exists())
1743 self.assertTrue(p.is_dir())
1744 with self.assertRaises(FileExistsError) as cm:
1745 p.mkdir()
1746 self.assertEqual(cm.exception.errno, errno.EEXIST)
1747 p.mkdir(exist_ok=True)
1748 self.assertTrue(p.exists())
1749 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1750
1751 def test_mkdir_exist_ok_with_parent(self):
1752 p = self.cls(BASE, 'dirC')
1753 self.assertTrue(p.exists())
1754 with self.assertRaises(FileExistsError) as cm:
1755 p.mkdir()
1756 self.assertEqual(cm.exception.errno, errno.EEXIST)
1757 p = p / 'newdirC'
1758 p.mkdir(parents=True)
1759 st_ctime_first = p.stat().st_ctime
1760 self.assertTrue(p.exists())
1761 with self.assertRaises(FileExistsError) as cm:
1762 p.mkdir(parents=True)
1763 self.assertEqual(cm.exception.errno, errno.EEXIST)
1764 p.mkdir(parents=True, exist_ok=True)
1765 self.assertTrue(p.exists())
1766 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1767
Serhiy Storchakaaf7b9ec2017-03-24 20:51:53 +02001768 def test_mkdir_exist_ok_root(self):
1769 # Issue #25803: A drive root could raise PermissionError on Windows
1770 self.cls('/').resolve().mkdir(exist_ok=True)
1771 self.cls('/').resolve().mkdir(parents=True, exist_ok=True)
1772
Steve Dowerd3c48532017-02-04 14:54:56 -08001773 @only_nt # XXX: not sure how to test this on POSIX
1774 def test_mkdir_with_unknown_drive(self):
1775 for d in 'ZYXWVUTSRQPONMLKJIHGFEDCBA':
1776 p = self.cls(d + ':\\')
1777 if not p.is_dir():
1778 break
1779 else:
1780 self.skipTest("cannot find a drive that doesn't exist")
1781 with self.assertRaises(OSError):
1782 (p / 'child' / 'path').mkdir(parents=True)
1783
Barry Warsaw7c549c42014-08-05 11:28:12 -04001784 def test_mkdir_with_child_file(self):
1785 p = self.cls(BASE, 'dirB', 'fileB')
1786 self.assertTrue(p.exists())
1787 # An exception is raised when the last path component is an existing
1788 # regular file, regardless of whether exist_ok is true or not.
1789 with self.assertRaises(FileExistsError) as cm:
1790 p.mkdir(parents=True)
1791 self.assertEqual(cm.exception.errno, errno.EEXIST)
1792 with self.assertRaises(FileExistsError) as cm:
1793 p.mkdir(parents=True, exist_ok=True)
1794 self.assertEqual(cm.exception.errno, errno.EEXIST)
1795
1796 def test_mkdir_no_parents_file(self):
1797 p = self.cls(BASE, 'fileA')
1798 self.assertTrue(p.exists())
1799 # An exception is raised when the last path component is an existing
1800 # regular file, regardless of whether exist_ok is true or not.
1801 with self.assertRaises(FileExistsError) as cm:
1802 p.mkdir()
1803 self.assertEqual(cm.exception.errno, errno.EEXIST)
1804 with self.assertRaises(FileExistsError) as cm:
1805 p.mkdir(exist_ok=True)
1806 self.assertEqual(cm.exception.errno, errno.EEXIST)
1807
Armin Rigo22a594a2017-04-13 20:08:15 +02001808 def test_mkdir_concurrent_parent_creation(self):
1809 for pattern_num in range(32):
1810 p = self.cls(BASE, 'dirCPC%d' % pattern_num)
1811 self.assertFalse(p.exists())
1812
1813 def my_mkdir(path, mode=0o777):
1814 path = str(path)
1815 # Emulate another process that would create the directory
1816 # just before we try to create it ourselves. We do it
1817 # in all possible pattern combinations, assuming that this
1818 # function is called at most 5 times (dirCPC/dir1/dir2,
1819 # dirCPC/dir1, dirCPC, dirCPC/dir1, dirCPC/dir1/dir2).
1820 if pattern.pop():
1821 os.mkdir(path, mode) # from another process
1822 concurrently_created.add(path)
1823 os.mkdir(path, mode) # our real call
1824
1825 pattern = [bool(pattern_num & (1 << n)) for n in range(5)]
1826 concurrently_created = set()
1827 p12 = p / 'dir1' / 'dir2'
1828 try:
1829 with mock.patch("pathlib._normal_accessor.mkdir", my_mkdir):
1830 p12.mkdir(parents=True, exist_ok=False)
1831 except FileExistsError:
1832 self.assertIn(str(p12), concurrently_created)
1833 else:
1834 self.assertNotIn(str(p12), concurrently_created)
1835 self.assertTrue(p.exists())
1836
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001837 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001838 def test_symlink_to(self):
1839 P = self.cls(BASE)
1840 target = P / 'fileA'
1841 # Symlinking a path target
1842 link = P / 'dirA' / 'linkAA'
1843 link.symlink_to(target)
1844 self.assertEqual(link.stat(), target.stat())
1845 self.assertNotEqual(link.lstat(), target.stat())
1846 # Symlinking a str target
1847 link = P / 'dirA' / 'linkAAA'
1848 link.symlink_to(str(target))
1849 self.assertEqual(link.stat(), target.stat())
1850 self.assertNotEqual(link.lstat(), target.stat())
1851 self.assertFalse(link.is_dir())
1852 # Symlinking to a directory
1853 target = P / 'dirB'
1854 link = P / 'dirA' / 'linkAAAA'
1855 link.symlink_to(target, target_is_directory=True)
1856 self.assertEqual(link.stat(), target.stat())
1857 self.assertNotEqual(link.lstat(), target.stat())
1858 self.assertTrue(link.is_dir())
1859 self.assertTrue(list(link.iterdir()))
1860
1861 def test_is_dir(self):
1862 P = self.cls(BASE)
1863 self.assertTrue((P / 'dirA').is_dir())
1864 self.assertFalse((P / 'fileA').is_dir())
1865 self.assertFalse((P / 'non-existing').is_dir())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001866 self.assertFalse((P / 'fileA' / 'bah').is_dir())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001867 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001868 self.assertFalse((P / 'linkA').is_dir())
1869 self.assertTrue((P / 'linkB').is_dir())
1870 self.assertFalse((P/ 'brokenLink').is_dir())
1871
1872 def test_is_file(self):
1873 P = self.cls(BASE)
1874 self.assertTrue((P / 'fileA').is_file())
1875 self.assertFalse((P / 'dirA').is_file())
1876 self.assertFalse((P / 'non-existing').is_file())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001877 self.assertFalse((P / 'fileA' / 'bah').is_file())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001878 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001879 self.assertTrue((P / 'linkA').is_file())
1880 self.assertFalse((P / 'linkB').is_file())
1881 self.assertFalse((P/ 'brokenLink').is_file())
1882
Cooper Lees173ff4a2017-08-01 15:35:45 -07001883 @only_posix
1884 def test_is_mount(self):
1885 P = self.cls(BASE)
1886 R = self.cls('/') # TODO: Work out windows
1887 self.assertFalse((P / 'fileA').is_mount())
1888 self.assertFalse((P / 'dirA').is_mount())
1889 self.assertFalse((P / 'non-existing').is_mount())
1890 self.assertFalse((P / 'fileA' / 'bah').is_mount())
1891 self.assertTrue(R.is_mount())
1892 if support.can_symlink():
1893 self.assertFalse((P / 'linkA').is_mount())
1894
Antoine Pitrou31119e42013-11-22 17:38:12 +01001895 def test_is_symlink(self):
1896 P = self.cls(BASE)
1897 self.assertFalse((P / 'fileA').is_symlink())
1898 self.assertFalse((P / 'dirA').is_symlink())
1899 self.assertFalse((P / 'non-existing').is_symlink())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001900 self.assertFalse((P / 'fileA' / 'bah').is_symlink())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001901 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001902 self.assertTrue((P / 'linkA').is_symlink())
1903 self.assertTrue((P / 'linkB').is_symlink())
1904 self.assertTrue((P/ 'brokenLink').is_symlink())
1905
1906 def test_is_fifo_false(self):
1907 P = self.cls(BASE)
1908 self.assertFalse((P / 'fileA').is_fifo())
1909 self.assertFalse((P / 'dirA').is_fifo())
1910 self.assertFalse((P / 'non-existing').is_fifo())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001911 self.assertFalse((P / 'fileA' / 'bah').is_fifo())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001912
1913 @unittest.skipUnless(hasattr(os, "mkfifo"), "os.mkfifo() required")
Xavier de Gaye3a4e9892016-12-13 10:00:01 +01001914 @unittest.skipIf(android_not_root, "mkfifo not allowed, non root user")
Antoine Pitrou31119e42013-11-22 17:38:12 +01001915 def test_is_fifo_true(self):
1916 P = self.cls(BASE, 'myfifo')
1917 os.mkfifo(str(P))
1918 self.assertTrue(P.is_fifo())
1919 self.assertFalse(P.is_socket())
1920 self.assertFalse(P.is_file())
1921
1922 def test_is_socket_false(self):
1923 P = self.cls(BASE)
1924 self.assertFalse((P / 'fileA').is_socket())
1925 self.assertFalse((P / 'dirA').is_socket())
1926 self.assertFalse((P / 'non-existing').is_socket())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001927 self.assertFalse((P / 'fileA' / 'bah').is_socket())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001928
1929 @unittest.skipUnless(hasattr(socket, "AF_UNIX"), "Unix sockets required")
1930 def test_is_socket_true(self):
1931 P = self.cls(BASE, 'mysock')
Antoine Pitrou330ce592013-11-22 18:05:06 +01001932 sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001933 self.addCleanup(sock.close)
Antoine Pitrou330ce592013-11-22 18:05:06 +01001934 try:
1935 sock.bind(str(P))
1936 except OSError as e:
Xavier de Gayee88ed052016-12-14 11:52:28 +01001937 if (isinstance(e, PermissionError) or
1938 "AF_UNIX path too long" in str(e)):
Antoine Pitrou330ce592013-11-22 18:05:06 +01001939 self.skipTest("cannot bind Unix socket: " + str(e))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001940 self.assertTrue(P.is_socket())
1941 self.assertFalse(P.is_fifo())
1942 self.assertFalse(P.is_file())
1943
1944 def test_is_block_device_false(self):
1945 P = self.cls(BASE)
1946 self.assertFalse((P / 'fileA').is_block_device())
1947 self.assertFalse((P / 'dirA').is_block_device())
1948 self.assertFalse((P / 'non-existing').is_block_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001949 self.assertFalse((P / 'fileA' / 'bah').is_block_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001950
1951 def test_is_char_device_false(self):
1952 P = self.cls(BASE)
1953 self.assertFalse((P / 'fileA').is_char_device())
1954 self.assertFalse((P / 'dirA').is_char_device())
1955 self.assertFalse((P / 'non-existing').is_char_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001956 self.assertFalse((P / 'fileA' / 'bah').is_char_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001957
1958 def test_is_char_device_true(self):
1959 # Under Unix, /dev/null should generally be a char device
1960 P = self.cls('/dev/null')
1961 if not P.exists():
1962 self.skipTest("/dev/null required")
1963 self.assertTrue(P.is_char_device())
1964 self.assertFalse(P.is_block_device())
1965 self.assertFalse(P.is_file())
1966
1967 def test_pickling_common(self):
1968 p = self.cls(BASE, 'fileA')
1969 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
1970 dumped = pickle.dumps(p, proto)
1971 pp = pickle.loads(dumped)
1972 self.assertEqual(pp.stat(), p.stat())
1973
1974 def test_parts_interning(self):
1975 P = self.cls
1976 p = P('/usr/bin/foo')
1977 q = P('/usr/local/bin')
1978 # 'usr'
1979 self.assertIs(p.parts[1], q.parts[1])
1980 # 'bin'
1981 self.assertIs(p.parts[2], q.parts[3])
1982
Antoine Pitrouc274fd22013-12-16 19:57:41 +01001983 def _check_complex_symlinks(self, link0_target):
1984 # Test solving a non-looping chain of symlinks (issue #19887)
1985 P = self.cls(BASE)
1986 self.dirlink(os.path.join('link0', 'link0'), join('link1'))
1987 self.dirlink(os.path.join('link1', 'link1'), join('link2'))
1988 self.dirlink(os.path.join('link2', 'link2'), join('link3'))
1989 self.dirlink(link0_target, join('link0'))
1990
1991 # Resolve absolute paths
1992 p = (P / 'link0').resolve()
1993 self.assertEqual(p, P)
1994 self.assertEqual(str(p), BASE)
1995 p = (P / 'link1').resolve()
1996 self.assertEqual(p, P)
1997 self.assertEqual(str(p), BASE)
1998 p = (P / 'link2').resolve()
1999 self.assertEqual(p, P)
2000 self.assertEqual(str(p), BASE)
2001 p = (P / 'link3').resolve()
2002 self.assertEqual(p, P)
2003 self.assertEqual(str(p), BASE)
2004
2005 # Resolve relative paths
2006 old_path = os.getcwd()
2007 os.chdir(BASE)
2008 try:
2009 p = self.cls('link0').resolve()
2010 self.assertEqual(p, P)
2011 self.assertEqual(str(p), BASE)
2012 p = self.cls('link1').resolve()
2013 self.assertEqual(p, P)
2014 self.assertEqual(str(p), BASE)
2015 p = self.cls('link2').resolve()
2016 self.assertEqual(p, P)
2017 self.assertEqual(str(p), BASE)
2018 p = self.cls('link3').resolve()
2019 self.assertEqual(p, P)
2020 self.assertEqual(str(p), BASE)
2021 finally:
2022 os.chdir(old_path)
2023
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002024 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002025 def test_complex_symlinks_absolute(self):
2026 self._check_complex_symlinks(BASE)
2027
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002028 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002029 def test_complex_symlinks_relative(self):
2030 self._check_complex_symlinks('.')
2031
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002032 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002033 def test_complex_symlinks_relative_dot_dot(self):
2034 self._check_complex_symlinks(os.path.join('dirA', '..'))
2035
Antoine Pitrou31119e42013-11-22 17:38:12 +01002036
2037class PathTest(_BasePathTest, unittest.TestCase):
2038 cls = pathlib.Path
2039
2040 def test_concrete_class(self):
2041 p = self.cls('a')
2042 self.assertIs(type(p),
2043 pathlib.WindowsPath if os.name == 'nt' else pathlib.PosixPath)
2044
2045 def test_unsupported_flavour(self):
2046 if os.name == 'nt':
2047 self.assertRaises(NotImplementedError, pathlib.PosixPath)
2048 else:
2049 self.assertRaises(NotImplementedError, pathlib.WindowsPath)
2050
Berker Peksag4a208e42016-01-30 17:50:48 +02002051 def test_glob_empty_pattern(self):
2052 p = self.cls()
2053 with self.assertRaisesRegex(ValueError, 'Unacceptable pattern'):
2054 list(p.glob(''))
2055
Antoine Pitrou31119e42013-11-22 17:38:12 +01002056
2057@only_posix
2058class PosixPathTest(_BasePathTest, unittest.TestCase):
2059 cls = pathlib.PosixPath
2060
Steve Dower98eb3602016-11-09 12:58:17 -08002061 def _check_symlink_loop(self, *args, strict=True):
Antoine Pitrou31119e42013-11-22 17:38:12 +01002062 path = self.cls(*args)
2063 with self.assertRaises(RuntimeError):
Steve Dower98eb3602016-11-09 12:58:17 -08002064 print(path.resolve(strict))
Antoine Pitrou31119e42013-11-22 17:38:12 +01002065
2066 def test_open_mode(self):
2067 old_mask = os.umask(0)
2068 self.addCleanup(os.umask, old_mask)
2069 p = self.cls(BASE)
2070 with (p / 'new_file').open('wb'):
2071 pass
2072 st = os.stat(join('new_file'))
2073 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
2074 os.umask(0o022)
2075 with (p / 'other_new_file').open('wb'):
2076 pass
2077 st = os.stat(join('other_new_file'))
2078 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
2079
2080 def test_touch_mode(self):
2081 old_mask = os.umask(0)
2082 self.addCleanup(os.umask, old_mask)
2083 p = self.cls(BASE)
2084 (p / 'new_file').touch()
2085 st = os.stat(join('new_file'))
2086 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
2087 os.umask(0o022)
2088 (p / 'other_new_file').touch()
2089 st = os.stat(join('other_new_file'))
2090 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
2091 (p / 'masked_new_file').touch(mode=0o750)
2092 st = os.stat(join('masked_new_file'))
2093 self.assertEqual(stat.S_IMODE(st.st_mode), 0o750)
2094
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002095 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01002096 def test_resolve_loop(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +01002097 # Loops with relative symlinks
2098 os.symlink('linkX/inside', join('linkX'))
2099 self._check_symlink_loop(BASE, 'linkX')
2100 os.symlink('linkY', join('linkY'))
2101 self._check_symlink_loop(BASE, 'linkY')
2102 os.symlink('linkZ/../linkZ', join('linkZ'))
2103 self._check_symlink_loop(BASE, 'linkZ')
Steve Dower98eb3602016-11-09 12:58:17 -08002104 # Non-strict
2105 self._check_symlink_loop(BASE, 'linkZ', 'foo', strict=False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01002106 # Loops with absolute symlinks
2107 os.symlink(join('linkU/inside'), join('linkU'))
2108 self._check_symlink_loop(BASE, 'linkU')
2109 os.symlink(join('linkV'), join('linkV'))
2110 self._check_symlink_loop(BASE, 'linkV')
2111 os.symlink(join('linkW/../linkW'), join('linkW'))
2112 self._check_symlink_loop(BASE, 'linkW')
Steve Dower98eb3602016-11-09 12:58:17 -08002113 # Non-strict
2114 self._check_symlink_loop(BASE, 'linkW', 'foo', strict=False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01002115
2116 def test_glob(self):
2117 P = self.cls
2118 p = P(BASE)
Brett Cannonfe77f4e2013-11-22 16:14:10 -05002119 given = set(p.glob("FILEa"))
2120 expect = set() if not support.fs_is_case_insensitive(BASE) else given
2121 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01002122 self.assertEqual(set(p.glob("FILEa*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01002123
2124 def test_rglob(self):
2125 P = self.cls
2126 p = P(BASE, "dirC")
Brett Cannonfe77f4e2013-11-22 16:14:10 -05002127 given = set(p.rglob("FILEd"))
2128 expect = set() if not support.fs_is_case_insensitive(BASE) else given
2129 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01002130 self.assertEqual(set(p.rglob("FILEd*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01002131
Xavier de Gayefb24eea2016-12-13 09:11:38 +01002132 @unittest.skipUnless(hasattr(pwd, 'getpwall'),
2133 'pwd module does not expose getpwall()')
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002134 def test_expanduser(self):
2135 P = self.cls
2136 support.import_module('pwd')
2137 import pwd
2138 pwdent = pwd.getpwuid(os.getuid())
2139 username = pwdent.pw_name
Serhiy Storchakaa3fd0b22016-05-03 21:17:03 +03002140 userhome = pwdent.pw_dir.rstrip('/') or '/'
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002141 # find arbitrary different user (if exists)
2142 for pwdent in pwd.getpwall():
2143 othername = pwdent.pw_name
2144 otherhome = pwdent.pw_dir.rstrip('/')
Victor Stinnere9694392015-01-10 09:00:20 +01002145 if othername != username and otherhome:
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002146 break
2147
2148 p1 = P('~/Documents')
2149 p2 = P('~' + username + '/Documents')
2150 p3 = P('~' + othername + '/Documents')
2151 p4 = P('../~' + username + '/Documents')
2152 p5 = P('/~' + username + '/Documents')
2153 p6 = P('')
2154 p7 = P('~fakeuser/Documents')
2155
2156 with support.EnvironmentVarGuard() as env:
2157 env.pop('HOME', None)
2158
2159 self.assertEqual(p1.expanduser(), P(userhome) / 'Documents')
2160 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2161 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2162 self.assertEqual(p4.expanduser(), p4)
2163 self.assertEqual(p5.expanduser(), p5)
2164 self.assertEqual(p6.expanduser(), p6)
2165 self.assertRaises(RuntimeError, p7.expanduser)
2166
2167 env['HOME'] = '/tmp'
2168 self.assertEqual(p1.expanduser(), P('/tmp/Documents'))
2169 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2170 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2171 self.assertEqual(p4.expanduser(), p4)
2172 self.assertEqual(p5.expanduser(), p5)
2173 self.assertEqual(p6.expanduser(), p6)
2174 self.assertRaises(RuntimeError, p7.expanduser)
2175
Antoine Pitrou31119e42013-11-22 17:38:12 +01002176
2177@only_nt
2178class WindowsPathTest(_BasePathTest, unittest.TestCase):
2179 cls = pathlib.WindowsPath
2180
2181 def test_glob(self):
2182 P = self.cls
2183 p = P(BASE)
2184 self.assertEqual(set(p.glob("FILEa")), { P(BASE, "fileA") })
2185
2186 def test_rglob(self):
2187 P = self.cls
2188 p = P(BASE, "dirC")
2189 self.assertEqual(set(p.rglob("FILEd")), { P(BASE, "dirC/dirD/fileD") })
2190
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002191 def test_expanduser(self):
2192 P = self.cls
2193 with support.EnvironmentVarGuard() as env:
2194 env.pop('HOME', None)
2195 env.pop('USERPROFILE', None)
2196 env.pop('HOMEPATH', None)
2197 env.pop('HOMEDRIVE', None)
2198 env['USERNAME'] = 'alice'
2199
2200 # test that the path returns unchanged
2201 p1 = P('~/My Documents')
2202 p2 = P('~alice/My Documents')
2203 p3 = P('~bob/My Documents')
2204 p4 = P('/~/My Documents')
2205 p5 = P('d:~/My Documents')
2206 p6 = P('')
2207 self.assertRaises(RuntimeError, p1.expanduser)
2208 self.assertRaises(RuntimeError, p2.expanduser)
2209 self.assertRaises(RuntimeError, p3.expanduser)
2210 self.assertEqual(p4.expanduser(), p4)
2211 self.assertEqual(p5.expanduser(), p5)
2212 self.assertEqual(p6.expanduser(), p6)
2213
2214 def check():
2215 env.pop('USERNAME', None)
2216 self.assertEqual(p1.expanduser(),
2217 P('C:/Users/alice/My Documents'))
2218 self.assertRaises(KeyError, p2.expanduser)
2219 env['USERNAME'] = 'alice'
2220 self.assertEqual(p2.expanduser(),
2221 P('C:/Users/alice/My Documents'))
2222 self.assertEqual(p3.expanduser(),
2223 P('C:/Users/bob/My Documents'))
2224 self.assertEqual(p4.expanduser(), p4)
2225 self.assertEqual(p5.expanduser(), p5)
2226 self.assertEqual(p6.expanduser(), p6)
2227
2228 # test the first lookup key in the env vars
2229 env['HOME'] = 'C:\\Users\\alice'
2230 check()
2231
2232 # test that HOMEPATH is available instead
2233 env.pop('HOME', None)
2234 env['HOMEPATH'] = 'C:\\Users\\alice'
2235 check()
2236
2237 env['HOMEDRIVE'] = 'C:\\'
2238 env['HOMEPATH'] = 'Users\\alice'
2239 check()
2240
2241 env.pop('HOMEDRIVE', None)
2242 env.pop('HOMEPATH', None)
2243 env['USERPROFILE'] = 'C:\\Users\\alice'
2244 check()
2245
Antoine Pitrou31119e42013-11-22 17:38:12 +01002246
2247if __name__ == "__main__":
2248 unittest.main()