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