blob: 1c53ab78ea4e11b1a705567bb80ae0d28afc6174 [file] [log] [blame]
Antoine Pitrou31119e42013-11-22 17:38:12 +01001import collections
2import 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
Antoine Pitrou31119e42013-11-22 17:38:12 +010011
12from test import support
13TESTFN = support.TESTFN
14
15try:
16 import grp, pwd
17except ImportError:
18 grp = pwd = None
19
20
21class _BaseFlavourTest(object):
22
23 def _check_parse_parts(self, arg, expected):
24 f = self.flavour.parse_parts
25 sep = self.flavour.sep
26 altsep = self.flavour.altsep
27 actual = f([x.replace('/', sep) for x in arg])
28 self.assertEqual(actual, expected)
29 if altsep:
30 actual = f([x.replace('/', altsep) for x in arg])
31 self.assertEqual(actual, expected)
32
33 def test_parse_parts_common(self):
34 check = self._check_parse_parts
35 sep = self.flavour.sep
36 # Unanchored parts
37 check([], ('', '', []))
38 check(['a'], ('', '', ['a']))
39 check(['a/'], ('', '', ['a']))
40 check(['a', 'b'], ('', '', ['a', 'b']))
41 # Expansion
42 check(['a/b'], ('', '', ['a', 'b']))
43 check(['a/b/'], ('', '', ['a', 'b']))
44 check(['a', 'b/c', 'd'], ('', '', ['a', 'b', 'c', 'd']))
45 # Collapsing and stripping excess slashes
46 check(['a', 'b//c', 'd'], ('', '', ['a', 'b', 'c', 'd']))
47 check(['a', 'b/c/', 'd'], ('', '', ['a', 'b', 'c', 'd']))
48 # Eliminating standalone dots
49 check(['.'], ('', '', []))
50 check(['.', '.', 'b'], ('', '', ['b']))
51 check(['a', '.', 'b'], ('', '', ['a', 'b']))
52 check(['a', '.', '.'], ('', '', ['a']))
53 # The first part is anchored
54 check(['/a/b'], ('', sep, [sep, 'a', 'b']))
55 check(['/a', 'b'], ('', sep, [sep, 'a', 'b']))
56 check(['/a/', 'b'], ('', sep, [sep, 'a', 'b']))
57 # Ignoring parts before an anchored part
58 check(['a', '/b', 'c'], ('', sep, [sep, 'b', 'c']))
59 check(['a', '/b', '/c'], ('', sep, [sep, 'c']))
60
61
62class PosixFlavourTest(_BaseFlavourTest, unittest.TestCase):
63 flavour = pathlib._posix_flavour
64
65 def test_parse_parts(self):
66 check = self._check_parse_parts
67 # Collapsing of excess leading slashes, except for the double-slash
68 # special case.
69 check(['//a', 'b'], ('', '//', ['//', 'a', 'b']))
70 check(['///a', 'b'], ('', '/', ['/', 'a', 'b']))
71 check(['////a', 'b'], ('', '/', ['/', 'a', 'b']))
72 # Paths which look like NT paths aren't treated specially
73 check(['c:a'], ('', '', ['c:a']))
74 check(['c:\\a'], ('', '', ['c:\\a']))
75 check(['\\a'], ('', '', ['\\a']))
76
77 def test_splitroot(self):
78 f = self.flavour.splitroot
79 self.assertEqual(f(''), ('', '', ''))
80 self.assertEqual(f('a'), ('', '', 'a'))
81 self.assertEqual(f('a/b'), ('', '', 'a/b'))
82 self.assertEqual(f('a/b/'), ('', '', 'a/b/'))
83 self.assertEqual(f('/a'), ('', '/', 'a'))
84 self.assertEqual(f('/a/b'), ('', '/', 'a/b'))
85 self.assertEqual(f('/a/b/'), ('', '/', 'a/b/'))
86 # The root is collapsed when there are redundant slashes
87 # except when there are exactly two leading slashes, which
88 # is a special case in POSIX.
89 self.assertEqual(f('//a'), ('', '//', 'a'))
90 self.assertEqual(f('///a'), ('', '/', 'a'))
91 self.assertEqual(f('///a/b'), ('', '/', 'a/b'))
92 # Paths which look like NT paths aren't treated specially
93 self.assertEqual(f('c:/a/b'), ('', '', 'c:/a/b'))
94 self.assertEqual(f('\\/a/b'), ('', '', '\\/a/b'))
95 self.assertEqual(f('\\a\\b'), ('', '', '\\a\\b'))
96
97
98class NTFlavourTest(_BaseFlavourTest, unittest.TestCase):
99 flavour = pathlib._windows_flavour
100
101 def test_parse_parts(self):
102 check = self._check_parse_parts
103 # First part is anchored
104 check(['c:'], ('c:', '', ['c:']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100105 check(['c:/'], ('c:', '\\', ['c:\\']))
106 check(['/'], ('', '\\', ['\\']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100107 check(['c:a'], ('c:', '', ['c:', 'a']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100108 check(['c:/a'], ('c:', '\\', ['c:\\', 'a']))
109 check(['/a'], ('', '\\', ['\\', 'a']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100110 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100111 check(['//a/b'], ('\\\\a\\b', '\\', ['\\\\a\\b\\']))
112 check(['//a/b/'], ('\\\\a\\b', '\\', ['\\\\a\\b\\']))
113 check(['//a/b/c'], ('\\\\a\\b', '\\', ['\\\\a\\b\\', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100114 # Second part is anchored, so that the first part is ignored
115 check(['a', 'Z:b', 'c'], ('Z:', '', ['Z:', 'b', 'c']))
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100116 check(['a', 'Z:/b', 'c'], ('Z:', '\\', ['Z:\\', 'b', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100117 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100118 check(['a', '//b/c', 'd'], ('\\\\b\\c', '\\', ['\\\\b\\c\\', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100119 # Collapsing and stripping excess slashes
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100120 check(['a', 'Z://b//c/', 'd/'], ('Z:', '\\', ['Z:\\', 'b', 'c', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100121 # UNC paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100122 check(['a', '//b/c//', 'd'], ('\\\\b\\c', '\\', ['\\\\b\\c\\', 'd']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100123 # Extended paths
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100124 check(['//?/c:/'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\']))
125 check(['//?/c:/a'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\', 'a']))
126 check(['//?/c:/a', '/b'], ('\\\\?\\c:', '\\', ['\\\\?\\c:\\', 'b']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100127 # Extended UNC paths (format is "\\?\UNC\server\share")
Antoine Pitrou57fffd62015-02-15 18:03:59 +0100128 check(['//?/UNC/b/c'], ('\\\\?\\UNC\\b\\c', '\\', ['\\\\?\\UNC\\b\\c\\']))
129 check(['//?/UNC/b/c/d'], ('\\\\?\\UNC\\b\\c', '\\', ['\\\\?\\UNC\\b\\c\\', 'd']))
130 # Second part has a root but not drive
131 check(['a', '/b', 'c'], ('', '\\', ['\\', 'b', 'c']))
132 check(['Z:/a', '/b', 'c'], ('Z:', '\\', ['Z:\\', 'b', 'c']))
133 check(['//?/Z:/a', '/b', 'c'], ('\\\\?\\Z:', '\\', ['\\\\?\\Z:\\', 'b', 'c']))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100134
135 def test_splitroot(self):
136 f = self.flavour.splitroot
137 self.assertEqual(f(''), ('', '', ''))
138 self.assertEqual(f('a'), ('', '', 'a'))
139 self.assertEqual(f('a\\b'), ('', '', 'a\\b'))
140 self.assertEqual(f('\\a'), ('', '\\', 'a'))
141 self.assertEqual(f('\\a\\b'), ('', '\\', 'a\\b'))
142 self.assertEqual(f('c:a\\b'), ('c:', '', 'a\\b'))
143 self.assertEqual(f('c:\\a\\b'), ('c:', '\\', 'a\\b'))
144 # Redundant slashes in the root are collapsed
145 self.assertEqual(f('\\\\a'), ('', '\\', 'a'))
146 self.assertEqual(f('\\\\\\a/b'), ('', '\\', 'a/b'))
147 self.assertEqual(f('c:\\\\a'), ('c:', '\\', 'a'))
148 self.assertEqual(f('c:\\\\\\a/b'), ('c:', '\\', 'a/b'))
149 # Valid UNC paths
150 self.assertEqual(f('\\\\a\\b'), ('\\\\a\\b', '\\', ''))
151 self.assertEqual(f('\\\\a\\b\\'), ('\\\\a\\b', '\\', ''))
152 self.assertEqual(f('\\\\a\\b\\c\\d'), ('\\\\a\\b', '\\', 'c\\d'))
153 # These are non-UNC paths (according to ntpath.py and test_ntpath)
154 # However, command.com says such paths are invalid, so it's
155 # difficult to know what the right semantics are
156 self.assertEqual(f('\\\\\\a\\b'), ('', '\\', 'a\\b'))
157 self.assertEqual(f('\\\\a'), ('', '\\', 'a'))
158
159
160#
161# Tests for the pure classes
162#
163
164class _BasePurePathTest(object):
165
166 # keys are canonical paths, values are list of tuples of arguments
167 # supposed to produce equal paths
168 equivalences = {
169 'a/b': [
170 ('a', 'b'), ('a/', 'b'), ('a', 'b/'), ('a/', 'b/'),
171 ('a/b/',), ('a//b',), ('a//b//',),
172 # empty components get removed
173 ('', 'a', 'b'), ('a', '', 'b'), ('a', 'b', ''),
174 ],
175 '/b/c/d': [
176 ('a', '/b/c', 'd'), ('a', '///b//c', 'd/'),
177 ('/a', '/b/c', 'd'),
178 # empty components get removed
179 ('/', 'b', '', 'c/d'), ('/', '', 'b/c/d'), ('', '/b/c/d'),
180 ],
181 }
182
183 def setUp(self):
184 p = self.cls('a')
185 self.flavour = p._flavour
186 self.sep = self.flavour.sep
187 self.altsep = self.flavour.altsep
188
189 def test_constructor_common(self):
190 P = self.cls
191 p = P('a')
192 self.assertIsInstance(p, P)
193 P('a', 'b', 'c')
194 P('/a', 'b', 'c')
195 P('a/b/c')
196 P('/a/b/c')
197 self.assertEqual(P(P('a')), P('a'))
198 self.assertEqual(P(P('a'), 'b'), P('a/b'))
199 self.assertEqual(P(P('a'), P('b')), P('a/b'))
200
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200201 def _check_str_subclass(self, *args):
202 # Issue #21127: it should be possible to construct a PurePath object
203 # from an str subclass instance, and it then gets converted to
204 # a pure str object.
205 class StrSubclass(str):
206 pass
207 P = self.cls
208 p = P(*(StrSubclass(x) for x in args))
209 self.assertEqual(p, P(*args))
210 for part in p.parts:
211 self.assertIs(type(part), str)
212
213 def test_str_subclass_common(self):
214 self._check_str_subclass('')
215 self._check_str_subclass('.')
216 self._check_str_subclass('a')
217 self._check_str_subclass('a/b.txt')
218 self._check_str_subclass('/a/b.txt')
219
Antoine Pitrou31119e42013-11-22 17:38:12 +0100220 def test_join_common(self):
221 P = self.cls
222 p = P('a/b')
223 pp = p.joinpath('c')
224 self.assertEqual(pp, P('a/b/c'))
225 self.assertIs(type(pp), type(p))
226 pp = p.joinpath('c', 'd')
227 self.assertEqual(pp, P('a/b/c/d'))
228 pp = p.joinpath(P('c'))
229 self.assertEqual(pp, P('a/b/c'))
230 pp = p.joinpath('/c')
231 self.assertEqual(pp, P('/c'))
232
233 def test_div_common(self):
234 # Basically the same as joinpath()
235 P = self.cls
236 p = P('a/b')
237 pp = p / 'c'
238 self.assertEqual(pp, P('a/b/c'))
239 self.assertIs(type(pp), type(p))
240 pp = p / 'c/d'
241 self.assertEqual(pp, P('a/b/c/d'))
242 pp = p / 'c' / 'd'
243 self.assertEqual(pp, P('a/b/c/d'))
244 pp = 'c' / p / 'd'
245 self.assertEqual(pp, P('c/a/b/d'))
246 pp = p / P('c')
247 self.assertEqual(pp, P('a/b/c'))
248 pp = p/ '/c'
249 self.assertEqual(pp, P('/c'))
250
251 def _check_str(self, expected, args):
252 p = self.cls(*args)
253 self.assertEqual(str(p), expected.replace('/', self.sep))
254
255 def test_str_common(self):
256 # Canonicalized paths roundtrip
257 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
258 self._check_str(pathstr, (pathstr,))
259 # Special case for the empty path
260 self._check_str('.', ('',))
261 # Other tests for str() are in test_equivalences()
262
263 def test_as_posix_common(self):
264 P = self.cls
265 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
266 self.assertEqual(P(pathstr).as_posix(), pathstr)
267 # Other tests for as_posix() are in test_equivalences()
268
269 def test_as_bytes_common(self):
270 sep = os.fsencode(self.sep)
271 P = self.cls
272 self.assertEqual(bytes(P('a/b')), b'a' + sep + b'b')
273
274 def test_as_uri_common(self):
275 P = self.cls
276 with self.assertRaises(ValueError):
277 P('a').as_uri()
278 with self.assertRaises(ValueError):
279 P().as_uri()
280
281 def test_repr_common(self):
282 for pathstr in ('a', 'a/b', 'a/b/c', '/', '/a/b', '/a/b/c'):
283 p = self.cls(pathstr)
284 clsname = p.__class__.__name__
285 r = repr(p)
286 # The repr() is in the form ClassName("forward-slashes path")
287 self.assertTrue(r.startswith(clsname + '('), r)
288 self.assertTrue(r.endswith(')'), r)
289 inner = r[len(clsname) + 1 : -1]
290 self.assertEqual(eval(inner), p.as_posix())
291 # The repr() roundtrips
292 q = eval(r, pathlib.__dict__)
293 self.assertIs(q.__class__, p.__class__)
294 self.assertEqual(q, p)
295 self.assertEqual(repr(q), r)
296
297 def test_eq_common(self):
298 P = self.cls
299 self.assertEqual(P('a/b'), P('a/b'))
300 self.assertEqual(P('a/b'), P('a', 'b'))
301 self.assertNotEqual(P('a/b'), P('a'))
302 self.assertNotEqual(P('a/b'), P('/a/b'))
303 self.assertNotEqual(P('a/b'), P())
304 self.assertNotEqual(P('/a/b'), P('/'))
305 self.assertNotEqual(P(), P('/'))
306 self.assertNotEqual(P(), "")
307 self.assertNotEqual(P(), {})
308 self.assertNotEqual(P(), int)
309
310 def test_match_common(self):
311 P = self.cls
312 self.assertRaises(ValueError, P('a').match, '')
313 self.assertRaises(ValueError, P('a').match, '.')
314 # Simple relative pattern
315 self.assertTrue(P('b.py').match('b.py'))
316 self.assertTrue(P('a/b.py').match('b.py'))
317 self.assertTrue(P('/a/b.py').match('b.py'))
318 self.assertFalse(P('a.py').match('b.py'))
319 self.assertFalse(P('b/py').match('b.py'))
320 self.assertFalse(P('/a.py').match('b.py'))
321 self.assertFalse(P('b.py/c').match('b.py'))
322 # Wilcard relative pattern
323 self.assertTrue(P('b.py').match('*.py'))
324 self.assertTrue(P('a/b.py').match('*.py'))
325 self.assertTrue(P('/a/b.py').match('*.py'))
326 self.assertFalse(P('b.pyc').match('*.py'))
327 self.assertFalse(P('b./py').match('*.py'))
328 self.assertFalse(P('b.py/c').match('*.py'))
329 # Multi-part relative pattern
330 self.assertTrue(P('ab/c.py').match('a*/*.py'))
331 self.assertTrue(P('/d/ab/c.py').match('a*/*.py'))
332 self.assertFalse(P('a.py').match('a*/*.py'))
333 self.assertFalse(P('/dab/c.py').match('a*/*.py'))
334 self.assertFalse(P('ab/c.py/d').match('a*/*.py'))
335 # Absolute pattern
336 self.assertTrue(P('/b.py').match('/*.py'))
337 self.assertFalse(P('b.py').match('/*.py'))
338 self.assertFalse(P('a/b.py').match('/*.py'))
339 self.assertFalse(P('/a/b.py').match('/*.py'))
340 # Multi-part absolute pattern
341 self.assertTrue(P('/a/b.py').match('/a/*.py'))
342 self.assertFalse(P('/ab.py').match('/a/*.py'))
343 self.assertFalse(P('/a/b/c.py').match('/a/*.py'))
344
345 def test_ordering_common(self):
346 # Ordering is tuple-alike
347 def assertLess(a, b):
348 self.assertLess(a, b)
349 self.assertGreater(b, a)
350 P = self.cls
351 a = P('a')
352 b = P('a/b')
353 c = P('abc')
354 d = P('b')
355 assertLess(a, b)
356 assertLess(a, c)
357 assertLess(a, d)
358 assertLess(b, c)
359 assertLess(c, d)
360 P = self.cls
361 a = P('/a')
362 b = P('/a/b')
363 c = P('/abc')
364 d = P('/b')
365 assertLess(a, b)
366 assertLess(a, c)
367 assertLess(a, d)
368 assertLess(b, c)
369 assertLess(c, d)
370 with self.assertRaises(TypeError):
371 P() < {}
372
373 def test_parts_common(self):
374 # `parts` returns a tuple
375 sep = self.sep
376 P = self.cls
377 p = P('a/b')
378 parts = p.parts
379 self.assertEqual(parts, ('a', 'b'))
380 # The object gets reused
381 self.assertIs(parts, p.parts)
382 # When the path is absolute, the anchor is a separate part
383 p = P('/a/b')
384 parts = p.parts
385 self.assertEqual(parts, (sep, 'a', 'b'))
386
387 def test_equivalences(self):
388 for k, tuples in self.equivalences.items():
389 canon = k.replace('/', self.sep)
390 posix = k.replace(self.sep, '/')
391 if canon != posix:
392 tuples = tuples + [
393 tuple(part.replace('/', self.sep) for part in t)
394 for t in tuples
395 ]
396 tuples.append((posix, ))
397 pcanon = self.cls(canon)
398 for t in tuples:
399 p = self.cls(*t)
400 self.assertEqual(p, pcanon, "failed with args {}".format(t))
401 self.assertEqual(hash(p), hash(pcanon))
402 self.assertEqual(str(p), canon)
403 self.assertEqual(p.as_posix(), posix)
404
405 def test_parent_common(self):
406 # Relative
407 P = self.cls
408 p = P('a/b/c')
409 self.assertEqual(p.parent, P('a/b'))
410 self.assertEqual(p.parent.parent, P('a'))
411 self.assertEqual(p.parent.parent.parent, P())
412 self.assertEqual(p.parent.parent.parent.parent, P())
413 # Anchored
414 p = P('/a/b/c')
415 self.assertEqual(p.parent, P('/a/b'))
416 self.assertEqual(p.parent.parent, P('/a'))
417 self.assertEqual(p.parent.parent.parent, P('/'))
418 self.assertEqual(p.parent.parent.parent.parent, P('/'))
419
420 def test_parents_common(self):
421 # Relative
422 P = self.cls
423 p = P('a/b/c')
424 par = p.parents
425 self.assertEqual(len(par), 3)
426 self.assertEqual(par[0], P('a/b'))
427 self.assertEqual(par[1], P('a'))
428 self.assertEqual(par[2], P('.'))
429 self.assertEqual(list(par), [P('a/b'), P('a'), P('.')])
430 with self.assertRaises(IndexError):
431 par[-1]
432 with self.assertRaises(IndexError):
433 par[3]
434 with self.assertRaises(TypeError):
435 par[0] = p
436 # Anchored
437 p = P('/a/b/c')
438 par = p.parents
439 self.assertEqual(len(par), 3)
440 self.assertEqual(par[0], P('/a/b'))
441 self.assertEqual(par[1], P('/a'))
442 self.assertEqual(par[2], P('/'))
443 self.assertEqual(list(par), [P('/a/b'), P('/a'), P('/')])
444 with self.assertRaises(IndexError):
445 par[3]
446
447 def test_drive_common(self):
448 P = self.cls
449 self.assertEqual(P('a/b').drive, '')
450 self.assertEqual(P('/a/b').drive, '')
451 self.assertEqual(P('').drive, '')
452
453 def test_root_common(self):
454 P = self.cls
455 sep = self.sep
456 self.assertEqual(P('').root, '')
457 self.assertEqual(P('a/b').root, '')
458 self.assertEqual(P('/').root, sep)
459 self.assertEqual(P('/a/b').root, sep)
460
461 def test_anchor_common(self):
462 P = self.cls
463 sep = self.sep
464 self.assertEqual(P('').anchor, '')
465 self.assertEqual(P('a/b').anchor, '')
466 self.assertEqual(P('/').anchor, sep)
467 self.assertEqual(P('/a/b').anchor, sep)
468
469 def test_name_common(self):
470 P = self.cls
471 self.assertEqual(P('').name, '')
472 self.assertEqual(P('.').name, '')
473 self.assertEqual(P('/').name, '')
474 self.assertEqual(P('a/b').name, 'b')
475 self.assertEqual(P('/a/b').name, 'b')
476 self.assertEqual(P('/a/b/.').name, 'b')
477 self.assertEqual(P('a/b.py').name, 'b.py')
478 self.assertEqual(P('/a/b.py').name, 'b.py')
479
480 def test_suffix_common(self):
481 P = self.cls
482 self.assertEqual(P('').suffix, '')
483 self.assertEqual(P('.').suffix, '')
484 self.assertEqual(P('..').suffix, '')
485 self.assertEqual(P('/').suffix, '')
486 self.assertEqual(P('a/b').suffix, '')
487 self.assertEqual(P('/a/b').suffix, '')
488 self.assertEqual(P('/a/b/.').suffix, '')
489 self.assertEqual(P('a/b.py').suffix, '.py')
490 self.assertEqual(P('/a/b.py').suffix, '.py')
491 self.assertEqual(P('a/.hgrc').suffix, '')
492 self.assertEqual(P('/a/.hgrc').suffix, '')
493 self.assertEqual(P('a/.hg.rc').suffix, '.rc')
494 self.assertEqual(P('/a/.hg.rc').suffix, '.rc')
495 self.assertEqual(P('a/b.tar.gz').suffix, '.gz')
496 self.assertEqual(P('/a/b.tar.gz').suffix, '.gz')
497 self.assertEqual(P('a/Some name. Ending with a dot.').suffix, '')
498 self.assertEqual(P('/a/Some name. Ending with a dot.').suffix, '')
499
500 def test_suffixes_common(self):
501 P = self.cls
502 self.assertEqual(P('').suffixes, [])
503 self.assertEqual(P('.').suffixes, [])
504 self.assertEqual(P('/').suffixes, [])
505 self.assertEqual(P('a/b').suffixes, [])
506 self.assertEqual(P('/a/b').suffixes, [])
507 self.assertEqual(P('/a/b/.').suffixes, [])
508 self.assertEqual(P('a/b.py').suffixes, ['.py'])
509 self.assertEqual(P('/a/b.py').suffixes, ['.py'])
510 self.assertEqual(P('a/.hgrc').suffixes, [])
511 self.assertEqual(P('/a/.hgrc').suffixes, [])
512 self.assertEqual(P('a/.hg.rc').suffixes, ['.rc'])
513 self.assertEqual(P('/a/.hg.rc').suffixes, ['.rc'])
514 self.assertEqual(P('a/b.tar.gz').suffixes, ['.tar', '.gz'])
515 self.assertEqual(P('/a/b.tar.gz').suffixes, ['.tar', '.gz'])
516 self.assertEqual(P('a/Some name. Ending with a dot.').suffixes, [])
517 self.assertEqual(P('/a/Some name. Ending with a dot.').suffixes, [])
518
519 def test_stem_common(self):
520 P = self.cls
521 self.assertEqual(P('').stem, '')
522 self.assertEqual(P('.').stem, '')
523 self.assertEqual(P('..').stem, '..')
524 self.assertEqual(P('/').stem, '')
525 self.assertEqual(P('a/b').stem, 'b')
526 self.assertEqual(P('a/b.py').stem, 'b')
527 self.assertEqual(P('a/.hgrc').stem, '.hgrc')
528 self.assertEqual(P('a/.hg.rc').stem, '.hg')
529 self.assertEqual(P('a/b.tar.gz').stem, 'b.tar')
530 self.assertEqual(P('a/Some name. Ending with a dot.').stem,
531 'Some name. Ending with a dot.')
532
533 def test_with_name_common(self):
534 P = self.cls
535 self.assertEqual(P('a/b').with_name('d.xml'), P('a/d.xml'))
536 self.assertEqual(P('/a/b').with_name('d.xml'), P('/a/d.xml'))
537 self.assertEqual(P('a/b.py').with_name('d.xml'), P('a/d.xml'))
538 self.assertEqual(P('/a/b.py').with_name('d.xml'), P('/a/d.xml'))
539 self.assertEqual(P('a/Dot ending.').with_name('d.xml'), P('a/d.xml'))
540 self.assertEqual(P('/a/Dot ending.').with_name('d.xml'), P('/a/d.xml'))
541 self.assertRaises(ValueError, P('').with_name, 'd.xml')
542 self.assertRaises(ValueError, P('.').with_name, 'd.xml')
543 self.assertRaises(ValueError, P('/').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400544 self.assertRaises(ValueError, P('a/b').with_name, '')
545 self.assertRaises(ValueError, P('a/b').with_name, '/c')
546 self.assertRaises(ValueError, P('a/b').with_name, 'c/')
547 self.assertRaises(ValueError, P('a/b').with_name, 'c/d')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100548
549 def test_with_suffix_common(self):
550 P = self.cls
551 self.assertEqual(P('a/b').with_suffix('.gz'), P('a/b.gz'))
552 self.assertEqual(P('/a/b').with_suffix('.gz'), P('/a/b.gz'))
553 self.assertEqual(P('a/b.py').with_suffix('.gz'), P('a/b.gz'))
554 self.assertEqual(P('/a/b.py').with_suffix('.gz'), P('/a/b.gz'))
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400555 # Stripping suffix
556 self.assertEqual(P('a/b.py').with_suffix(''), P('a/b'))
557 self.assertEqual(P('/a/b').with_suffix(''), P('/a/b'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100558 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100559 self.assertRaises(ValueError, P('').with_suffix, '.gz')
560 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
561 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100562 # Invalid suffix
563 self.assertRaises(ValueError, P('a/b').with_suffix, 'gz')
564 self.assertRaises(ValueError, P('a/b').with_suffix, '/')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400565 self.assertRaises(ValueError, P('a/b').with_suffix, '.')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100566 self.assertRaises(ValueError, P('a/b').with_suffix, '/.gz')
567 self.assertRaises(ValueError, P('a/b').with_suffix, 'c/d')
568 self.assertRaises(ValueError, P('a/b').with_suffix, '.c/.d')
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400569 self.assertRaises(ValueError, P('a/b').with_suffix, './.d')
570 self.assertRaises(ValueError, P('a/b').with_suffix, '.d/.')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100571
572 def test_relative_to_common(self):
573 P = self.cls
574 p = P('a/b')
575 self.assertRaises(TypeError, p.relative_to)
Antoine Pitrou156b3612013-12-28 19:49:04 +0100576 self.assertRaises(TypeError, p.relative_to, b'a')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100577 self.assertEqual(p.relative_to(P()), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100578 self.assertEqual(p.relative_to(''), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100579 self.assertEqual(p.relative_to(P('a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100580 self.assertEqual(p.relative_to('a'), P('b'))
581 self.assertEqual(p.relative_to('a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100582 self.assertEqual(p.relative_to(P('a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100583 self.assertEqual(p.relative_to('a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100584 # With several args
585 self.assertEqual(p.relative_to('a', 'b'), P())
586 # Unrelated paths
587 self.assertRaises(ValueError, p.relative_to, P('c'))
588 self.assertRaises(ValueError, p.relative_to, P('a/b/c'))
589 self.assertRaises(ValueError, p.relative_to, P('a/c'))
590 self.assertRaises(ValueError, p.relative_to, P('/a'))
591 p = P('/a/b')
592 self.assertEqual(p.relative_to(P('/')), P('a/b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100593 self.assertEqual(p.relative_to('/'), P('a/b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100594 self.assertEqual(p.relative_to(P('/a')), P('b'))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100595 self.assertEqual(p.relative_to('/a'), P('b'))
596 self.assertEqual(p.relative_to('/a/'), P('b'))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100597 self.assertEqual(p.relative_to(P('/a/b')), P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100598 self.assertEqual(p.relative_to('/a/b'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +0100599 # Unrelated paths
600 self.assertRaises(ValueError, p.relative_to, P('/c'))
601 self.assertRaises(ValueError, p.relative_to, P('/a/b/c'))
602 self.assertRaises(ValueError, p.relative_to, P('/a/c'))
603 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +0100604 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100605 self.assertRaises(ValueError, p.relative_to, P('a'))
606
607 def test_pickling_common(self):
608 P = self.cls
609 p = P('/a/b')
610 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
611 dumped = pickle.dumps(p, proto)
612 pp = pickle.loads(dumped)
613 self.assertIs(pp.__class__, p.__class__)
614 self.assertEqual(pp, p)
615 self.assertEqual(hash(pp), hash(p))
616 self.assertEqual(str(pp), str(p))
617
618
619class PurePosixPathTest(_BasePurePathTest, unittest.TestCase):
620 cls = pathlib.PurePosixPath
621
622 def test_root(self):
623 P = self.cls
624 self.assertEqual(P('/a/b').root, '/')
625 self.assertEqual(P('///a/b').root, '/')
626 # POSIX special case for two leading slashes
627 self.assertEqual(P('//a/b').root, '//')
628
629 def test_eq(self):
630 P = self.cls
631 self.assertNotEqual(P('a/b'), P('A/b'))
632 self.assertEqual(P('/a'), P('///a'))
633 self.assertNotEqual(P('/a'), P('//a'))
634
635 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100636 P = self.cls
637 self.assertEqual(P('/').as_uri(), 'file:///')
638 self.assertEqual(P('/a/b.c').as_uri(), 'file:///a/b.c')
639 self.assertEqual(P('/a/b%#c').as_uri(), 'file:///a/b%25%23c')
Antoine Pitrou29eac422013-11-22 17:57:03 +0100640
641 def test_as_uri_non_ascii(self):
642 from urllib.parse import quote_from_bytes
643 P = self.cls
644 try:
645 os.fsencode('\xe9')
646 except UnicodeEncodeError:
647 self.skipTest("\\xe9 cannot be encoded to the filesystem encoding")
Antoine Pitrou31119e42013-11-22 17:38:12 +0100648 self.assertEqual(P('/a/b\xe9').as_uri(),
649 'file:///a/b' + quote_from_bytes(os.fsencode('\xe9')))
650
651 def test_match(self):
652 P = self.cls
653 self.assertFalse(P('A.py').match('a.PY'))
654
655 def test_is_absolute(self):
656 P = self.cls
657 self.assertFalse(P().is_absolute())
658 self.assertFalse(P('a').is_absolute())
659 self.assertFalse(P('a/b/').is_absolute())
660 self.assertTrue(P('/').is_absolute())
661 self.assertTrue(P('/a').is_absolute())
662 self.assertTrue(P('/a/b/').is_absolute())
663 self.assertTrue(P('//a').is_absolute())
664 self.assertTrue(P('//a/b').is_absolute())
665
666 def test_is_reserved(self):
667 P = self.cls
668 self.assertIs(False, P('').is_reserved())
669 self.assertIs(False, P('/').is_reserved())
670 self.assertIs(False, P('/foo/bar').is_reserved())
671 self.assertIs(False, P('/dev/con/PRN/NUL').is_reserved())
672
673 def test_join(self):
674 P = self.cls
675 p = P('//a')
676 pp = p.joinpath('b')
677 self.assertEqual(pp, P('//a/b'))
678 pp = P('/a').joinpath('//c')
679 self.assertEqual(pp, P('//c'))
680 pp = P('//a').joinpath('/c')
681 self.assertEqual(pp, P('/c'))
682
683 def test_div(self):
684 # Basically the same as joinpath()
685 P = self.cls
686 p = P('//a')
687 pp = p / 'b'
688 self.assertEqual(pp, P('//a/b'))
689 pp = P('/a') / '//c'
690 self.assertEqual(pp, P('//c'))
691 pp = P('//a') / '/c'
692 self.assertEqual(pp, P('/c'))
693
694
695class PureWindowsPathTest(_BasePurePathTest, unittest.TestCase):
696 cls = pathlib.PureWindowsPath
697
698 equivalences = _BasePurePathTest.equivalences.copy()
699 equivalences.update({
700 'c:a': [ ('c:', 'a'), ('c:', 'a/'), ('/', 'c:', 'a') ],
701 'c:/a': [
702 ('c:/', 'a'), ('c:', '/', 'a'), ('c:', '/a'),
703 ('/z', 'c:/', 'a'), ('//x/y', 'c:/', 'a'),
704 ],
705 '//a/b/': [ ('//a/b',) ],
706 '//a/b/c': [
707 ('//a/b', 'c'), ('//a/b/', 'c'),
708 ],
709 })
710
711 def test_str(self):
712 p = self.cls('a/b/c')
713 self.assertEqual(str(p), 'a\\b\\c')
714 p = self.cls('c:/a/b/c')
715 self.assertEqual(str(p), 'c:\\a\\b\\c')
716 p = self.cls('//a/b')
717 self.assertEqual(str(p), '\\\\a\\b\\')
718 p = self.cls('//a/b/c')
719 self.assertEqual(str(p), '\\\\a\\b\\c')
720 p = self.cls('//a/b/c/d')
721 self.assertEqual(str(p), '\\\\a\\b\\c\\d')
722
Antoine Pitroucb5ec772014-04-23 00:34:15 +0200723 def test_str_subclass(self):
724 self._check_str_subclass('c:')
725 self._check_str_subclass('c:a')
726 self._check_str_subclass('c:a\\b.txt')
727 self._check_str_subclass('c:\\')
728 self._check_str_subclass('c:\\a')
729 self._check_str_subclass('c:\\a\\b.txt')
730 self._check_str_subclass('\\\\some\\share')
731 self._check_str_subclass('\\\\some\\share\\a')
732 self._check_str_subclass('\\\\some\\share\\a\\b.txt')
733
Antoine Pitrou31119e42013-11-22 17:38:12 +0100734 def test_eq(self):
735 P = self.cls
736 self.assertEqual(P('c:a/b'), P('c:a/b'))
737 self.assertEqual(P('c:a/b'), P('c:', 'a', 'b'))
738 self.assertNotEqual(P('c:a/b'), P('d:a/b'))
739 self.assertNotEqual(P('c:a/b'), P('c:/a/b'))
740 self.assertNotEqual(P('/a/b'), P('c:/a/b'))
741 # Case-insensitivity
742 self.assertEqual(P('a/B'), P('A/b'))
743 self.assertEqual(P('C:a/B'), P('c:A/b'))
744 self.assertEqual(P('//Some/SHARE/a/B'), P('//somE/share/A/b'))
745
746 def test_as_uri(self):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100747 P = self.cls
748 with self.assertRaises(ValueError):
749 P('/a/b').as_uri()
750 with self.assertRaises(ValueError):
751 P('c:a/b').as_uri()
752 self.assertEqual(P('c:/').as_uri(), 'file:///c:/')
753 self.assertEqual(P('c:/a/b.c').as_uri(), 'file:///c:/a/b.c')
754 self.assertEqual(P('c:/a/b%#c').as_uri(), 'file:///c:/a/b%25%23c')
755 self.assertEqual(P('c:/a/b\xe9').as_uri(), 'file:///c:/a/b%C3%A9')
756 self.assertEqual(P('//some/share/').as_uri(), 'file://some/share/')
757 self.assertEqual(P('//some/share/a/b.c').as_uri(),
758 'file://some/share/a/b.c')
759 self.assertEqual(P('//some/share/a/b%#c\xe9').as_uri(),
760 'file://some/share/a/b%25%23c%C3%A9')
761
762 def test_match_common(self):
763 P = self.cls
764 # Absolute patterns
765 self.assertTrue(P('c:/b.py').match('/*.py'))
766 self.assertTrue(P('c:/b.py').match('c:*.py'))
767 self.assertTrue(P('c:/b.py').match('c:/*.py'))
768 self.assertFalse(P('d:/b.py').match('c:/*.py')) # wrong drive
769 self.assertFalse(P('b.py').match('/*.py'))
770 self.assertFalse(P('b.py').match('c:*.py'))
771 self.assertFalse(P('b.py').match('c:/*.py'))
772 self.assertFalse(P('c:b.py').match('/*.py'))
773 self.assertFalse(P('c:b.py').match('c:/*.py'))
774 self.assertFalse(P('/b.py').match('c:*.py'))
775 self.assertFalse(P('/b.py').match('c:/*.py'))
776 # UNC patterns
777 self.assertTrue(P('//some/share/a.py').match('/*.py'))
778 self.assertTrue(P('//some/share/a.py').match('//some/share/*.py'))
779 self.assertFalse(P('//other/share/a.py').match('//some/share/*.py'))
780 self.assertFalse(P('//some/share/a/b.py').match('//some/share/*.py'))
781 # Case-insensitivity
782 self.assertTrue(P('B.py').match('b.PY'))
783 self.assertTrue(P('c:/a/B.Py').match('C:/A/*.pY'))
784 self.assertTrue(P('//Some/Share/B.Py').match('//somE/sharE/*.pY'))
785
786 def test_ordering_common(self):
787 # Case-insensitivity
788 def assertOrderedEqual(a, b):
789 self.assertLessEqual(a, b)
790 self.assertGreaterEqual(b, a)
791 P = self.cls
792 p = P('c:A/b')
793 q = P('C:a/B')
794 assertOrderedEqual(p, q)
795 self.assertFalse(p < q)
796 self.assertFalse(p > q)
797 p = P('//some/Share/A/b')
798 q = P('//Some/SHARE/a/B')
799 assertOrderedEqual(p, q)
800 self.assertFalse(p < q)
801 self.assertFalse(p > q)
802
803 def test_parts(self):
804 P = self.cls
805 p = P('c:a/b')
806 parts = p.parts
807 self.assertEqual(parts, ('c:', 'a', 'b'))
808 p = P('c:/a/b')
809 parts = p.parts
810 self.assertEqual(parts, ('c:\\', 'a', 'b'))
811 p = P('//a/b/c/d')
812 parts = p.parts
813 self.assertEqual(parts, ('\\\\a\\b\\', 'c', 'd'))
814
815 def test_parent(self):
816 # Anchored
817 P = self.cls
818 p = P('z:a/b/c')
819 self.assertEqual(p.parent, P('z:a/b'))
820 self.assertEqual(p.parent.parent, P('z:a'))
821 self.assertEqual(p.parent.parent.parent, P('z:'))
822 self.assertEqual(p.parent.parent.parent.parent, P('z:'))
823 p = P('z:/a/b/c')
824 self.assertEqual(p.parent, P('z:/a/b'))
825 self.assertEqual(p.parent.parent, P('z:/a'))
826 self.assertEqual(p.parent.parent.parent, P('z:/'))
827 self.assertEqual(p.parent.parent.parent.parent, P('z:/'))
828 p = P('//a/b/c/d')
829 self.assertEqual(p.parent, P('//a/b/c'))
830 self.assertEqual(p.parent.parent, P('//a/b'))
831 self.assertEqual(p.parent.parent.parent, P('//a/b'))
832
833 def test_parents(self):
834 # Anchored
835 P = self.cls
836 p = P('z:a/b/')
837 par = p.parents
838 self.assertEqual(len(par), 2)
839 self.assertEqual(par[0], P('z:a'))
840 self.assertEqual(par[1], P('z:'))
841 self.assertEqual(list(par), [P('z:a'), P('z:')])
842 with self.assertRaises(IndexError):
843 par[2]
844 p = P('z:/a/b/')
845 par = p.parents
846 self.assertEqual(len(par), 2)
847 self.assertEqual(par[0], P('z:/a'))
848 self.assertEqual(par[1], P('z:/'))
849 self.assertEqual(list(par), [P('z:/a'), P('z:/')])
850 with self.assertRaises(IndexError):
851 par[2]
852 p = P('//a/b/c/d')
853 par = p.parents
854 self.assertEqual(len(par), 2)
855 self.assertEqual(par[0], P('//a/b/c'))
856 self.assertEqual(par[1], P('//a/b'))
857 self.assertEqual(list(par), [P('//a/b/c'), P('//a/b')])
858 with self.assertRaises(IndexError):
859 par[2]
860
861 def test_drive(self):
862 P = self.cls
863 self.assertEqual(P('c:').drive, 'c:')
864 self.assertEqual(P('c:a/b').drive, 'c:')
865 self.assertEqual(P('c:/').drive, 'c:')
866 self.assertEqual(P('c:/a/b/').drive, 'c:')
867 self.assertEqual(P('//a/b').drive, '\\\\a\\b')
868 self.assertEqual(P('//a/b/').drive, '\\\\a\\b')
869 self.assertEqual(P('//a/b/c/d').drive, '\\\\a\\b')
870
871 def test_root(self):
872 P = self.cls
873 self.assertEqual(P('c:').root, '')
874 self.assertEqual(P('c:a/b').root, '')
875 self.assertEqual(P('c:/').root, '\\')
876 self.assertEqual(P('c:/a/b/').root, '\\')
877 self.assertEqual(P('//a/b').root, '\\')
878 self.assertEqual(P('//a/b/').root, '\\')
879 self.assertEqual(P('//a/b/c/d').root, '\\')
880
881 def test_anchor(self):
882 P = self.cls
883 self.assertEqual(P('c:').anchor, 'c:')
884 self.assertEqual(P('c:a/b').anchor, 'c:')
885 self.assertEqual(P('c:/').anchor, 'c:\\')
886 self.assertEqual(P('c:/a/b/').anchor, 'c:\\')
887 self.assertEqual(P('//a/b').anchor, '\\\\a\\b\\')
888 self.assertEqual(P('//a/b/').anchor, '\\\\a\\b\\')
889 self.assertEqual(P('//a/b/c/d').anchor, '\\\\a\\b\\')
890
891 def test_name(self):
892 P = self.cls
893 self.assertEqual(P('c:').name, '')
894 self.assertEqual(P('c:/').name, '')
895 self.assertEqual(P('c:a/b').name, 'b')
896 self.assertEqual(P('c:/a/b').name, 'b')
897 self.assertEqual(P('c:a/b.py').name, 'b.py')
898 self.assertEqual(P('c:/a/b.py').name, 'b.py')
899 self.assertEqual(P('//My.py/Share.php').name, '')
900 self.assertEqual(P('//My.py/Share.php/a/b').name, 'b')
901
902 def test_suffix(self):
903 P = self.cls
904 self.assertEqual(P('c:').suffix, '')
905 self.assertEqual(P('c:/').suffix, '')
906 self.assertEqual(P('c:a/b').suffix, '')
907 self.assertEqual(P('c:/a/b').suffix, '')
908 self.assertEqual(P('c:a/b.py').suffix, '.py')
909 self.assertEqual(P('c:/a/b.py').suffix, '.py')
910 self.assertEqual(P('c:a/.hgrc').suffix, '')
911 self.assertEqual(P('c:/a/.hgrc').suffix, '')
912 self.assertEqual(P('c:a/.hg.rc').suffix, '.rc')
913 self.assertEqual(P('c:/a/.hg.rc').suffix, '.rc')
914 self.assertEqual(P('c:a/b.tar.gz').suffix, '.gz')
915 self.assertEqual(P('c:/a/b.tar.gz').suffix, '.gz')
916 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffix, '')
917 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffix, '')
918 self.assertEqual(P('//My.py/Share.php').suffix, '')
919 self.assertEqual(P('//My.py/Share.php/a/b').suffix, '')
920
921 def test_suffixes(self):
922 P = self.cls
923 self.assertEqual(P('c:').suffixes, [])
924 self.assertEqual(P('c:/').suffixes, [])
925 self.assertEqual(P('c:a/b').suffixes, [])
926 self.assertEqual(P('c:/a/b').suffixes, [])
927 self.assertEqual(P('c:a/b.py').suffixes, ['.py'])
928 self.assertEqual(P('c:/a/b.py').suffixes, ['.py'])
929 self.assertEqual(P('c:a/.hgrc').suffixes, [])
930 self.assertEqual(P('c:/a/.hgrc').suffixes, [])
931 self.assertEqual(P('c:a/.hg.rc').suffixes, ['.rc'])
932 self.assertEqual(P('c:/a/.hg.rc').suffixes, ['.rc'])
933 self.assertEqual(P('c:a/b.tar.gz').suffixes, ['.tar', '.gz'])
934 self.assertEqual(P('c:/a/b.tar.gz').suffixes, ['.tar', '.gz'])
935 self.assertEqual(P('//My.py/Share.php').suffixes, [])
936 self.assertEqual(P('//My.py/Share.php/a/b').suffixes, [])
937 self.assertEqual(P('c:a/Some name. Ending with a dot.').suffixes, [])
938 self.assertEqual(P('c:/a/Some name. Ending with a dot.').suffixes, [])
939
940 def test_stem(self):
941 P = self.cls
942 self.assertEqual(P('c:').stem, '')
943 self.assertEqual(P('c:.').stem, '')
944 self.assertEqual(P('c:..').stem, '..')
945 self.assertEqual(P('c:/').stem, '')
946 self.assertEqual(P('c:a/b').stem, 'b')
947 self.assertEqual(P('c:a/b.py').stem, 'b')
948 self.assertEqual(P('c:a/.hgrc').stem, '.hgrc')
949 self.assertEqual(P('c:a/.hg.rc').stem, '.hg')
950 self.assertEqual(P('c:a/b.tar.gz').stem, 'b.tar')
951 self.assertEqual(P('c:a/Some name. Ending with a dot.').stem,
952 'Some name. Ending with a dot.')
953
954 def test_with_name(self):
955 P = self.cls
956 self.assertEqual(P('c:a/b').with_name('d.xml'), P('c:a/d.xml'))
957 self.assertEqual(P('c:/a/b').with_name('d.xml'), P('c:/a/d.xml'))
958 self.assertEqual(P('c:a/Dot ending.').with_name('d.xml'), P('c:a/d.xml'))
959 self.assertEqual(P('c:/a/Dot ending.').with_name('d.xml'), P('c:/a/d.xml'))
960 self.assertRaises(ValueError, P('c:').with_name, 'd.xml')
961 self.assertRaises(ValueError, P('c:/').with_name, 'd.xml')
962 self.assertRaises(ValueError, P('//My/Share').with_name, 'd.xml')
Antoine Pitrou7084e732014-07-06 21:31:12 -0400963 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:')
964 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:e')
965 self.assertRaises(ValueError, P('c:a/b').with_name, 'd:/e')
966 self.assertRaises(ValueError, P('c:a/b').with_name, '//My/Share')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100967
968 def test_with_suffix(self):
969 P = self.cls
970 self.assertEqual(P('c:a/b').with_suffix('.gz'), P('c:a/b.gz'))
971 self.assertEqual(P('c:/a/b').with_suffix('.gz'), P('c:/a/b.gz'))
972 self.assertEqual(P('c:a/b.py').with_suffix('.gz'), P('c:a/b.gz'))
973 self.assertEqual(P('c:/a/b.py').with_suffix('.gz'), P('c:/a/b.gz'))
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100974 # Path doesn't have a "filename" component
Antoine Pitrou31119e42013-11-22 17:38:12 +0100975 self.assertRaises(ValueError, P('').with_suffix, '.gz')
976 self.assertRaises(ValueError, P('.').with_suffix, '.gz')
977 self.assertRaises(ValueError, P('/').with_suffix, '.gz')
978 self.assertRaises(ValueError, P('//My/Share').with_suffix, '.gz')
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100979 # Invalid suffix
980 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'gz')
981 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/')
982 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\')
983 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:')
984 self.assertRaises(ValueError, P('c:a/b').with_suffix, '/.gz')
985 self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\.gz')
986 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:.gz')
987 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c/d')
988 self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c\\d')
989 self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c/d')
990 self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c\\d')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100991
992 def test_relative_to(self):
993 P = self.cls
Antoine Pitrou156b3612013-12-28 19:49:04 +0100994 p = P('C:Foo/Bar')
995 self.assertEqual(p.relative_to(P('c:')), P('Foo/Bar'))
996 self.assertEqual(p.relative_to('c:'), P('Foo/Bar'))
997 self.assertEqual(p.relative_to(P('c:foO')), P('Bar'))
998 self.assertEqual(p.relative_to('c:foO'), P('Bar'))
999 self.assertEqual(p.relative_to('c:foO/'), P('Bar'))
1000 self.assertEqual(p.relative_to(P('c:foO/baR')), P())
1001 self.assertEqual(p.relative_to('c:foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001002 # Unrelated paths
1003 self.assertRaises(ValueError, p.relative_to, P())
Antoine Pitrou156b3612013-12-28 19:49:04 +01001004 self.assertRaises(ValueError, p.relative_to, '')
Antoine Pitrou31119e42013-11-22 17:38:12 +01001005 self.assertRaises(ValueError, p.relative_to, P('d:'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001006 self.assertRaises(ValueError, p.relative_to, P('/'))
1007 self.assertRaises(ValueError, p.relative_to, P('Foo'))
1008 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1009 self.assertRaises(ValueError, p.relative_to, P('C:/Foo'))
1010 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Bar/Baz'))
1011 self.assertRaises(ValueError, p.relative_to, P('C:Foo/Baz'))
1012 p = P('C:/Foo/Bar')
1013 self.assertEqual(p.relative_to(P('c:')), P('/Foo/Bar'))
1014 self.assertEqual(p.relative_to('c:'), P('/Foo/Bar'))
1015 self.assertEqual(str(p.relative_to(P('c:'))), '\\Foo\\Bar')
1016 self.assertEqual(str(p.relative_to('c:')), '\\Foo\\Bar')
1017 self.assertEqual(p.relative_to(P('c:/')), P('Foo/Bar'))
1018 self.assertEqual(p.relative_to('c:/'), P('Foo/Bar'))
1019 self.assertEqual(p.relative_to(P('c:/foO')), P('Bar'))
1020 self.assertEqual(p.relative_to('c:/foO'), P('Bar'))
1021 self.assertEqual(p.relative_to('c:/foO/'), P('Bar'))
1022 self.assertEqual(p.relative_to(P('c:/foO/baR')), P())
1023 self.assertEqual(p.relative_to('c:/foO/baR'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001024 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001025 self.assertRaises(ValueError, p.relative_to, P('C:/Baz'))
1026 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Bar/Baz'))
1027 self.assertRaises(ValueError, p.relative_to, P('C:/Foo/Baz'))
1028 self.assertRaises(ValueError, p.relative_to, P('C:Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001029 self.assertRaises(ValueError, p.relative_to, P('d:'))
1030 self.assertRaises(ValueError, p.relative_to, P('d:/'))
Antoine Pitrou156b3612013-12-28 19:49:04 +01001031 self.assertRaises(ValueError, p.relative_to, P('/'))
1032 self.assertRaises(ValueError, p.relative_to, P('/Foo'))
1033 self.assertRaises(ValueError, p.relative_to, P('//C/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001034 # UNC paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001035 p = P('//Server/Share/Foo/Bar')
1036 self.assertEqual(p.relative_to(P('//sErver/sHare')), P('Foo/Bar'))
1037 self.assertEqual(p.relative_to('//sErver/sHare'), P('Foo/Bar'))
1038 self.assertEqual(p.relative_to('//sErver/sHare/'), P('Foo/Bar'))
1039 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo')), P('Bar'))
1040 self.assertEqual(p.relative_to('//sErver/sHare/Foo'), P('Bar'))
1041 self.assertEqual(p.relative_to('//sErver/sHare/Foo/'), P('Bar'))
1042 self.assertEqual(p.relative_to(P('//sErver/sHare/Foo/Bar')), P())
1043 self.assertEqual(p.relative_to('//sErver/sHare/Foo/Bar'), P())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001044 # Unrelated paths
Antoine Pitrou156b3612013-12-28 19:49:04 +01001045 self.assertRaises(ValueError, p.relative_to, P('/Server/Share/Foo'))
1046 self.assertRaises(ValueError, p.relative_to, P('c:/Server/Share/Foo'))
1047 self.assertRaises(ValueError, p.relative_to, P('//z/Share/Foo'))
1048 self.assertRaises(ValueError, p.relative_to, P('//Server/z/Foo'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001049
1050 def test_is_absolute(self):
1051 P = self.cls
1052 # Under NT, only paths with both a drive and a root are absolute
1053 self.assertFalse(P().is_absolute())
1054 self.assertFalse(P('a').is_absolute())
1055 self.assertFalse(P('a/b/').is_absolute())
1056 self.assertFalse(P('/').is_absolute())
1057 self.assertFalse(P('/a').is_absolute())
1058 self.assertFalse(P('/a/b/').is_absolute())
1059 self.assertFalse(P('c:').is_absolute())
1060 self.assertFalse(P('c:a').is_absolute())
1061 self.assertFalse(P('c:a/b/').is_absolute())
1062 self.assertTrue(P('c:/').is_absolute())
1063 self.assertTrue(P('c:/a').is_absolute())
1064 self.assertTrue(P('c:/a/b/').is_absolute())
1065 # UNC paths are absolute by definition
1066 self.assertTrue(P('//a/b').is_absolute())
1067 self.assertTrue(P('//a/b/').is_absolute())
1068 self.assertTrue(P('//a/b/c').is_absolute())
1069 self.assertTrue(P('//a/b/c/d').is_absolute())
1070
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001071 def test_join(self):
1072 P = self.cls
1073 p = P('C:/a/b')
1074 pp = p.joinpath('x/y')
1075 self.assertEqual(pp, P('C:/a/b/x/y'))
1076 pp = p.joinpath('/x/y')
1077 self.assertEqual(pp, P('C:/x/y'))
1078 # Joining with a different drive => the first path is ignored, even
1079 # if the second path is relative.
1080 pp = p.joinpath('D:x/y')
1081 self.assertEqual(pp, P('D:x/y'))
1082 pp = p.joinpath('D:/x/y')
1083 self.assertEqual(pp, P('D:/x/y'))
1084 pp = p.joinpath('//host/share/x/y')
1085 self.assertEqual(pp, P('//host/share/x/y'))
1086 # Joining with the same drive => the first path is appended to if
1087 # the second path is relative.
1088 pp = p.joinpath('c:x/y')
1089 self.assertEqual(pp, P('C:/a/b/x/y'))
1090 pp = p.joinpath('c:/x/y')
1091 self.assertEqual(pp, P('C:/x/y'))
1092
1093 def test_div(self):
1094 # Basically the same as joinpath()
1095 P = self.cls
1096 p = P('C:/a/b')
1097 self.assertEqual(p / 'x/y', P('C:/a/b/x/y'))
1098 self.assertEqual(p / 'x' / 'y', P('C:/a/b/x/y'))
1099 self.assertEqual(p / '/x/y', P('C:/x/y'))
1100 self.assertEqual(p / '/x' / 'y', P('C:/x/y'))
1101 # Joining with a different drive => the first path is ignored, even
1102 # if the second path is relative.
1103 self.assertEqual(p / 'D:x/y', P('D:x/y'))
1104 self.assertEqual(p / 'D:' / 'x/y', P('D:x/y'))
1105 self.assertEqual(p / 'D:/x/y', P('D:/x/y'))
1106 self.assertEqual(p / 'D:' / '/x/y', P('D:/x/y'))
1107 self.assertEqual(p / '//host/share/x/y', P('//host/share/x/y'))
1108 # Joining with the same drive => the first path is appended to if
1109 # the second path is relative.
Serhiy Storchaka010ff582013-12-06 17:25:51 +02001110 self.assertEqual(p / 'c:x/y', P('C:/a/b/x/y'))
1111 self.assertEqual(p / 'c:/x/y', P('C:/x/y'))
Serhiy Storchakaa9939022013-12-06 17:14:12 +02001112
Antoine Pitrou31119e42013-11-22 17:38:12 +01001113 def test_is_reserved(self):
1114 P = self.cls
1115 self.assertIs(False, P('').is_reserved())
1116 self.assertIs(False, P('/').is_reserved())
1117 self.assertIs(False, P('/foo/bar').is_reserved())
1118 self.assertIs(True, P('con').is_reserved())
1119 self.assertIs(True, P('NUL').is_reserved())
1120 self.assertIs(True, P('NUL.txt').is_reserved())
1121 self.assertIs(True, P('com1').is_reserved())
1122 self.assertIs(True, P('com9.bar').is_reserved())
1123 self.assertIs(False, P('bar.com9').is_reserved())
1124 self.assertIs(True, P('lpt1').is_reserved())
1125 self.assertIs(True, P('lpt9.bar').is_reserved())
1126 self.assertIs(False, P('bar.lpt9').is_reserved())
1127 # Only the last component matters
1128 self.assertIs(False, P('c:/NUL/con/baz').is_reserved())
1129 # UNC paths are never reserved
1130 self.assertIs(False, P('//my/share/nul/con/aux').is_reserved())
1131
1132
1133class PurePathTest(_BasePurePathTest, unittest.TestCase):
1134 cls = pathlib.PurePath
1135
1136 def test_concrete_class(self):
1137 p = self.cls('a')
1138 self.assertIs(type(p),
1139 pathlib.PureWindowsPath if os.name == 'nt' else pathlib.PurePosixPath)
1140
1141 def test_different_flavours_unequal(self):
1142 p = pathlib.PurePosixPath('a')
1143 q = pathlib.PureWindowsPath('a')
1144 self.assertNotEqual(p, q)
1145
1146 def test_different_flavours_unordered(self):
1147 p = pathlib.PurePosixPath('a')
1148 q = pathlib.PureWindowsPath('a')
1149 with self.assertRaises(TypeError):
1150 p < q
1151 with self.assertRaises(TypeError):
1152 p <= q
1153 with self.assertRaises(TypeError):
1154 p > q
1155 with self.assertRaises(TypeError):
1156 p >= q
1157
1158
1159#
1160# Tests for the concrete classes
1161#
1162
1163# Make sure any symbolic links in the base test path are resolved
1164BASE = os.path.realpath(TESTFN)
1165join = lambda *x: os.path.join(BASE, *x)
1166rel_join = lambda *x: os.path.join(TESTFN, *x)
1167
1168def symlink_skip_reason():
1169 if not pathlib.supports_symlinks:
1170 return "no system support for symlinks"
1171 try:
1172 os.symlink(__file__, BASE)
1173 except OSError as e:
1174 return str(e)
1175 else:
1176 support.unlink(BASE)
1177 return None
1178
1179symlink_skip_reason = symlink_skip_reason()
1180
1181only_nt = unittest.skipIf(os.name != 'nt',
1182 'test requires a Windows-compatible system')
1183only_posix = unittest.skipIf(os.name == 'nt',
1184 'test requires a POSIX-compatible system')
1185with_symlinks = unittest.skipIf(symlink_skip_reason, symlink_skip_reason)
1186
1187
1188@only_posix
1189class PosixPathAsPureTest(PurePosixPathTest):
1190 cls = pathlib.PosixPath
1191
1192@only_nt
1193class WindowsPathAsPureTest(PureWindowsPathTest):
1194 cls = pathlib.WindowsPath
1195
1196
1197class _BasePathTest(object):
1198 """Tests for the FS-accessing functionalities of the Path classes."""
1199
1200 # (BASE)
1201 # |
1202 # |-- dirA/
1203 # |-- linkC -> "../dirB"
1204 # |-- dirB/
1205 # | |-- fileB
1206 # |-- linkD -> "../dirB"
1207 # |-- dirC/
1208 # | |-- fileC
1209 # | |-- fileD
1210 # |-- fileA
1211 # |-- linkA -> "fileA"
1212 # |-- linkB -> "dirB"
1213 #
1214
1215 def setUp(self):
1216 os.mkdir(BASE)
Victor Stinnerec864692014-07-21 19:19:05 +02001217 self.addCleanup(support.rmtree, BASE)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001218 os.mkdir(join('dirA'))
1219 os.mkdir(join('dirB'))
1220 os.mkdir(join('dirC'))
1221 os.mkdir(join('dirC', 'dirD'))
1222 with open(join('fileA'), 'wb') as f:
1223 f.write(b"this is file A\n")
1224 with open(join('dirB', 'fileB'), 'wb') as f:
1225 f.write(b"this is file B\n")
1226 with open(join('dirC', 'fileC'), 'wb') as f:
1227 f.write(b"this is file C\n")
1228 with open(join('dirC', 'dirD', 'fileD'), 'wb') as f:
1229 f.write(b"this is file D\n")
1230 if not symlink_skip_reason:
Antoine Pitrou31119e42013-11-22 17:38:12 +01001231 # Relative symlinks
1232 os.symlink('fileA', join('linkA'))
1233 os.symlink('non-existing', join('brokenLink'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001234 self.dirlink('dirB', join('linkB'))
1235 self.dirlink(os.path.join('..', 'dirB'), join('dirA', 'linkC'))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001236 # This one goes upwards but doesn't create a loop
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001237 self.dirlink(os.path.join('..', 'dirB'), join('dirB', 'linkD'))
1238
1239 if os.name == 'nt':
1240 # Workaround for http://bugs.python.org/issue13772
1241 def dirlink(self, src, dest):
1242 os.symlink(src, dest, target_is_directory=True)
1243 else:
1244 def dirlink(self, src, dest):
1245 os.symlink(src, dest)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001246
1247 def assertSame(self, path_a, path_b):
1248 self.assertTrue(os.path.samefile(str(path_a), str(path_b)),
1249 "%r and %r don't point to the same file" %
1250 (path_a, path_b))
1251
1252 def assertFileNotFound(self, func, *args, **kwargs):
1253 with self.assertRaises(FileNotFoundError) as cm:
1254 func(*args, **kwargs)
1255 self.assertEqual(cm.exception.errno, errno.ENOENT)
1256
1257 def _test_cwd(self, p):
1258 q = self.cls(os.getcwd())
1259 self.assertEqual(p, q)
1260 self.assertEqual(str(p), str(q))
1261 self.assertIs(type(p), type(q))
1262 self.assertTrue(p.is_absolute())
1263
1264 def test_cwd(self):
1265 p = self.cls.cwd()
1266 self._test_cwd(p)
1267
Antoine Pitrou17cba7d2015-01-12 21:03:41 +01001268 def _test_home(self, p):
1269 q = self.cls(os.path.expanduser('~'))
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_home(self):
1276 p = self.cls.home()
1277 self._test_home(p)
1278
Antoine Pitrou43e3d942014-05-13 10:50:15 +02001279 def test_samefile(self):
1280 fileA_path = os.path.join(BASE, 'fileA')
1281 fileB_path = os.path.join(BASE, 'dirB', 'fileB')
1282 p = self.cls(fileA_path)
1283 pp = self.cls(fileA_path)
1284 q = self.cls(fileB_path)
1285 self.assertTrue(p.samefile(fileA_path))
1286 self.assertTrue(p.samefile(pp))
1287 self.assertFalse(p.samefile(fileB_path))
1288 self.assertFalse(p.samefile(q))
1289 # Test the non-existent file case
1290 non_existent = os.path.join(BASE, 'foo')
1291 r = self.cls(non_existent)
1292 self.assertRaises(FileNotFoundError, p.samefile, r)
1293 self.assertRaises(FileNotFoundError, p.samefile, non_existent)
1294 self.assertRaises(FileNotFoundError, r.samefile, p)
1295 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1296 self.assertRaises(FileNotFoundError, r.samefile, r)
1297 self.assertRaises(FileNotFoundError, r.samefile, non_existent)
1298
Antoine Pitrou31119e42013-11-22 17:38:12 +01001299 def test_empty_path(self):
1300 # The empty path points to '.'
1301 p = self.cls('')
1302 self.assertEqual(p.stat(), os.stat('.'))
1303
Antoine Pitrou8477ed62014-12-30 20:54:45 +01001304 def test_expanduser_common(self):
1305 P = self.cls
1306 p = P('~')
1307 self.assertEqual(p.expanduser(), P(os.path.expanduser('~')))
1308 p = P('foo')
1309 self.assertEqual(p.expanduser(), p)
1310 p = P('/~')
1311 self.assertEqual(p.expanduser(), p)
1312 p = P('../~')
1313 self.assertEqual(p.expanduser(), p)
1314 p = P(P('').absolute().anchor) / '~'
1315 self.assertEqual(p.expanduser(), p)
1316
Antoine Pitrou31119e42013-11-22 17:38:12 +01001317 def test_exists(self):
1318 P = self.cls
1319 p = P(BASE)
1320 self.assertIs(True, p.exists())
1321 self.assertIs(True, (p / 'dirA').exists())
1322 self.assertIs(True, (p / 'fileA').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001323 self.assertIs(False, (p / 'fileA' / 'bah').exists())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001324 if not symlink_skip_reason:
1325 self.assertIs(True, (p / 'linkA').exists())
1326 self.assertIs(True, (p / 'linkB').exists())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001327 self.assertIs(True, (p / 'linkB' / 'fileB').exists())
1328 self.assertIs(False, (p / 'linkA' / 'bah').exists())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001329 self.assertIs(False, (p / 'foo').exists())
1330 self.assertIs(False, P('/xyzzy').exists())
1331
1332 def test_open_common(self):
1333 p = self.cls(BASE)
1334 with (p / 'fileA').open('r') as f:
1335 self.assertIsInstance(f, io.TextIOBase)
1336 self.assertEqual(f.read(), "this is file A\n")
1337 with (p / 'fileA').open('rb') as f:
1338 self.assertIsInstance(f, io.BufferedIOBase)
1339 self.assertEqual(f.read().strip(), b"this is file A")
1340 with (p / 'fileA').open('rb', buffering=0) as f:
1341 self.assertIsInstance(f, io.RawIOBase)
1342 self.assertEqual(f.read().strip(), b"this is file A")
1343
Georg Brandlea683982014-10-01 19:12:33 +02001344 def test_read_write_bytes(self):
1345 p = self.cls(BASE)
1346 (p / 'fileA').write_bytes(b'abcdefg')
1347 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1348 # check that trying to write str does not truncate the file
1349 self.assertRaises(TypeError, (p / 'fileA').write_bytes, 'somestr')
1350 self.assertEqual((p / 'fileA').read_bytes(), b'abcdefg')
1351
1352 def test_read_write_text(self):
1353 p = self.cls(BASE)
1354 (p / 'fileA').write_text('äbcdefg', encoding='latin-1')
1355 self.assertEqual((p / 'fileA').read_text(
1356 encoding='utf-8', errors='ignore'), 'bcdefg')
1357 # check that trying to write bytes does not truncate the file
1358 self.assertRaises(TypeError, (p / 'fileA').write_text, b'somebytes')
1359 self.assertEqual((p / 'fileA').read_text(encoding='latin-1'), 'äbcdefg')
1360
Antoine Pitrou31119e42013-11-22 17:38:12 +01001361 def test_iterdir(self):
1362 P = self.cls
1363 p = P(BASE)
1364 it = p.iterdir()
1365 paths = set(it)
1366 expected = ['dirA', 'dirB', 'dirC', 'fileA']
1367 if not symlink_skip_reason:
1368 expected += ['linkA', 'linkB', 'brokenLink']
1369 self.assertEqual(paths, { P(BASE, q) for q in expected })
1370
1371 @with_symlinks
1372 def test_iterdir_symlink(self):
1373 # __iter__ on a symlink to a directory
1374 P = self.cls
1375 p = P(BASE, 'linkB')
1376 paths = set(p.iterdir())
1377 expected = { P(BASE, 'linkB', q) for q in ['fileB', 'linkD'] }
1378 self.assertEqual(paths, expected)
1379
1380 def test_iterdir_nodir(self):
1381 # __iter__ on something that is not a directory
1382 p = self.cls(BASE, 'fileA')
1383 with self.assertRaises(OSError) as cm:
1384 next(p.iterdir())
1385 # ENOENT or EINVAL under Windows, ENOTDIR otherwise
1386 # (see issue #12802)
1387 self.assertIn(cm.exception.errno, (errno.ENOTDIR,
1388 errno.ENOENT, errno.EINVAL))
1389
1390 def test_glob_common(self):
1391 def _check(glob, expected):
1392 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1393 P = self.cls
1394 p = P(BASE)
1395 it = p.glob("fileA")
1396 self.assertIsInstance(it, collections.Iterator)
1397 _check(it, ["fileA"])
1398 _check(p.glob("fileB"), [])
1399 _check(p.glob("dir*/file*"), ["dirB/fileB", "dirC/fileC"])
1400 if symlink_skip_reason:
1401 _check(p.glob("*A"), ['dirA', 'fileA'])
1402 else:
1403 _check(p.glob("*A"), ['dirA', 'fileA', 'linkA'])
1404 if symlink_skip_reason:
1405 _check(p.glob("*B/*"), ['dirB/fileB'])
1406 else:
1407 _check(p.glob("*B/*"), ['dirB/fileB', 'dirB/linkD',
1408 'linkB/fileB', 'linkB/linkD'])
1409 if symlink_skip_reason:
1410 _check(p.glob("*/fileB"), ['dirB/fileB'])
1411 else:
1412 _check(p.glob("*/fileB"), ['dirB/fileB', 'linkB/fileB'])
1413
1414 def test_rglob_common(self):
1415 def _check(glob, expected):
1416 self.assertEqual(set(glob), { P(BASE, q) for q in expected })
1417 P = self.cls
1418 p = P(BASE)
1419 it = p.rglob("fileA")
1420 self.assertIsInstance(it, collections.Iterator)
1421 # XXX cannot test because of symlink loops in the test setup
1422 #_check(it, ["fileA"])
1423 #_check(p.rglob("fileB"), ["dirB/fileB"])
1424 #_check(p.rglob("*/fileA"), [""])
1425 #_check(p.rglob("*/fileB"), ["dirB/fileB"])
1426 #_check(p.rglob("file*"), ["fileA", "dirB/fileB"])
1427 # No symlink loops here
1428 p = P(BASE, "dirC")
1429 _check(p.rglob("file*"), ["dirC/fileC", "dirC/dirD/fileD"])
1430 _check(p.rglob("*/*"), ["dirC/dirD/fileD"])
1431
1432 def test_glob_dotdot(self):
1433 # ".." is not special in globs
1434 P = self.cls
1435 p = P(BASE)
1436 self.assertEqual(set(p.glob("..")), { P(BASE, "..") })
1437 self.assertEqual(set(p.glob("dirA/../file*")), { P(BASE, "dirA/../fileA") })
1438 self.assertEqual(set(p.glob("../xyzzy")), set())
1439
1440 def _check_resolve_relative(self, p, expected):
1441 q = p.resolve()
1442 self.assertEqual(q, expected)
1443
1444 def _check_resolve_absolute(self, p, expected):
1445 q = p.resolve()
1446 self.assertEqual(q, expected)
1447
1448 @with_symlinks
1449 def test_resolve_common(self):
1450 P = self.cls
1451 p = P(BASE, 'foo')
1452 with self.assertRaises(OSError) as cm:
1453 p.resolve()
1454 self.assertEqual(cm.exception.errno, errno.ENOENT)
1455 # These are all relative symlinks
1456 p = P(BASE, 'dirB', 'fileB')
1457 self._check_resolve_relative(p, p)
1458 p = P(BASE, 'linkA')
1459 self._check_resolve_relative(p, P(BASE, 'fileA'))
1460 p = P(BASE, 'dirA', 'linkC', 'fileB')
1461 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
1462 p = P(BASE, 'dirB', 'linkD', 'fileB')
1463 self._check_resolve_relative(p, P(BASE, 'dirB', 'fileB'))
1464 # Now create absolute symlinks
1465 d = tempfile.mkdtemp(suffix='-dirD')
Victor Stinnerec864692014-07-21 19:19:05 +02001466 self.addCleanup(support.rmtree, d)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001467 os.symlink(os.path.join(d), join('dirA', 'linkX'))
1468 os.symlink(join('dirB'), os.path.join(d, 'linkY'))
1469 p = P(BASE, 'dirA', 'linkX', 'linkY', 'fileB')
1470 self._check_resolve_absolute(p, P(BASE, 'dirB', 'fileB'))
1471
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001472 @with_symlinks
1473 def test_resolve_dot(self):
1474 # See https://bitbucket.org/pitrou/pathlib/issue/9/pathresolve-fails-on-complex-symlinks
1475 p = self.cls(BASE)
1476 self.dirlink('.', join('0'))
Antoine Pitroucc157512013-12-03 17:13:13 +01001477 self.dirlink(os.path.join('0', '0'), join('1'))
1478 self.dirlink(os.path.join('1', '1'), join('2'))
Antoine Pitrou51af82c2013-12-03 11:01:08 +01001479 q = p / '2'
1480 self.assertEqual(q.resolve(), p)
1481
Antoine Pitrou31119e42013-11-22 17:38:12 +01001482 def test_with(self):
1483 p = self.cls(BASE)
1484 it = p.iterdir()
1485 it2 = p.iterdir()
1486 next(it2)
1487 with p:
1488 pass
1489 # I/O operation on closed path
1490 self.assertRaises(ValueError, next, it)
1491 self.assertRaises(ValueError, next, it2)
1492 self.assertRaises(ValueError, p.open)
1493 self.assertRaises(ValueError, p.resolve)
1494 self.assertRaises(ValueError, p.absolute)
1495 self.assertRaises(ValueError, p.__enter__)
1496
1497 def test_chmod(self):
1498 p = self.cls(BASE) / 'fileA'
1499 mode = p.stat().st_mode
1500 # Clear writable bit
1501 new_mode = mode & ~0o222
1502 p.chmod(new_mode)
1503 self.assertEqual(p.stat().st_mode, new_mode)
1504 # Set writable bit
1505 new_mode = mode | 0o222
1506 p.chmod(new_mode)
1507 self.assertEqual(p.stat().st_mode, new_mode)
1508
1509 # XXX also need a test for lchmod
1510
1511 def test_stat(self):
1512 p = self.cls(BASE) / 'fileA'
1513 st = p.stat()
1514 self.assertEqual(p.stat(), st)
1515 # Change file mode by flipping write bit
1516 p.chmod(st.st_mode ^ 0o222)
1517 self.addCleanup(p.chmod, st.st_mode)
1518 self.assertNotEqual(p.stat(), st)
1519
1520 @with_symlinks
1521 def test_lstat(self):
1522 p = self.cls(BASE)/ 'linkA'
1523 st = p.stat()
1524 self.assertNotEqual(st, p.lstat())
1525
1526 def test_lstat_nosymlink(self):
1527 p = self.cls(BASE) / 'fileA'
1528 st = p.stat()
1529 self.assertEqual(st, p.lstat())
1530
1531 @unittest.skipUnless(pwd, "the pwd module is needed for this test")
1532 def test_owner(self):
1533 p = self.cls(BASE) / 'fileA'
1534 uid = p.stat().st_uid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001535 try:
1536 name = pwd.getpwuid(uid).pw_name
1537 except KeyError:
1538 self.skipTest(
1539 "user %d doesn't have an entry in the system database" % uid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001540 self.assertEqual(name, p.owner())
1541
1542 @unittest.skipUnless(grp, "the grp module is needed for this test")
1543 def test_group(self):
1544 p = self.cls(BASE) / 'fileA'
1545 gid = p.stat().st_gid
Antoine Pitrou2cf4b0f2013-11-25 19:51:53 +01001546 try:
1547 name = grp.getgrgid(gid).gr_name
1548 except KeyError:
1549 self.skipTest(
1550 "group %d doesn't have an entry in the system database" % gid)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001551 self.assertEqual(name, p.group())
1552
1553 def test_unlink(self):
1554 p = self.cls(BASE) / 'fileA'
1555 p.unlink()
1556 self.assertFileNotFound(p.stat)
1557 self.assertFileNotFound(p.unlink)
1558
1559 def test_rmdir(self):
1560 p = self.cls(BASE) / 'dirA'
1561 for q in p.iterdir():
1562 q.unlink()
1563 p.rmdir()
1564 self.assertFileNotFound(p.stat)
1565 self.assertFileNotFound(p.unlink)
1566
1567 def test_rename(self):
1568 P = self.cls(BASE)
1569 p = P / 'fileA'
1570 size = p.stat().st_size
1571 # Renaming to another path
1572 q = P / 'dirA' / 'fileAA'
1573 p.rename(q)
1574 self.assertEqual(q.stat().st_size, size)
1575 self.assertFileNotFound(p.stat)
1576 # Renaming to a str of a relative path
1577 r = rel_join('fileAAA')
1578 q.rename(r)
1579 self.assertEqual(os.stat(r).st_size, size)
1580 self.assertFileNotFound(q.stat)
1581
1582 def test_replace(self):
1583 P = self.cls(BASE)
1584 p = P / 'fileA'
1585 size = p.stat().st_size
1586 # Replacing a non-existing path
1587 q = P / 'dirA' / 'fileAA'
1588 p.replace(q)
1589 self.assertEqual(q.stat().st_size, size)
1590 self.assertFileNotFound(p.stat)
1591 # Replacing another (existing) path
1592 r = rel_join('dirB', 'fileB')
1593 q.replace(r)
1594 self.assertEqual(os.stat(r).st_size, size)
1595 self.assertFileNotFound(q.stat)
1596
1597 def test_touch_common(self):
1598 P = self.cls(BASE)
1599 p = P / 'newfileA'
1600 self.assertFalse(p.exists())
1601 p.touch()
1602 self.assertTrue(p.exists())
Antoine Pitrou0f575642013-11-22 23:20:08 +01001603 st = p.stat()
1604 old_mtime = st.st_mtime
1605 old_mtime_ns = st.st_mtime_ns
Antoine Pitrou31119e42013-11-22 17:38:12 +01001606 # Rewind the mtime sufficiently far in the past to work around
1607 # filesystem-specific timestamp granularity.
1608 os.utime(str(p), (old_mtime - 10, old_mtime - 10))
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001609 # The file mtime should be refreshed by calling touch() again
Antoine Pitrou31119e42013-11-22 17:38:12 +01001610 p.touch()
Antoine Pitrou0f575642013-11-22 23:20:08 +01001611 st = p.stat()
Antoine Pitrou2cf39172013-11-23 15:25:59 +01001612 self.assertGreaterEqual(st.st_mtime_ns, old_mtime_ns)
1613 self.assertGreaterEqual(st.st_mtime, old_mtime)
Antoine Pitroubb6694d2013-11-23 01:32:53 +01001614 # Now with exist_ok=False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001615 p = P / 'newfileB'
1616 self.assertFalse(p.exists())
1617 p.touch(mode=0o700, exist_ok=False)
1618 self.assertTrue(p.exists())
1619 self.assertRaises(OSError, p.touch, exist_ok=False)
1620
Antoine Pitrou8b784932013-11-23 14:52:39 +01001621 def test_touch_nochange(self):
1622 P = self.cls(BASE)
1623 p = P / 'fileA'
1624 p.touch()
1625 with p.open('rb') as f:
1626 self.assertEqual(f.read().strip(), b"this is file A")
1627
Antoine Pitrou31119e42013-11-22 17:38:12 +01001628 def test_mkdir(self):
1629 P = self.cls(BASE)
1630 p = P / 'newdirA'
1631 self.assertFalse(p.exists())
1632 p.mkdir()
1633 self.assertTrue(p.exists())
1634 self.assertTrue(p.is_dir())
1635 with self.assertRaises(OSError) as cm:
1636 p.mkdir()
1637 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001638
1639 def test_mkdir_parents(self):
1640 # Creating a chain of directories
1641 p = self.cls(BASE, 'newdirB', 'newdirC')
1642 self.assertFalse(p.exists())
1643 with self.assertRaises(OSError) as cm:
1644 p.mkdir()
1645 self.assertEqual(cm.exception.errno, errno.ENOENT)
1646 p.mkdir(parents=True)
1647 self.assertTrue(p.exists())
1648 self.assertTrue(p.is_dir())
1649 with self.assertRaises(OSError) as cm:
1650 p.mkdir(parents=True)
1651 self.assertEqual(cm.exception.errno, errno.EEXIST)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001652 # test `mode` arg
1653 mode = stat.S_IMODE(p.stat().st_mode) # default mode
1654 p = self.cls(BASE, 'newdirD', 'newdirE')
1655 p.mkdir(0o555, parents=True)
1656 self.assertTrue(p.exists())
1657 self.assertTrue(p.is_dir())
1658 if os.name != 'nt':
1659 # the directory's permissions follow the mode argument
Gregory P. Smithb599c612014-01-20 01:10:33 -08001660 self.assertEqual(stat.S_IMODE(p.stat().st_mode), 0o7555 & mode)
Antoine Pitrou0048c982013-12-16 20:22:37 +01001661 # the parent's permissions follow the default process settings
1662 self.assertEqual(stat.S_IMODE(p.parent.stat().st_mode), mode)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001663
Barry Warsaw7c549c42014-08-05 11:28:12 -04001664 def test_mkdir_exist_ok(self):
1665 p = self.cls(BASE, 'dirB')
1666 st_ctime_first = p.stat().st_ctime
1667 self.assertTrue(p.exists())
1668 self.assertTrue(p.is_dir())
1669 with self.assertRaises(FileExistsError) as cm:
1670 p.mkdir()
1671 self.assertEqual(cm.exception.errno, errno.EEXIST)
1672 p.mkdir(exist_ok=True)
1673 self.assertTrue(p.exists())
1674 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1675
1676 def test_mkdir_exist_ok_with_parent(self):
1677 p = self.cls(BASE, 'dirC')
1678 self.assertTrue(p.exists())
1679 with self.assertRaises(FileExistsError) as cm:
1680 p.mkdir()
1681 self.assertEqual(cm.exception.errno, errno.EEXIST)
1682 p = p / 'newdirC'
1683 p.mkdir(parents=True)
1684 st_ctime_first = p.stat().st_ctime
1685 self.assertTrue(p.exists())
1686 with self.assertRaises(FileExistsError) as cm:
1687 p.mkdir(parents=True)
1688 self.assertEqual(cm.exception.errno, errno.EEXIST)
1689 p.mkdir(parents=True, exist_ok=True)
1690 self.assertTrue(p.exists())
1691 self.assertEqual(p.stat().st_ctime, st_ctime_first)
1692
1693 def test_mkdir_with_child_file(self):
1694 p = self.cls(BASE, 'dirB', 'fileB')
1695 self.assertTrue(p.exists())
1696 # An exception is raised when the last path component is an existing
1697 # regular file, regardless of whether exist_ok is true or not.
1698 with self.assertRaises(FileExistsError) as cm:
1699 p.mkdir(parents=True)
1700 self.assertEqual(cm.exception.errno, errno.EEXIST)
1701 with self.assertRaises(FileExistsError) as cm:
1702 p.mkdir(parents=True, exist_ok=True)
1703 self.assertEqual(cm.exception.errno, errno.EEXIST)
1704
1705 def test_mkdir_no_parents_file(self):
1706 p = self.cls(BASE, 'fileA')
1707 self.assertTrue(p.exists())
1708 # An exception is raised when the last path component is an existing
1709 # regular file, regardless of whether exist_ok is true or not.
1710 with self.assertRaises(FileExistsError) as cm:
1711 p.mkdir()
1712 self.assertEqual(cm.exception.errno, errno.EEXIST)
1713 with self.assertRaises(FileExistsError) as cm:
1714 p.mkdir(exist_ok=True)
1715 self.assertEqual(cm.exception.errno, errno.EEXIST)
1716
Antoine Pitrou31119e42013-11-22 17:38:12 +01001717 @with_symlinks
1718 def test_symlink_to(self):
1719 P = self.cls(BASE)
1720 target = P / 'fileA'
1721 # Symlinking a path target
1722 link = P / 'dirA' / 'linkAA'
1723 link.symlink_to(target)
1724 self.assertEqual(link.stat(), target.stat())
1725 self.assertNotEqual(link.lstat(), target.stat())
1726 # Symlinking a str target
1727 link = P / 'dirA' / 'linkAAA'
1728 link.symlink_to(str(target))
1729 self.assertEqual(link.stat(), target.stat())
1730 self.assertNotEqual(link.lstat(), target.stat())
1731 self.assertFalse(link.is_dir())
1732 # Symlinking to a directory
1733 target = P / 'dirB'
1734 link = P / 'dirA' / 'linkAAAA'
1735 link.symlink_to(target, target_is_directory=True)
1736 self.assertEqual(link.stat(), target.stat())
1737 self.assertNotEqual(link.lstat(), target.stat())
1738 self.assertTrue(link.is_dir())
1739 self.assertTrue(list(link.iterdir()))
1740
1741 def test_is_dir(self):
1742 P = self.cls(BASE)
1743 self.assertTrue((P / 'dirA').is_dir())
1744 self.assertFalse((P / 'fileA').is_dir())
1745 self.assertFalse((P / 'non-existing').is_dir())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001746 self.assertFalse((P / 'fileA' / 'bah').is_dir())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001747 if not symlink_skip_reason:
1748 self.assertFalse((P / 'linkA').is_dir())
1749 self.assertTrue((P / 'linkB').is_dir())
1750 self.assertFalse((P/ 'brokenLink').is_dir())
1751
1752 def test_is_file(self):
1753 P = self.cls(BASE)
1754 self.assertTrue((P / 'fileA').is_file())
1755 self.assertFalse((P / 'dirA').is_file())
1756 self.assertFalse((P / 'non-existing').is_file())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001757 self.assertFalse((P / 'fileA' / 'bah').is_file())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001758 if not symlink_skip_reason:
1759 self.assertTrue((P / 'linkA').is_file())
1760 self.assertFalse((P / 'linkB').is_file())
1761 self.assertFalse((P/ 'brokenLink').is_file())
1762
1763 def test_is_symlink(self):
1764 P = self.cls(BASE)
1765 self.assertFalse((P / 'fileA').is_symlink())
1766 self.assertFalse((P / 'dirA').is_symlink())
1767 self.assertFalse((P / 'non-existing').is_symlink())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001768 self.assertFalse((P / 'fileA' / 'bah').is_symlink())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001769 if not symlink_skip_reason:
1770 self.assertTrue((P / 'linkA').is_symlink())
1771 self.assertTrue((P / 'linkB').is_symlink())
1772 self.assertTrue((P/ 'brokenLink').is_symlink())
1773
1774 def test_is_fifo_false(self):
1775 P = self.cls(BASE)
1776 self.assertFalse((P / 'fileA').is_fifo())
1777 self.assertFalse((P / 'dirA').is_fifo())
1778 self.assertFalse((P / 'non-existing').is_fifo())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001779 self.assertFalse((P / 'fileA' / 'bah').is_fifo())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001780
1781 @unittest.skipUnless(hasattr(os, "mkfifo"), "os.mkfifo() required")
1782 def test_is_fifo_true(self):
1783 P = self.cls(BASE, 'myfifo')
1784 os.mkfifo(str(P))
1785 self.assertTrue(P.is_fifo())
1786 self.assertFalse(P.is_socket())
1787 self.assertFalse(P.is_file())
1788
1789 def test_is_socket_false(self):
1790 P = self.cls(BASE)
1791 self.assertFalse((P / 'fileA').is_socket())
1792 self.assertFalse((P / 'dirA').is_socket())
1793 self.assertFalse((P / 'non-existing').is_socket())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001794 self.assertFalse((P / 'fileA' / 'bah').is_socket())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001795
1796 @unittest.skipUnless(hasattr(socket, "AF_UNIX"), "Unix sockets required")
1797 def test_is_socket_true(self):
1798 P = self.cls(BASE, 'mysock')
Antoine Pitrou330ce592013-11-22 18:05:06 +01001799 sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001800 self.addCleanup(sock.close)
Antoine Pitrou330ce592013-11-22 18:05:06 +01001801 try:
1802 sock.bind(str(P))
1803 except OSError as e:
1804 if "AF_UNIX path too long" in str(e):
1805 self.skipTest("cannot bind Unix socket: " + str(e))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001806 self.assertTrue(P.is_socket())
1807 self.assertFalse(P.is_fifo())
1808 self.assertFalse(P.is_file())
1809
1810 def test_is_block_device_false(self):
1811 P = self.cls(BASE)
1812 self.assertFalse((P / 'fileA').is_block_device())
1813 self.assertFalse((P / 'dirA').is_block_device())
1814 self.assertFalse((P / 'non-existing').is_block_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001815 self.assertFalse((P / 'fileA' / 'bah').is_block_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001816
1817 def test_is_char_device_false(self):
1818 P = self.cls(BASE)
1819 self.assertFalse((P / 'fileA').is_char_device())
1820 self.assertFalse((P / 'dirA').is_char_device())
1821 self.assertFalse((P / 'non-existing').is_char_device())
Antoine Pitrou2b2852b2014-10-30 23:14:03 +01001822 self.assertFalse((P / 'fileA' / 'bah').is_char_device())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001823
1824 def test_is_char_device_true(self):
1825 # Under Unix, /dev/null should generally be a char device
1826 P = self.cls('/dev/null')
1827 if not P.exists():
1828 self.skipTest("/dev/null required")
1829 self.assertTrue(P.is_char_device())
1830 self.assertFalse(P.is_block_device())
1831 self.assertFalse(P.is_file())
1832
1833 def test_pickling_common(self):
1834 p = self.cls(BASE, 'fileA')
1835 for proto in range(0, pickle.HIGHEST_PROTOCOL + 1):
1836 dumped = pickle.dumps(p, proto)
1837 pp = pickle.loads(dumped)
1838 self.assertEqual(pp.stat(), p.stat())
1839
1840 def test_parts_interning(self):
1841 P = self.cls
1842 p = P('/usr/bin/foo')
1843 q = P('/usr/local/bin')
1844 # 'usr'
1845 self.assertIs(p.parts[1], q.parts[1])
1846 # 'bin'
1847 self.assertIs(p.parts[2], q.parts[3])
1848
Antoine Pitrouc274fd22013-12-16 19:57:41 +01001849 def _check_complex_symlinks(self, link0_target):
1850 # Test solving a non-looping chain of symlinks (issue #19887)
1851 P = self.cls(BASE)
1852 self.dirlink(os.path.join('link0', 'link0'), join('link1'))
1853 self.dirlink(os.path.join('link1', 'link1'), join('link2'))
1854 self.dirlink(os.path.join('link2', 'link2'), join('link3'))
1855 self.dirlink(link0_target, join('link0'))
1856
1857 # Resolve absolute paths
1858 p = (P / 'link0').resolve()
1859 self.assertEqual(p, P)
1860 self.assertEqual(str(p), BASE)
1861 p = (P / 'link1').resolve()
1862 self.assertEqual(p, P)
1863 self.assertEqual(str(p), BASE)
1864 p = (P / 'link2').resolve()
1865 self.assertEqual(p, P)
1866 self.assertEqual(str(p), BASE)
1867 p = (P / 'link3').resolve()
1868 self.assertEqual(p, P)
1869 self.assertEqual(str(p), BASE)
1870
1871 # Resolve relative paths
1872 old_path = os.getcwd()
1873 os.chdir(BASE)
1874 try:
1875 p = self.cls('link0').resolve()
1876 self.assertEqual(p, P)
1877 self.assertEqual(str(p), BASE)
1878 p = self.cls('link1').resolve()
1879 self.assertEqual(p, P)
1880 self.assertEqual(str(p), BASE)
1881 p = self.cls('link2').resolve()
1882 self.assertEqual(p, P)
1883 self.assertEqual(str(p), BASE)
1884 p = self.cls('link3').resolve()
1885 self.assertEqual(p, P)
1886 self.assertEqual(str(p), BASE)
1887 finally:
1888 os.chdir(old_path)
1889
1890 @with_symlinks
1891 def test_complex_symlinks_absolute(self):
1892 self._check_complex_symlinks(BASE)
1893
1894 @with_symlinks
1895 def test_complex_symlinks_relative(self):
1896 self._check_complex_symlinks('.')
1897
1898 @with_symlinks
1899 def test_complex_symlinks_relative_dot_dot(self):
1900 self._check_complex_symlinks(os.path.join('dirA', '..'))
1901
Antoine Pitrou31119e42013-11-22 17:38:12 +01001902
1903class PathTest(_BasePathTest, unittest.TestCase):
1904 cls = pathlib.Path
1905
1906 def test_concrete_class(self):
1907 p = self.cls('a')
1908 self.assertIs(type(p),
1909 pathlib.WindowsPath if os.name == 'nt' else pathlib.PosixPath)
1910
1911 def test_unsupported_flavour(self):
1912 if os.name == 'nt':
1913 self.assertRaises(NotImplementedError, pathlib.PosixPath)
1914 else:
1915 self.assertRaises(NotImplementedError, pathlib.WindowsPath)
1916
1917
1918@only_posix
1919class PosixPathTest(_BasePathTest, unittest.TestCase):
1920 cls = pathlib.PosixPath
1921
1922 def _check_symlink_loop(self, *args):
1923 path = self.cls(*args)
1924 with self.assertRaises(RuntimeError):
1925 print(path.resolve())
1926
1927 def test_open_mode(self):
1928 old_mask = os.umask(0)
1929 self.addCleanup(os.umask, old_mask)
1930 p = self.cls(BASE)
1931 with (p / 'new_file').open('wb'):
1932 pass
1933 st = os.stat(join('new_file'))
1934 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
1935 os.umask(0o022)
1936 with (p / 'other_new_file').open('wb'):
1937 pass
1938 st = os.stat(join('other_new_file'))
1939 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
1940
1941 def test_touch_mode(self):
1942 old_mask = os.umask(0)
1943 self.addCleanup(os.umask, old_mask)
1944 p = self.cls(BASE)
1945 (p / 'new_file').touch()
1946 st = os.stat(join('new_file'))
1947 self.assertEqual(stat.S_IMODE(st.st_mode), 0o666)
1948 os.umask(0o022)
1949 (p / 'other_new_file').touch()
1950 st = os.stat(join('other_new_file'))
1951 self.assertEqual(stat.S_IMODE(st.st_mode), 0o644)
1952 (p / 'masked_new_file').touch(mode=0o750)
1953 st = os.stat(join('masked_new_file'))
1954 self.assertEqual(stat.S_IMODE(st.st_mode), 0o750)
1955
1956 @with_symlinks
1957 def test_resolve_loop(self):
1958 # Loop detection for broken symlinks under POSIX
Antoine Pitrou31119e42013-11-22 17:38:12 +01001959 # Loops with relative symlinks
1960 os.symlink('linkX/inside', join('linkX'))
1961 self._check_symlink_loop(BASE, 'linkX')
1962 os.symlink('linkY', join('linkY'))
1963 self._check_symlink_loop(BASE, 'linkY')
1964 os.symlink('linkZ/../linkZ', join('linkZ'))
1965 self._check_symlink_loop(BASE, 'linkZ')
1966 # Loops with absolute symlinks
1967 os.symlink(join('linkU/inside'), join('linkU'))
1968 self._check_symlink_loop(BASE, 'linkU')
1969 os.symlink(join('linkV'), join('linkV'))
1970 self._check_symlink_loop(BASE, 'linkV')
1971 os.symlink(join('linkW/../linkW'), join('linkW'))
1972 self._check_symlink_loop(BASE, 'linkW')
1973
1974 def test_glob(self):
1975 P = self.cls
1976 p = P(BASE)
Brett Cannonfe77f4e2013-11-22 16:14:10 -05001977 given = set(p.glob("FILEa"))
1978 expect = set() if not support.fs_is_case_insensitive(BASE) else given
1979 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01001980 self.assertEqual(set(p.glob("FILEa*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001981
1982 def test_rglob(self):
1983 P = self.cls
1984 p = P(BASE, "dirC")
Brett Cannonfe77f4e2013-11-22 16:14:10 -05001985 given = set(p.rglob("FILEd"))
1986 expect = set() if not support.fs_is_case_insensitive(BASE) else given
1987 self.assertEqual(given, expect)
Antoine Pitrou2dd38fb2013-11-22 22:26:01 +01001988 self.assertEqual(set(p.rglob("FILEd*")), set())
Antoine Pitrou31119e42013-11-22 17:38:12 +01001989
Antoine Pitrou8477ed62014-12-30 20:54:45 +01001990 def test_expanduser(self):
1991 P = self.cls
1992 support.import_module('pwd')
1993 import pwd
1994 pwdent = pwd.getpwuid(os.getuid())
1995 username = pwdent.pw_name
1996 userhome = pwdent.pw_dir.rstrip('/')
1997 # find arbitrary different user (if exists)
1998 for pwdent in pwd.getpwall():
1999 othername = pwdent.pw_name
2000 otherhome = pwdent.pw_dir.rstrip('/')
Victor Stinnere9694392015-01-10 09:00:20 +01002001 if othername != username and otherhome:
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002002 break
2003
2004 p1 = P('~/Documents')
2005 p2 = P('~' + username + '/Documents')
2006 p3 = P('~' + othername + '/Documents')
2007 p4 = P('../~' + username + '/Documents')
2008 p5 = P('/~' + username + '/Documents')
2009 p6 = P('')
2010 p7 = P('~fakeuser/Documents')
2011
2012 with support.EnvironmentVarGuard() as env:
2013 env.pop('HOME', None)
2014
2015 self.assertEqual(p1.expanduser(), P(userhome) / 'Documents')
2016 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2017 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2018 self.assertEqual(p4.expanduser(), p4)
2019 self.assertEqual(p5.expanduser(), p5)
2020 self.assertEqual(p6.expanduser(), p6)
2021 self.assertRaises(RuntimeError, p7.expanduser)
2022
2023 env['HOME'] = '/tmp'
2024 self.assertEqual(p1.expanduser(), P('/tmp/Documents'))
2025 self.assertEqual(p2.expanduser(), P(userhome) / 'Documents')
2026 self.assertEqual(p3.expanduser(), P(otherhome) / 'Documents')
2027 self.assertEqual(p4.expanduser(), p4)
2028 self.assertEqual(p5.expanduser(), p5)
2029 self.assertEqual(p6.expanduser(), p6)
2030 self.assertRaises(RuntimeError, p7.expanduser)
2031
Antoine Pitrou31119e42013-11-22 17:38:12 +01002032
2033@only_nt
2034class WindowsPathTest(_BasePathTest, unittest.TestCase):
2035 cls = pathlib.WindowsPath
2036
2037 def test_glob(self):
2038 P = self.cls
2039 p = P(BASE)
2040 self.assertEqual(set(p.glob("FILEa")), { P(BASE, "fileA") })
2041
2042 def test_rglob(self):
2043 P = self.cls
2044 p = P(BASE, "dirC")
2045 self.assertEqual(set(p.rglob("FILEd")), { P(BASE, "dirC/dirD/fileD") })
2046
Antoine Pitrou8477ed62014-12-30 20:54:45 +01002047 def test_expanduser(self):
2048 P = self.cls
2049 with support.EnvironmentVarGuard() as env:
2050 env.pop('HOME', None)
2051 env.pop('USERPROFILE', None)
2052 env.pop('HOMEPATH', None)
2053 env.pop('HOMEDRIVE', None)
2054 env['USERNAME'] = 'alice'
2055
2056 # test that the path returns unchanged
2057 p1 = P('~/My Documents')
2058 p2 = P('~alice/My Documents')
2059 p3 = P('~bob/My Documents')
2060 p4 = P('/~/My Documents')
2061 p5 = P('d:~/My Documents')
2062 p6 = P('')
2063 self.assertRaises(RuntimeError, p1.expanduser)
2064 self.assertRaises(RuntimeError, p2.expanduser)
2065 self.assertRaises(RuntimeError, p3.expanduser)
2066 self.assertEqual(p4.expanduser(), p4)
2067 self.assertEqual(p5.expanduser(), p5)
2068 self.assertEqual(p6.expanduser(), p6)
2069
2070 def check():
2071 env.pop('USERNAME', None)
2072 self.assertEqual(p1.expanduser(),
2073 P('C:/Users/alice/My Documents'))
2074 self.assertRaises(KeyError, p2.expanduser)
2075 env['USERNAME'] = 'alice'
2076 self.assertEqual(p2.expanduser(),
2077 P('C:/Users/alice/My Documents'))
2078 self.assertEqual(p3.expanduser(),
2079 P('C:/Users/bob/My Documents'))
2080 self.assertEqual(p4.expanduser(), p4)
2081 self.assertEqual(p5.expanduser(), p5)
2082 self.assertEqual(p6.expanduser(), p6)
2083
2084 # test the first lookup key in the env vars
2085 env['HOME'] = 'C:\\Users\\alice'
2086 check()
2087
2088 # test that HOMEPATH is available instead
2089 env.pop('HOME', None)
2090 env['HOMEPATH'] = 'C:\\Users\\alice'
2091 check()
2092
2093 env['HOMEDRIVE'] = 'C:\\'
2094 env['HOMEPATH'] = 'Users\\alice'
2095 check()
2096
2097 env.pop('HOMEDRIVE', None)
2098 env.pop('HOMEPATH', None)
2099 env['USERPROFILE'] = 'C:\\Users\\alice'
2100 check()
2101
Antoine Pitrou31119e42013-11-22 17:38:12 +01002102
2103if __name__ == "__main__":
2104 unittest.main()