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