Skip to content

Commit 02e8217

Browse files
committed
pythonGH-130614: Add test suites for readable and writable pathlib paths
Add `test.test_pathlib.test_read`, which tests `ReadablePath` subclasses. Tests are run against `ReadableZipPath`, `ReadableLocalPath`, and `Path`. This test suite is mostly incomplete. Add `test.test_pathlib.test_write`, which tests `WritablePath` subclasses. Tests are run against `WritableZipPath`, `WritableLocalPath`, and `Path`. This test suite is mostly complete. Add `test.test_pathlib.support.zip_path`, which provides the `*ZipPath` classes mentioned above. It also provides a `ZipPathGround` class that's used by tests to make assertions about the contents of the zip file. Add `test.test_pathlib.support.local_path`, which provides the `*LocalPath` classes mentioned above. They're stripped-back versions of `pathlib.Path`, basically. It also provides `LocalPathGround` class that's used by tests to access the local filesystem.
1 parent fda056e commit 02e8217

File tree

6 files changed

+601
-52
lines changed

6 files changed

+601
-52
lines changed

Lib/test/test_pathlib/support/__init__.py

Whitespace-only changes.
+181
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,181 @@
1+
import os
2+
import pathlib._abc
3+
import pathlib.types
4+
import posixpath
5+
6+
from test.support import os_helper
7+
8+
9+
class LocalPathInfo(pathlib.types.PathInfo):
10+
"""
11+
Simple implementation of PathInfo for a local path
12+
"""
13+
def __init__(self, path):
14+
self._path = str(path)
15+
self._exists = None
16+
self._is_dir = None
17+
self._is_file = None
18+
self._is_symlink = None
19+
20+
def exists(self, *, follow_symlinks=True):
21+
"""Whether this path exists."""
22+
if not follow_symlinks and self.is_symlink():
23+
return True
24+
if self._exists is None:
25+
self._exists = os.path.exists(self._path)
26+
return self._exists
27+
28+
def is_dir(self, *, follow_symlinks=True):
29+
"""Whether this path is a directory."""
30+
if not follow_symlinks and self.is_symlink():
31+
return False
32+
if self._is_dir is None:
33+
self._is_dir = os.path.isdir(self._path)
34+
return self._is_dir
35+
36+
def is_file(self, *, follow_symlinks=True):
37+
"""Whether this path is a regular file."""
38+
if not follow_symlinks and self.is_symlink():
39+
return False
40+
if self._is_file is None:
41+
self._is_file = os.path.isfile(self._path)
42+
return self._is_file
43+
44+
def is_symlink(self):
45+
"""Whether this path is a symbolic link."""
46+
if self._is_symlink is None:
47+
self._is_symlink = os.path.islink(self._path)
48+
return self._is_symlink
49+
50+
51+
class ReadableLocalPath(pathlib._abc.ReadablePath):
52+
"""
53+
Simple implementation of a ReadablePath class for local filesystem paths.
54+
"""
55+
parser = os.path
56+
57+
def __init__(self, *pathsegments):
58+
self._segments = pathsegments
59+
self._info = None
60+
61+
def __str__(self):
62+
if not self._segments:
63+
return ''
64+
return self.parser.join(*self._segments)
65+
66+
def __fspath__(self):
67+
return str(self)
68+
69+
def with_segments(self, *pathsegments):
70+
return type(self)(*pathsegments)
71+
72+
def __open_rb__(self, buffering=-1):
73+
return open(self, 'rb')
74+
75+
def iterdir(self):
76+
return (self / name for name in os.listdir(self))
77+
78+
def readlink(self):
79+
return self.with_segments(os.readlink(self))
80+
81+
@property
82+
def info(self):
83+
if self._info is None:
84+
self._info = LocalPathInfo(self)
85+
return self._info
86+
87+
88+
class WritableLocalPath(pathlib._abc.WritablePath):
89+
"""
90+
Simple implementation of a WritablePath class for local filesystem paths.
91+
"""
92+
93+
__slots__ = ('_segments',)
94+
parser = posixpath
95+
96+
def __init__(self, *pathsegments):
97+
self._segments = pathsegments
98+
99+
def __str__(self):
100+
if not self._segments:
101+
return ''
102+
return self.parser.join(*self._segments)
103+
104+
def __fspath__(self):
105+
return str(self)
106+
107+
def with_segments(self, *pathsegments):
108+
return type(self)(*pathsegments)
109+
110+
def __open_wb__(self, buffering=-1):
111+
return open(self, 'wb')
112+
113+
def mkdir(self, mode=0o777):
114+
os.mkdir(self, mode)
115+
116+
def symlink_to(self, target, target_is_directory=False):
117+
os.symlink(target, self, target_is_directory)
118+
119+
120+
class LocalPathGround:
121+
can_symlink = os_helper.can_symlink()
122+
123+
def __init__(self, path_cls):
124+
self.path_cls = path_cls
125+
126+
def setup(self):
127+
root = self.path_cls(os_helper.TESTFN)
128+
os.mkdir(root)
129+
return root
130+
131+
def teardown(self, root):
132+
os_helper.rmtree(root)
133+
134+
def create_file(self, p, data=b''):
135+
with open(p, 'wb') as f:
136+
f.write(data)
137+
138+
def create_hierarchy(self, p):
139+
os.mkdir(os.path.join(p, 'dirA'))
140+
os.mkdir(os.path.join(p, 'dirB'))
141+
os.mkdir(os.path.join(p, 'dirC'))
142+
os.mkdir(os.path.join(p, 'dirC', 'dirD'))
143+
with open(os.path.join(p, 'fileA'), 'wb') as f:
144+
f.write(b"this is file A\n")
145+
with open(os.path.join(p, 'dirB', 'fileB'), 'wb') as f:
146+
f.write(b"this is file B\n")
147+
with open(os.path.join(p, 'dirC', 'fileC'), 'wb') as f:
148+
f.write(b"this is file C\n")
149+
with open(os.path.join(p, 'dirC', 'novel.txt'), 'wb') as f:
150+
f.write(b"this is a novel\n")
151+
with open(os.path.join(p, 'dirC', 'dirD', 'fileD'), 'wb') as f:
152+
f.write(b"this is file D\n")
153+
if self.can_symlink:
154+
# Relative symlinks.
155+
os.symlink('fileA', os.path.join(p, 'linkA'))
156+
os.symlink('non-existing', os.path.join(p, 'brokenLink'))
157+
os.symlink('dirB',
158+
os.path.join(p, 'linkB'),
159+
target_is_directory=True)
160+
os.symlink(os.path.join('..', 'dirB'),
161+
os.path.join(p, 'dirA', 'linkC'),
162+
target_is_directory=True)
163+
# This one goes upwards, creating a loop.
164+
os.symlink(os.path.join('..', 'dirB'),
165+
os.path.join(p, 'dirB', 'linkD'),
166+
target_is_directory=True)
167+
# Broken symlink (pointing to itself).
168+
os.symlink('brokenLinkLoop', os.path.join(p, 'brokenLinkLoop'))
169+
170+
isdir = staticmethod(os.path.isdir)
171+
isfile = staticmethod(os.path.isfile)
172+
islink = staticmethod(os.path.islink)
173+
readlink = staticmethod(os.readlink)
174+
175+
def readtext(self, p):
176+
with open(p, 'r') as f:
177+
return f.read()
178+
179+
def readbytes(self, p):
180+
with open(p, 'rb') as f:
181+
return f.read()

0 commit comments

Comments
 (0)