blob: 53215550b4354ec74e601d390a65ffc0c13a5871 [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
Miss Islington (bot)a13b6542018-03-02 02:17:51 -080014from test.support import TESTFN, FakePath
Antoine Pitrou31119e42013-11-22 17:38:12 +010015
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)
194 P('a', 'b', 'c')
195 P('/a', 'b', 'c')
196 P('a/b/c')
197 P('/a/b/c')
Miss Islington (bot)a13b6542018-03-02 02:17:51 -0800198 P(FakePath("a/b/c"))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100199 self.assertEqual(P(P('a')), P('a'))
200 self.assertEqual(P(P('a'), 'b'), P('a/b'))
201 self.assertEqual(P(P('a'), P('b')), P('a/b'))
Miss Islington (bot)a13b6542018-03-02 02:17:51 -0800202 self.assertEqual(P(P('a'), P('b'), P('c')), P(FakePath("a/b/c")))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100203
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200204 def _check_str_subclass(self, *args):
205 # Issue #21127: it should be possible to construct a PurePath object
Serhiy Storchaka6a7b3a72016-04-17 08:32:47 +0300206 # from a str subclass instance, and it then gets converted to
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200207 # a pure str object.
208 class StrSubclass(str):
209 pass
210 P = self.cls
211 p = P(*(StrSubclass(x) for x in args))
212 self.assertEqual(p, P(*args))
213 for part in p.parts:
214 self.assertIs(type(part), str)
215
216 def test_str_subclass_common(self):
217 self._check_str_subclass('')
218 self._check_str_subclass('.')
219 self._check_str_subclass('a')
220 self._check_str_subclass('a/b.txt')
221 self._check_str_subclass('/a/b.txt')
222
Antoine Pitrou31119e42013-11-22 17:38:12 +0100223 def test_join_common(self):
224 P = self.cls
225 p = P('a/b')
226 pp = p.joinpath('c')
227 self.assertEqual(pp, P('a/b/c'))
228 self.assertIs(type(pp), type(p))
229 pp = p.joinpath('c', 'd')
230 self.assertEqual(pp, P('a/b/c/d'))
231 pp = p.joinpath(P('c'))
232 self.assertEqual(pp, P('a/b/c'))
233 pp = p.joinpath('/c')
234 self.assertEqual(pp, P('/c'))
235
236 def test_div_common(self):
237 # Basically the same as joinpath()
238 P = self.cls
239 p = P('a/b')
240 pp = p / 'c'
241 self.assertEqual(pp, P('a/b/c'))
242 self.assertIs(type(pp), type(p))
243 pp = p / 'c/d'
244 self.assertEqual(pp, P('a/b/c/d'))
245 pp = p / 'c' / 'd'
246 self.assertEqual(pp, P('a/b/c/d'))
247 pp = 'c' / p / 'd'
248 self.assertEqual(pp, P('c/a/b/d'))
249 pp = p / P('c')
250 self.assertEqual(pp, P('a/b/c'))
251 pp = p/ '/c'
252 self.assertEqual(pp, P('/c'))
253
254 def _check_str(self, expected, args):
255 p = self.cls(*args)
256 self.assertEqual(str(p), expected.replace('/', self.sep))
257
258 def test_str_common(self):
259 # Canonicalized paths roundtrip
260 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
261 self._check_str(pathstr, (pathstr,))
262 # Special case for the empty path
263 self._check_str('.', ('',))
264 # Other tests for str() are in test_equivalences()
265
266 def test_as_posix_common(self):
267 P = self.cls
268 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
269 self.assertEqual(P(pathstr).as_posix(), pathstr)
270 # Other tests for as_posix() are in test_equivalences()
271
272 def test_as_bytes_common(self):
273 sep = os.fsencode(self.sep)
274 P = self.cls
275 self.assertEqual(bytes(P('a/b')), b'a' + sep + b'b')
276
277 def test_as_uri_common(self):
278 P = self.cls
279 with self.assertRaises(ValueError):
280 P('a').as_uri()
281 with self.assertRaises(ValueError):
282 P().as_uri()
283
284 def test_repr_common(self):
285 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
286 p = self.cls(pathstr)
287 clsname = p.__class__.__name__
288 r = repr(p)
289 # The repr() is in the form ClassName("forward-slashes path")
290 self.assertTrue(r.startswith(clsname + '('), r)
291 self.assertTrue(r.endswith(')'), r)
292 inner = r[len(clsname) + 1 : -1]
293 self.assertEqual(eval(inner), p.as_posix())
294 # The repr() roundtrips
295 q = eval(r, pathlib.__dict__)
296 self.assertIs(q.__class__, p.__class__)
297 self.assertEqual(q, p)
298 self.assertEqual(repr(q), r)
299
300 def test_eq_common(self):
301 P = self.cls
302 self.assertEqual(P('a/b'), P('a/b'))
303 self.assertEqual(P('a/b'), P('a', 'b'))
304 self.assertNotEqual(P('a/b'), P('a'))
305 self.assertNotEqual(P('a/b'), P('/a/b'))
306 self.assertNotEqual(P('a/b'), P())
307 self.assertNotEqual(P('/a/b'), P('/'))
308 self.assertNotEqual(P(), P('/'))
309 self.assertNotEqual(P(), "")
310 self.assertNotEqual(P(), {})
311 self.assertNotEqual(P(), int)
312
313 def test_match_common(self):
314 P = self.cls
315 self.assertRaises(ValueError, P('a').match, '')
316 self.assertRaises(ValueError, P('a').match, '.')
317 # Simple relative pattern
318 self.assertTrue(P('b.py').match('b.py'))
319 self.assertTrue(P('a/b.py').match('b.py'))
320 self.assertTrue(P('/a/b.py').match('b.py'))
321 self.assertFalse(P('a.py').match('b.py'))
322 self.assertFalse(P('b/py').match('b.py'))
323 self.assertFalse(P('/a.py').match('b.py'))
324 self.assertFalse(P('b.py/c').match('b.py'))
325 # Wilcard relative pattern
326 self.assertTrue(P('b.py').match('*.py'))
327 self.assertTrue(P('a/b.py').match('*.py'))
328 self.assertTrue(P('/a/b.py').match('*.py'))
329 self.assertFalse(P('b.pyc').match('*.py'))
330 self.assertFalse(P('b./py').match('*.py'))
331 self.assertFalse(P('b.py/c').match('*.py'))
332 # Multi-part relative pattern
333 self.assertTrue(P('ab/c.py').match('a*/*.py'))
334 self.assertTrue(P('/d/ab/c.py').match('a*/*.py'))
335 self.assertFalse(P('a.py').match('a*/*.py'))
336 self.assertFalse(P('/dab/c.py').match('a*/*.py'))
337 self.assertFalse(P('ab/c.py/d').match('a*/*.py'))
338 # Absolute pattern
339 self.assertTrue(P('/b.py').match('/*.py'))
340 self.assertFalse(P('b.py').match('/*.py'))
341 self.assertFalse(P('a/b.py').match('/*.py'))
342 self.assertFalse(P('/a/b.py').match('/*.py'))
343 # Multi-part absolute pattern
344 self.assertTrue(P('/a/b.py').match('/a/*.py'))
345 self.assertFalse(P('/ab.py').match('/a/*.py'))
346 self.assertFalse(P('/a/b/c.py').match('/a/*.py'))
347
348 def test_ordering_common(self):
349 # Ordering is tuple-alike
350 def assertLess(a, b):
351 self.assertLess(a, b)
352 self.assertGreater(b, a)
353 P = self.cls
354 a = P('a')
355 b = P('a/b')
356 c = P('abc')
357 d = P('b')
358 assertLess(a, b)
359 assertLess(a, c)
360 assertLess(a, d)
361 assertLess(b, c)
362 assertLess(c, d)
363 P = self.cls
364 a = P('/a')
365 b = P('/a/b')
366 c = P('/abc')
367 d = P('/b')
368 assertLess(a, b)
369 assertLess(a, c)
370 assertLess(a, d)
371 assertLess(b, c)
372 assertLess(c, d)
373 with self.assertRaises(TypeError):
374 P() < {}
375
376 def test_parts_common(self):
377 # `parts` returns a tuple
378 sep = self.sep
379 P = self.cls
380 p = P('a/b')
381 parts = p.parts
382 self.assertEqual(parts, ('a', 'b'))
383 # The object gets reused
384 self.assertIs(parts, p.parts)
385 # When the path is absolute, the anchor is a separate part
386 p = P('/a/b')
387 parts = p.parts
388 self.assertEqual(parts, (sep, 'a', 'b'))
389
Brett Cannon568be632016-06-10 12:20:49 -0700390 def test_fspath_common(self):
391 P = self.cls
392 p = P('a/b')
393 self._check_str(p.__fspath__(), ('a/b',))
394 self._check_str(os.fspath(p), ('a/b',))
395
Antoine Pitrou31119e42013-11-22 17:38:12 +0100396 def test_equivalences(self):
397 for k, tuples in self.equivalences.items():
398 canon = k.replace('/', self.sep)
399 posix = k.replace(self.sep, '/')
400 if canon != posix:
401 tuples = tuples + [
402 tuple(part.replace('/', self.sep) for part in t)
403 for t in tuples
404 ]
405 tuples.append((posix, ))
406 pcanon = self.cls(canon)
407 for t in tuples:
408 p = self.cls(*t)
409 self.assertEqual(p, pcanon, "failed with args {}".format(t))
410 self.assertEqual(hash(p), hash(pcanon))
411 self.assertEqual(str(p), canon)
412 self.assertEqual(p.as_posix(), posix)
413
414 def test_parent_common(self):
415 # Relative
416 P = self.cls
417 p = P('a/b/c')
418 self.assertEqual(p.parent, P('a/b'))
419 self.assertEqual(p.parent.parent, P('a'))
420 self.assertEqual(p.parent.parent.parent, P())
421 self.assertEqual(p.parent.parent.parent.parent, P())
422 # Anchored
423 p = P('/a/b/c')
424 self.assertEqual(p.parent, P('/a/b'))
425 self.assertEqual(p.parent.parent, P('/a'))
426 self.assertEqual(p.parent.parent.parent, P('/'))
427 self.assertEqual(p.parent.parent.parent.parent, P('/'))
428
429 def test_parents_common(self):
430 # Relative
431 P = self.cls
432 p = P('a/b/c')
433 par = p.parents
434 self.assertEqual(len(par), 3)
435 self.assertEqual(par[0], P('a/b'))
436 self.assertEqual(par[1], P('a'))
437 self.assertEqual(par[2], P('.'))
438 self.assertEqual(list(par), [P('a/b'), P('a'), P('.')])
439 with self.assertRaises(IndexError):
440 par[-1]
441 with self.assertRaises(IndexError):
442 par[3]
443 with self.assertRaises(TypeError):
444 par[0] = p
445 # Anchored
446 p = P('/a/b/c')
447 par = p.parents
448 self.assertEqual(len(par), 3)
449 self.assertEqual(par[0], P('/a/b'))
450 self.assertEqual(par[1], P('/a'))
451 self.assertEqual(par[2], P('/'))
452 self.assertEqual(list(par), [P('/a/b'), P('/a'), P('/')])
453 with self.assertRaises(IndexError):
454 par[3]
455
456 def test_drive_common(self):
457 P = self.cls
458 self.assertEqual(P('a/b').drive, '')
459 self.assertEqual(P('/a/b').drive, '')
460 self.assertEqual(P('').drive, '')
461
462 def test_root_common(self):
463 P = self.cls
464 sep = self.sep
465 self.assertEqual(P('').root, '')
466 self.assertEqual(P('a/b').root, '')
467 self.assertEqual(P('/').root, sep)
468 self.assertEqual(P('/a/b').root, sep)
469
470 def test_anchor_common(self):
471 P = self.cls
472 sep = self.sep
473 self.assertEqual(P('').anchor, '')
474 self.assertEqual(P('a/b').anchor, '')
475 self.assertEqual(P('/').anchor, sep)
476 self.assertEqual(P('/a/b').anchor, sep)
477
478 def test_name_common(self):
479 P = self.cls
480 self.assertEqual(P('').name, '')
481 self.assertEqual(P('.').name, '')
482 self.assertEqual(P('/').name, '')
483 self.assertEqual(P('a/b').name, 'b')
484 self.assertEqual(P('/a/b').name, 'b')
485 self.assertEqual(P('/a/b/.').name, 'b')
486 self.assertEqual(P('a/b.py').name, 'b.py')
487 self.assertEqual(P('/a/b.py').name, 'b.py')
488
489 def test_suffix_common(self):
490 P = self.cls
491 self.assertEqual(P('').suffix, '')
492 self.assertEqual(P('.').suffix, '')
493 self.assertEqual(P('..').suffix, '')
494 self.assertEqual(P('/').suffix, '')
495 self.assertEqual(P('a/b').suffix, '')
496 self.assertEqual(P('/a/b').suffix, '')
497 self.assertEqual(P('/a/b/.').suffix, '')
498 self.assertEqual(P('a/b.py').suffix, '.py')
499 self.assertEqual(P('/a/b.py').suffix, '.py')
500 self.assertEqual(P('a/.hgrc').suffix, '')
501 self.assertEqual(P('/a/.hgrc').suffix, '')
502 self.assertEqual(P('a/.hg.rc').suffix, '.rc')
503 self.assertEqual(P('/a/.hg.rc').suffix, '.rc')
504 self.assertEqual(P('a/b.tar.gz').suffix, '.gz')
505 self.assertEqual(P('/a/b.tar.gz').suffix, '.gz')
506 self.assertEqual(P('a/Some name. Ending with a dot.').suffix, '')
507 self.assertEqual(P('/a/Some name. Ending with a dot.').suffix, '')
508
509 def test_suffixes_common(self):
510 P = self.cls
511 self.assertEqual(P('').suffixes, [])
512 self.assertEqual(P('.').suffixes, [])
513 self.assertEqual(P('/').suffixes, [])
514 self.assertEqual(P('a/b').suffixes, [])
515 self.assertEqual(P('/a/b').suffixes, [])
516 self.assertEqual(P('/a/b/.').suffixes, [])
517 self.assertEqual(P('a/b.py').suffixes, ['.py'])
518 self.assertEqual(P('/a/b.py').suffixes, ['.py'])
519 self.assertEqual(P('a/.hgrc').suffixes, [])
520 self.assertEqual(P('/a/.hgrc').suffixes, [])
521 self.assertEqual(P('a/.hg.rc').suffixes, ['.rc'])
522 self.assertEqual(P('/a/.hg.rc').suffixes, ['.rc'])
523 self.assertEqual(P('a/b.tar.gz').suffixes, ['.tar', '.gz'])
524 self.assertEqual(P('/a/b.tar.gz').suffixes, ['.tar', '.gz'])
525 self.assertEqual(P('a/Some name. Ending with a dot.').suffixes, [])
526 self.assertEqual(P('/a/Some name. Ending with a dot.').suffixes, [])
527
528 def test_stem_common(self):
529 P = self.cls
530 self.assertEqual(P('').stem, '')
531 self.assertEqual(P('.').stem, '')
532 self.assertEqual(P('..').stem, '..')
533 self.assertEqual(P('/').stem, '')
534 self.assertEqual(P('a/b').stem, 'b')
535 self.assertEqual(P('a/b.py').stem, 'b')
536 self.assertEqual(P('a/.hgrc').stem, '.hgrc')
537 self.assertEqual(P('a/.hg.rc').stem, '.hg')
538 self.assertEqual(P('a/b.tar.gz').stem, 'b.tar')
539 self.assertEqual(P('a/Some name. Ending with a dot.').stem,
540 'Some name. Ending with a dot.')
541
542 def test_with_name_common(self):
543 P = self.cls
544 self.assertEqual(P('a/b').with_name('d.xml'), P('a/d.xml'))
545 self.assertEqual(P('/a/b').with_name('d.xml'), P('/a/d.xml'))
546 self.assertEqual(P('a/b.py').with_name('d.xml'), P('a/d.xml'))
547 self.assertEqual(P('/a/b.py').with_name('d.xml'), P('/a/d.xml'))
548 self.assertEqual(P('a/Dot ending.').with_name('d.xml'), P('a/d.xml'))
549 self.assertEqual(P('/a/Dot ending.').with_name('d.xml'), P('/a/d.xml'))
550 self.assertRaises(ValueError, P('').with_name, 'd.xml')
551 self.assertRaises(ValueError, P('.').with_name, 'd.xml')
552 self.assertRaises(ValueError, P('/').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400553 self.assertRaises(ValueError, P('a/b').with_name, '')
554 self.assertRaises(ValueError, P('a/b').with_name, '/c')
555 self.assertRaises(ValueError, P('a/b').with_name, 'c/')
556 self.assertRaises(ValueError, P('a/b').with_name, 'c/d')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100557
558 def test_with_suffix_common(self):
559 P = self.cls
560 self.assertEqual(P('a/b').with_suffix('.gz'), P('a/b.gz'))
561 self.assertEqual(P('/a/b').with_suffix('.gz'), P('/a/b.gz'))
562 self.assertEqual(P('a/b.py').with_suffix('.gz'), P('a/b.gz'))
563 self.assertEqual(P('/a/b.py').with_suffix('.gz'), P('/a/b.gz'))
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400564 # Stripping suffix
565 self.assertEqual(P('a/b.py').with_suffix(''), P('a/b'))
566 self.assertEqual(P('/a/b').with_suffix(''), P('/a/b'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100567 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100568 self.assertRaises(ValueError, P('').with_suffix, '.gz')
569 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
570 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100571 # Invalid suffix
572 self.assertRaises(ValueError, P('a/b').with_suffix, 'gz')
573 self.assertRaises(ValueError, P('a/b').with_suffix, '/')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400574 self.assertRaises(ValueError, P('a/b').with_suffix, '.')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100575 self.assertRaises(ValueError, P('a/b').with_suffix, '/.gz')
576 self.assertRaises(ValueError, P('a/b').with_suffix, 'c/d')
577 self.assertRaises(ValueError, P('a/b').with_suffix, '.c/.d')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400578 self.assertRaises(ValueError, P('a/b').with_suffix, './.d')
579 self.assertRaises(ValueError, P('a/b').with_suffix, '.d/.')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100580
581 def test_relative_to_common(self):
582 P = self.cls
583 p = P('a/b')
584 self.assertRaises(TypeError, p.relative_to)
Antoine Pitrou156b3612013-12-28 19:49:04 +0100585 self.assertRaises(TypeError, p.relative_to, b'a')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100586 self.assertEqual(p.relative_to(P()), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100587 self.assertEqual(p.relative_to(''), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100588 self.assertEqual(p.relative_to(P('a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100589 self.assertEqual(p.relative_to('a'), P('b'))
590 self.assertEqual(p.relative_to('a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100591 self.assertEqual(p.relative_to(P('a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100592 self.assertEqual(p.relative_to('a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100593 # With several args
594 self.assertEqual(p.relative_to('a', 'b'), P())
595 # Unrelated paths
596 self.assertRaises(ValueError, p.relative_to, P('c'))
597 self.assertRaises(ValueError, p.relative_to, P('a/b/c'))
598 self.assertRaises(ValueError, p.relative_to, P('a/c'))
599 self.assertRaises(ValueError, p.relative_to, P('/a'))
600 p = P('/a/b')
601 self.assertEqual(p.relative_to(P('/')), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100602 self.assertEqual(p.relative_to('/'), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100603 self.assertEqual(p.relative_to(P('/a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100604 self.assertEqual(p.relative_to('/a'), P('b'))
605 self.assertEqual(p.relative_to('/a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100606 self.assertEqual(p.relative_to(P('/a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100607 self.assertEqual(p.relative_to('/a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100608 # Unrelated paths
609 self.assertRaises(ValueError, p.relative_to, P('/c'))
610 self.assertRaises(ValueError, p.relative_to, P('/a/b/c'))
611 self.assertRaises(ValueError, p.relative_to, P('/a/c'))
612 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100613 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100614 self.assertRaises(ValueError, p.relative_to, P('a'))
615
616 def test_pickling_common(self):
617 P = self.cls
618 p = P('/a/b')
619 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
620 dumped = pickle.dumps(p, proto)
621 pp = pickle.loads(dumped)
622 self.assertIs(pp.__class__, p.__class__)
623 self.assertEqual(pp, p)
624 self.assertEqual(hash(pp), hash(p))
625 self.assertEqual(str(pp), str(p))
626
627
628class PurePosixPathTest(_BasePurePathTest, unittest.TestCase):
629 cls = pathlib.PurePosixPath
630
631 def test_root(self):
632 P = self.cls
633 self.assertEqual(P('/a/b').root, '/')
634 self.assertEqual(P('///a/b').root, '/')
635 # POSIX special case for two leading slashes
636 self.assertEqual(P('//a/b').root, '//')
637
638 def test_eq(self):
639 P = self.cls
640 self.assertNotEqual(P('a/b'), P('A/b'))
641 self.assertEqual(P('/a'), P('///a'))
642 self.assertNotEqual(P('/a'), P('//a'))
643
644 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100645 P = self.cls
646 self.assertEqual(P('/').as_uri(), 'file:///')
647 self.assertEqual(P('/a/b.c').as_uri(), 'file:///a/b.c')
648 self.assertEqual(P('/a/b%#c').as_uri(), 'file:///a/b%25%23c')
Antoine Pitrou29eac422013-11-22 17:57:03 +0100649
650 def test_as_uri_non_ascii(self):
651 from urllib.parse import quote_from_bytes
652 P = self.cls
653 try:
654 os.fsencode('\xe9')
655 except UnicodeEncodeError:
656 self.skipTest("\\xe9 cannot be encoded to the filesystem encoding")
Antoine Pitrou31119e42013-11-22 17:38:12 +0100657 self.assertEqual(P('/a/b\xe9').as_uri(),
658 'file:///a/b' + quote_from_bytes(os.fsencode('\xe9')))
659
660 def test_match(self):
661 P = self.cls
662 self.assertFalse(P('A.py').match('a.PY'))
663
664 def test_is_absolute(self):
665 P = self.cls
666 self.assertFalse(P().is_absolute())
667 self.assertFalse(P('a').is_absolute())
668 self.assertFalse(P('a/b/').is_absolute())
669 self.assertTrue(P('/').is_absolute())
670 self.assertTrue(P('/a').is_absolute())
671 self.assertTrue(P('/a/b/').is_absolute())
672 self.assertTrue(P('//a').is_absolute())
673 self.assertTrue(P('//a/b').is_absolute())
674
675 def test_is_reserved(self):
676 P = self.cls
677 self.assertIs(False, P('').is_reserved())
678 self.assertIs(False, P('/').is_reserved())
679 self.assertIs(False, P('/foo/bar').is_reserved())
680 self.assertIs(False, P('/dev/con/PRN/NUL').is_reserved())
681
682 def test_join(self):
683 P = self.cls
684 p = P('//a')
685 pp = p.joinpath('b')
686 self.assertEqual(pp, P('//a/b'))
687 pp = P('/a').joinpath('//c')
688 self.assertEqual(pp, P('//c'))
689 pp = P('//a').joinpath('/c')
690 self.assertEqual(pp, P('/c'))
691
692 def test_div(self):
693 # Basically the same as joinpath()
694 P = self.cls
695 p = P('//a')
696 pp = p / 'b'
697 self.assertEqual(pp, P('//a/b'))
698 pp = P('/a') / '//c'
699 self.assertEqual(pp, P('//c'))
700 pp = P('//a') / '/c'
701 self.assertEqual(pp, P('/c'))
702
703
704class PureWindowsPathTest(_BasePurePathTest, unittest.TestCase):
705 cls = pathlib.PureWindowsPath
706
707 equivalences = _BasePurePathTest.equivalences.copy()
708 equivalences.update({
709 'c:a': [ ('c:', 'a'), ('c:', 'a/'), ('/', 'c:', 'a') ],
710 'c:/a': [
711 ('c:/', 'a'), ('c:', '/', 'a'), ('c:', '/a'),
712 ('/z', 'c:/', 'a'), ('//x/y', 'c:/', 'a'),
713 ],
714 '//a/b/': [ ('//a/b',) ],
715 '//a/b/c': [
716 ('//a/b', 'c'), ('//a/b/', 'c'),
717 ],
718 })
719
720 def test_str(self):
721 p = self.cls('a/b/c')
722 self.assertEqual(str(p), 'a\\b\\c')
723 p = self.cls('c:/a/b/c')
724 self.assertEqual(str(p), 'c:\\a\\b\\c')
725 p = self.cls('//a/b')
726 self.assertEqual(str(p), '\\\\a\\b\\')
727 p = self.cls('//a/b/c')
728 self.assertEqual(str(p), '\\\\a\\b\\c')
729 p = self.cls('//a/b/c/d')
730 self.assertEqual(str(p), '\\\\a\\b\\c\\d')
731
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200732 def test_str_subclass(self):
733 self._check_str_subclass('c:')
734 self._check_str_subclass('c:a')
735 self._check_str_subclass('c:a\\b.txt')
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('\\\\some\\share')
740 self._check_str_subclass('\\\\some\\share\\a')
741 self._check_str_subclass('\\\\some\\share\\a\\b.txt')
742
Antoine Pitrou31119e42013-11-22 17:38:12 +0100743 def test_eq(self):
744 P = self.cls
745 self.assertEqual(P('c:a/b'), P('c:a/b'))
746 self.assertEqual(P('c:a/b'), P('c:', 'a', 'b'))
747 self.assertNotEqual(P('c:a/b'), P('d:a/b'))
748 self.assertNotEqual(P('c:a/b'), P('c:/a/b'))
749 self.assertNotEqual(P('/a/b'), P('c:/a/b'))
750 # Case-insensitivity
751 self.assertEqual(P('a/B'), P('A/b'))
752 self.assertEqual(P('C:a/B'), P('c:A/b'))
753 self.assertEqual(P('//Some/SHARE/a/B'), P('//somE/share/A/b'))
754
755 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100756 P = self.cls
757 with self.assertRaises(ValueError):
758 P('/a/b').as_uri()
759 with self.assertRaises(ValueError):
760 P('c:a/b').as_uri()
761 self.assertEqual(P('c:/').as_uri(), 'file:///c:/')
762 self.assertEqual(P('c:/a/b.c').as_uri(), 'file:///c:/a/b.c')
763 self.assertEqual(P('c:/a/b%#c').as_uri(), 'file:///c:/a/b%25%23c')
764 self.assertEqual(P('c:/a/b\xe9').as_uri(), 'file:///c:/a/b%C3%A9')
765 self.assertEqual(P('//some/share/').as_uri(), 'file://some/share/')
766 self.assertEqual(P('//some/share/a/b.c').as_uri(),
767 'file://some/share/a/b.c')
768 self.assertEqual(P('//some/share/a/b%#c\xe9').as_uri(),
769 'file://some/share/a/b%25%23c%C3%A9')
770
771 def test_match_common(self):
772 P = self.cls
773 # Absolute patterns
774 self.assertTrue(P('c:/b.py').match('/*.py'))
775 self.assertTrue(P('c:/b.py').match('c:*.py'))
776 self.assertTrue(P('c:/b.py').match('c:/*.py'))
777 self.assertFalse(P('d:/b.py').match('c:/*.py')) # wrong drive
778 self.assertFalse(P('b.py').match('/*.py'))
779 self.assertFalse(P('b.py').match('c:*.py'))
780 self.assertFalse(P('b.py').match('c:/*.py'))
781 self.assertFalse(P('c:b.py').match('/*.py'))
782 self.assertFalse(P('c:b.py').match('c:/*.py'))
783 self.assertFalse(P('/b.py').match('c:*.py'))
784 self.assertFalse(P('/b.py').match('c:/*.py'))
785 # UNC patterns
786 self.assertTrue(P('//some/share/a.py').match('/*.py'))
787 self.assertTrue(P('//some/share/a.py').match('//some/share/*.py'))
788 self.assertFalse(P('//other/share/a.py').match('//some/share/*.py'))
789 self.assertFalse(P('//some/share/a/b.py').match('//some/share/*.py'))
790 # Case-insensitivity
791 self.assertTrue(P('B.py').match('b.PY'))
792 self.assertTrue(P('c:/a/B.Py').match('C:/A/*.pY'))
793 self.assertTrue(P('//Some/Share/B.Py').match('//somE/sharE/*.pY'))
794
795 def test_ordering_common(self):
796 # Case-insensitivity
797 def assertOrderedEqual(a, b):
798 self.assertLessEqual(a, b)
799 self.assertGreaterEqual(b, a)
800 P = self.cls
801 p = P('c:A/b')
802 q = P('C:a/B')
803 assertOrderedEqual(p, q)
804 self.assertFalse(p < q)
805 self.assertFalse(p > q)
806 p = P('//some/Share/A/b')
807 q = P('//Some/SHARE/a/B')
808 assertOrderedEqual(p, q)
809 self.assertFalse(p < q)
810 self.assertFalse(p > q)
811
812 def test_parts(self):
813 P = self.cls
814 p = P('c:a/b')
815 parts = p.parts
816 self.assertEqual(parts, ('c:', 'a', 'b'))
817 p = P('c:/a/b')
818 parts = p.parts
819 self.assertEqual(parts, ('c:\\', 'a', 'b'))
820 p = P('//a/b/c/d')
821 parts = p.parts
822 self.assertEqual(parts, ('\\\\a\\b\\', 'c', 'd'))
823
824 def test_parent(self):
825 # Anchored
826 P = self.cls
827 p = P('z:a/b/c')
828 self.assertEqual(p.parent, P('z:a/b'))
829 self.assertEqual(p.parent.parent, P('z:a'))
830 self.assertEqual(p.parent.parent.parent, P('z:'))
831 self.assertEqual(p.parent.parent.parent.parent, P('z:'))
832 p = P('z:/a/b/c')
833 self.assertEqual(p.parent, P('z:/a/b'))
834 self.assertEqual(p.parent.parent, P('z:/a'))
835 self.assertEqual(p.parent.parent.parent, P('z:/'))
836 self.assertEqual(p.parent.parent.parent.parent, P('z:/'))
837 p = P('//a/b/c/d')
838 self.assertEqual(p.parent, P('//a/b/c'))
839 self.assertEqual(p.parent.parent, P('//a/b'))
840 self.assertEqual(p.parent.parent.parent, P('//a/b'))
841
842 def test_parents(self):
843 # Anchored
844 P = self.cls
845 p = P('z:a/b/')
846 par = p.parents
847 self.assertEqual(len(par), 2)
848 self.assertEqual(par[0], P('z:a'))
849 self.assertEqual(par[1], P('z:'))
850 self.assertEqual(list(par), [P('z:a'), P('z:')])
851 with self.assertRaises(IndexError):
852 par[2]
853 p = P('z:/a/b/')
854 par = p.parents
855 self.assertEqual(len(par), 2)
856 self.assertEqual(par[0], P('z:/a'))
857 self.assertEqual(par[1], P('z:/'))
858 self.assertEqual(list(par), [P('z:/a'), P('z:/')])
859 with self.assertRaises(IndexError):
860 par[2]
861 p = P('//a/b/c/d')
862 par = p.parents
863 self.assertEqual(len(par), 2)
864 self.assertEqual(par[0], P('//a/b/c'))
865 self.assertEqual(par[1], P('//a/b'))
866 self.assertEqual(list(par), [P('//a/b/c'), P('//a/b')])
867 with self.assertRaises(IndexError):
868 par[2]
869
870 def test_drive(self):
871 P = self.cls
872 self.assertEqual(P('c:').drive, 'c:')
873 self.assertEqual(P('c:a/b').drive, 'c:')
874 self.assertEqual(P('c:/').drive, 'c:')
875 self.assertEqual(P('c:/a/b/').drive, 'c:')
876 self.assertEqual(P('//a/b').drive, '\\\\a\\b')
877 self.assertEqual(P('//a/b/').drive, '\\\\a\\b')
878 self.assertEqual(P('//a/b/c/d').drive, '\\\\a\\b')
879
880 def test_root(self):
881 P = self.cls
882 self.assertEqual(P('c:').root, '')
883 self.assertEqual(P('c:a/b').root, '')
884 self.assertEqual(P('c:/').root, '\\')
885 self.assertEqual(P('c:/a/b/').root, '\\')
886 self.assertEqual(P('//a/b').root, '\\')
887 self.assertEqual(P('//a/b/').root, '\\')
888 self.assertEqual(P('//a/b/c/d').root, '\\')
889
890 def test_anchor(self):
891 P = self.cls
892 self.assertEqual(P('c:').anchor, 'c:')
893 self.assertEqual(P('c:a/b').anchor, 'c:')
894 self.assertEqual(P('c:/').anchor, 'c:\\')
895 self.assertEqual(P('c:/a/b/').anchor, 'c:\\')
896 self.assertEqual(P('//a/b').anchor, '\\\\a\\b\\')
897 self.assertEqual(P('//a/b/').anchor, '\\\\a\\b\\')
898 self.assertEqual(P('//a/b/c/d').anchor, '\\\\a\\b\\')
899
900 def test_name(self):
901 P = self.cls
902 self.assertEqual(P('c:').name, '')
903 self.assertEqual(P('c:/').name, '')
904 self.assertEqual(P('c:a/b').name, 'b')
905 self.assertEqual(P('c:/a/b').name, 'b')
906 self.assertEqual(P('c:a/b.py').name, 'b.py')
907 self.assertEqual(P('c:/a/b.py').name, 'b.py')
908 self.assertEqual(P('//My.py/Share.php').name, '')
909 self.assertEqual(P('//My.py/Share.php/a/b').name, 'b')
910
911 def test_suffix(self):
912 P = self.cls
913 self.assertEqual(P('c:').suffix, '')
914 self.assertEqual(P('c:/').suffix, '')
915 self.assertEqual(P('c:a/b').suffix, '')
916 self.assertEqual(P('c:/a/b').suffix, '')
917 self.assertEqual(P('c:a/b.py').suffix, '.py')
918 self.assertEqual(P('c:/a/b.py').suffix, '.py')
919 self.assertEqual(P('c:a/.hgrc').suffix, '')
920 self.assertEqual(P('c:/a/.hgrc').suffix, '')
921 self.assertEqual(P('c:a/.hg.rc').suffix, '.rc')
922 self.assertEqual(P('c:/a/.hg.rc').suffix, '.rc')
923 self.assertEqual(P('c:a/b.tar.gz').suffix, '.gz')
924 self.assertEqual(P('c:/a/b.tar.gz').suffix, '.gz')
925 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffix, '')
926 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffix, '')
927 self.assertEqual(P('//My.py/Share.php').suffix, '')
928 self.assertEqual(P('//My.py/Share.php/a/b').suffix, '')
929
930 def test_suffixes(self):
931 P = self.cls
932 self.assertEqual(P('c:').suffixes, [])
933 self.assertEqual(P('c:/').suffixes, [])
934 self.assertEqual(P('c:a/b').suffixes, [])
935 self.assertEqual(P('c:/a/b').suffixes, [])
936 self.assertEqual(P('c:a/b.py').suffixes, ['.py'])
937 self.assertEqual(P('c:/a/b.py').suffixes, ['.py'])
938 self.assertEqual(P('c:a/.hgrc').suffixes, [])
939 self.assertEqual(P('c:/a/.hgrc').suffixes, [])
940 self.assertEqual(P('c:a/.hg.rc').suffixes, ['.rc'])
941 self.assertEqual(P('c:/a/.hg.rc').suffixes, ['.rc'])
942 self.assertEqual(P('c:a/b.tar.gz').suffixes, ['.tar', '.gz'])
943 self.assertEqual(P('c:/a/b.tar.gz').suffixes, ['.tar', '.gz'])
944 self.assertEqual(P('//My.py/Share.php').suffixes, [])
945 self.assertEqual(P('//My.py/Share.php/a/b').suffixes, [])
946 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffixes, [])
947 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffixes, [])
948
949 def test_stem(self):
950 P = self.cls
951 self.assertEqual(P('c:').stem, '')
952 self.assertEqual(P('c:.').stem, '')
953 self.assertEqual(P('c:..').stem, '..')
954 self.assertEqual(P('c:/').stem, '')
955 self.assertEqual(P('c:a/b').stem, 'b')
956 self.assertEqual(P('c:a/b.py').stem, 'b')
957 self.assertEqual(P('c:a/.hgrc').stem, '.hgrc')
958 self.assertEqual(P('c:a/.hg.rc').stem, '.hg')
959 self.assertEqual(P('c:a/b.tar.gz').stem, 'b.tar')
960 self.assertEqual(P('c:a/Some name. Ending with a dot.').stem,
961 'Some name. Ending with a dot.')
962
963 def test_with_name(self):
964 P = self.cls
965 self.assertEqual(P('c:a/b').with_name('d.xml'), P('c:a/d.xml'))
966 self.assertEqual(P('c:/a/b').with_name('d.xml'), P('c:/a/d.xml'))
967 self.assertEqual(P('c:a/Dot ending.').with_name('d.xml'), P('c:a/d.xml'))
968 self.assertEqual(P('c:/a/Dot ending.').with_name('d.xml'), P('c:/a/d.xml'))
969 self.assertRaises(ValueError, P('c:').with_name, 'd.xml')
970 self.assertRaises(ValueError, P('c:/').with_name, 'd.xml')
971 self.assertRaises(ValueError, P('//My/Share').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400972 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:')
973 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:e')
974 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:/e')
975 self.assertRaises(ValueError, P('c:a/b').with_name, '//My/Share')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100976
977 def test_with_suffix(self):
978 P = self.cls
979 self.assertEqual(P('c:a/b').with_suffix('.gz'), P('c:a/b.gz'))
980 self.assertEqual(P('c:/a/b').with_suffix('.gz'), P('c:/a/b.gz'))
981 self.assertEqual(P('c:a/b.py').with_suffix('.gz'), P('c:a/b.gz'))
982 self.assertEqual(P('c:/a/b.py').with_suffix('.gz'), P('c:/a/b.gz'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100983 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100984 self.assertRaises(ValueError, P('').with_suffix, '.gz')
985 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
986 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
987 self.assertRaises(ValueError, P('//My/Share').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100988 # Invalid suffix
989 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'gz')
990 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/')
991 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\')
992 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:')
993 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/.gz')
994 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\.gz')
995 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:.gz')
996 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c/d')
997 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c\\d')
998 self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c/d')
999 self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c\\d')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001000
1001 def test_relative_to(self):
1002 P = self.cls
Antoine Pitrou156b3612013-12-28 19:49:04 +01001003 p = P('C:Foo/Bar')
1004 self.assertEqual(p.relative_to(P('c:')), P('Foo/Bar'))
1005 self.assertEqual(p.relative_to('c:'), P('Foo/Bar'))
1006 self.assertEqual(p.relative_to(P('c:foO')), P('Bar'))
1007 self.assertEqual(p.relative_to('c:foO'), P('Bar'))
1008 self.assertEqual(p.relative_to('c:foO/'), P('Bar'))
1009 self.assertEqual(p.relative_to(P('c:foO/baR')), P())
1010 self.assertEqual(p.relative_to('c:foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001011 # Unrelated paths
1012 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +01001013 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001014 self.assertRaises(ValueError, p.relative_to, P('d:'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001015 self.assertRaises(ValueError, p.relative_to, P('/'))
1016 self.assertRaises(ValueError, p.relative_to, P('Foo'))
1017 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1018 self.assertRaises(ValueError, p.relative_to, P('C:/Foo'))
1019 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Bar/Baz'))
1020 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Baz'))
1021 p = P('C:/Foo/Bar')
1022 self.assertEqual(p.relative_to(P('c:')), P('/Foo/Bar'))
1023 self.assertEqual(p.relative_to('c:'), P('/Foo/Bar'))
1024 self.assertEqual(str(p.relative_to(P('c:'))), '\\Foo\\Bar')
1025 self.assertEqual(str(p.relative_to('c:')), '\\Foo\\Bar')
1026 self.assertEqual(p.relative_to(P('c:/')), P('Foo/Bar'))
1027 self.assertEqual(p.relative_to('c:/'), P('Foo/Bar'))
1028 self.assertEqual(p.relative_to(P('c:/foO')), P('Bar'))
1029 self.assertEqual(p.relative_to('c:/foO'), P('Bar'))
1030 self.assertEqual(p.relative_to('c:/foO/'), P('Bar'))
1031 self.assertEqual(p.relative_to(P('c:/foO/baR')), P())
1032 self.assertEqual(p.relative_to('c:/foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001033 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001034 self.assertRaises(ValueError, p.relative_to, P('C:/Baz'))
1035 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Bar/Baz'))
1036 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Baz'))
1037 self.assertRaises(ValueError, p.relative_to, P('C:Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001038 self.assertRaises(ValueError, p.relative_to, P('d:'))
1039 self.assertRaises(ValueError, p.relative_to, P('d:/'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001040 self.assertRaises(ValueError, p.relative_to, P('/'))
1041 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1042 self.assertRaises(ValueError, p.relative_to, P('//C/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001043 # UNC paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001044 p = P('//Server/Share/Foo/Bar')
1045 self.assertEqual(p.relative_to(P('//sErver/sHare')), P('Foo/Bar'))
1046 self.assertEqual(p.relative_to('//sErver/sHare'), P('Foo/Bar'))
1047 self.assertEqual(p.relative_to('//sErver/sHare/'), P('Foo/Bar'))
1048 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo')), P('Bar'))
1049 self.assertEqual(p.relative_to('//sErver/sHare/Foo'), P('Bar'))
1050 self.assertEqual(p.relative_to('//sErver/sHare/Foo/'), P('Bar'))
1051 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo/Bar')), P())
1052 self.assertEqual(p.relative_to('//sErver/sHare/Foo/Bar'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001053 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001054 self.assertRaises(ValueError, p.relative_to, P('/Server/Share/Foo'))
1055 self.assertRaises(ValueError, p.relative_to, P('c:/Server/Share/Foo'))
1056 self.assertRaises(ValueError, p.relative_to, P('//z/Share/Foo'))
1057 self.assertRaises(ValueError, p.relative_to, P('//Server/z/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001058
1059 def test_is_absolute(self):
1060 P = self.cls
1061 # Under NT, only paths with both a drive and a root are absolute
1062 self.assertFalse(P().is_absolute())
1063 self.assertFalse(P('a').is_absolute())
1064 self.assertFalse(P('a/b/').is_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('c:').is_absolute())
1069 self.assertFalse(P('c:a').is_absolute())
1070 self.assertFalse(P('c:a/b/').is_absolute())
1071 self.assertTrue(P('c:/').is_absolute())
1072 self.assertTrue(P('c:/a').is_absolute())
1073 self.assertTrue(P('c:/a/b/').is_absolute())
1074 # UNC paths are absolute by definition
1075 self.assertTrue(P('//a/b').is_absolute())
1076 self.assertTrue(P('//a/b/').is_absolute())
1077 self.assertTrue(P('//a/b/c').is_absolute())
1078 self.assertTrue(P('//a/b/c/d').is_absolute())
1079
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001080 def test_join(self):
1081 P = self.cls
1082 p = P('C:/a/b')
1083 pp = p.joinpath('x/y')
1084 self.assertEqual(pp, P('C:/a/b/x/y'))
1085 pp = p.joinpath('/x/y')
1086 self.assertEqual(pp, P('C:/x/y'))
1087 # Joining with a different drive => the first path is ignored, even
1088 # if the second path is relative.
1089 pp = p.joinpath('D:x/y')
1090 self.assertEqual(pp, P('D:x/y'))
1091 pp = p.joinpath('D:/x/y')
1092 self.assertEqual(pp, P('D:/x/y'))
1093 pp = p.joinpath('//host/share/x/y')
1094 self.assertEqual(pp, P('//host/share/x/y'))
1095 # Joining with the same drive => the first path is appended to if
1096 # the second path is relative.
1097 pp = p.joinpath('c:x/y')
1098 self.assertEqual(pp, P('C:/a/b/x/y'))
1099 pp = p.joinpath('c:/x/y')
1100 self.assertEqual(pp, P('C:/x/y'))
1101
1102 def test_div(self):
1103 # Basically the same as joinpath()
1104 P = self.cls
1105 p = P('C:/a/b')
1106 self.assertEqual(p / 'x/y', P('C:/a/b/x/y'))
1107 self.assertEqual(p / 'x' / 'y', P('C:/a/b/x/y'))
1108 self.assertEqual(p / '/x/y', P('C:/x/y'))
1109 self.assertEqual(p / '/x' / 'y', P('C:/x/y'))
1110 # Joining with a different drive => the first path is ignored, even
1111 # if the second path is relative.
1112 self.assertEqual(p / 'D:x/y', P('D:x/y'))
1113 self.assertEqual(p / 'D:' / 'x/y', P('D:x/y'))
1114 self.assertEqual(p / 'D:/x/y', P('D:/x/y'))
1115 self.assertEqual(p / 'D:' / '/x/y', P('D:/x/y'))
1116 self.assertEqual(p / '//host/share/x/y', P('//host/share/x/y'))
1117 # Joining with the same drive => the first path is appended to if
1118 # the second path is relative.
Serhiy Storchaka010ff582013-12-06 17:25:51 +02001119 self.assertEqual(p / 'c:x/y', P('C:/a/b/x/y'))
1120 self.assertEqual(p / 'c:/x/y', P('C:/x/y'))
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001121
Antoine Pitrou31119e42013-11-22 17:38:12 +01001122 def test_is_reserved(self):
1123 P = self.cls
1124 self.assertIs(False, P('').is_reserved())
1125 self.assertIs(False, P('/').is_reserved())
1126 self.assertIs(False, P('/foo/bar').is_reserved())
1127 self.assertIs(True, P('con').is_reserved())
1128 self.assertIs(True, P('NUL').is_reserved())
1129 self.assertIs(True, P('NUL.txt').is_reserved())
1130 self.assertIs(True, P('com1').is_reserved())
1131 self.assertIs(True, P('com9.bar').is_reserved())
1132 self.assertIs(False, P('bar.com9').is_reserved())
1133 self.assertIs(True, P('lpt1').is_reserved())
1134 self.assertIs(True, P('lpt9.bar').is_reserved())
1135 self.assertIs(False, P('bar.lpt9').is_reserved())
1136 # Only the last component matters
1137 self.assertIs(False, P('c:/NUL/con/baz').is_reserved())
1138 # UNC paths are never reserved
1139 self.assertIs(False, P('//my/share/nul/con/aux').is_reserved())
1140
Antoine Pitrou31119e42013-11-22 17:38:12 +01001141class PurePathTest(_BasePurePathTest, unittest.TestCase):
1142 cls = pathlib.PurePath
1143
1144 def test_concrete_class(self):
1145 p = self.cls('a')
1146 self.assertIs(type(p),
1147 pathlib.PureWindowsPath if os.name == 'nt' else pathlib.PurePosixPath)
1148
1149 def test_different_flavours_unequal(self):
1150 p = pathlib.PurePosixPath('a')
1151 q = pathlib.PureWindowsPath('a')
1152 self.assertNotEqual(p, q)
1153
1154 def test_different_flavours_unordered(self):
1155 p = pathlib.PurePosixPath('a')
1156 q = pathlib.PureWindowsPath('a')
1157 with self.assertRaises(TypeError):
1158 p < q
1159 with self.assertRaises(TypeError):
1160 p <= q
1161 with self.assertRaises(TypeError):
1162 p > q
1163 with self.assertRaises(TypeError):
1164 p >= q
1165
1166
1167#
1168# Tests for the concrete classes
1169#
1170
1171# Make sure any symbolic links in the base test path are resolved
1172BASE = os.path.realpath(TESTFN)
1173join = lambda *x: os.path.join(BASE, *x)
1174rel_join = lambda *x: os.path.join(TESTFN, *x)
1175
Antoine Pitrou31119e42013-11-22 17:38:12 +01001176only_nt = unittest.skipIf(os.name != 'nt',
1177 'test requires a Windows-compatible system')
1178only_posix = unittest.skipIf(os.name == 'nt',
1179 'test requires a POSIX-compatible system')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001180
1181@only_posix
1182class PosixPathAsPureTest(PurePosixPathTest):
1183 cls = pathlib.PosixPath
1184
1185@only_nt
1186class WindowsPathAsPureTest(PureWindowsPathTest):
1187 cls = pathlib.WindowsPath
1188
Victor Stinnerd7569632016-03-11 22:53:00 +01001189 def test_owner(self):
1190 P = self.cls
1191 with self.assertRaises(NotImplementedError):
1192 P('c:/').owner()
1193
1194 def test_group(self):
1195 P = self.cls
1196 with self.assertRaises(NotImplementedError):
1197 P('c:/').group()
1198
Antoine Pitrou31119e42013-11-22 17:38:12 +01001199
1200class _BasePathTest(object):
1201 """Tests for the FS-accessing functionalities of the Path classes."""
1202
1203 # (BASE)
1204 # |
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001205 # |-- brokenLink -> non-existing
1206 # |-- dirA
1207 # | `-- linkC -> ../dirB
1208 # |-- dirB
1209 # | |-- fileB
1210 # | `-- linkD -> ../dirB
1211 # |-- dirC
1212 # | |-- dirD
1213 # | | `-- fileD
1214 # | `-- fileC
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001215 # |-- dirE # No permissions
Antoine Pitrou31119e42013-11-22 17:38:12 +01001216 # |-- fileA
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001217 # |-- linkA -> fileA
1218 # `-- linkB -> dirB
Antoine Pitrou31119e42013-11-22 17:38:12 +01001219 #
1220
1221 def setUp(self):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001222 def cleanup():
1223 os.chmod(join('dirE'), 0o777)
1224 support.rmtree(BASE)
1225 self.addCleanup(cleanup)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001226 os.mkdir(BASE)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001227 os.mkdir(join('dirA'))
1228 os.mkdir(join('dirB'))
1229 os.mkdir(join('dirC'))
1230 os.mkdir(join('dirC', 'dirD'))
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001231 os.mkdir(join('dirE'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001232 with open(join('fileA'), 'wb') as f:
1233 f.write(b"this is file A\n")
1234 with open(join('dirB', 'fileB'), 'wb') as f:
1235 f.write(b"this is file B\n")
1236 with open(join('dirC', 'fileC'), 'wb') as f:
1237 f.write(b"this is file C\n")
1238 with open(join('dirC', 'dirD', 'fileD'), 'wb') as f:
1239 f.write(b"this is file D\n")
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001240 os.chmod(join('dirE'), 0)
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001241 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001242 # Relative symlinks
1243 os.symlink('fileA', join('linkA'))
1244 os.symlink('non-existing', join('brokenLink'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001245 self.dirlink('dirB', join('linkB'))
1246 self.dirlink(os.path.join('..', 'dirB'), join('dirA', 'linkC'))
Guido van Rossum69bfb152016-01-06 10:31:33 -08001247 # This one goes upwards, creating a loop
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001248 self.dirlink(os.path.join('..', 'dirB'), join('dirB', 'linkD'))
1249
1250 if os.name == 'nt':
1251 # Workaround for http://bugs.python.org/issue13772
1252 def dirlink(self, src, dest):
1253 os.symlink(src, dest, target_is_directory=True)
1254 else:
1255 def dirlink(self, src, dest):
1256 os.symlink(src, dest)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001257
1258 def assertSame(self, path_a, path_b):
1259 self.assertTrue(os.path.samefile(str(path_a), str(path_b)),
1260 "%r and %r don't point to the same file" %
1261 (path_a, path_b))
1262
1263 def assertFileNotFound(self, func, *args, **kwargs):
1264 with self.assertRaises(FileNotFoundError) as cm:
1265 func(*args, **kwargs)
1266 self.assertEqual(cm.exception.errno, errno.ENOENT)
1267
1268 def _test_cwd(self, p):
1269 q = self.cls(os.getcwd())
1270 self.assertEqual(p, q)
1271 self.assertEqual(str(p), str(q))
1272 self.assertIs(type(p), type(q))
1273 self.assertTrue(p.is_absolute())
1274
1275 def test_cwd(self):
1276 p = self.cls.cwd()
1277 self._test_cwd(p)
1278
Antoine Pitrou17cba7d2015-01-12 21:03:41 +01001279 def _test_home(self, p):
1280 q = self.cls(os.path.expanduser('~'))
1281 self.assertEqual(p, q)
1282 self.assertEqual(str(p), str(q))
1283 self.assertIs(type(p), type(q))
1284 self.assertTrue(p.is_absolute())
1285
1286 def test_home(self):
1287 p = self.cls.home()
1288 self._test_home(p)
1289
Antoine Pitrou43e3d942014-05-13 10:50:15 +02001290 def test_samefile(self):
1291 fileA_path = os.path.join(BASE, 'fileA')
1292 fileB_path = os.path.join(BASE, 'dirB', 'fileB')
1293 p = self.cls(fileA_path)
1294 pp = self.cls(fileA_path)
1295 q = self.cls(fileB_path)
1296 self.assertTrue(p.samefile(fileA_path))
1297 self.assertTrue(p.samefile(pp))
1298 self.assertFalse(p.samefile(fileB_path))
1299 self.assertFalse(p.samefile(q))
1300 # Test the non-existent file case
1301 non_existent = os.path.join(BASE, 'foo')
1302 r = self.cls(non_existent)
1303 self.assertRaises(FileNotFoundError, p.samefile, r)
1304 self.assertRaises(FileNotFoundError, p.samefile, non_existent)
1305 self.assertRaises(FileNotFoundError, r.samefile, p)
1306 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1307 self.assertRaises(FileNotFoundError, r.samefile, r)
1308 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1309
Antoine Pitrou31119e42013-11-22 17:38:12 +01001310 def test_empty_path(self):
1311 # The empty path points to '.'
1312 p = self.cls('')
1313 self.assertEqual(p.stat(), os.stat('.'))
1314
Antoine Pitrou8477ed62014-12-30 20:54:45 +01001315 def test_expanduser_common(self):
1316 P = self.cls
1317 p = P('~')
1318 self.assertEqual(p.expanduser(), P(os.path.expanduser('~')))
1319 p = P('foo')
1320 self.assertEqual(p.expanduser(), p)
1321 p = P('/~')
1322 self.assertEqual(p.expanduser(), p)
1323 p = P('../~')
1324 self.assertEqual(p.expanduser(), p)
1325 p = P(P('').absolute().anchor) / '~'
1326 self.assertEqual(p.expanduser(), p)
1327
Antoine Pitrou31119e42013-11-22 17:38:12 +01001328 def test_exists(self):
1329 P = self.cls
1330 p = P(BASE)
1331 self.assertIs(True, p.exists())
1332 self.assertIs(True, (p / 'dirA').exists())
1333 self.assertIs(True, (p / 'fileA').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001334 self.assertIs(False, (p / 'fileA' / 'bah').exists())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001335 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001336 self.assertIs(True, (p / 'linkA').exists())
1337 self.assertIs(True, (p / 'linkB').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001338 self.assertIs(True, (p / 'linkB' / 'fileB').exists())
1339 self.assertIs(False, (p / 'linkA' / 'bah').exists())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001340 self.assertIs(False, (p / 'foo').exists())
1341 self.assertIs(False, P('/xyzzy').exists())
1342
1343 def test_open_common(self):
1344 p = self.cls(BASE)
1345 with (p / 'fileA').open('r') as f:
1346 self.assertIsInstance(f, io.TextIOBase)
1347 self.assertEqual(f.read(), "this is file A\n")
1348 with (p / 'fileA').open('rb') as f:
1349 self.assertIsInstance(f, io.BufferedIOBase)
1350 self.assertEqual(f.read().strip(), b"this is file A")
1351 with (p / 'fileA').open('rb', buffering=0) as f:
1352 self.assertIsInstance(f, io.RawIOBase)
1353 self.assertEqual(f.read().strip(), b"this is file A")
1354
Georg Brandlea683982014-10-01 19:12:33 +02001355 def test_read_write_bytes(self):
1356 p = self.cls(BASE)
1357 (p / 'fileA').write_bytes(b'abcdefg')
1358 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1359 # check that trying to write str does not truncate the file
1360 self.assertRaises(TypeError, (p / 'fileA').write_bytes, 'somestr')
1361 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1362
1363 def test_read_write_text(self):
1364 p = self.cls(BASE)
1365 (p / 'fileA').write_text('äbcdefg', encoding='latin-1')
1366 self.assertEqual((p / 'fileA').read_text(
1367 encoding='utf-8', errors='ignore'), 'bcdefg')
1368 # check that trying to write bytes does not truncate the file
1369 self.assertRaises(TypeError, (p / 'fileA').write_text, b'somebytes')
1370 self.assertEqual((p / 'fileA').read_text(encoding='latin-1'), 'äbcdefg')
1371
Antoine Pitrou31119e42013-11-22 17:38:12 +01001372 def test_iterdir(self):
1373 P = self.cls
1374 p = P(BASE)
1375 it = p.iterdir()
1376 paths = set(it)
Guido van Rossum6c2d33a2016-01-06 09:42:07 -08001377 expected = ['dirA', 'dirB', 'dirC', 'dirE', 'fileA']
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001378 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001379 expected += ['linkA', 'linkB', 'brokenLink']
1380 self.assertEqual(paths, { P(BASE, q) for q in expected })
1381
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001382 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001383 def test_iterdir_symlink(self):
1384 # __iter__ on a symlink to a directory
1385 P = self.cls
1386 p = P(BASE, 'linkB')
1387 paths = set(p.iterdir())
1388 expected = { P(BASE, 'linkB', q) for q in ['fileB', 'linkD'] }
1389 self.assertEqual(paths, expected)
1390
1391 def test_iterdir_nodir(self):
1392 # __iter__ on something that is not a directory
1393 p = self.cls(BASE, 'fileA')
1394 with self.assertRaises(OSError) as cm:
1395 next(p.iterdir())
1396 # ENOENT or EINVAL under Windows, ENOTDIR otherwise
1397 # (see issue #12802)
1398 self.assertIn(cm.exception.errno, (errno.ENOTDIR,
1399 errno.ENOENT, errno.EINVAL))
1400
1401 def test_glob_common(self):
1402 def _check(glob, expected):
1403 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1404 P = self.cls
1405 p = P(BASE)
1406 it = p.glob("fileA")
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001407 self.assertIsInstance(it, collections.abc.Iterator)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001408 _check(it, ["fileA"])
1409 _check(p.glob("fileB"), [])
1410 _check(p.glob("dir*/file*"), ["dirB/fileB", "dirC/fileC"])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001411 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001412 _check(p.glob("*A"), ['dirA', 'fileA'])
1413 else:
1414 _check(p.glob("*A"), ['dirA', 'fileA', 'linkA'])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001415 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001416 _check(p.glob("*B/*"), ['dirB/fileB'])
1417 else:
1418 _check(p.glob("*B/*"), ['dirB/fileB', 'dirB/linkD',
1419 'linkB/fileB', 'linkB/linkD'])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001420 if not support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001421 _check(p.glob("*/fileB"), ['dirB/fileB'])
1422 else:
1423 _check(p.glob("*/fileB"), ['dirB/fileB', 'linkB/fileB'])
1424
1425 def test_rglob_common(self):
1426 def _check(glob, expected):
1427 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1428 P = self.cls
1429 p = P(BASE)
1430 it = p.rglob("fileA")
Serhiy Storchaka2e576f52017-04-24 09:05:00 +03001431 self.assertIsInstance(it, collections.abc.Iterator)
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001432 _check(it, ["fileA"])
1433 _check(p.rglob("fileB"), ["dirB/fileB"])
1434 _check(p.rglob("*/fileA"), [])
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001435 if not support.can_symlink():
Guido van Rossum9c39b672016-01-07 13:12:34 -08001436 _check(p.rglob("*/fileB"), ["dirB/fileB"])
1437 else:
1438 _check(p.rglob("*/fileB"), ["dirB/fileB", "dirB/linkD/fileB",
1439 "linkB/fileB", "dirA/linkC/fileB"])
Guido van Rossumbc9fdda2016-01-07 10:56:36 -08001440 _check(p.rglob("file*"), ["fileA", "dirB/fileB",
1441 "dirC/fileC", "dirC/dirD/fileD"])
Antoine Pitrou31119e42013-11-22 17:38:12 +01001442 p = P(BASE, "dirC")
1443 _check(p.rglob("file*"), ["dirC/fileC", "dirC/dirD/fileD"])
1444 _check(p.rglob("*/*"), ["dirC/dirD/fileD"])
1445
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001446 @support.skip_unless_symlink
Guido van Rossum69bfb152016-01-06 10:31:33 -08001447 def test_rglob_symlink_loop(self):
1448 # Don't get fooled by symlink loops (Issue #26012)
1449 P = self.cls
1450 p = P(BASE)
1451 given = set(p.rglob('*'))
1452 expect = {'brokenLink',
1453 'dirA', 'dirA/linkC',
1454 'dirB', 'dirB/fileB', 'dirB/linkD',
1455 'dirC', 'dirC/dirD', 'dirC/dirD/fileD', 'dirC/fileC',
1456 'dirE',
1457 'fileA',
1458 'linkA',
1459 'linkB',
1460 }
1461 self.assertEqual(given, {p / x for x in expect})
1462
Antoine Pitrou31119e42013-11-22 17:38:12 +01001463 def test_glob_dotdot(self):
1464 # ".." is not special in globs
1465 P = self.cls
1466 p = P(BASE)
1467 self.assertEqual(set(p.glob("..")), { P(BASE, "..") })
1468 self.assertEqual(set(p.glob("dirA/../file*")), { P(BASE, "dirA/../fileA") })
1469 self.assertEqual(set(p.glob("../xyzzy")), set())
1470
Ezio Melottiac1e7f62015-12-28 23:50:19 +02001471
Steve Dower98eb3602016-11-09 12:58:17 -08001472 def _check_resolve(self, p, expected, strict=True):
1473 q = p.resolve(strict)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001474 self.assertEqual(q, expected)
1475
Ezio Melottiac1e7f62015-12-28 23:50:19 +02001476 # this can be used to check both relative and absolute resolutions
1477 _check_resolve_relative = _check_resolve_absolute = _check_resolve
Antoine Pitrou31119e42013-11-22 17:38:12 +01001478
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001479 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001480 def test_resolve_common(self):
1481 P = self.cls
1482 p = P(BASE, 'foo')
1483 with self.assertRaises(OSError) as cm:
Steve Dower98eb3602016-11-09 12:58:17 -08001484 p.resolve(strict=True)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001485 self.assertEqual(cm.exception.errno, errno.ENOENT)
Steve Dower98eb3602016-11-09 12:58:17 -08001486 # Non-strict
1487 self.assertEqual(str(p.resolve(strict=False)),
1488 os.path.join(BASE, 'foo'))
1489 p = P(BASE, 'foo', 'in', 'spam')
1490 self.assertEqual(str(p.resolve(strict=False)),
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001491 os.path.join(BASE, 'foo', 'in', 'spam'))
Steve Dower98eb3602016-11-09 12:58:17 -08001492 p = P(BASE, '..', 'foo', 'in', 'spam')
1493 self.assertEqual(str(p.resolve(strict=False)),
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001494 os.path.abspath(os.path.join('foo', 'in', 'spam')))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001495 # These are all relative symlinks
1496 p = P(BASE, 'dirB', 'fileB')
1497 self._check_resolve_relative(p, p)
1498 p = P(BASE, 'linkA')
1499 self._check_resolve_relative(p, P(BASE, 'fileA'))
1500 p = P(BASE, 'dirA', 'linkC', 'fileB')
1501 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
1502 p = P(BASE, 'dirB', 'linkD', 'fileB')
1503 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
Steve Dower98eb3602016-11-09 12:58:17 -08001504 # Non-strict
1505 p = P(BASE, 'dirA', 'linkC', 'fileB', 'foo', 'in', 'spam')
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001506 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB', 'foo', 'in',
1507 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001508 p = P(BASE, 'dirA', 'linkC', '..', 'foo', 'in', 'spam')
1509 if os.name == 'nt':
1510 # In Windows, if linkY points to dirB, 'dirA\linkY\..'
1511 # resolves to 'dirA' without resolving linkY first.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001512 self._check_resolve_relative(p, P(BASE, 'dirA', 'foo', 'in',
1513 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001514 else:
1515 # In Posix, if linkY points to dirB, 'dirA/linkY/..'
1516 # resolves to 'dirB/..' first before resolving to parent of dirB.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001517 self._check_resolve_relative(p, P(BASE, 'foo', 'in', 'spam'), False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001518 # Now create absolute symlinks
1519 d = tempfile.mkdtemp(suffix='-dirD')
Victor Stinnerec864692014-07-21 19:19:05 +02001520 self.addCleanup(support.rmtree, d)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001521 os.symlink(os.path.join(d), join('dirA', 'linkX'))
1522 os.symlink(join('dirB'), os.path.join(d, 'linkY'))
1523 p = P(BASE, 'dirA', 'linkX', 'linkY', 'fileB')
1524 self._check_resolve_absolute(p, P(BASE, 'dirB', 'fileB'))
Steve Dower98eb3602016-11-09 12:58:17 -08001525 # Non-strict
1526 p = P(BASE, 'dirA', 'linkX', 'linkY', 'foo', 'in', 'spam')
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001527 self._check_resolve_relative(p, P(BASE, 'dirB', 'foo', 'in', 'spam'),
1528 False)
Steve Dower98eb3602016-11-09 12:58:17 -08001529 p = P(BASE, 'dirA', 'linkX', 'linkY', '..', 'foo', 'in', 'spam')
1530 if os.name == 'nt':
1531 # In Windows, if linkY points to dirB, 'dirA\linkY\..'
1532 # resolves to 'dirA' without resolving linkY first.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001533 self._check_resolve_relative(p, P(d, 'foo', 'in', 'spam'), False)
Steve Dower98eb3602016-11-09 12:58:17 -08001534 else:
1535 # In Posix, if linkY points to dirB, 'dirA/linkY/..'
1536 # resolves to 'dirB/..' first before resolving to parent of dirB.
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001537 self._check_resolve_relative(p, P(BASE, 'foo', 'in', 'spam'), False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001538
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001539 @support.skip_unless_symlink
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001540 def test_resolve_dot(self):
1541 # See https://bitbucket.org/pitrou/pathlib/issue/9/pathresolve-fails-on-complex-symlinks
1542 p = self.cls(BASE)
1543 self.dirlink('.', join('0'))
Antoine Pitroucc157512013-12-03 17:13:13 +01001544 self.dirlink(os.path.join('0', '0'), join('1'))
1545 self.dirlink(os.path.join('1', '1'), join('2'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001546 q = p / '2'
Steve Dower98eb3602016-11-09 12:58:17 -08001547 self.assertEqual(q.resolve(strict=True), p)
1548 r = q / '3' / '4'
1549 self.assertRaises(FileNotFoundError, r.resolve, strict=True)
1550 # Non-strict
Antoine Pietriadd98eb2017-06-07 17:29:17 +02001551 self.assertEqual(r.resolve(strict=False), p / '3' / '4')
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001552
Antoine Pitrou31119e42013-11-22 17:38:12 +01001553 def test_with(self):
1554 p = self.cls(BASE)
1555 it = p.iterdir()
1556 it2 = p.iterdir()
1557 next(it2)
1558 with p:
1559 pass
1560 # I/O operation on closed path
1561 self.assertRaises(ValueError, next, it)
1562 self.assertRaises(ValueError, next, it2)
1563 self.assertRaises(ValueError, p.open)
1564 self.assertRaises(ValueError, p.resolve)
1565 self.assertRaises(ValueError, p.absolute)
1566 self.assertRaises(ValueError, p.__enter__)
1567
1568 def test_chmod(self):
1569 p = self.cls(BASE) / 'fileA'
1570 mode = p.stat().st_mode
1571 # Clear writable bit
1572 new_mode = mode & ~0o222
1573 p.chmod(new_mode)
1574 self.assertEqual(p.stat().st_mode, new_mode)
1575 # Set writable bit
1576 new_mode = mode | 0o222
1577 p.chmod(new_mode)
1578 self.assertEqual(p.stat().st_mode, new_mode)
1579
1580 # XXX also need a test for lchmod
1581
1582 def test_stat(self):
1583 p = self.cls(BASE) / 'fileA'
1584 st = p.stat()
1585 self.assertEqual(p.stat(), st)
1586 # Change file mode by flipping write bit
1587 p.chmod(st.st_mode ^ 0o222)
1588 self.addCleanup(p.chmod, st.st_mode)
1589 self.assertNotEqual(p.stat(), st)
1590
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001591 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001592 def test_lstat(self):
1593 p = self.cls(BASE)/ 'linkA'
1594 st = p.stat()
1595 self.assertNotEqual(st, p.lstat())
1596
1597 def test_lstat_nosymlink(self):
1598 p = self.cls(BASE) / 'fileA'
1599 st = p.stat()
1600 self.assertEqual(st, p.lstat())
1601
1602 @unittest.skipUnless(pwd, "the pwd module is needed for this test")
1603 def test_owner(self):
1604 p = self.cls(BASE) / 'fileA'
1605 uid = p.stat().st_uid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001606 try:
1607 name = pwd.getpwuid(uid).pw_name
1608 except KeyError:
1609 self.skipTest(
1610 "user %d doesn't have an entry in the system database" % uid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001611 self.assertEqual(name, p.owner())
1612
1613 @unittest.skipUnless(grp, "the grp module is needed for this test")
1614 def test_group(self):
1615 p = self.cls(BASE) / 'fileA'
1616 gid = p.stat().st_gid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001617 try:
1618 name = grp.getgrgid(gid).gr_name
1619 except KeyError:
1620 self.skipTest(
1621 "group %d doesn't have an entry in the system database" % gid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001622 self.assertEqual(name, p.group())
1623
1624 def test_unlink(self):
1625 p = self.cls(BASE) / 'fileA'
1626 p.unlink()
1627 self.assertFileNotFound(p.stat)
1628 self.assertFileNotFound(p.unlink)
1629
1630 def test_rmdir(self):
1631 p = self.cls(BASE) / 'dirA'
1632 for q in p.iterdir():
1633 q.unlink()
1634 p.rmdir()
1635 self.assertFileNotFound(p.stat)
1636 self.assertFileNotFound(p.unlink)
1637
1638 def test_rename(self):
1639 P = self.cls(BASE)
1640 p = P / 'fileA'
1641 size = p.stat().st_size
1642 # Renaming to another path
1643 q = P / 'dirA' / 'fileAA'
1644 p.rename(q)
1645 self.assertEqual(q.stat().st_size, size)
1646 self.assertFileNotFound(p.stat)
1647 # Renaming to a str of a relative path
1648 r = rel_join('fileAAA')
1649 q.rename(r)
1650 self.assertEqual(os.stat(r).st_size, size)
1651 self.assertFileNotFound(q.stat)
1652
1653 def test_replace(self):
1654 P = self.cls(BASE)
1655 p = P / 'fileA'
1656 size = p.stat().st_size
1657 # Replacing a non-existing path
1658 q = P / 'dirA' / 'fileAA'
1659 p.replace(q)
1660 self.assertEqual(q.stat().st_size, size)
1661 self.assertFileNotFound(p.stat)
1662 # Replacing another (existing) path
1663 r = rel_join('dirB', 'fileB')
1664 q.replace(r)
1665 self.assertEqual(os.stat(r).st_size, size)
1666 self.assertFileNotFound(q.stat)
1667
1668 def test_touch_common(self):
1669 P = self.cls(BASE)
1670 p = P / 'newfileA'
1671 self.assertFalse(p.exists())
1672 p.touch()
1673 self.assertTrue(p.exists())
Antoine Pitrou0f575642013-11-22 23:20:08 +01001674 st = p.stat()
1675 old_mtime = st.st_mtime
1676 old_mtime_ns = st.st_mtime_ns
Antoine Pitrou31119e42013-11-22 17:38:12 +01001677 # Rewind the mtime sufficiently far in the past to work around
1678 # filesystem-specific timestamp granularity.
1679 os.utime(str(p), (old_mtime - 10, old_mtime - 10))
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001680 # The file mtime should be refreshed by calling touch() again
Antoine Pitrou31119e42013-11-22 17:38:12 +01001681 p.touch()
Antoine Pitrou0f575642013-11-22 23:20:08 +01001682 st = p.stat()
Antoine Pitrou2cf39172013-11-23 15:25:59 +01001683 self.assertGreaterEqual(st.st_mtime_ns, old_mtime_ns)
1684 self.assertGreaterEqual(st.st_mtime, old_mtime)
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001685 # Now with exist_ok=False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001686 p = P / 'newfileB'
1687 self.assertFalse(p.exists())
1688 p.touch(mode=0o700, exist_ok=False)
1689 self.assertTrue(p.exists())
1690 self.assertRaises(OSError, p.touch, exist_ok=False)
1691
Antoine Pitrou8b784932013-11-23 14:52:39 +01001692 def test_touch_nochange(self):
1693 P = self.cls(BASE)
1694 p = P / 'fileA'
1695 p.touch()
1696 with p.open('rb') as f:
1697 self.assertEqual(f.read().strip(), b"this is file A")
1698
Antoine Pitrou31119e42013-11-22 17:38:12 +01001699 def test_mkdir(self):
1700 P = self.cls(BASE)
1701 p = P / 'newdirA'
1702 self.assertFalse(p.exists())
1703 p.mkdir()
1704 self.assertTrue(p.exists())
1705 self.assertTrue(p.is_dir())
1706 with self.assertRaises(OSError) as cm:
1707 p.mkdir()
1708 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001709
1710 def test_mkdir_parents(self):
1711 # Creating a chain of directories
1712 p = self.cls(BASE, 'newdirB', 'newdirC')
1713 self.assertFalse(p.exists())
1714 with self.assertRaises(OSError) as cm:
1715 p.mkdir()
1716 self.assertEqual(cm.exception.errno, errno.ENOENT)
1717 p.mkdir(parents=True)
1718 self.assertTrue(p.exists())
1719 self.assertTrue(p.is_dir())
1720 with self.assertRaises(OSError) as cm:
1721 p.mkdir(parents=True)
1722 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001723 # test `mode` arg
1724 mode = stat.S_IMODE(p.stat().st_mode) # default mode
1725 p = self.cls(BASE, 'newdirD', 'newdirE')
1726 p.mkdir(0o555, parents=True)
1727 self.assertTrue(p.exists())
1728 self.assertTrue(p.is_dir())
1729 if os.name != 'nt':
1730 # the directory's permissions follow the mode argument
Gregory P. Smithb599c612014-01-20 01:10:33 -08001731 self.assertEqual(stat.S_IMODE(p.stat().st_mode), 0o7555 & mode)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001732 # the parent's permissions follow the default process settings
1733 self.assertEqual(stat.S_IMODE(p.parent.stat().st_mode), mode)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001734
Barry Warsaw7c549c42014-08-05 11:28:12 -04001735 def test_mkdir_exist_ok(self):
1736 p = self.cls(BASE, 'dirB')
1737 st_ctime_first = p.stat().st_ctime
1738 self.assertTrue(p.exists())
1739 self.assertTrue(p.is_dir())
1740 with self.assertRaises(FileExistsError) as cm:
1741 p.mkdir()
1742 self.assertEqual(cm.exception.errno, errno.EEXIST)
1743 p.mkdir(exist_ok=True)
1744 self.assertTrue(p.exists())
1745 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1746
1747 def test_mkdir_exist_ok_with_parent(self):
1748 p = self.cls(BASE, 'dirC')
1749 self.assertTrue(p.exists())
1750 with self.assertRaises(FileExistsError) as cm:
1751 p.mkdir()
1752 self.assertEqual(cm.exception.errno, errno.EEXIST)
1753 p = p / 'newdirC'
1754 p.mkdir(parents=True)
1755 st_ctime_first = p.stat().st_ctime
1756 self.assertTrue(p.exists())
1757 with self.assertRaises(FileExistsError) as cm:
1758 p.mkdir(parents=True)
1759 self.assertEqual(cm.exception.errno, errno.EEXIST)
1760 p.mkdir(parents=True, exist_ok=True)
1761 self.assertTrue(p.exists())
1762 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1763
Serhiy Storchakaaf7b9ec2017-03-24 20:51:53 +02001764 def test_mkdir_exist_ok_root(self):
1765 # Issue #25803: A drive root could raise PermissionError on Windows
1766 self.cls('/').resolve().mkdir(exist_ok=True)
1767 self.cls('/').resolve().mkdir(parents=True, exist_ok=True)
1768
Steve Dowerd3c48532017-02-04 14:54:56 -08001769 @only_nt # XXX: not sure how to test this on POSIX
1770 def test_mkdir_with_unknown_drive(self):
1771 for d in 'ZYXWVUTSRQPONMLKJIHGFEDCBA':
1772 p = self.cls(d + ':\\')
1773 if not p.is_dir():
1774 break
1775 else:
1776 self.skipTest("cannot find a drive that doesn't exist")
1777 with self.assertRaises(OSError):
1778 (p / 'child' / 'path').mkdir(parents=True)
1779
Barry Warsaw7c549c42014-08-05 11:28:12 -04001780 def test_mkdir_with_child_file(self):
1781 p = self.cls(BASE, 'dirB', 'fileB')
1782 self.assertTrue(p.exists())
1783 # An exception is raised when the last path component is an existing
1784 # regular file, regardless of whether exist_ok is true or not.
1785 with self.assertRaises(FileExistsError) as cm:
1786 p.mkdir(parents=True)
1787 self.assertEqual(cm.exception.errno, errno.EEXIST)
1788 with self.assertRaises(FileExistsError) as cm:
1789 p.mkdir(parents=True, exist_ok=True)
1790 self.assertEqual(cm.exception.errno, errno.EEXIST)
1791
1792 def test_mkdir_no_parents_file(self):
1793 p = self.cls(BASE, 'fileA')
1794 self.assertTrue(p.exists())
1795 # An exception is raised when the last path component is an existing
1796 # regular file, regardless of whether exist_ok is true or not.
1797 with self.assertRaises(FileExistsError) as cm:
1798 p.mkdir()
1799 self.assertEqual(cm.exception.errno, errno.EEXIST)
1800 with self.assertRaises(FileExistsError) as cm:
1801 p.mkdir(exist_ok=True)
1802 self.assertEqual(cm.exception.errno, errno.EEXIST)
1803
Armin Rigo22a594a2017-04-13 20:08:15 +02001804 def test_mkdir_concurrent_parent_creation(self):
1805 for pattern_num in range(32):
1806 p = self.cls(BASE, 'dirCPC%d' % pattern_num)
1807 self.assertFalse(p.exists())
1808
1809 def my_mkdir(path, mode=0o777):
1810 path = str(path)
1811 # Emulate another process that would create the directory
1812 # just before we try to create it ourselves. We do it
1813 # in all possible pattern combinations, assuming that this
1814 # function is called at most 5 times (dirCPC/dir1/dir2,
1815 # dirCPC/dir1, dirCPC, dirCPC/dir1, dirCPC/dir1/dir2).
1816 if pattern.pop():
1817 os.mkdir(path, mode) # from another process
1818 concurrently_created.add(path)
1819 os.mkdir(path, mode) # our real call
1820
1821 pattern = [bool(pattern_num & (1 << n)) for n in range(5)]
1822 concurrently_created = set()
1823 p12 = p / 'dir1' / 'dir2'
1824 try:
1825 with mock.patch("pathlib._normal_accessor.mkdir", my_mkdir):
1826 p12.mkdir(parents=True, exist_ok=False)
1827 except FileExistsError:
1828 self.assertIn(str(p12), concurrently_created)
1829 else:
1830 self.assertNotIn(str(p12), concurrently_created)
1831 self.assertTrue(p.exists())
1832
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001833 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01001834 def test_symlink_to(self):
1835 P = self.cls(BASE)
1836 target = P / 'fileA'
1837 # Symlinking a path target
1838 link = P / 'dirA' / 'linkAA'
1839 link.symlink_to(target)
1840 self.assertEqual(link.stat(), target.stat())
1841 self.assertNotEqual(link.lstat(), target.stat())
1842 # Symlinking a str target
1843 link = P / 'dirA' / 'linkAAA'
1844 link.symlink_to(str(target))
1845 self.assertEqual(link.stat(), target.stat())
1846 self.assertNotEqual(link.lstat(), target.stat())
1847 self.assertFalse(link.is_dir())
1848 # Symlinking to a directory
1849 target = P / 'dirB'
1850 link = P / 'dirA' / 'linkAAAA'
1851 link.symlink_to(target, target_is_directory=True)
1852 self.assertEqual(link.stat(), target.stat())
1853 self.assertNotEqual(link.lstat(), target.stat())
1854 self.assertTrue(link.is_dir())
1855 self.assertTrue(list(link.iterdir()))
1856
1857 def test_is_dir(self):
1858 P = self.cls(BASE)
1859 self.assertTrue((P / 'dirA').is_dir())
1860 self.assertFalse((P / 'fileA').is_dir())
1861 self.assertFalse((P / 'non-existing').is_dir())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001862 self.assertFalse((P / 'fileA' / 'bah').is_dir())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001863 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001864 self.assertFalse((P / 'linkA').is_dir())
1865 self.assertTrue((P / 'linkB').is_dir())
1866 self.assertFalse((P/ 'brokenLink').is_dir())
1867
1868 def test_is_file(self):
1869 P = self.cls(BASE)
1870 self.assertTrue((P / 'fileA').is_file())
1871 self.assertFalse((P / 'dirA').is_file())
1872 self.assertFalse((P / 'non-existing').is_file())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001873 self.assertFalse((P / 'fileA' / 'bah').is_file())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001874 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001875 self.assertTrue((P / 'linkA').is_file())
1876 self.assertFalse((P / 'linkB').is_file())
1877 self.assertFalse((P/ 'brokenLink').is_file())
1878
Cooper Lees173ff4a2017-08-01 15:35:45 -07001879 @only_posix
1880 def test_is_mount(self):
1881 P = self.cls(BASE)
1882 R = self.cls('/') # TODO: Work out windows
1883 self.assertFalse((P / 'fileA').is_mount())
1884 self.assertFalse((P / 'dirA').is_mount())
1885 self.assertFalse((P / 'non-existing').is_mount())
1886 self.assertFalse((P / 'fileA' / 'bah').is_mount())
1887 self.assertTrue(R.is_mount())
1888 if support.can_symlink():
1889 self.assertFalse((P / 'linkA').is_mount())
1890
Antoine Pitrou31119e42013-11-22 17:38:12 +01001891 def test_is_symlink(self):
1892 P = self.cls(BASE)
1893 self.assertFalse((P / 'fileA').is_symlink())
1894 self.assertFalse((P / 'dirA').is_symlink())
1895 self.assertFalse((P / 'non-existing').is_symlink())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001896 self.assertFalse((P / 'fileA' / 'bah').is_symlink())
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07001897 if support.can_symlink():
Antoine Pitrou31119e42013-11-22 17:38:12 +01001898 self.assertTrue((P / 'linkA').is_symlink())
1899 self.assertTrue((P / 'linkB').is_symlink())
1900 self.assertTrue((P/ 'brokenLink').is_symlink())
1901
1902 def test_is_fifo_false(self):
1903 P = self.cls(BASE)
1904 self.assertFalse((P / 'fileA').is_fifo())
1905 self.assertFalse((P / 'dirA').is_fifo())
1906 self.assertFalse((P / 'non-existing').is_fifo())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001907 self.assertFalse((P / 'fileA' / 'bah').is_fifo())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001908
1909 @unittest.skipUnless(hasattr(os, "mkfifo"), "os.mkfifo() required")
1910 def test_is_fifo_true(self):
1911 P = self.cls(BASE, 'myfifo')
xdegaye92c2ca72017-11-12 17:31:07 +01001912 try:
1913 os.mkfifo(str(P))
1914 except PermissionError as e:
1915 self.skipTest('os.mkfifo(): %s' % e)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001916 self.assertTrue(P.is_fifo())
1917 self.assertFalse(P.is_socket())
1918 self.assertFalse(P.is_file())
1919
1920 def test_is_socket_false(self):
1921 P = self.cls(BASE)
1922 self.assertFalse((P / 'fileA').is_socket())
1923 self.assertFalse((P / 'dirA').is_socket())
1924 self.assertFalse((P / 'non-existing').is_socket())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001925 self.assertFalse((P / 'fileA' / 'bah').is_socket())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001926
1927 @unittest.skipUnless(hasattr(socket, "AF_UNIX"), "Unix sockets required")
1928 def test_is_socket_true(self):
1929 P = self.cls(BASE, 'mysock')
Antoine Pitrou330ce592013-11-22 18:05:06 +01001930 sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001931 self.addCleanup(sock.close)
Antoine Pitrou330ce592013-11-22 18:05:06 +01001932 try:
1933 sock.bind(str(P))
1934 except OSError as e:
Xavier de Gayee88ed052016-12-14 11:52:28 +01001935 if (isinstance(e, PermissionError) or
1936 "AF_UNIX path too long" in str(e)):
Antoine Pitrou330ce592013-11-22 18:05:06 +01001937 self.skipTest("cannot bind Unix socket: " + str(e))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001938 self.assertTrue(P.is_socket())
1939 self.assertFalse(P.is_fifo())
1940 self.assertFalse(P.is_file())
1941
1942 def test_is_block_device_false(self):
1943 P = self.cls(BASE)
1944 self.assertFalse((P / 'fileA').is_block_device())
1945 self.assertFalse((P / 'dirA').is_block_device())
1946 self.assertFalse((P / 'non-existing').is_block_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001947 self.assertFalse((P / 'fileA' / 'bah').is_block_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001948
1949 def test_is_char_device_false(self):
1950 P = self.cls(BASE)
1951 self.assertFalse((P / 'fileA').is_char_device())
1952 self.assertFalse((P / 'dirA').is_char_device())
1953 self.assertFalse((P / 'non-existing').is_char_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001954 self.assertFalse((P / 'fileA' / 'bah').is_char_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001955
1956 def test_is_char_device_true(self):
1957 # Under Unix, /dev/null should generally be a char device
1958 P = self.cls('/dev/null')
1959 if not P.exists():
1960 self.skipTest("/dev/null required")
1961 self.assertTrue(P.is_char_device())
1962 self.assertFalse(P.is_block_device())
1963 self.assertFalse(P.is_file())
1964
1965 def test_pickling_common(self):
1966 p = self.cls(BASE, 'fileA')
1967 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
1968 dumped = pickle.dumps(p, proto)
1969 pp = pickle.loads(dumped)
1970 self.assertEqual(pp.stat(), p.stat())
1971
1972 def test_parts_interning(self):
1973 P = self.cls
1974 p = P('/usr/bin/foo')
1975 q = P('/usr/local/bin')
1976 # 'usr'
1977 self.assertIs(p.parts[1], q.parts[1])
1978 # 'bin'
1979 self.assertIs(p.parts[2], q.parts[3])
1980
Antoine Pitrouc274fd22013-12-16 19:57:41 +01001981 def _check_complex_symlinks(self, link0_target):
1982 # Test solving a non-looping chain of symlinks (issue #19887)
1983 P = self.cls(BASE)
1984 self.dirlink(os.path.join('link0', 'link0'), join('link1'))
1985 self.dirlink(os.path.join('link1', 'link1'), join('link2'))
1986 self.dirlink(os.path.join('link2', 'link2'), join('link3'))
1987 self.dirlink(link0_target, join('link0'))
1988
1989 # Resolve absolute paths
1990 p = (P / 'link0').resolve()
1991 self.assertEqual(p, P)
1992 self.assertEqual(str(p), BASE)
1993 p = (P / 'link1').resolve()
1994 self.assertEqual(p, P)
1995 self.assertEqual(str(p), BASE)
1996 p = (P / 'link2').resolve()
1997 self.assertEqual(p, P)
1998 self.assertEqual(str(p), BASE)
1999 p = (P / 'link3').resolve()
2000 self.assertEqual(p, P)
2001 self.assertEqual(str(p), BASE)
2002
2003 # Resolve relative paths
2004 old_path = os.getcwd()
2005 os.chdir(BASE)
2006 try:
2007 p = self.cls('link0').resolve()
2008 self.assertEqual(p, P)
2009 self.assertEqual(str(p), BASE)
2010 p = self.cls('link1').resolve()
2011 self.assertEqual(p, P)
2012 self.assertEqual(str(p), BASE)
2013 p = self.cls('link2').resolve()
2014 self.assertEqual(p, P)
2015 self.assertEqual(str(p), BASE)
2016 p = self.cls('link3').resolve()
2017 self.assertEqual(p, P)
2018 self.assertEqual(str(p), BASE)
2019 finally:
2020 os.chdir(old_path)
2021
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002022 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002023 def test_complex_symlinks_absolute(self):
2024 self._check_complex_symlinks(BASE)
2025
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002026 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002027 def test_complex_symlinks_relative(self):
2028 self._check_complex_symlinks('.')
2029
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002030 @support.skip_unless_symlink
Antoine Pitrouc274fd22013-12-16 19:57:41 +01002031 def test_complex_symlinks_relative_dot_dot(self):
2032 self._check_complex_symlinks(os.path.join('dirA', '..'))
2033
Antoine Pitrou31119e42013-11-22 17:38:12 +01002034
2035class PathTest(_BasePathTest, unittest.TestCase):
2036 cls = pathlib.Path
2037
2038 def test_concrete_class(self):
2039 p = self.cls('a')
2040 self.assertIs(type(p),
2041 pathlib.WindowsPath if os.name == 'nt' else pathlib.PosixPath)
2042
2043 def test_unsupported_flavour(self):
2044 if os.name == 'nt':
2045 self.assertRaises(NotImplementedError, pathlib.PosixPath)
2046 else:
2047 self.assertRaises(NotImplementedError, pathlib.WindowsPath)
2048
Berker Peksag4a208e42016-01-30 17:50:48 +02002049 def test_glob_empty_pattern(self):
2050 p = self.cls()
2051 with self.assertRaisesRegex(ValueError, 'Unacceptable pattern'):
2052 list(p.glob(''))
2053
Antoine Pitrou31119e42013-11-22 17:38:12 +01002054
2055@only_posix
2056class PosixPathTest(_BasePathTest, unittest.TestCase):
2057 cls = pathlib.PosixPath
2058
Steve Dower98eb3602016-11-09 12:58:17 -08002059 def _check_symlink_loop(self, *args, strict=True):
Antoine Pitrou31119e42013-11-22 17:38:12 +01002060 path = self.cls(*args)
2061 with self.assertRaises(RuntimeError):
Steve Dower98eb3602016-11-09 12:58:17 -08002062 print(path.resolve(strict))
Antoine Pitrou31119e42013-11-22 17:38:12 +01002063
2064 def test_open_mode(self):
2065 old_mask = os.umask(0)
2066 self.addCleanup(os.umask, old_mask)
2067 p = self.cls(BASE)
2068 with (p / 'new_file').open('wb'):
2069 pass
2070 st = os.stat(join('new_file'))
2071 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
2072 os.umask(0o022)
2073 with (p / 'other_new_file').open('wb'):
2074 pass
2075 st = os.stat(join('other_new_file'))
2076 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
2077
2078 def test_touch_mode(self):
2079 old_mask = os.umask(0)
2080 self.addCleanup(os.umask, old_mask)
2081 p = self.cls(BASE)
2082 (p / 'new_file').touch()
2083 st = os.stat(join('new_file'))
2084 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
2085 os.umask(0o022)
2086 (p / 'other_new_file').touch()
2087 st = os.stat(join('other_new_file'))
2088 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
2089 (p / 'masked_new_file').touch(mode=0o750)
2090 st = os.stat(join('masked_new_file'))
2091 self.assertEqual(stat.S_IMODE(st.st_mode), 0o750)
2092
Vajrasky Kokec1f5df2017-03-29 02:32:35 +07002093 @support.skip_unless_symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +01002094 def test_resolve_loop(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +01002095 # Loops with relative symlinks
2096 os.symlink('linkX/inside', join('linkX'))
2097 self._check_symlink_loop(BASE, 'linkX')
2098 os.symlink('linkY', join('linkY'))
2099 self._check_symlink_loop(BASE, 'linkY')
2100 os.symlink('linkZ/../linkZ', join('linkZ'))
2101 self._check_symlink_loop(BASE, 'linkZ')
Steve Dower98eb3602016-11-09 12:58:17 -08002102 # Non-strict
2103 self._check_symlink_loop(BASE, 'linkZ', 'foo', strict=False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01002104 # Loops with absolute symlinks
2105 os.symlink(join('linkU/inside'), join('linkU'))
2106 self._check_symlink_loop(BASE, 'linkU')
2107 os.symlink(join('linkV'), join('linkV'))
2108 self._check_symlink_loop(BASE, 'linkV')
2109 os.symlink(join('linkW/../linkW'), join('linkW'))
2110 self._check_symlink_loop(BASE, 'linkW')
Steve Dower98eb3602016-11-09 12:58:17 -08002111 # Non-strict
2112 self._check_symlink_loop(BASE, 'linkW', 'foo', strict=False)
Antoine Pitrou31119e42013-11-22 17:38:12 +01002113
2114 def test_glob(self):
2115 P = self.cls
2116 p = P(BASE)
Brett Cannonfe77f4e2013-11-22 16:14:10 -05002117 given = set(p.glob("FILEa"))
2118 expect = set() if not support.fs_is_case_insensitive(BASE) else given
2119 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01002120 self.assertEqual(set(p.glob("FILEa*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01002121
2122 def test_rglob(self):
2123 P = self.cls
2124 p = P(BASE, "dirC")
Brett Cannonfe77f4e2013-11-22 16:14:10 -05002125 given = set(p.rglob("FILEd"))
2126 expect = set() if not support.fs_is_case_insensitive(BASE) else given
2127 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01002128 self.assertEqual(set(p.rglob("FILEd*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01002129
Xavier de Gayefb24eea2016-12-13 09:11:38 +01002130 @unittest.skipUnless(hasattr(pwd, 'getpwall'),
2131 'pwd module does not expose getpwall()')
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002132 def test_expanduser(self):
2133 P = self.cls
2134 support.import_module('pwd')
2135 import pwd
2136 pwdent = pwd.getpwuid(os.getuid())
2137 username = pwdent.pw_name
Serhiy Storchakaa3fd0b22016-05-03 21:17:03 +03002138 userhome = pwdent.pw_dir.rstrip('/') or '/'
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002139 # find arbitrary different user (if exists)
2140 for pwdent in pwd.getpwall():
2141 othername = pwdent.pw_name
2142 otherhome = pwdent.pw_dir.rstrip('/')
Victor Stinnere9694392015-01-10 09:00:20 +01002143 if othername != username and otherhome:
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002144 break
2145
2146 p1 = P('~/Documents')
2147 p2 = P('~' + username + '/Documents')
2148 p3 = P('~' + othername + '/Documents')
2149 p4 = P('../~' + username + '/Documents')
2150 p5 = P('/~' + username + '/Documents')
2151 p6 = P('')
2152 p7 = P('~fakeuser/Documents')
2153
2154 with support.EnvironmentVarGuard() as env:
2155 env.pop('HOME', None)
2156
2157 self.assertEqual(p1.expanduser(), P(userhome) / 'Documents')
2158 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2159 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2160 self.assertEqual(p4.expanduser(), p4)
2161 self.assertEqual(p5.expanduser(), p5)
2162 self.assertEqual(p6.expanduser(), p6)
2163 self.assertRaises(RuntimeError, p7.expanduser)
2164
2165 env['HOME'] = '/tmp'
2166 self.assertEqual(p1.expanduser(), P('/tmp/Documents'))
2167 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2168 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2169 self.assertEqual(p4.expanduser(), p4)
2170 self.assertEqual(p5.expanduser(), p5)
2171 self.assertEqual(p6.expanduser(), p6)
2172 self.assertRaises(RuntimeError, p7.expanduser)
2173
Antoine Pitrou31119e42013-11-22 17:38:12 +01002174
2175@only_nt
2176class WindowsPathTest(_BasePathTest, unittest.TestCase):
2177 cls = pathlib.WindowsPath
2178
2179 def test_glob(self):
2180 P = self.cls
2181 p = P(BASE)
2182 self.assertEqual(set(p.glob("FILEa")), { P(BASE, "fileA") })
2183
2184 def test_rglob(self):
2185 P = self.cls
2186 p = P(BASE, "dirC")
2187 self.assertEqual(set(p.rglob("FILEd")), { P(BASE, "dirC/dirD/fileD") })
2188
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002189 def test_expanduser(self):
2190 P = self.cls
2191 with support.EnvironmentVarGuard() as env:
2192 env.pop('HOME', None)
2193 env.pop('USERPROFILE', None)
2194 env.pop('HOMEPATH', None)
2195 env.pop('HOMEDRIVE', None)
2196 env['USERNAME'] = 'alice'
2197
2198 # test that the path returns unchanged
2199 p1 = P('~/My Documents')
2200 p2 = P('~alice/My Documents')
2201 p3 = P('~bob/My Documents')
2202 p4 = P('/~/My Documents')
2203 p5 = P('d:~/My Documents')
2204 p6 = P('')
2205 self.assertRaises(RuntimeError, p1.expanduser)
2206 self.assertRaises(RuntimeError, p2.expanduser)
2207 self.assertRaises(RuntimeError, p3.expanduser)
2208 self.assertEqual(p4.expanduser(), p4)
2209 self.assertEqual(p5.expanduser(), p5)
2210 self.assertEqual(p6.expanduser(), p6)
2211
2212 def check():
2213 env.pop('USERNAME', None)
2214 self.assertEqual(p1.expanduser(),
2215 P('C:/Users/alice/My Documents'))
2216 self.assertRaises(KeyError, p2.expanduser)
2217 env['USERNAME'] = 'alice'
2218 self.assertEqual(p2.expanduser(),
2219 P('C:/Users/alice/My Documents'))
2220 self.assertEqual(p3.expanduser(),
2221 P('C:/Users/bob/My Documents'))
2222 self.assertEqual(p4.expanduser(), p4)
2223 self.assertEqual(p5.expanduser(), p5)
2224 self.assertEqual(p6.expanduser(), p6)
2225
2226 # test the first lookup key in the env vars
2227 env['HOME'] = 'C:\\Users\\alice'
2228 check()
2229
2230 # test that HOMEPATH is available instead
2231 env.pop('HOME', None)
2232 env['HOMEPATH'] = 'C:\\Users\\alice'
2233 check()
2234
2235 env['HOMEDRIVE'] = 'C:\\'
2236 env['HOMEPATH'] = 'Users\\alice'
2237 check()
2238
2239 env.pop('HOMEDRIVE', None)
2240 env.pop('HOMEPATH', None)
2241 env['USERPROFILE'] = 'C:\\Users\\alice'
2242 check()
2243
Antoine Pitrou31119e42013-11-22 17:38:12 +01002244
2245if __name__ == "__main__":
2246 unittest.main()