blob: e56e0d20844ec5728784841ae88809035378005d [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
14TESTFN = support.TESTFN
15
16try:
17 import grp, pwd
18except ImportError:
19 grp = pwd = None
20
21
22class _BaseFlavourTest(object):
23
24 def _check_parse_parts(self, arg, expected):
25 f = self.flavour.parse_parts
26 sep = self.flavour.sep
27 altsep = self.flavour.altsep
28 actual = f([x.replace('/', sep) for x in arg])
29 self.assertEqual(actual, expected)
30 if altsep:
31 actual = f([x.replace('/', altsep) for x in arg])
32 self.assertEqual(actual, expected)
33
34 def test_parse_parts_common(self):
35 check = self._check_parse_parts
36 sep = self.flavour.sep
37 # Unanchored parts
38 check([], ('', '', []))
39 check(['a'], ('', '', ['a']))
40 check(['a/'], ('', '', ['a']))
41 check(['a', 'b'], ('', '', ['a', 'b']))
42 # Expansion
43 check(['a/b'], ('', '', ['a', 'b']))
44 check(['a/b/'], ('', '', ['a', 'b']))
45 check(['a', 'b/c', 'd'], ('', '', ['a', 'b', 'c', 'd']))
46 # Collapsing and stripping excess slashes
47 check(['a', 'b//c', 'd'], ('', '', ['a', 'b', 'c', 'd']))
48 check(['a', 'b/c/', 'd'], ('', '', ['a', 'b', 'c', 'd']))
49 # Eliminating standalone dots
50 check(['.'], ('', '', []))
51 check(['.', '.', 'b'], ('', '', ['b']))
52 check(['a', '.', 'b'], ('', '', ['a', 'b']))
53 check(['a', '.', '.'], ('', '', ['a']))
54 # The first part is anchored
55 check(['/a/b'], ('', sep, [sep, 'a', 'b']))
56 check(['/a', 'b'], ('', sep, [sep, 'a', 'b']))
57 check(['/a/', 'b'], ('', sep, [sep, 'a', 'b']))
58 # Ignoring parts before an anchored part
59 check(['a', '/b', 'c'], ('', sep, [sep, 'b', 'c']))
60 check(['a', '/b', '/c'], ('', sep, [sep, 'c']))
61
62
63class PosixFlavourTest(_BaseFlavourTest, unittest.TestCase):
64 flavour = pathlib._posix_flavour
65
66 def test_parse_parts(self):
67 check = self._check_parse_parts
68 # Collapsing of excess leading slashes, except for the double-slash
69 # special case.
70 check(['//a', 'b'], ('', '//', ['//', 'a', 'b']))
71 check(['///a', 'b'], ('', '/', ['/', 'a', 'b']))
72 check(['////a', 'b'], ('', '/', ['/', 'a', 'b']))
73 # Paths which look like NT paths aren't treated specially
74 check(['c:a'], ('', '', ['c:a']))
75 check(['c:\\a'], ('', '', ['c:\\a']))
76 check(['\\a'], ('', '', ['\\a']))
77
78 def test_splitroot(self):
79 f = self.flavour.splitroot
80 self.assertEqual(f(''), ('', '', ''))
81 self.assertEqual(f('a'), ('', '', 'a'))
82 self.assertEqual(f('a/b'), ('', '', 'a/b'))
83 self.assertEqual(f('a/b/'), ('', '', 'a/b/'))
84 self.assertEqual(f('/a'), ('', '/', 'a'))
85 self.assertEqual(f('/a/b'), ('', '/', 'a/b'))
86 self.assertEqual(f('/a/b/'), ('', '/', 'a/b/'))
87 # The root is collapsed when there are redundant slashes
88 # except when there are exactly two leading slashes, which
89 # is a special case in POSIX.
90 self.assertEqual(f('//a'), ('', '//', 'a'))
91 self.assertEqual(f('///a'), ('', '/', 'a'))
92 self.assertEqual(f('///a/b'), ('', '/', 'a/b'))
93 # Paths which look like NT paths aren't treated specially
94 self.assertEqual(f('c:/a/b'), ('', '', 'c:/a/b'))
95 self.assertEqual(f('\\/a/b'), ('', '', '\\/a/b'))
96 self.assertEqual(f('\\a\\b'), ('', '', '\\a\\b'))
97
98
99class NTFlavourTest(_BaseFlavourTest, unittest.TestCase):
100 flavour = pathlib._windows_flavour
101
102 def test_parse_parts(self):
103 check = self._check_parse_parts
104 # First part is anchored
105 check(['c:'], ('c:', '', ['c:']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100106 check(['c:/'], ('c:', '\\', ['c:\\']))
107 check(['/'], ('', '\\', ['\\']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100108 check(['c:a'], ('c:', '', ['c:', 'a']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100109 check(['c:/a'], ('c:', '\\', ['c:\\', 'a']))
110 check(['/a'], ('', '\\', ['\\', 'a']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100111 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100112 check(['//a/b'], ('\\\\a\\b', '\\', ['\\\\a\\b\\']))
113 check(['//a/b/'], ('\\\\a\\b', '\\', ['\\\\a\\b\\']))
114 check(['//a/b/c'], ('\\\\a\\b', '\\', ['\\\\a\\b\\', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100115 # Second part is anchored, so that the first part is ignored
116 check(['a', 'Z:b', 'c'], ('Z:', '', ['Z:', 'b', 'c']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100117 check(['a', 'Z:/b', 'c'], ('Z:', '\\', ['Z:\\', 'b', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100118 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100119 check(['a', '//b/c', 'd'], ('\\\\b\\c', '\\', ['\\\\b\\c\\', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100120 # Collapsing and stripping excess slashes
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100121 check(['a', 'Z://b//c/', 'd/'], ('Z:', '\\', ['Z:\\', 'b', 'c', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100122 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100123 check(['a', '//b/c//', 'd'], ('\\\\b\\c', '\\', ['\\\\b\\c\\', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100124 # Extended paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100125 check(['//?/c:/'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\']))
126 check(['//?/c:/a'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\', 'a']))
127 check(['//?/c:/a', '/b'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\', 'b']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100128 # Extended UNC paths (format is "\\?\UNC\server\share")
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100129 check(['//?/UNC/b/c'], ('\\\\?\\UNC\\b\\c', '\\', ['\\\\?\\UNC\\b\\c\\']))
130 check(['//?/UNC/b/c/d'], ('\\\\?\\UNC\\b\\c', '\\', ['\\\\?\\UNC\\b\\c\\', 'd']))
131 # Second part has a root but not drive
132 check(['a', '/b', 'c'], ('', '\\', ['\\', 'b', 'c']))
133 check(['Z:/a', '/b', 'c'], ('Z:', '\\', ['Z:\\', 'b', 'c']))
134 check(['//?/Z:/a', '/b', 'c'], ('\\\\?\\Z:', '\\', ['\\\\?\\Z:\\', 'b', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100135
136 def test_splitroot(self):
137 f = self.flavour.splitroot
138 self.assertEqual(f(''), ('', '', ''))
139 self.assertEqual(f('a'), ('', '', 'a'))
140 self.assertEqual(f('a\\b'), ('', '', 'a\\b'))
141 self.assertEqual(f('\\a'), ('', '\\', 'a'))
142 self.assertEqual(f('\\a\\b'), ('', '\\', 'a\\b'))
143 self.assertEqual(f('c:a\\b'), ('c:', '', 'a\\b'))
144 self.assertEqual(f('c:\\a\\b'), ('c:', '\\', 'a\\b'))
145 # Redundant slashes in the root are collapsed
146 self.assertEqual(f('\\\\a'), ('', '\\', 'a'))
147 self.assertEqual(f('\\\\\\a/b'), ('', '\\', 'a/b'))
148 self.assertEqual(f('c:\\\\a'), ('c:', '\\', 'a'))
149 self.assertEqual(f('c:\\\\\\a/b'), ('c:', '\\', 'a/b'))
150 # Valid UNC paths
151 self.assertEqual(f('\\\\a\\b'), ('\\\\a\\b', '\\', ''))
152 self.assertEqual(f('\\\\a\\b\\'), ('\\\\a\\b', '\\', ''))
153 self.assertEqual(f('\\\\a\\b\\c\\d'), ('\\\\a\\b', '\\', 'c\\d'))
154 # These are non-UNC paths (according to ntpath.py and test_ntpath)
155 # However, command.com says such paths are invalid, so it's
156 # difficult to know what the right semantics are
157 self.assertEqual(f('\\\\\\a\\b'), ('', '\\', 'a\\b'))
158 self.assertEqual(f('\\\\a'), ('', '\\', 'a'))
159
160
161#
162# Tests for the pure classes
163#
164
165class _BasePurePathTest(object):
166
167 # keys are canonical paths, values are list of tuples of arguments
168 # supposed to produce equal paths
169 equivalences = {
170 'a/b': [
171 ('a', 'b'), ('a/', 'b'), ('a', 'b/'), ('a/', 'b/'),
172 ('a/b/',), ('a//b',), ('a//b//',),
173 # empty components get removed
174 ('', 'a', 'b'), ('a', '', 'b'), ('a', 'b', ''),
175 ],
176 '/b/c/d': [
177 ('a', '/b/c', 'd'), ('a', '///b//c', 'd/'),
178 ('/a', '/b/c', 'd'),
179 # empty components get removed
180 ('/', 'b', '', 'c/d'), ('/', '', 'b/c/d'), ('', '/b/c/d'),
181 ],
182 }
183
184 def setUp(self):
185 p = self.cls('a')
186 self.flavour = p._flavour
187 self.sep = self.flavour.sep
188 self.altsep = self.flavour.altsep
189
190 def test_constructor_common(self):
191 P = self.cls
192 p = P('a')
193 self.assertIsInstance(p, P)
Brett Cannon568be632016-06-10 12:20:49 -0700194 class PathLike:
195 def __fspath__(self):
196 return "a/b/c"
Antoine Pitrou31119e42013-11-22 17:38:12 +0100197 P('a', 'b', 'c')
198 P('/a', 'b', 'c')
199 P('a/b/c')
200 P('/a/b/c')
Brett Cannon568be632016-06-10 12:20:49 -0700201 P(PathLike())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100202 self.assertEqual(P(P('a')), P('a'))
203 self.assertEqual(P(P('a'), 'b'), P('a/b'))
204 self.assertEqual(P(P('a'), P('b')), P('a/b'))
Brett Cannon568be632016-06-10 12:20:49 -0700205 self.assertEqual(P(P('a'), P('b'), P('c')), P(PathLike()))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100206
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200207 def _check_str_subclass(self, *args):
208 # Issue #21127: it should be possible to construct a PurePath object
Serhiy Storchaka6a7b3a72016-04-17 08:32:47 +0300209 # from a str subclass instance, and it then gets converted to
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200210 # a pure str object.
211 class StrSubclass(str):
212 pass
213 P = self.cls
214 p = P(*(StrSubclass(x) for x in args))
215 self.assertEqual(p, P(*args))
216 for part in p.parts:
217 self.assertIs(type(part), str)
218
219 def test_str_subclass_common(self):
220 self._check_str_subclass('')
221 self._check_str_subclass('.')
222 self._check_str_subclass('a')
223 self._check_str_subclass('a/b.txt')
224 self._check_str_subclass('/a/b.txt')
225
Antoine Pitrou31119e42013-11-22 17:38:12 +0100226 def test_join_common(self):
227 P = self.cls
228 p = P('a/b')
229 pp = p.joinpath('c')
230 self.assertEqual(pp, P('a/b/c'))
231 self.assertIs(type(pp), type(p))
232 pp = p.joinpath('c', 'd')
233 self.assertEqual(pp, P('a/b/c/d'))
234 pp = p.joinpath(P('c'))
235 self.assertEqual(pp, P('a/b/c'))
236 pp = p.joinpath('/c')
237 self.assertEqual(pp, P('/c'))
238
239 def test_div_common(self):
240 # Basically the same as joinpath()
241 P = self.cls
242 p = P('a/b')
243 pp = p / 'c'
244 self.assertEqual(pp, P('a/b/c'))
245 self.assertIs(type(pp), type(p))
246 pp = p / 'c/d'
247 self.assertEqual(pp, P('a/b/c/d'))
248 pp = p / 'c' / 'd'
249 self.assertEqual(pp, P('a/b/c/d'))
250 pp = 'c' / p / 'd'
251 self.assertEqual(pp, P('c/a/b/d'))
252 pp = p / P('c')
253 self.assertEqual(pp, P('a/b/c'))
254 pp = p/ '/c'
255 self.assertEqual(pp, P('/c'))
256
257 def _check_str(self, expected, args):
258 p = self.cls(*args)
259 self.assertEqual(str(p), expected.replace('/', self.sep))
260
261 def test_str_common(self):
262 # Canonicalized paths roundtrip
263 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
264 self._check_str(pathstr, (pathstr,))
265 # Special case for the empty path
266 self._check_str('.', ('',))
267 # Other tests for str() are in test_equivalences()
268
269 def test_as_posix_common(self):
270 P = self.cls
271 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
272 self.assertEqual(P(pathstr).as_posix(), pathstr)
273 # Other tests for as_posix() are in test_equivalences()
274
275 def test_as_bytes_common(self):
276 sep = os.fsencode(self.sep)
277 P = self.cls
278 self.assertEqual(bytes(P('a/b')), b'a' + sep + b'b')
279
280 def test_as_uri_common(self):
281 P = self.cls
282 with self.assertRaises(ValueError):
283 P('a').as_uri()
284 with self.assertRaises(ValueError):
285 P().as_uri()
286
287 def test_repr_common(self):
288 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
289 p = self.cls(pathstr)
290 clsname = p.__class__.__name__
291 r = repr(p)
292 # The repr() is in the form ClassName("forward-slashes path")
293 self.assertTrue(r.startswith(clsname + '('), r)
294 self.assertTrue(r.endswith(')'), r)
295 inner = r[len(clsname) + 1 : -1]
296 self.assertEqual(eval(inner), p.as_posix())
297 # The repr() roundtrips
298 q = eval(r, pathlib.__dict__)
299 self.assertIs(q.__class__, p.__class__)
300 self.assertEqual(q, p)
301 self.assertEqual(repr(q), r)
302
303 def test_eq_common(self):
304 P = self.cls
305 self.assertEqual(P('a/b'), P('a/b'))
306 self.assertEqual(P('a/b'), P('a', 'b'))
307 self.assertNotEqual(P('a/b'), P('a'))
308 self.assertNotEqual(P('a/b'), P('/a/b'))
309 self.assertNotEqual(P('a/b'), P())
310 self.assertNotEqual(P('/a/b'), P('/'))
311 self.assertNotEqual(P(), P('/'))
312 self.assertNotEqual(P(), "")
313 self.assertNotEqual(P(), {})
314 self.assertNotEqual(P(), int)
315
316 def test_match_common(self):
317 P = self.cls
318 self.assertRaises(ValueError, P('a').match, '')
319 self.assertRaises(ValueError, P('a').match, '.')
320 # Simple relative pattern
321 self.assertTrue(P('b.py').match('b.py'))
322 self.assertTrue(P('a/b.py').match('b.py'))
323 self.assertTrue(P('/a/b.py').match('b.py'))
324 self.assertFalse(P('a.py').match('b.py'))
325 self.assertFalse(P('b/py').match('b.py'))
326 self.assertFalse(P('/a.py').match('b.py'))
327 self.assertFalse(P('b.py/c').match('b.py'))
328 # Wilcard relative pattern
329 self.assertTrue(P('b.py').match('*.py'))
330 self.assertTrue(P('a/b.py').match('*.py'))
331 self.assertTrue(P('/a/b.py').match('*.py'))
332 self.assertFalse(P('b.pyc').match('*.py'))
333 self.assertFalse(P('b./py').match('*.py'))
334 self.assertFalse(P('b.py/c').match('*.py'))
335 # Multi-part relative pattern
336 self.assertTrue(P('ab/c.py').match('a*/*.py'))
337 self.assertTrue(P('/d/ab/c.py').match('a*/*.py'))
338 self.assertFalse(P('a.py').match('a*/*.py'))
339 self.assertFalse(P('/dab/c.py').match('a*/*.py'))
340 self.assertFalse(P('ab/c.py/d').match('a*/*.py'))
341 # Absolute pattern
342 self.assertTrue(P('/b.py').match('/*.py'))
343 self.assertFalse(P('b.py').match('/*.py'))
344 self.assertFalse(P('a/b.py').match('/*.py'))
345 self.assertFalse(P('/a/b.py').match('/*.py'))
346 # Multi-part absolute pattern
347 self.assertTrue(P('/a/b.py').match('/a/*.py'))
348 self.assertFalse(P('/ab.py').match('/a/*.py'))
349 self.assertFalse(P('/a/b/c.py').match('/a/*.py'))
350
351 def test_ordering_common(self):
352 # Ordering is tuple-alike
353 def assertLess(a, b):
354 self.assertLess(a, b)
355 self.assertGreater(b, a)
356 P = self.cls
357 a = P('a')
358 b = P('a/b')
359 c = P('abc')
360 d = P('b')
361 assertLess(a, b)
362 assertLess(a, c)
363 assertLess(a, d)
364 assertLess(b, c)
365 assertLess(c, d)
366 P = self.cls
367 a = P('/a')
368 b = P('/a/b')
369 c = P('/abc')
370 d = P('/b')
371 assertLess(a, b)
372 assertLess(a, c)
373 assertLess(a, d)
374 assertLess(b, c)
375 assertLess(c, d)
376 with self.assertRaises(TypeError):
377 P() < {}
378
379 def test_parts_common(self):
380 # `parts` returns a tuple
381 sep = self.sep
382 P = self.cls
383 p = P('a/b')
384 parts = p.parts
385 self.assertEqual(parts, ('a', 'b'))
386 # The object gets reused
387 self.assertIs(parts, p.parts)
388 # When the path is absolute, the anchor is a separate part
389 p = P('/a/b')
390 parts = p.parts
391 self.assertEqual(parts, (sep, 'a', 'b'))
392
Brett Cannon568be632016-06-10 12:20:49 -0700393 def test_fspath_common(self):
394 P = self.cls
395 p = P('a/b')
396 self._check_str(p.__fspath__(), ('a/b',))
397 self._check_str(os.fspath(p), ('a/b',))
398
Antoine Pitrou31119e42013-11-22 17:38:12 +0100399 def test_equivalences(self):
400 for k, tuples in self.equivalences.items():
401 canon = k.replace('/', self.sep)
402 posix = k.replace(self.sep, '/')
403 if canon != posix:
404 tuples = tuples + [
405 tuple(part.replace('/', self.sep) for part in t)
406 for t in tuples
407 ]
408 tuples.append((posix, ))
409 pcanon = self.cls(canon)
410 for t in tuples:
411 p = self.cls(*t)
412 self.assertEqual(p, pcanon, "failed with args {}".format(t))
413 self.assertEqual(hash(p), hash(pcanon))
414 self.assertEqual(str(p), canon)
415 self.assertEqual(p.as_posix(), posix)
416
417 def test_parent_common(self):
418 # Relative
419 P = self.cls
420 p = P('a/b/c')
421 self.assertEqual(p.parent, P('a/b'))
422 self.assertEqual(p.parent.parent, P('a'))
423 self.assertEqual(p.parent.parent.parent, P())
424 self.assertEqual(p.parent.parent.parent.parent, P())
425 # Anchored
426 p = P('/a/b/c')
427 self.assertEqual(p.parent, P('/a/b'))
428 self.assertEqual(p.parent.parent, P('/a'))
429 self.assertEqual(p.parent.parent.parent, P('/'))
430 self.assertEqual(p.parent.parent.parent.parent, P('/'))
431
432 def test_parents_common(self):
433 # Relative
434 P = self.cls
435 p = P('a/b/c')
436 par = p.parents
437 self.assertEqual(len(par), 3)
438 self.assertEqual(par[0], P('a/b'))
439 self.assertEqual(par[1], P('a'))
440 self.assertEqual(par[2], P('.'))
441 self.assertEqual(list(par), [P('a/b'), P('a'), P('.')])
442 with self.assertRaises(IndexError):
443 par[-1]
444 with self.assertRaises(IndexError):
445 par[3]
446 with self.assertRaises(TypeError):
447 par[0] = p
448 # Anchored
449 p = P('/a/b/c')
450 par = p.parents
451 self.assertEqual(len(par), 3)
452 self.assertEqual(par[0], P('/a/b'))
453 self.assertEqual(par[1], P('/a'))
454 self.assertEqual(par[2], P('/'))
455 self.assertEqual(list(par), [P('/a/b'), P('/a'), P('/')])
456 with self.assertRaises(IndexError):
457 par[3]
458
459 def test_drive_common(self):
460 P = self.cls
461 self.assertEqual(P('a/b').drive, '')
462 self.assertEqual(P('/a/b').drive, '')
463 self.assertEqual(P('').drive, '')
464
465 def test_root_common(self):
466 P = self.cls
467 sep = self.sep
468 self.assertEqual(P('').root, '')
469 self.assertEqual(P('a/b').root, '')
470 self.assertEqual(P('/').root, sep)
471 self.assertEqual(P('/a/b').root, sep)
472
473 def test_anchor_common(self):
474 P = self.cls
475 sep = self.sep
476 self.assertEqual(P('').anchor, '')
477 self.assertEqual(P('a/b').anchor, '')
478 self.assertEqual(P('/').anchor, sep)
479 self.assertEqual(P('/a/b').anchor, sep)
480
481 def test_name_common(self):
482 P = self.cls
483 self.assertEqual(P('').name, '')
484 self.assertEqual(P('.').name, '')
485 self.assertEqual(P('/').name, '')
486 self.assertEqual(P('a/b').name, 'b')
487 self.assertEqual(P('/a/b').name, 'b')
488 self.assertEqual(P('/a/b/.').name, 'b')
489 self.assertEqual(P('a/b.py').name, 'b.py')
490 self.assertEqual(P('/a/b.py').name, 'b.py')
491
492 def test_suffix_common(self):
493 P = self.cls
494 self.assertEqual(P('').suffix, '')
495 self.assertEqual(P('.').suffix, '')
496 self.assertEqual(P('..').suffix, '')
497 self.assertEqual(P('/').suffix, '')
498 self.assertEqual(P('a/b').suffix, '')
499 self.assertEqual(P('/a/b').suffix, '')
500 self.assertEqual(P('/a/b/.').suffix, '')
501 self.assertEqual(P('a/b.py').suffix, '.py')
502 self.assertEqual(P('/a/b.py').suffix, '.py')
503 self.assertEqual(P('a/.hgrc').suffix, '')
504 self.assertEqual(P('/a/.hgrc').suffix, '')
505 self.assertEqual(P('a/.hg.rc').suffix, '.rc')
506 self.assertEqual(P('/a/.hg.rc').suffix, '.rc')
507 self.assertEqual(P('a/b.tar.gz').suffix, '.gz')
508 self.assertEqual(P('/a/b.tar.gz').suffix, '.gz')
509 self.assertEqual(P('a/Some name. Ending with a dot.').suffix, '')
510 self.assertEqual(P('/a/Some name. Ending with a dot.').suffix, '')
511
512 def test_suffixes_common(self):
513 P = self.cls
514 self.assertEqual(P('').suffixes, [])
515 self.assertEqual(P('.').suffixes, [])
516 self.assertEqual(P('/').suffixes, [])
517 self.assertEqual(P('a/b').suffixes, [])
518 self.assertEqual(P('/a/b').suffixes, [])
519 self.assertEqual(P('/a/b/.').suffixes, [])
520 self.assertEqual(P('a/b.py').suffixes, ['.py'])
521 self.assertEqual(P('/a/b.py').suffixes, ['.py'])
522 self.assertEqual(P('a/.hgrc').suffixes, [])
523 self.assertEqual(P('/a/.hgrc').suffixes, [])
524 self.assertEqual(P('a/.hg.rc').suffixes, ['.rc'])
525 self.assertEqual(P('/a/.hg.rc').suffixes, ['.rc'])
526 self.assertEqual(P('a/b.tar.gz').suffixes, ['.tar', '.gz'])
527 self.assertEqual(P('/a/b.tar.gz').suffixes, ['.tar', '.gz'])
528 self.assertEqual(P('a/Some name. Ending with a dot.').suffixes, [])
529 self.assertEqual(P('/a/Some name. Ending with a dot.').suffixes, [])
530
531 def test_stem_common(self):
532 P = self.cls
533 self.assertEqual(P('').stem, '')
534 self.assertEqual(P('.').stem, '')
535 self.assertEqual(P('..').stem, '..')
536 self.assertEqual(P('/').stem, '')
537 self.assertEqual(P('a/b').stem, 'b')
538 self.assertEqual(P('a/b.py').stem, 'b')
539 self.assertEqual(P('a/.hgrc').stem, '.hgrc')
540 self.assertEqual(P('a/.hg.rc').stem, '.hg')
541 self.assertEqual(P('a/b.tar.gz').stem, 'b.tar')
542 self.assertEqual(P('a/Some name. Ending with a dot.').stem,
543 'Some name. Ending with a dot.')
544
545 def test_with_name_common(self):
546 P = self.cls
547 self.assertEqual(P('a/b').with_name('d.xml'), P('a/d.xml'))
548 self.assertEqual(P('/a/b').with_name('d.xml'), P('/a/d.xml'))
549 self.assertEqual(P('a/b.py').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/Dot ending.').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.assertRaises(ValueError, P('').with_name, 'd.xml')
554 self.assertRaises(ValueError, P('.').with_name, 'd.xml')
555 self.assertRaises(ValueError, P('/').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400556 self.assertRaises(ValueError, P('a/b').with_name, '')
557 self.assertRaises(ValueError, P('a/b').with_name, '/c')
558 self.assertRaises(ValueError, P('a/b').with_name, 'c/')
559 self.assertRaises(ValueError, P('a/b').with_name, 'c/d')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100560
561 def test_with_suffix_common(self):
562 P = self.cls
563 self.assertEqual(P('a/b').with_suffix('.gz'), P('a/b.gz'))
564 self.assertEqual(P('/a/b').with_suffix('.gz'), P('/a/b.gz'))
565 self.assertEqual(P('a/b.py').with_suffix('.gz'), P('a/b.gz'))
566 self.assertEqual(P('/a/b.py').with_suffix('.gz'), P('/a/b.gz'))
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400567 # Stripping suffix
568 self.assertEqual(P('a/b.py').with_suffix(''), P('a/b'))
569 self.assertEqual(P('/a/b').with_suffix(''), P('/a/b'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100570 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100571 self.assertRaises(ValueError, P('').with_suffix, '.gz')
572 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
573 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100574 # Invalid suffix
575 self.assertRaises(ValueError, P('a/b').with_suffix, 'gz')
576 self.assertRaises(ValueError, P('a/b').with_suffix, '/')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400577 self.assertRaises(ValueError, P('a/b').with_suffix, '.')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100578 self.assertRaises(ValueError, P('a/b').with_suffix, '/.gz')
579 self.assertRaises(ValueError, P('a/b').with_suffix, 'c/d')
580 self.assertRaises(ValueError, P('a/b').with_suffix, '.c/.d')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400581 self.assertRaises(ValueError, P('a/b').with_suffix, './.d')
582 self.assertRaises(ValueError, P('a/b').with_suffix, '.d/.')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100583
584 def test_relative_to_common(self):
585 P = self.cls
586 p = P('a/b')
587 self.assertRaises(TypeError, p.relative_to)
Antoine Pitrou156b3612013-12-28 19:49:04 +0100588 self.assertRaises(TypeError, p.relative_to, b'a')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100589 self.assertEqual(p.relative_to(P()), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100590 self.assertEqual(p.relative_to(''), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100591 self.assertEqual(p.relative_to(P('a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100592 self.assertEqual(p.relative_to('a'), P('b'))
593 self.assertEqual(p.relative_to('a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100594 self.assertEqual(p.relative_to(P('a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100595 self.assertEqual(p.relative_to('a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100596 # With several args
597 self.assertEqual(p.relative_to('a', 'b'), P())
598 # Unrelated paths
599 self.assertRaises(ValueError, p.relative_to, P('c'))
600 self.assertRaises(ValueError, p.relative_to, P('a/b/c'))
601 self.assertRaises(ValueError, p.relative_to, P('a/c'))
602 self.assertRaises(ValueError, p.relative_to, P('/a'))
603 p = P('/a/b')
604 self.assertEqual(p.relative_to(P('/')), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100605 self.assertEqual(p.relative_to('/'), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100606 self.assertEqual(p.relative_to(P('/a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100607 self.assertEqual(p.relative_to('/a'), P('b'))
608 self.assertEqual(p.relative_to('/a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100609 self.assertEqual(p.relative_to(P('/a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100610 self.assertEqual(p.relative_to('/a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100611 # Unrelated paths
612 self.assertRaises(ValueError, p.relative_to, P('/c'))
613 self.assertRaises(ValueError, p.relative_to, P('/a/b/c'))
614 self.assertRaises(ValueError, p.relative_to, P('/a/c'))
615 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100616 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100617 self.assertRaises(ValueError, p.relative_to, P('a'))
618
619 def test_pickling_common(self):
620 P = self.cls
621 p = P('/a/b')
622 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
623 dumped = pickle.dumps(p, proto)
624 pp = pickle.loads(dumped)
625 self.assertIs(pp.__class__, p.__class__)
626 self.assertEqual(pp, p)
627 self.assertEqual(hash(pp), hash(p))
628 self.assertEqual(str(pp), str(p))
629
630
631class PurePosixPathTest(_BasePurePathTest, unittest.TestCase):
632 cls = pathlib.PurePosixPath
633
634 def test_root(self):
635 P = self.cls
636 self.assertEqual(P('/a/b').root, '/')
637 self.assertEqual(P('///a/b').root, '/')
638 # POSIX special case for two leading slashes
639 self.assertEqual(P('//a/b').root, '//')
640
641 def test_eq(self):
642 P = self.cls
643 self.assertNotEqual(P('a/b'), P('A/b'))
644 self.assertEqual(P('/a'), P('///a'))
645 self.assertNotEqual(P('/a'), P('//a'))
646
647 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100648 P = self.cls
649 self.assertEqual(P('/').as_uri(), 'file:///')
650 self.assertEqual(P('/a/b.c').as_uri(), 'file:///a/b.c')
651 self.assertEqual(P('/a/b%#c').as_uri(), 'file:///a/b%25%23c')
Antoine Pitrou29eac422013-11-22 17:57:03 +0100652
653 def test_as_uri_non_ascii(self):
654 from urllib.parse import quote_from_bytes
655 P = self.cls
656 try:
657 os.fsencode('\xe9')
658 except UnicodeEncodeError:
659 self.skipTest("\\xe9 cannot be encoded to the filesystem encoding")
Antoine Pitrou31119e42013-11-22 17:38:12 +0100660 self.assertEqual(P('/a/b\xe9').as_uri(),
661 'file:///a/b' + quote_from_bytes(os.fsencode('\xe9')))
662
663 def test_match(self):
664 P = self.cls
665 self.assertFalse(P('A.py').match('a.PY'))
666
667 def test_is_absolute(self):
668 P = self.cls
669 self.assertFalse(P().is_absolute())
670 self.assertFalse(P('a').is_absolute())
671 self.assertFalse(P('a/b/').is_absolute())
672 self.assertTrue(P('/').is_absolute())
673 self.assertTrue(P('/a').is_absolute())
674 self.assertTrue(P('/a/b/').is_absolute())
675 self.assertTrue(P('//a').is_absolute())
676 self.assertTrue(P('//a/b').is_absolute())
677
678 def test_is_reserved(self):
679 P = self.cls
680 self.assertIs(False, P('').is_reserved())
681 self.assertIs(False, P('/').is_reserved())
682 self.assertIs(False, P('/foo/bar').is_reserved())
683 self.assertIs(False, P('/dev/con/PRN/NUL').is_reserved())
684
685 def test_join(self):
686 P = self.cls
687 p = P('//a')
688 pp = p.joinpath('b')
689 self.assertEqual(pp, P('//a/b'))
690 pp = P('/a').joinpath('//c')
691 self.assertEqual(pp, P('//c'))
692 pp = P('//a').joinpath('/c')
693 self.assertEqual(pp, P('/c'))
694
695 def test_div(self):
696 # Basically the same as joinpath()
697 P = self.cls
698 p = P('//a')
699 pp = p / 'b'
700 self.assertEqual(pp, P('//a/b'))
701 pp = P('/a') / '//c'
702 self.assertEqual(pp, P('//c'))
703 pp = P('//a') / '/c'
704 self.assertEqual(pp, P('/c'))
705
706
707class PureWindowsPathTest(_BasePurePathTest, unittest.TestCase):
708 cls = pathlib.PureWindowsPath
709
710 equivalences = _BasePurePathTest.equivalences.copy()
711 equivalences.update({
712 'c:a': [ ('c:', 'a'), ('c:', 'a/'), ('/', 'c:', 'a') ],
713 'c:/a': [
714 ('c:/', 'a'), ('c:', '/', 'a'), ('c:', '/a'),
715 ('/z', 'c:/', 'a'), ('//x/y', 'c:/', 'a'),
716 ],
717 '//a/b/': [ ('//a/b',) ],
718 '//a/b/c': [
719 ('//a/b', 'c'), ('//a/b/', 'c'),
720 ],
721 })
722
723 def test_str(self):
724 p = self.cls('a/b/c')
725 self.assertEqual(str(p), 'a\\b\\c')
726 p = self.cls('c:/a/b/c')
727 self.assertEqual(str(p), 'c:\\a\\b\\c')
728 p = self.cls('//a/b')
729 self.assertEqual(str(p), '\\\\a\\b\\')
730 p = self.cls('//a/b/c')
731 self.assertEqual(str(p), '\\\\a\\b\\c')
732 p = self.cls('//a/b/c/d')
733 self.assertEqual(str(p), '\\\\a\\b\\c\\d')
734
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200735 def test_str_subclass(self):
736 self._check_str_subclass('c:')
737 self._check_str_subclass('c:a')
738 self._check_str_subclass('c:a\\b.txt')
739 self._check_str_subclass('c:\\')
740 self._check_str_subclass('c:\\a')
741 self._check_str_subclass('c:\\a\\b.txt')
742 self._check_str_subclass('\\\\some\\share')
743 self._check_str_subclass('\\\\some\\share\\a')
744 self._check_str_subclass('\\\\some\\share\\a\\b.txt')
745
Antoine Pitrou31119e42013-11-22 17:38:12 +0100746 def test_eq(self):
747 P = self.cls
748 self.assertEqual(P('c:a/b'), P('c:a/b'))
749 self.assertEqual(P('c:a/b'), P('c:', 'a', 'b'))
750 self.assertNotEqual(P('c:a/b'), P('d:a/b'))
751 self.assertNotEqual(P('c:a/b'), P('c:/a/b'))
752 self.assertNotEqual(P('/a/b'), P('c:/a/b'))
753 # Case-insensitivity
754 self.assertEqual(P('a/B'), P('A/b'))
755 self.assertEqual(P('C:a/B'), P('c:A/b'))
756 self.assertEqual(P('//Some/SHARE/a/B'), P('//somE/share/A/b'))
757
758 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100759 P = self.cls
760 with self.assertRaises(ValueError):
761 P('/a/b').as_uri()
762 with self.assertRaises(ValueError):
763 P('c:a/b').as_uri()
764 self.assertEqual(P('c:/').as_uri(), 'file:///c:/')
765 self.assertEqual(P('c:/a/b.c').as_uri(), 'file:///c:/a/b.c')
766 self.assertEqual(P('c:/a/b%#c').as_uri(), 'file:///c:/a/b%25%23c')
767 self.assertEqual(P('c:/a/b\xe9').as_uri(), 'file:///c:/a/b%C3%A9')
768 self.assertEqual(P('//some/share/').as_uri(), 'file://some/share/')
769 self.assertEqual(P('//some/share/a/b.c').as_uri(),
770 'file://some/share/a/b.c')
771 self.assertEqual(P('//some/share/a/b%#c\xe9').as_uri(),
772 'file://some/share/a/b%25%23c%C3%A9')
773
774 def test_match_common(self):
775 P = self.cls
776 # Absolute patterns
777 self.assertTrue(P('c:/b.py').match('/*.py'))
778 self.assertTrue(P('c:/b.py').match('c:*.py'))
779 self.assertTrue(P('c:/b.py').match('c:/*.py'))
780 self.assertFalse(P('d:/b.py').match('c:/*.py')) # wrong drive
781 self.assertFalse(P('b.py').match('/*.py'))
782 self.assertFalse(P('b.py').match('c:*.py'))
783 self.assertFalse(P('b.py').match('c:/*.py'))
784 self.assertFalse(P('c:b.py').match('/*.py'))
785 self.assertFalse(P('c:b.py').match('c:/*.py'))
786 self.assertFalse(P('/b.py').match('c:*.py'))
787 self.assertFalse(P('/b.py').match('c:/*.py'))
788 # UNC patterns
789 self.assertTrue(P('//some/share/a.py').match('/*.py'))
790 self.assertTrue(P('//some/share/a.py').match('//some/share/*.py'))
791 self.assertFalse(P('//other/share/a.py').match('//some/share/*.py'))
792 self.assertFalse(P('//some/share/a/b.py').match('//some/share/*.py'))
793 # Case-insensitivity
794 self.assertTrue(P('B.py').match('b.PY'))
795 self.assertTrue(P('c:/a/B.Py').match('C:/A/*.pY'))
796 self.assertTrue(P('//Some/Share/B.Py').match('//somE/sharE/*.pY'))
797
798 def test_ordering_common(self):
799 # Case-insensitivity
800 def assertOrderedEqual(a, b):
801 self.assertLessEqual(a, b)
802 self.assertGreaterEqual(b, a)
803 P = self.cls
804 p = P('c:A/b')
805 q = P('C:a/B')
806 assertOrderedEqual(p, q)
807 self.assertFalse(p < q)
808 self.assertFalse(p > q)
809 p = P('//some/Share/A/b')
810 q = P('//Some/SHARE/a/B')
811 assertOrderedEqual(p, q)
812 self.assertFalse(p < q)
813 self.assertFalse(p > q)
814
815 def test_parts(self):
816 P = self.cls
817 p = P('c:a/b')
818 parts = p.parts
819 self.assertEqual(parts, ('c:', 'a', 'b'))
820 p = P('c:/a/b')
821 parts = p.parts
822 self.assertEqual(parts, ('c:\\', 'a', 'b'))
823 p = P('//a/b/c/d')
824 parts = p.parts
825 self.assertEqual(parts, ('\\\\a\\b\\', 'c', 'd'))
826
827 def test_parent(self):
828 # Anchored
829 P = self.cls
830 p = P('z:a/b/c')
831 self.assertEqual(p.parent, P('z:a/b'))
832 self.assertEqual(p.parent.parent, P('z:a'))
833 self.assertEqual(p.parent.parent.parent, P('z:'))
834 self.assertEqual(p.parent.parent.parent.parent, P('z:'))
835 p = P('z:/a/b/c')
836 self.assertEqual(p.parent, P('z:/a/b'))
837 self.assertEqual(p.parent.parent, P('z:/a'))
838 self.assertEqual(p.parent.parent.parent, P('z:/'))
839 self.assertEqual(p.parent.parent.parent.parent, P('z:/'))
840 p = P('//a/b/c/d')
841 self.assertEqual(p.parent, P('//a/b/c'))
842 self.assertEqual(p.parent.parent, P('//a/b'))
843 self.assertEqual(p.parent.parent.parent, P('//a/b'))
844
845 def test_parents(self):
846 # Anchored
847 P = self.cls
848 p = P('z:a/b/')
849 par = p.parents
850 self.assertEqual(len(par), 2)
851 self.assertEqual(par[0], P('z:a'))
852 self.assertEqual(par[1], P('z:'))
853 self.assertEqual(list(par), [P('z:a'), P('z:')])
854 with self.assertRaises(IndexError):
855 par[2]
856 p = P('z:/a/b/')
857 par = p.parents
858 self.assertEqual(len(par), 2)
859 self.assertEqual(par[0], P('z:/a'))
860 self.assertEqual(par[1], P('z:/'))
861 self.assertEqual(list(par), [P('z:/a'), P('z:/')])
862 with self.assertRaises(IndexError):
863 par[2]
864 p = P('//a/b/c/d')
865 par = p.parents
866 self.assertEqual(len(par), 2)
867 self.assertEqual(par[0], P('//a/b/c'))
868 self.assertEqual(par[1], P('//a/b'))
869 self.assertEqual(list(par), [P('//a/b/c'), P('//a/b')])
870 with self.assertRaises(IndexError):
871 par[2]
872
873 def test_drive(self):
874 P = self.cls
875 self.assertEqual(P('c:').drive, 'c:')
876 self.assertEqual(P('c:a/b').drive, 'c:')
877 self.assertEqual(P('c:/').drive, 'c:')
878 self.assertEqual(P('c:/a/b/').drive, 'c:')
879 self.assertEqual(P('//a/b').drive, '\\\\a\\b')
880 self.assertEqual(P('//a/b/').drive, '\\\\a\\b')
881 self.assertEqual(P('//a/b/c/d').drive, '\\\\a\\b')
882
883 def test_root(self):
884 P = self.cls
885 self.assertEqual(P('c:').root, '')
886 self.assertEqual(P('c:a/b').root, '')
887 self.assertEqual(P('c:/').root, '\\')
888 self.assertEqual(P('c:/a/b/').root, '\\')
889 self.assertEqual(P('//a/b').root, '\\')
890 self.assertEqual(P('//a/b/').root, '\\')
891 self.assertEqual(P('//a/b/c/d').root, '\\')
892
893 def test_anchor(self):
894 P = self.cls
895 self.assertEqual(P('c:').anchor, 'c:')
896 self.assertEqual(P('c:a/b').anchor, 'c:')
897 self.assertEqual(P('c:/').anchor, 'c:\\')
898 self.assertEqual(P('c:/a/b/').anchor, 'c:\\')
899 self.assertEqual(P('//a/b').anchor, '\\\\a\\b\\')
900 self.assertEqual(P('//a/b/').anchor, '\\\\a\\b\\')
901 self.assertEqual(P('//a/b/c/d').anchor, '\\\\a\\b\\')
902
903 def test_name(self):
904 P = self.cls
905 self.assertEqual(P('c:').name, '')
906 self.assertEqual(P('c:/').name, '')
907 self.assertEqual(P('c:a/b').name, 'b')
908 self.assertEqual(P('c:/a/b').name, 'b')
909 self.assertEqual(P('c:a/b.py').name, 'b.py')
910 self.assertEqual(P('c:/a/b.py').name, 'b.py')
911 self.assertEqual(P('//My.py/Share.php').name, '')
912 self.assertEqual(P('//My.py/Share.php/a/b').name, 'b')
913
914 def test_suffix(self):
915 P = self.cls
916 self.assertEqual(P('c:').suffix, '')
917 self.assertEqual(P('c:/').suffix, '')
918 self.assertEqual(P('c:a/b').suffix, '')
919 self.assertEqual(P('c:/a/b').suffix, '')
920 self.assertEqual(P('c:a/b.py').suffix, '.py')
921 self.assertEqual(P('c:/a/b.py').suffix, '.py')
922 self.assertEqual(P('c:a/.hgrc').suffix, '')
923 self.assertEqual(P('c:/a/.hgrc').suffix, '')
924 self.assertEqual(P('c:a/.hg.rc').suffix, '.rc')
925 self.assertEqual(P('c:/a/.hg.rc').suffix, '.rc')
926 self.assertEqual(P('c:a/b.tar.gz').suffix, '.gz')
927 self.assertEqual(P('c:/a/b.tar.gz').suffix, '.gz')
928 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffix, '')
929 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffix, '')
930 self.assertEqual(P('//My.py/Share.php').suffix, '')
931 self.assertEqual(P('//My.py/Share.php/a/b').suffix, '')
932
933 def test_suffixes(self):
934 P = self.cls
935 self.assertEqual(P('c:').suffixes, [])
936 self.assertEqual(P('c:/').suffixes, [])
937 self.assertEqual(P('c:a/b').suffixes, [])
938 self.assertEqual(P('c:/a/b').suffixes, [])
939 self.assertEqual(P('c:a/b.py').suffixes, ['.py'])
940 self.assertEqual(P('c:/a/b.py').suffixes, ['.py'])
941 self.assertEqual(P('c:a/.hgrc').suffixes, [])
942 self.assertEqual(P('c:/a/.hgrc').suffixes, [])
943 self.assertEqual(P('c:a/.hg.rc').suffixes, ['.rc'])
944 self.assertEqual(P('c:/a/.hg.rc').suffixes, ['.rc'])
945 self.assertEqual(P('c:a/b.tar.gz').suffixes, ['.tar', '.gz'])
946 self.assertEqual(P('c:/a/b.tar.gz').suffixes, ['.tar', '.gz'])
947 self.assertEqual(P('//My.py/Share.php').suffixes, [])
948 self.assertEqual(P('//My.py/Share.php/a/b').suffixes, [])
949 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffixes, [])
950 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffixes, [])
951
952 def test_stem(self):
953 P = self.cls
954 self.assertEqual(P('c:').stem, '')
955 self.assertEqual(P('c:.').stem, '')
956 self.assertEqual(P('c:..').stem, '..')
957 self.assertEqual(P('c:/').stem, '')
958 self.assertEqual(P('c:a/b').stem, 'b')
959 self.assertEqual(P('c:a/b.py').stem, 'b')
960 self.assertEqual(P('c:a/.hgrc').stem, '.hgrc')
961 self.assertEqual(P('c:a/.hg.rc').stem, '.hg')
962 self.assertEqual(P('c:a/b.tar.gz').stem, 'b.tar')
963 self.assertEqual(P('c:a/Some name. Ending with a dot.').stem,
964 'Some name. Ending with a dot.')
965
966 def test_with_name(self):
967 P = self.cls
968 self.assertEqual(P('c:a/b').with_name('d.xml'), P('c:a/d.xml'))
969 self.assertEqual(P('c:/a/b').with_name('d.xml'), P('c:/a/d.xml'))
970 self.assertEqual(P('c:a/Dot ending.').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.assertRaises(ValueError, P('c:').with_name, 'd.xml')
973 self.assertRaises(ValueError, P('c:/').with_name, 'd.xml')
974 self.assertRaises(ValueError, P('//My/Share').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400975 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:')
976 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:e')
977 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:/e')
978 self.assertRaises(ValueError, P('c:a/b').with_name, '//My/Share')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100979
980 def test_with_suffix(self):
981 P = self.cls
982 self.assertEqual(P('c:a/b').with_suffix('.gz'), P('c:a/b.gz'))
983 self.assertEqual(P('c:/a/b').with_suffix('.gz'), P('c:/a/b.gz'))
984 self.assertEqual(P('c:a/b.py').with_suffix('.gz'), P('c:a/b.gz'))
985 self.assertEqual(P('c:/a/b.py').with_suffix('.gz'), P('c:/a/b.gz'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100986 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100987 self.assertRaises(ValueError, P('').with_suffix, '.gz')
988 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
989 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
990 self.assertRaises(ValueError, P('//My/Share').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100991 # Invalid suffix
992 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'gz')
993 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/')
994 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\')
995 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:')
996 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/.gz')
997 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\.gz')
998 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:.gz')
999 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c/d')
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')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001003
1004 def test_relative_to(self):
1005 P = self.cls
Antoine Pitrou156b3612013-12-28 19:49:04 +01001006 p = P('C:Foo/Bar')
1007 self.assertEqual(p.relative_to(P('c:')), P('Foo/Bar'))
1008 self.assertEqual(p.relative_to('c:'), P('Foo/Bar'))
1009 self.assertEqual(p.relative_to(P('c:foO')), P('Bar'))
1010 self.assertEqual(p.relative_to('c:foO'), P('Bar'))
1011 self.assertEqual(p.relative_to('c:foO/'), P('Bar'))
1012 self.assertEqual(p.relative_to(P('c:foO/baR')), P())
1013 self.assertEqual(p.relative_to('c:foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001014 # Unrelated paths
1015 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +01001016 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001017 self.assertRaises(ValueError, p.relative_to, P('d:'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001018 self.assertRaises(ValueError, p.relative_to, P('/'))
1019 self.assertRaises(ValueError, p.relative_to, P('Foo'))
1020 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1021 self.assertRaises(ValueError, p.relative_to, P('C:/Foo'))
1022 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Bar/Baz'))
1023 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Baz'))
1024 p = P('C:/Foo/Bar')
1025 self.assertEqual(p.relative_to(P('c:')), P('/Foo/Bar'))
1026 self.assertEqual(p.relative_to('c:'), P('/Foo/Bar'))
1027 self.assertEqual(str(p.relative_to(P('c:'))), '\\Foo\\Bar')
1028 self.assertEqual(str(p.relative_to('c:')), '\\Foo\\Bar')
1029 self.assertEqual(p.relative_to(P('c:/')), P('Foo/Bar'))
1030 self.assertEqual(p.relative_to('c:/'), P('Foo/Bar'))
1031 self.assertEqual(p.relative_to(P('c:/foO')), P('Bar'))
1032 self.assertEqual(p.relative_to('c:/foO'), P('Bar'))
1033 self.assertEqual(p.relative_to('c:/foO/'), P('Bar'))
1034 self.assertEqual(p.relative_to(P('c:/foO/baR')), P())
1035 self.assertEqual(p.relative_to('c:/foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001036 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001037 self.assertRaises(ValueError, p.relative_to, P('C:/Baz'))
1038 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Bar/Baz'))
1039 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Baz'))
1040 self.assertRaises(ValueError, p.relative_to, P('C:Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001041 self.assertRaises(ValueError, p.relative_to, P('d:'))
1042 self.assertRaises(ValueError, p.relative_to, P('d:/'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001043 self.assertRaises(ValueError, p.relative_to, P('/'))
1044 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1045 self.assertRaises(ValueError, p.relative_to, P('//C/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001046 # UNC paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001047 p = P('//Server/Share/Foo/Bar')
1048 self.assertEqual(p.relative_to(P('//sErver/sHare')), P('Foo/Bar'))
1049 self.assertEqual(p.relative_to('//sErver/sHare'), P('Foo/Bar'))
1050 self.assertEqual(p.relative_to('//sErver/sHare/'), P('Foo/Bar'))
1051 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo')), P('Bar'))
1052 self.assertEqual(p.relative_to('//sErver/sHare/Foo'), P('Bar'))
1053 self.assertEqual(p.relative_to('//sErver/sHare/Foo/'), P('Bar'))
1054 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo/Bar')), P())
1055 self.assertEqual(p.relative_to('//sErver/sHare/Foo/Bar'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001056 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001057 self.assertRaises(ValueError, p.relative_to, P('/Server/Share/Foo'))
1058 self.assertRaises(ValueError, p.relative_to, P('c:/Server/Share/Foo'))
1059 self.assertRaises(ValueError, p.relative_to, P('//z/Share/Foo'))
1060 self.assertRaises(ValueError, p.relative_to, P('//Server/z/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001061
1062 def test_is_absolute(self):
1063 P = self.cls
1064 # Under NT, only paths with both a drive and a root are absolute
1065 self.assertFalse(P().is_absolute())
1066 self.assertFalse(P('a').is_absolute())
1067 self.assertFalse(P('a/b/').is_absolute())
1068 self.assertFalse(P('/').is_absolute())
1069 self.assertFalse(P('/a').is_absolute())
1070 self.assertFalse(P('/a/b/').is_absolute())
1071 self.assertFalse(P('c:').is_absolute())
1072 self.assertFalse(P('c:a').is_absolute())
1073 self.assertFalse(P('c:a/b/').is_absolute())
1074 self.assertTrue(P('c:/').is_absolute())
1075 self.assertTrue(P('c:/a').is_absolute())
1076 self.assertTrue(P('c:/a/b/').is_absolute())
1077 # UNC paths are absolute by definition
1078 self.assertTrue(P('//a/b').is_absolute())
1079 self.assertTrue(P('//a/b/').is_absolute())
1080 self.assertTrue(P('//a/b/c').is_absolute())
1081 self.assertTrue(P('//a/b/c/d').is_absolute())
1082
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001083 def test_join(self):
1084 P = self.cls
1085 p = P('C:/a/b')
1086 pp = p.joinpath('x/y')
1087 self.assertEqual(pp, P('C:/a/b/x/y'))
1088 pp = p.joinpath('/x/y')
1089 self.assertEqual(pp, P('C:/x/y'))
1090 # Joining with a different drive => the first path is ignored, even
1091 # if the second path is relative.
1092 pp = p.joinpath('D:x/y')
1093 self.assertEqual(pp, P('D:x/y'))
1094 pp = p.joinpath('D:/x/y')
1095 self.assertEqual(pp, P('D:/x/y'))
1096 pp = p.joinpath('//host/share/x/y')
1097 self.assertEqual(pp, P('//host/share/x/y'))
1098 # Joining with the same drive => the first path is appended to if
1099 # the second path is relative.
1100 pp = p.joinpath('c:x/y')
1101 self.assertEqual(pp, P('C:/a/b/x/y'))
1102 pp = p.joinpath('c:/x/y')
1103 self.assertEqual(pp, P('C:/x/y'))
1104
1105 def test_div(self):
1106 # Basically the same as joinpath()
1107 P = self.cls
1108 p = P('C:/a/b')
1109 self.assertEqual(p / 'x/y', P('C:/a/b/x/y'))
1110 self.assertEqual(p / 'x' / 'y', P('C:/a/b/x/y'))
1111 self.assertEqual(p / '/x/y', P('C:/x/y'))
1112 self.assertEqual(p / '/x' / 'y', P('C:/x/y'))
1113 # Joining with a different drive => the first path is ignored, even
1114 # if the second path is relative.
1115 self.assertEqual(p / 'D:x/y', P('D:x/y'))
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 / '//host/share/x/y', P('//host/share/x/y'))
1120 # Joining with the same drive => the first path is appended to if
1121 # the second path is relative.
Serhiy Storchaka010ff582013-12-06 17:25:51 +02001122 self.assertEqual(p / 'c:x/y', P('C:/a/b/x/y'))
1123 self.assertEqual(p / 'c:/x/y', P('C:/x/y'))
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001124
Antoine Pitrou31119e42013-11-22 17:38:12 +01001125 def test_is_reserved(self):
1126 P = self.cls
1127 self.assertIs(False, P('').is_reserved())
1128 self.assertIs(False, P('/').is_reserved())
1129 self.assertIs(False, P('/foo/bar').is_reserved())
1130 self.assertIs(True, P('con').is_reserved())
1131 self.assertIs(True, P('NUL').is_reserved())
1132 self.assertIs(True, P('NUL.txt').is_reserved())
1133 self.assertIs(True, P('com1').is_reserved())
1134 self.assertIs(True, P('com9.bar').is_reserved())
1135 self.assertIs(False, P('bar.com9').is_reserved())
1136 self.assertIs(True, P('lpt1').is_reserved())
1137 self.assertIs(True, P('lpt9.bar').is_reserved())
1138 self.assertIs(False, P('bar.lpt9').is_reserved())
1139 # Only the last component matters
1140 self.assertIs(False, P('c:/NUL/con/baz').is_reserved())
1141 # UNC paths are never reserved
1142 self.assertIs(False, P('//my/share/nul/con/aux').is_reserved())
1143
Antoine Pitrou31119e42013-11-22 17:38:12 +01001144class PurePathTest(_BasePurePathTest, unittest.TestCase):
1145 cls = pathlib.PurePath
1146
1147 def test_concrete_class(self):
1148 p = self.cls('a')
1149 self.assertIs(type(p),
1150 pathlib.PureWindowsPath if os.name == 'nt' else pathlib.PurePosixPath)
1151
1152 def test_different_flavours_unequal(self):
1153 p = pathlib.PurePosixPath('a')
1154 q = pathlib.PureWindowsPath('a')
1155 self.assertNotEqual(p, q)
1156
1157 def test_different_flavours_unordered(self):
1158 p = pathlib.PurePosixPath('a')
1159 q = pathlib.PureWindowsPath('a')
1160 with self.assertRaises(TypeError):
1161 p < q
1162 with self.assertRaises(TypeError):
1163 p <= q
1164 with self.assertRaises(TypeError):
1165 p > q
1166 with self.assertRaises(TypeError):
1167 p >= q
1168
1169
1170#
1171# Tests for the concrete classes
1172#
1173
1174# Make sure any symbolic links in the base test path are resolved
1175BASE = os.path.realpath(TESTFN)
1176join = lambda *x: os.path.join(BASE, *x)
1177rel_join = lambda *x: os.path.join(TESTFN, *x)
1178
Antoine Pitrou31119e42013-11-22 17:38:12 +01001179only_nt = unittest.skipIf(os.name != 'nt',
1180 'test requires a Windows-compatible system')
1181only_posix = unittest.skipIf(os.name == 'nt',
1182 'test requires a POSIX-compatible system')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001183
1184@only_posix
1185class PosixPathAsPureTest(PurePosixPathTest):
1186 cls = pathlib.PosixPath
1187
1188@only_nt
1189class WindowsPathAsPureTest(PureWindowsPathTest):
1190 cls = pathlib.WindowsPath
1191
Victor Stinnerd7569632016-03-11 22:53:00 +01001192 def test_owner(self):
1193 P = self.cls
1194 with self.assertRaises(NotImplementedError):
1195 P('c:/').owner()
1196
1197 def test_group(self):
1198 P = self.cls
1199 with self.assertRaises(NotImplementedError):
1200 P('c:/').group()
1201
Antoine Pitrou31119e42013-11-22 17:38:12 +01001202
1203class _BasePathTest(object):
1204 """Tests for the FS-accessing functionalities of the Path classes."""
1205
1206 # (BASE)
1207 # |
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001208 # |-- brokenLink -> non-existing
1209 # |-- dirA
1210 # | `-- linkC -> ../dirB
1211 # |-- dirB
1212 # | |-- fileB
1213 # | `-- linkD -> ../dirB
1214 # |-- dirC
1215 # | |-- dirD
1216 # | | `-- fileD
1217 # | `-- fileC
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001218 # |-- dirE # No permissions
Antoine Pitrou31119e42013-11-22 17:38:12 +01001219 # |-- fileA
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001220 # |-- linkA -> fileA
1221 # `-- linkB -> dirB
Antoine Pitrou31119e42013-11-22 17:38:12 +01001222 #
1223
1224 def setUp(self):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001225 def cleanup():
1226 os.chmod(join('dirE'), 0o777)
1227 support.rmtree(BASE)
1228 self.addCleanup(cleanup)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001229 os.mkdir(BASE)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001230 os.mkdir(join('dirA'))
1231 os.mkdir(join('dirB'))
1232 os.mkdir(join('dirC'))
1233 os.mkdir(join('dirC', 'dirD'))
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001234 os.mkdir(join('dirE'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001235 with open(join('fileA'), 'wb') as f:
1236 f.write(b"this is file A\n")
1237 with open(join('dirB', 'fileB'), 'wb') as f:
1238 f.write(b"this is file B\n")
1239 with open(join('dirC', 'fileC'), 'wb') as f:
1240 f.write(b"this is file C\n")
1241 with open(join('dirC', 'dirD', 'fileD'), 'wb') as f:
1242 f.write(b"this is file D\n")
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001243 os.chmod(join('dirE'), 0)
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001244 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001245 # Relative symlinks
1246 os.symlink('fileA', join('linkA'))
1247 os.symlink('non-existing', join('brokenLink'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001248 self.dirlink('dirB', join('linkB'))
1249 self.dirlink(os.path.join('..', 'dirB'), join('dirA', 'linkC'))
Guido van Rossum69bfb152016-01-06 10:31:33 -08001250 # This one goes upwards, creating a loop
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001251 self.dirlink(os.path.join('..', 'dirB'), join('dirB', 'linkD'))
1252
1253 if os.name == 'nt':
1254 # Workaround for http://bugs.python.org/issue13772
1255 def dirlink(self, src, dest):
1256 os.symlink(src, dest, target_is_directory=True)
1257 else:
1258 def dirlink(self, src, dest):
1259 os.symlink(src, dest)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001260
1261 def assertSame(self, path_a, path_b):
1262 self.assertTrue(os.path.samefile(str(path_a), str(path_b)),
1263 "%r and %r don't point to the same file" %
1264 (path_a, path_b))
1265
1266 def assertFileNotFound(self, func, *args, **kwargs):
1267 with self.assertRaises(FileNotFoundError) as cm:
1268 func(*args, **kwargs)
1269 self.assertEqual(cm.exception.errno, errno.ENOENT)
1270
1271 def _test_cwd(self, p):
1272 q = self.cls(os.getcwd())
1273 self.assertEqual(p, q)
1274 self.assertEqual(str(p), str(q))
1275 self.assertIs(type(p), type(q))
1276 self.assertTrue(p.is_absolute())
1277
1278 def test_cwd(self):
1279 p = self.cls.cwd()
1280 self._test_cwd(p)
1281
Antoine Pitrou17cba7d2015-01-12 21:03:41 +01001282 def _test_home(self, p):
1283 q = self.cls(os.path.expanduser('~'))
1284 self.assertEqual(p, q)
1285 self.assertEqual(str(p), str(q))
1286 self.assertIs(type(p), type(q))
1287 self.assertTrue(p.is_absolute())
1288
1289 def test_home(self):
1290 p = self.cls.home()
1291 self._test_home(p)
1292
Antoine Pitrou43e3d942014-05-13 10:50:15 +02001293 def test_samefile(self):
1294 fileA_path = os.path.join(BASE, 'fileA')
1295 fileB_path = os.path.join(BASE, 'dirB', 'fileB')
1296 p = self.cls(fileA_path)
1297 pp = self.cls(fileA_path)
1298 q = self.cls(fileB_path)
1299 self.assertTrue(p.samefile(fileA_path))
1300 self.assertTrue(p.samefile(pp))
1301 self.assertFalse(p.samefile(fileB_path))
1302 self.assertFalse(p.samefile(q))
1303 # Test the non-existent file case
1304 non_existent = os.path.join(BASE, 'foo')
1305 r = self.cls(non_existent)
1306 self.assertRaises(FileNotFoundError, p.samefile, r)
1307 self.assertRaises(FileNotFoundError, p.samefile, non_existent)
1308 self.assertRaises(FileNotFoundError, r.samefile, p)
1309 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1310 self.assertRaises(FileNotFoundError, r.samefile, r)
1311 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1312
Antoine Pitrou31119e42013-11-22 17:38:12 +01001313 def test_empty_path(self):
1314 # The empty path points to '.'
1315 p = self.cls('')
1316 self.assertEqual(p.stat(), os.stat('.'))
1317
Antoine Pitrou8477ed62014-12-30 20:54:45 +01001318 def test_expanduser_common(self):
1319 P = self.cls
1320 p = P('~')
1321 self.assertEqual(p.expanduser(), P(os.path.expanduser('~')))
1322 p = P('foo')
1323 self.assertEqual(p.expanduser(), p)
1324 p = P('/~')
1325 self.assertEqual(p.expanduser(), p)
1326 p = P('../~')
1327 self.assertEqual(p.expanduser(), p)
1328 p = P(P('').absolute().anchor) / '~'
1329 self.assertEqual(p.expanduser(), p)
1330
Antoine Pitrou31119e42013-11-22 17:38:12 +01001331 def test_exists(self):
1332 P = self.cls
1333 p = P(BASE)
1334 self.assertIs(True, p.exists())
1335 self.assertIs(True, (p / 'dirA').exists())
1336 self.assertIs(True, (p / 'fileA').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001337 self.assertIs(False, (p / 'fileA' / 'bah').exists())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001338 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001339 self.assertIs(True, (p / 'linkA').exists())
1340 self.assertIs(True, (p / 'linkB').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001341 self.assertIs(True, (p / 'linkB' / 'fileB').exists())
1342 self.assertIs(False, (p / 'linkA' / 'bah').exists())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001343 self.assertIs(False, (p / 'foo').exists())
1344 self.assertIs(False, P('/xyzzy').exists())
1345
1346 def test_open_common(self):
1347 p = self.cls(BASE)
1348 with (p / 'fileA').open('r') as f:
1349 self.assertIsInstance(f, io.TextIOBase)
1350 self.assertEqual(f.read(), "this is file A\n")
1351 with (p / 'fileA').open('rb') as f:
1352 self.assertIsInstance(f, io.BufferedIOBase)
1353 self.assertEqual(f.read().strip(), b"this is file A")
1354 with (p / 'fileA').open('rb', buffering=0) as f:
1355 self.assertIsInstance(f, io.RawIOBase)
1356 self.assertEqual(f.read().strip(), b"this is file A")
1357
Georg Brandlea683982014-10-01 19:12:33 +02001358 def test_read_write_bytes(self):
1359 p = self.cls(BASE)
1360 (p / 'fileA').write_bytes(b'abcdefg')
1361 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1362 # check that trying to write str does not truncate the file
1363 self.assertRaises(TypeError, (p / 'fileA').write_bytes, 'somestr')
1364 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1365
1366 def test_read_write_text(self):
1367 p = self.cls(BASE)
1368 (p / 'fileA').write_text('äbcdefg', encoding='latin-1')
1369 self.assertEqual((p / 'fileA').read_text(
1370 encoding='utf-8', errors='ignore'), 'bcdefg')
1371 # check that trying to write bytes does not truncate the file
1372 self.assertRaises(TypeError, (p / 'fileA').write_text, b'somebytes')
1373 self.assertEqual((p / 'fileA').read_text(encoding='latin-1'), 'äbcdefg')
1374
Antoine Pitrou31119e42013-11-22 17:38:12 +01001375 def test_iterdir(self):
1376 P = self.cls
1377 p = P(BASE)
1378 it = p.iterdir()
1379 paths = set(it)
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001380 expected = ['dirA', 'dirB', 'dirC', 'dirE', 'fileA']
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001381 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001382 expected += ['linkA', 'linkB', 'brokenLink']
1383 self.assertEqual(paths, { P(BASE, q) for q in expected })
1384
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001385 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001386 def test_iterdir_symlink(self):
1387 # __iter__ on a symlink to a directory
1388 P = self.cls
1389 p = P(BASE, 'linkB')
1390 paths = set(p.iterdir())
1391 expected = { P(BASE, 'linkB', q) for q in ['fileB', 'linkD'] }
1392 self.assertEqual(paths, expected)
1393
1394 def test_iterdir_nodir(self):
1395 # __iter__ on something that is not a directory
1396 p = self.cls(BASE, 'fileA')
1397 with self.assertRaises(OSError) as cm:
1398 next(p.iterdir())
1399 # ENOENT or EINVAL under Windows, ENOTDIR otherwise
1400 # (see issue #12802)
1401 self.assertIn(cm.exception.errno, (errno.ENOTDIR,
1402 errno.ENOENT, errno.EINVAL))
1403
1404 def test_glob_common(self):
1405 def _check(glob, expected):
1406 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1407 P = self.cls
1408 p = P(BASE)
1409 it = p.glob("fileA")
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001410 self.assertIsInstance(it, collections.abc.Iterator)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001411 _check(it, ["fileA"])
1412 _check(p.glob("fileB"), [])
1413 _check(p.glob("dir*/file*"), ["dirB/fileB", "dirC/fileC"])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001414 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001415 _check(p.glob("*A"), ['dirA', 'fileA'])
1416 else:
1417 _check(p.glob("*A"), ['dirA', 'fileA', 'linkA'])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001418 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001419 _check(p.glob("*B/*"), ['dirB/fileB'])
1420 else:
1421 _check(p.glob("*B/*"), ['dirB/fileB', 'dirB/linkD',
1422 'linkB/fileB', 'linkB/linkD'])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001423 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001424 _check(p.glob("*/fileB"), ['dirB/fileB'])
1425 else:
1426 _check(p.glob("*/fileB"), ['dirB/fileB', 'linkB/fileB'])
1427
1428 def test_rglob_common(self):
1429 def _check(glob, expected):
1430 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1431 P = self.cls
1432 p = P(BASE)
1433 it = p.rglob("fileA")
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001434 self.assertIsInstance(it, collections.abc.Iterator)
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001435 _check(it, ["fileA"])
1436 _check(p.rglob("fileB"), ["dirB/fileB"])
1437 _check(p.rglob("*/fileA"), [])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001438 if not support.can_symlink():
Guido van Rossum9c39b672016-01-07 13:12:34 -08001439 _check(p.rglob("*/fileB"), ["dirB/fileB"])
1440 else:
1441 _check(p.rglob("*/fileB"), ["dirB/fileB", "dirB/linkD/fileB",
1442 "linkB/fileB", "dirA/linkC/fileB"])
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001443 _check(p.rglob("file*"), ["fileA", "dirB/fileB",
1444 "dirC/fileC", "dirC/dirD/fileD"])
Antoine Pitrou31119e42013-11-22 17:38:12 +01001445 p = P(BASE, "dirC")
1446 _check(p.rglob("file*"), ["dirC/fileC", "dirC/dirD/fileD"])
1447 _check(p.rglob("*/*"), ["dirC/dirD/fileD"])
1448
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001449 @support.skip_unless_symlink
Guido van Rossum69bfb152016-01-06 10:31:33 -08001450 def test_rglob_symlink_loop(self):
1451 # Don't get fooled by symlink loops (Issue #26012)
1452 P = self.cls
1453 p = P(BASE)
1454 given = set(p.rglob('*'))
1455 expect = {'brokenLink',
1456 'dirA', 'dirA/linkC',
1457 'dirB', 'dirB/fileB', 'dirB/linkD',
1458 'dirC', 'dirC/dirD', 'dirC/dirD/fileD', 'dirC/fileC',
1459 'dirE',
1460 'fileA',
1461 'linkA',
1462 'linkB',
1463 }
1464 self.assertEqual(given, {p / x for x in expect})
1465
Antoine Pitrou31119e42013-11-22 17:38:12 +01001466 def test_glob_dotdot(self):
1467 # ".." is not special in globs
1468 P = self.cls
1469 p = P(BASE)
1470 self.assertEqual(set(p.glob("..")), { P(BASE, "..") })
1471 self.assertEqual(set(p.glob("dirA/../file*")), { P(BASE, "dirA/../fileA") })
1472 self.assertEqual(set(p.glob("../xyzzy")), set())
1473
Ezio Melottiac1e7f62015-12-28 23:50:19 +02001474
Steve Dower98eb3602016-11-09 12:58:17 -08001475 def _check_resolve(self, p, expected, strict=True):
1476 q = p.resolve(strict)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001477 self.assertEqual(q, expected)
1478
Ezio Melottiac1e7f62015-12-28 23:50:19 +02001479 # this can be used to check both relative and absolute resolutions
1480 _check_resolve_relative = _check_resolve_absolute = _check_resolve
Antoine Pitrou31119e42013-11-22 17:38:12 +01001481
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001482 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001483 def test_resolve_common(self):
1484 P = self.cls
1485 p = P(BASE, 'foo')
1486 with self.assertRaises(OSError) as cm:
Steve Dower98eb3602016-11-09 12:58:17 -08001487 p.resolve(strict=True)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001488 self.assertEqual(cm.exception.errno, errno.ENOENT)
Steve Dower98eb3602016-11-09 12:58:17 -08001489 # Non-strict
1490 self.assertEqual(str(p.resolve(strict=False)),
1491 os.path.join(BASE, 'foo'))
1492 p = P(BASE, 'foo', 'in', 'spam')
1493 self.assertEqual(str(p.resolve(strict=False)),
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001494 os.path.join(BASE, 'foo', 'in', 'spam'))
Steve Dower98eb3602016-11-09 12:58:17 -08001495 p = P(BASE, '..', 'foo', 'in', 'spam')
1496 self.assertEqual(str(p.resolve(strict=False)),
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001497 os.path.abspath(os.path.join('foo', 'in', 'spam')))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001498 # These are all relative symlinks
1499 p = P(BASE, 'dirB', 'fileB')
1500 self._check_resolve_relative(p, p)
1501 p = P(BASE, 'linkA')
1502 self._check_resolve_relative(p, P(BASE, 'fileA'))
1503 p = P(BASE, 'dirA', 'linkC', 'fileB')
1504 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
1505 p = P(BASE, 'dirB', 'linkD', 'fileB')
1506 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
Steve Dower98eb3602016-11-09 12:58:17 -08001507 # Non-strict
1508 p = P(BASE, 'dirA', 'linkC', 'fileB', 'foo', 'in', 'spam')
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001509 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB', 'foo', 'in',
1510 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001511 p = P(BASE, 'dirA', 'linkC', '..', 'foo', 'in', 'spam')
1512 if os.name == 'nt':
1513 # In Windows, if linkY points to dirB, 'dirA\linkY\..'
1514 # resolves to 'dirA' without resolving linkY first.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001515 self._check_resolve_relative(p, P(BASE, 'dirA', 'foo', 'in',
1516 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001517 else:
1518 # In Posix, if linkY points to dirB, 'dirA/linkY/..'
1519 # resolves to 'dirB/..' first before resolving to parent of dirB.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001520 self._check_resolve_relative(p, P(BASE, 'foo', 'in', 'spam'), False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001521 # Now create absolute symlinks
1522 d = tempfile.mkdtemp(suffix='-dirD')
Victor Stinnerec864692014-07-21 19:19:05 +02001523 self.addCleanup(support.rmtree, d)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001524 os.symlink(os.path.join(d), join('dirA', 'linkX'))
1525 os.symlink(join('dirB'), os.path.join(d, 'linkY'))
1526 p = P(BASE, 'dirA', 'linkX', 'linkY', 'fileB')
1527 self._check_resolve_absolute(p, P(BASE, 'dirB', 'fileB'))
Steve Dower98eb3602016-11-09 12:58:17 -08001528 # Non-strict
1529 p = P(BASE, 'dirA', 'linkX', 'linkY', 'foo', 'in', 'spam')
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001530 self._check_resolve_relative(p, P(BASE, 'dirB', 'foo', 'in', 'spam'),
1531 False)
Steve Dower98eb3602016-11-09 12:58:17 -08001532 p = P(BASE, 'dirA', 'linkX', 'linkY', '..', 'foo', 'in', 'spam')
1533 if os.name == 'nt':
1534 # In Windows, if linkY points to dirB, 'dirA\linkY\..'
1535 # resolves to 'dirA' without resolving linkY first.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001536 self._check_resolve_relative(p, P(d, 'foo', 'in', 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001537 else:
1538 # In Posix, if linkY points to dirB, 'dirA/linkY/..'
1539 # resolves to 'dirB/..' first before resolving to parent of dirB.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001540 self._check_resolve_relative(p, P(BASE, 'foo', 'in', 'spam'), False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001541
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001542 @support.skip_unless_symlink
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001543 def test_resolve_dot(self):
1544 # See https://bitbucket.org/pitrou/pathlib/issue/9/pathresolve-fails-on-complex-symlinks
1545 p = self.cls(BASE)
1546 self.dirlink('.', join('0'))
Antoine Pitroucc157512013-12-03 17:13:13 +01001547 self.dirlink(os.path.join('0', '0'), join('1'))
1548 self.dirlink(os.path.join('1', '1'), join('2'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001549 q = p / '2'
Steve Dower98eb3602016-11-09 12:58:17 -08001550 self.assertEqual(q.resolve(strict=True), p)
1551 r = q / '3' / '4'
1552 self.assertRaises(FileNotFoundError, r.resolve, strict=True)
1553 # Non-strict
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001554 self.assertEqual(r.resolve(strict=False), p / '3' / '4')
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001555
Antoine Pitrou31119e42013-11-22 17:38:12 +01001556 def test_with(self):
1557 p = self.cls(BASE)
1558 it = p.iterdir()
1559 it2 = p.iterdir()
1560 next(it2)
1561 with p:
1562 pass
1563 # I/O operation on closed path
1564 self.assertRaises(ValueError, next, it)
1565 self.assertRaises(ValueError, next, it2)
1566 self.assertRaises(ValueError, p.open)
1567 self.assertRaises(ValueError, p.resolve)
1568 self.assertRaises(ValueError, p.absolute)
1569 self.assertRaises(ValueError, p.__enter__)
1570
1571 def test_chmod(self):
1572 p = self.cls(BASE) / 'fileA'
1573 mode = p.stat().st_mode
1574 # Clear writable bit
1575 new_mode = mode & ~0o222
1576 p.chmod(new_mode)
1577 self.assertEqual(p.stat().st_mode, new_mode)
1578 # Set writable bit
1579 new_mode = mode | 0o222
1580 p.chmod(new_mode)
1581 self.assertEqual(p.stat().st_mode, new_mode)
1582
1583 # XXX also need a test for lchmod
1584
1585 def test_stat(self):
1586 p = self.cls(BASE) / 'fileA'
1587 st = p.stat()
1588 self.assertEqual(p.stat(), st)
1589 # Change file mode by flipping write bit
1590 p.chmod(st.st_mode ^ 0o222)
1591 self.addCleanup(p.chmod, st.st_mode)
1592 self.assertNotEqual(p.stat(), st)
1593
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001594 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001595 def test_lstat(self):
1596 p = self.cls(BASE)/ 'linkA'
1597 st = p.stat()
1598 self.assertNotEqual(st, p.lstat())
1599
1600 def test_lstat_nosymlink(self):
1601 p = self.cls(BASE) / 'fileA'
1602 st = p.stat()
1603 self.assertEqual(st, p.lstat())
1604
1605 @unittest.skipUnless(pwd, "the pwd module is needed for this test")
1606 def test_owner(self):
1607 p = self.cls(BASE) / 'fileA'
1608 uid = p.stat().st_uid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001609 try:
1610 name = pwd.getpwuid(uid).pw_name
1611 except KeyError:
1612 self.skipTest(
1613 "user %d doesn't have an entry in the system database" % uid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001614 self.assertEqual(name, p.owner())
1615
1616 @unittest.skipUnless(grp, "the grp module is needed for this test")
1617 def test_group(self):
1618 p = self.cls(BASE) / 'fileA'
1619 gid = p.stat().st_gid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001620 try:
1621 name = grp.getgrgid(gid).gr_name
1622 except KeyError:
1623 self.skipTest(
1624 "group %d doesn't have an entry in the system database" % gid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001625 self.assertEqual(name, p.group())
1626
1627 def test_unlink(self):
1628 p = self.cls(BASE) / 'fileA'
1629 p.unlink()
1630 self.assertFileNotFound(p.stat)
1631 self.assertFileNotFound(p.unlink)
1632
1633 def test_rmdir(self):
1634 p = self.cls(BASE) / 'dirA'
1635 for q in p.iterdir():
1636 q.unlink()
1637 p.rmdir()
1638 self.assertFileNotFound(p.stat)
1639 self.assertFileNotFound(p.unlink)
1640
1641 def test_rename(self):
1642 P = self.cls(BASE)
1643 p = P / 'fileA'
1644 size = p.stat().st_size
1645 # Renaming to another path
1646 q = P / 'dirA' / 'fileAA'
1647 p.rename(q)
1648 self.assertEqual(q.stat().st_size, size)
1649 self.assertFileNotFound(p.stat)
1650 # Renaming to a str of a relative path
1651 r = rel_join('fileAAA')
1652 q.rename(r)
1653 self.assertEqual(os.stat(r).st_size, size)
1654 self.assertFileNotFound(q.stat)
1655
1656 def test_replace(self):
1657 P = self.cls(BASE)
1658 p = P / 'fileA'
1659 size = p.stat().st_size
1660 # Replacing a non-existing path
1661 q = P / 'dirA' / 'fileAA'
1662 p.replace(q)
1663 self.assertEqual(q.stat().st_size, size)
1664 self.assertFileNotFound(p.stat)
1665 # Replacing another (existing) path
1666 r = rel_join('dirB', 'fileB')
1667 q.replace(r)
1668 self.assertEqual(os.stat(r).st_size, size)
1669 self.assertFileNotFound(q.stat)
1670
1671 def test_touch_common(self):
1672 P = self.cls(BASE)
1673 p = P / 'newfileA'
1674 self.assertFalse(p.exists())
1675 p.touch()
1676 self.assertTrue(p.exists())
Antoine Pitrou0f575642013-11-22 23:20:08 +01001677 st = p.stat()
1678 old_mtime = st.st_mtime
1679 old_mtime_ns = st.st_mtime_ns
Antoine Pitrou31119e42013-11-22 17:38:12 +01001680 # Rewind the mtime sufficiently far in the past to work around
1681 # filesystem-specific timestamp granularity.
1682 os.utime(str(p), (old_mtime - 10, old_mtime - 10))
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001683 # The file mtime should be refreshed by calling touch() again
Antoine Pitrou31119e42013-11-22 17:38:12 +01001684 p.touch()
Antoine Pitrou0f575642013-11-22 23:20:08 +01001685 st = p.stat()
Antoine Pitrou2cf39172013-11-23 15:25:59 +01001686 self.assertGreaterEqual(st.st_mtime_ns, old_mtime_ns)
1687 self.assertGreaterEqual(st.st_mtime, old_mtime)
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001688 # Now with exist_ok=False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001689 p = P / 'newfileB'
1690 self.assertFalse(p.exists())
1691 p.touch(mode=0o700, exist_ok=False)
1692 self.assertTrue(p.exists())
1693 self.assertRaises(OSError, p.touch, exist_ok=False)
1694
Antoine Pitrou8b784932013-11-23 14:52:39 +01001695 def test_touch_nochange(self):
1696 P = self.cls(BASE)
1697 p = P / 'fileA'
1698 p.touch()
1699 with p.open('rb') as f:
1700 self.assertEqual(f.read().strip(), b"this is file A")
1701
Antoine Pitrou31119e42013-11-22 17:38:12 +01001702 def test_mkdir(self):
1703 P = self.cls(BASE)
1704 p = P / 'newdirA'
1705 self.assertFalse(p.exists())
1706 p.mkdir()
1707 self.assertTrue(p.exists())
1708 self.assertTrue(p.is_dir())
1709 with self.assertRaises(OSError) as cm:
1710 p.mkdir()
1711 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001712
1713 def test_mkdir_parents(self):
1714 # Creating a chain of directories
1715 p = self.cls(BASE, 'newdirB', 'newdirC')
1716 self.assertFalse(p.exists())
1717 with self.assertRaises(OSError) as cm:
1718 p.mkdir()
1719 self.assertEqual(cm.exception.errno, errno.ENOENT)
1720 p.mkdir(parents=True)
1721 self.assertTrue(p.exists())
1722 self.assertTrue(p.is_dir())
1723 with self.assertRaises(OSError) as cm:
1724 p.mkdir(parents=True)
1725 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001726 # test `mode` arg
1727 mode = stat.S_IMODE(p.stat().st_mode) # default mode
1728 p = self.cls(BASE, 'newdirD', 'newdirE')
1729 p.mkdir(0o555, parents=True)
1730 self.assertTrue(p.exists())
1731 self.assertTrue(p.is_dir())
1732 if os.name != 'nt':
1733 # the directory's permissions follow the mode argument
Gregory P. Smithb599c612014-01-20 01:10:33 -08001734 self.assertEqual(stat.S_IMODE(p.stat().st_mode), 0o7555 & mode)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001735 # the parent's permissions follow the default process settings
1736 self.assertEqual(stat.S_IMODE(p.parent.stat().st_mode), mode)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001737
Barry Warsaw7c549c42014-08-05 11:28:12 -04001738 def test_mkdir_exist_ok(self):
1739 p = self.cls(BASE, 'dirB')
1740 st_ctime_first = p.stat().st_ctime
1741 self.assertTrue(p.exists())
1742 self.assertTrue(p.is_dir())
1743 with self.assertRaises(FileExistsError) as cm:
1744 p.mkdir()
1745 self.assertEqual(cm.exception.errno, errno.EEXIST)
1746 p.mkdir(exist_ok=True)
1747 self.assertTrue(p.exists())
1748 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1749
1750 def test_mkdir_exist_ok_with_parent(self):
1751 p = self.cls(BASE, 'dirC')
1752 self.assertTrue(p.exists())
1753 with self.assertRaises(FileExistsError) as cm:
1754 p.mkdir()
1755 self.assertEqual(cm.exception.errno, errno.EEXIST)
1756 p = p / 'newdirC'
1757 p.mkdir(parents=True)
1758 st_ctime_first = p.stat().st_ctime
1759 self.assertTrue(p.exists())
1760 with self.assertRaises(FileExistsError) as cm:
1761 p.mkdir(parents=True)
1762 self.assertEqual(cm.exception.errno, errno.EEXIST)
1763 p.mkdir(parents=True, exist_ok=True)
1764 self.assertTrue(p.exists())
1765 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1766
Serhiy Storchakaaf7b9ec2017-03-24 20:51:53 +02001767 def test_mkdir_exist_ok_root(self):
1768 # Issue #25803: A drive root could raise PermissionError on Windows
1769 self.cls('/').resolve().mkdir(exist_ok=True)
1770 self.cls('/').resolve().mkdir(parents=True, exist_ok=True)
1771
Steve Dowerd3c48532017-02-04 14:54:56 -08001772 @only_nt # XXX: not sure how to test this on POSIX
1773 def test_mkdir_with_unknown_drive(self):
1774 for d in 'ZYXWVUTSRQPONMLKJIHGFEDCBA':
1775 p = self.cls(d + ':\\')
1776 if not p.is_dir():
1777 break
1778 else:
1779 self.skipTest("cannot find a drive that doesn't exist")
1780 with self.assertRaises(OSError):
1781 (p / 'child' / 'path').mkdir(parents=True)
1782
Barry Warsaw7c549c42014-08-05 11:28:12 -04001783 def test_mkdir_with_child_file(self):
1784 p = self.cls(BASE, 'dirB', 'fileB')
1785 self.assertTrue(p.exists())
1786 # An exception is raised when the last path component is an existing
1787 # regular file, regardless of whether exist_ok is true or not.
1788 with self.assertRaises(FileExistsError) as cm:
1789 p.mkdir(parents=True)
1790 self.assertEqual(cm.exception.errno, errno.EEXIST)
1791 with self.assertRaises(FileExistsError) as cm:
1792 p.mkdir(parents=True, exist_ok=True)
1793 self.assertEqual(cm.exception.errno, errno.EEXIST)
1794
1795 def test_mkdir_no_parents_file(self):
1796 p = self.cls(BASE, 'fileA')
1797 self.assertTrue(p.exists())
1798 # An exception is raised when the last path component is an existing
1799 # regular file, regardless of whether exist_ok is true or not.
1800 with self.assertRaises(FileExistsError) as cm:
1801 p.mkdir()
1802 self.assertEqual(cm.exception.errno, errno.EEXIST)
1803 with self.assertRaises(FileExistsError) as cm:
1804 p.mkdir(exist_ok=True)
1805 self.assertEqual(cm.exception.errno, errno.EEXIST)
1806
Armin Rigo22a594a2017-04-13 20:08:15 +02001807 def test_mkdir_concurrent_parent_creation(self):
1808 for pattern_num in range(32):
1809 p = self.cls(BASE, 'dirCPC%d' % pattern_num)
1810 self.assertFalse(p.exists())
1811
1812 def my_mkdir(path, mode=0o777):
1813 path = str(path)
1814 # Emulate another process that would create the directory
1815 # just before we try to create it ourselves. We do it
1816 # in all possible pattern combinations, assuming that this
1817 # function is called at most 5 times (dirCPC/dir1/dir2,
1818 # dirCPC/dir1, dirCPC, dirCPC/dir1, dirCPC/dir1/dir2).
1819 if pattern.pop():
1820 os.mkdir(path, mode) # from another process
1821 concurrently_created.add(path)
1822 os.mkdir(path, mode) # our real call
1823
1824 pattern = [bool(pattern_num & (1 << n)) for n in range(5)]
1825 concurrently_created = set()
1826 p12 = p / 'dir1' / 'dir2'
1827 try:
1828 with mock.patch("pathlib._normal_accessor.mkdir", my_mkdir):
1829 p12.mkdir(parents=True, exist_ok=False)
1830 except FileExistsError:
1831 self.assertIn(str(p12), concurrently_created)
1832 else:
1833 self.assertNotIn(str(p12), concurrently_created)
1834 self.assertTrue(p.exists())
1835
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001836 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001837 def test_symlink_to(self):
1838 P = self.cls(BASE)
1839 target = P / 'fileA'
1840 # Symlinking a path target
1841 link = P / 'dirA' / 'linkAA'
1842 link.symlink_to(target)
1843 self.assertEqual(link.stat(), target.stat())
1844 self.assertNotEqual(link.lstat(), target.stat())
1845 # Symlinking a str target
1846 link = P / 'dirA' / 'linkAAA'
1847 link.symlink_to(str(target))
1848 self.assertEqual(link.stat(), target.stat())
1849 self.assertNotEqual(link.lstat(), target.stat())
1850 self.assertFalse(link.is_dir())
1851 # Symlinking to a directory
1852 target = P / 'dirB'
1853 link = P / 'dirA' / 'linkAAAA'
1854 link.symlink_to(target, target_is_directory=True)
1855 self.assertEqual(link.stat(), target.stat())
1856 self.assertNotEqual(link.lstat(), target.stat())
1857 self.assertTrue(link.is_dir())
1858 self.assertTrue(list(link.iterdir()))
1859
1860 def test_is_dir(self):
1861 P = self.cls(BASE)
1862 self.assertTrue((P / 'dirA').is_dir())
1863 self.assertFalse((P / 'fileA').is_dir())
1864 self.assertFalse((P / 'non-existing').is_dir())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001865 self.assertFalse((P / 'fileA' / 'bah').is_dir())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001866 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001867 self.assertFalse((P / 'linkA').is_dir())
1868 self.assertTrue((P / 'linkB').is_dir())
1869 self.assertFalse((P/ 'brokenLink').is_dir())
1870
1871 def test_is_file(self):
1872 P = self.cls(BASE)
1873 self.assertTrue((P / 'fileA').is_file())
1874 self.assertFalse((P / 'dirA').is_file())
1875 self.assertFalse((P / 'non-existing').is_file())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001876 self.assertFalse((P / 'fileA' / 'bah').is_file())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001877 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001878 self.assertTrue((P / 'linkA').is_file())
1879 self.assertFalse((P / 'linkB').is_file())
1880 self.assertFalse((P/ 'brokenLink').is_file())
1881
Cooper Lees173ff4a2017-08-01 15:35:45 -07001882 @only_posix
1883 def test_is_mount(self):
1884 P = self.cls(BASE)
1885 R = self.cls('/') # TODO: Work out windows
1886 self.assertFalse((P / 'fileA').is_mount())
1887 self.assertFalse((P / 'dirA').is_mount())
1888 self.assertFalse((P / 'non-existing').is_mount())
1889 self.assertFalse((P / 'fileA' / 'bah').is_mount())
1890 self.assertTrue(R.is_mount())
1891 if support.can_symlink():
1892 self.assertFalse((P / 'linkA').is_mount())
1893
Antoine Pitrou31119e42013-11-22 17:38:12 +01001894 def test_is_symlink(self):
1895 P = self.cls(BASE)
1896 self.assertFalse((P / 'fileA').is_symlink())
1897 self.assertFalse((P / 'dirA').is_symlink())
1898 self.assertFalse((P / 'non-existing').is_symlink())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001899 self.assertFalse((P / 'fileA' / 'bah').is_symlink())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001900 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001901 self.assertTrue((P / 'linkA').is_symlink())
1902 self.assertTrue((P / 'linkB').is_symlink())
1903 self.assertTrue((P/ 'brokenLink').is_symlink())
1904
1905 def test_is_fifo_false(self):
1906 P = self.cls(BASE)
1907 self.assertFalse((P / 'fileA').is_fifo())
1908 self.assertFalse((P / 'dirA').is_fifo())
1909 self.assertFalse((P / 'non-existing').is_fifo())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001910 self.assertFalse((P / 'fileA' / 'bah').is_fifo())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001911
1912 @unittest.skipUnless(hasattr(os, "mkfifo"), "os.mkfifo() required")
1913 def test_is_fifo_true(self):
1914 P = self.cls(BASE, 'myfifo')
xdegaye92c2ca72017-11-12 17:31:07 +01001915 try:
1916 os.mkfifo(str(P))
1917 except PermissionError as e:
1918 self.skipTest('os.mkfifo(): %s' % e)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001919 self.assertTrue(P.is_fifo())
1920 self.assertFalse(P.is_socket())
1921 self.assertFalse(P.is_file())
1922
1923 def test_is_socket_false(self):
1924 P = self.cls(BASE)
1925 self.assertFalse((P / 'fileA').is_socket())
1926 self.assertFalse((P / 'dirA').is_socket())
1927 self.assertFalse((P / 'non-existing').is_socket())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001928 self.assertFalse((P / 'fileA' / 'bah').is_socket())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001929
1930 @unittest.skipUnless(hasattr(socket, "AF_UNIX"), "Unix sockets required")
1931 def test_is_socket_true(self):
1932 P = self.cls(BASE, 'mysock')
Antoine Pitrou330ce592013-11-22 18:05:06 +01001933 sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001934 self.addCleanup(sock.close)
Antoine Pitrou330ce592013-11-22 18:05:06 +01001935 try:
1936 sock.bind(str(P))
1937 except OSError as e:
Xavier de Gayee88ed052016-12-14 11:52:28 +01001938 if (isinstance(e, PermissionError) or
1939 "AF_UNIX path too long" in str(e)):
Antoine Pitrou330ce592013-11-22 18:05:06 +01001940 self.skipTest("cannot bind Unix socket: " + str(e))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001941 self.assertTrue(P.is_socket())
1942 self.assertFalse(P.is_fifo())
1943 self.assertFalse(P.is_file())
1944
1945 def test_is_block_device_false(self):
1946 P = self.cls(BASE)
1947 self.assertFalse((P / 'fileA').is_block_device())
1948 self.assertFalse((P / 'dirA').is_block_device())
1949 self.assertFalse((P / 'non-existing').is_block_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001950 self.assertFalse((P / 'fileA' / 'bah').is_block_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001951
1952 def test_is_char_device_false(self):
1953 P = self.cls(BASE)
1954 self.assertFalse((P / 'fileA').is_char_device())
1955 self.assertFalse((P / 'dirA').is_char_device())
1956 self.assertFalse((P / 'non-existing').is_char_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001957 self.assertFalse((P / 'fileA' / 'bah').is_char_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001958
1959 def test_is_char_device_true(self):
1960 # Under Unix, /dev/null should generally be a char device
1961 P = self.cls('/dev/null')
1962 if not P.exists():
1963 self.skipTest("/dev/null required")
1964 self.assertTrue(P.is_char_device())
1965 self.assertFalse(P.is_block_device())
1966 self.assertFalse(P.is_file())
1967
1968 def test_pickling_common(self):
1969 p = self.cls(BASE, 'fileA')
1970 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
1971 dumped = pickle.dumps(p, proto)
1972 pp = pickle.loads(dumped)
1973 self.assertEqual(pp.stat(), p.stat())
1974
1975 def test_parts_interning(self):
1976 P = self.cls
1977 p = P('/usr/bin/foo')
1978 q = P('/usr/local/bin')
1979 # 'usr'
1980 self.assertIs(p.parts[1], q.parts[1])
1981 # 'bin'
1982 self.assertIs(p.parts[2], q.parts[3])
1983
Antoine Pitrouc274fd22013-12-16 19:57:41 +01001984 def _check_complex_symlinks(self, link0_target):
1985 # Test solving a non-looping chain of symlinks (issue #19887)
1986 P = self.cls(BASE)
1987 self.dirlink(os.path.join('link0', 'link0'), join('link1'))
1988 self.dirlink(os.path.join('link1', 'link1'), join('link2'))
1989 self.dirlink(os.path.join('link2', 'link2'), join('link3'))
1990 self.dirlink(link0_target, join('link0'))
1991
1992 # Resolve absolute paths
1993 p = (P / 'link0').resolve()
1994 self.assertEqual(p, P)
1995 self.assertEqual(str(p), BASE)
1996 p = (P / 'link1').resolve()
1997 self.assertEqual(p, P)
1998 self.assertEqual(str(p), BASE)
1999 p = (P / 'link2').resolve()
2000 self.assertEqual(p, P)
2001 self.assertEqual(str(p), BASE)
2002 p = (P / 'link3').resolve()
2003 self.assertEqual(p, P)
2004 self.assertEqual(str(p), BASE)
2005
2006 # Resolve relative paths
2007 old_path = os.getcwd()
2008 os.chdir(BASE)
2009 try:
2010 p = self.cls('link0').resolve()
2011 self.assertEqual(p, P)
2012 self.assertEqual(str(p), BASE)
2013 p = self.cls('link1').resolve()
2014 self.assertEqual(p, P)
2015 self.assertEqual(str(p), BASE)
2016 p = self.cls('link2').resolve()
2017 self.assertEqual(p, P)
2018 self.assertEqual(str(p), BASE)
2019 p = self.cls('link3').resolve()
2020 self.assertEqual(p, P)
2021 self.assertEqual(str(p), BASE)
2022 finally:
2023 os.chdir(old_path)
2024
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002025 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002026 def test_complex_symlinks_absolute(self):
2027 self._check_complex_symlinks(BASE)
2028
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002029 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002030 def test_complex_symlinks_relative(self):
2031 self._check_complex_symlinks('.')
2032
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002033 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002034 def test_complex_symlinks_relative_dot_dot(self):
2035 self._check_complex_symlinks(os.path.join('dirA', '..'))
2036
Antoine Pitrou31119e42013-11-22 17:38:12 +01002037
2038class PathTest(_BasePathTest, unittest.TestCase):
2039 cls = pathlib.Path
2040
2041 def test_concrete_class(self):
2042 p = self.cls('a')
2043 self.assertIs(type(p),
2044 pathlib.WindowsPath if os.name == 'nt' else pathlib.PosixPath)
2045
2046 def test_unsupported_flavour(self):
2047 if os.name == 'nt':
2048 self.assertRaises(NotImplementedError, pathlib.PosixPath)
2049 else:
2050 self.assertRaises(NotImplementedError, pathlib.WindowsPath)
2051
Berker Peksag4a208e42016-01-30 17:50:48 +02002052 def test_glob_empty_pattern(self):
2053 p = self.cls()
2054 with self.assertRaisesRegex(ValueError, 'Unacceptable pattern'):
2055 list(p.glob(''))
2056
Antoine Pitrou31119e42013-11-22 17:38:12 +01002057
2058@only_posix
2059class PosixPathTest(_BasePathTest, unittest.TestCase):
2060 cls = pathlib.PosixPath
2061
Steve Dower98eb3602016-11-09 12:58:17 -08002062 def _check_symlink_loop(self, *args, strict=True):
Antoine Pitrou31119e42013-11-22 17:38:12 +01002063 path = self.cls(*args)
2064 with self.assertRaises(RuntimeError):
Steve Dower98eb3602016-11-09 12:58:17 -08002065 print(path.resolve(strict))
Antoine Pitrou31119e42013-11-22 17:38:12 +01002066
2067 def test_open_mode(self):
2068 old_mask = os.umask(0)
2069 self.addCleanup(os.umask, old_mask)
2070 p = self.cls(BASE)
2071 with (p / 'new_file').open('wb'):
2072 pass
2073 st = os.stat(join('new_file'))
2074 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
2075 os.umask(0o022)
2076 with (p / 'other_new_file').open('wb'):
2077 pass
2078 st = os.stat(join('other_new_file'))
2079 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
2080
2081 def test_touch_mode(self):
2082 old_mask = os.umask(0)
2083 self.addCleanup(os.umask, old_mask)
2084 p = self.cls(BASE)
2085 (p / 'new_file').touch()
2086 st = os.stat(join('new_file'))
2087 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
2088 os.umask(0o022)
2089 (p / 'other_new_file').touch()
2090 st = os.stat(join('other_new_file'))
2091 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
2092 (p / 'masked_new_file').touch(mode=0o750)
2093 st = os.stat(join('masked_new_file'))
2094 self.assertEqual(stat.S_IMODE(st.st_mode), 0o750)
2095
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002096 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01002097 def test_resolve_loop(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +01002098 # Loops with relative symlinks
2099 os.symlink('linkX/inside', join('linkX'))
2100 self._check_symlink_loop(BASE, 'linkX')
2101 os.symlink('linkY', join('linkY'))
2102 self._check_symlink_loop(BASE, 'linkY')
2103 os.symlink('linkZ/../linkZ', join('linkZ'))
2104 self._check_symlink_loop(BASE, 'linkZ')
Steve Dower98eb3602016-11-09 12:58:17 -08002105 # Non-strict
2106 self._check_symlink_loop(BASE, 'linkZ', 'foo', strict=False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01002107 # Loops with absolute symlinks
2108 os.symlink(join('linkU/inside'), join('linkU'))
2109 self._check_symlink_loop(BASE, 'linkU')
2110 os.symlink(join('linkV'), join('linkV'))
2111 self._check_symlink_loop(BASE, 'linkV')
2112 os.symlink(join('linkW/../linkW'), join('linkW'))
2113 self._check_symlink_loop(BASE, 'linkW')
Steve Dower98eb3602016-11-09 12:58:17 -08002114 # Non-strict
2115 self._check_symlink_loop(BASE, 'linkW', 'foo', strict=False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01002116
2117 def test_glob(self):
2118 P = self.cls
2119 p = P(BASE)
Brett Cannonfe77f4e2013-11-22 16:14:10 -05002120 given = set(p.glob("FILEa"))
2121 expect = set() if not support.fs_is_case_insensitive(BASE) else given
2122 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01002123 self.assertEqual(set(p.glob("FILEa*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01002124
2125 def test_rglob(self):
2126 P = self.cls
2127 p = P(BASE, "dirC")
Brett Cannonfe77f4e2013-11-22 16:14:10 -05002128 given = set(p.rglob("FILEd"))
2129 expect = set() if not support.fs_is_case_insensitive(BASE) else given
2130 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01002131 self.assertEqual(set(p.rglob("FILEd*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01002132
Xavier de Gayefb24eea2016-12-13 09:11:38 +01002133 @unittest.skipUnless(hasattr(pwd, 'getpwall'),
2134 'pwd module does not expose getpwall()')
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002135 def test_expanduser(self):
2136 P = self.cls
2137 support.import_module('pwd')
2138 import pwd
2139 pwdent = pwd.getpwuid(os.getuid())
2140 username = pwdent.pw_name
Serhiy Storchakaa3fd0b22016-05-03 21:17:03 +03002141 userhome = pwdent.pw_dir.rstrip('/') or '/'
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002142 # find arbitrary different user (if exists)
2143 for pwdent in pwd.getpwall():
2144 othername = pwdent.pw_name
2145 otherhome = pwdent.pw_dir.rstrip('/')
Victor Stinnere9694392015-01-10 09:00:20 +01002146 if othername != username and otherhome:
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002147 break
2148
2149 p1 = P('~/Documents')
2150 p2 = P('~' + username + '/Documents')
2151 p3 = P('~' + othername + '/Documents')
2152 p4 = P('../~' + username + '/Documents')
2153 p5 = P('/~' + username + '/Documents')
2154 p6 = P('')
2155 p7 = P('~fakeuser/Documents')
2156
2157 with support.EnvironmentVarGuard() as env:
2158 env.pop('HOME', None)
2159
2160 self.assertEqual(p1.expanduser(), P(userhome) / 'Documents')
2161 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2162 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2163 self.assertEqual(p4.expanduser(), p4)
2164 self.assertEqual(p5.expanduser(), p5)
2165 self.assertEqual(p6.expanduser(), p6)
2166 self.assertRaises(RuntimeError, p7.expanduser)
2167
2168 env['HOME'] = '/tmp'
2169 self.assertEqual(p1.expanduser(), P('/tmp/Documents'))
2170 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2171 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2172 self.assertEqual(p4.expanduser(), p4)
2173 self.assertEqual(p5.expanduser(), p5)
2174 self.assertEqual(p6.expanduser(), p6)
2175 self.assertRaises(RuntimeError, p7.expanduser)
2176
Antoine Pitrou31119e42013-11-22 17:38:12 +01002177
2178@only_nt
2179class WindowsPathTest(_BasePathTest, unittest.TestCase):
2180 cls = pathlib.WindowsPath
2181
2182 def test_glob(self):
2183 P = self.cls
2184 p = P(BASE)
2185 self.assertEqual(set(p.glob("FILEa")), { P(BASE, "fileA") })
2186
2187 def test_rglob(self):
2188 P = self.cls
2189 p = P(BASE, "dirC")
2190 self.assertEqual(set(p.rglob("FILEd")), { P(BASE, "dirC/dirD/fileD") })
2191
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002192 def test_expanduser(self):
2193 P = self.cls
2194 with support.EnvironmentVarGuard() as env:
2195 env.pop('HOME', None)
2196 env.pop('USERPROFILE', None)
2197 env.pop('HOMEPATH', None)
2198 env.pop('HOMEDRIVE', None)
2199 env['USERNAME'] = 'alice'
2200
2201 # test that the path returns unchanged
2202 p1 = P('~/My Documents')
2203 p2 = P('~alice/My Documents')
2204 p3 = P('~bob/My Documents')
2205 p4 = P('/~/My Documents')
2206 p5 = P('d:~/My Documents')
2207 p6 = P('')
2208 self.assertRaises(RuntimeError, p1.expanduser)
2209 self.assertRaises(RuntimeError, p2.expanduser)
2210 self.assertRaises(RuntimeError, p3.expanduser)
2211 self.assertEqual(p4.expanduser(), p4)
2212 self.assertEqual(p5.expanduser(), p5)
2213 self.assertEqual(p6.expanduser(), p6)
2214
2215 def check():
2216 env.pop('USERNAME', None)
2217 self.assertEqual(p1.expanduser(),
2218 P('C:/Users/alice/My Documents'))
2219 self.assertRaises(KeyError, p2.expanduser)
2220 env['USERNAME'] = 'alice'
2221 self.assertEqual(p2.expanduser(),
2222 P('C:/Users/alice/My Documents'))
2223 self.assertEqual(p3.expanduser(),
2224 P('C:/Users/bob/My Documents'))
2225 self.assertEqual(p4.expanduser(), p4)
2226 self.assertEqual(p5.expanduser(), p5)
2227 self.assertEqual(p6.expanduser(), p6)
2228
2229 # test the first lookup key in the env vars
2230 env['HOME'] = 'C:\\Users\\alice'
2231 check()
2232
2233 # test that HOMEPATH is available instead
2234 env.pop('HOME', None)
2235 env['HOMEPATH'] = 'C:\\Users\\alice'
2236 check()
2237
2238 env['HOMEDRIVE'] = 'C:\\'
2239 env['HOMEPATH'] = 'Users\\alice'
2240 check()
2241
2242 env.pop('HOMEDRIVE', None)
2243 env.pop('HOMEPATH', None)
2244 env['USERPROFILE'] = 'C:\\Users\\alice'
2245 check()
2246
Antoine Pitrou31119e42013-11-22 17:38:12 +01002247
2248if __name__ == "__main__":
2249 unittest.main()