1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
import sys, os
import py
from rpython.tool.udir import udir
from pypy.module.posix.test import test_posix2
def _make_dir(dirname, content):
d = os.path.join(str(udir), dirname)
os.mkdir(d)
for key, value in content.items():
filename = os.path.join(d, key)
if value == 'dir':
os.mkdir(filename)
elif value == 'file':
with open(filename, 'w') as f:
pass
elif value == 'symlink-file':
os.symlink(str(udir.ensure('some_file')), filename)
elif value == 'symlink-dir':
os.symlink(str(udir), filename)
elif value == 'symlink-broken':
os.symlink(filename + '-broken', filename)
elif value == 'symlink-error':
os.symlink(filename, filename)
else:
raise NotImplementedError(repr(value))
return d.decode(sys.getfilesystemencoding())
class AppTestScandir(object):
spaceconfig = {'usemodules': test_posix2.USEMODULES}
def setup_class(cls):
space = cls.space
cls.w_WIN32 = space.wrap(sys.platform == 'win32')
cls.w_sep = space.wrap(os.sep)
cls.w_posix = space.appexec([], test_posix2.GET_POSIX)
cls.w_dir_empty = space.wrap(_make_dir('empty', {}))
cls.w_dir0 = space.wrap(_make_dir('dir0', {'f1': 'file',
'f2': 'file',
'f3': 'file'}))
cls.w_dir1 = space.wrap(_make_dir('dir1', {'file1': 'file'}))
cls.w_dir2 = space.wrap(_make_dir('dir2', {'subdir2': 'dir'}))
if sys.platform != 'win32':
cls.w_dir3 = space.wrap(_make_dir('dir3', {'sfile3': 'symlink-file'}))
cls.w_dir4 = space.wrap(_make_dir('dir4', {'sdir4': 'symlink-dir'}))
cls.w_dir5 = space.wrap(_make_dir('dir5', {'sbrok5': 'symlink-broken'}))
cls.w_dir6 = space.wrap(_make_dir('dir6', {'serr6': 'symlink-error'}))
def test_scandir_empty(self):
posix = self.posix
sd = posix.scandir(self.dir_empty)
assert list(sd) == []
assert list(sd) == []
def test_scandir_files(self):
posix = self.posix
sd = posix.scandir(self.dir0)
names = [d.name for d in sd]
assert sorted(names) == ['f1', 'f2', 'f3']
def test_unicode_versus_bytes(self):
posix = self.posix
d = next(posix.scandir())
assert type(d.name) is str
assert type(d.path) is str
assert d.path == '.' + self.sep + d.name
d = next(posix.scandir(None))
assert type(d.name) is str
assert type(d.path) is str
assert d.path == '.' + self.sep + d.name
d = next(posix.scandir(u'.'))
assert type(d.name) is str
assert type(d.path) is str
assert d.path == '.' + self.sep + d.name
d = next(posix.scandir(self.sep))
assert type(d.name) is str
assert type(d.path) is str
assert d.path == self.sep + d.name
if not self.WIN32:
d = next(posix.scandir(b'.'))
assert type(d.name) is bytes
assert type(d.path) is bytes
assert d.path == b'./' + d.name
d = next(posix.scandir(b'/'))
assert type(d.name) is bytes
assert type(d.path) is bytes
assert d.path == b'/' + d.name
else:
raises(TypeError, posix.scandir, b'.')
raises(TypeError, posix.scandir, b'/')
raises(TypeError, posix.scandir, b'\\')
def test_stat1(self):
posix = self.posix
d = next(posix.scandir(self.dir1))
assert d.name == 'file1'
assert d.stat().st_mode & 0o170000 == 0o100000 # S_IFREG
assert d.stat().st_size == 0
@py.test.mark.skipif(sys.platform == "win32", reason="no symlink support so far")
def test_stat4(self):
posix = self.posix
d = next(posix.scandir(self.dir4))
assert d.name == 'sdir4'
assert d.stat().st_mode & 0o170000 == 0o040000 # S_IFDIR
assert d.stat(follow_symlinks=True).st_mode &0o170000 == 0o040000
assert d.stat(follow_symlinks=False).st_mode&0o170000 == 0o120000 #IFLNK
def test_dir1(self):
posix = self.posix
d = next(posix.scandir(self.dir1))
assert d.name == 'file1'
assert d.is_file()
assert not d.is_dir()
assert not d.is_symlink()
raises(TypeError, d.is_file, True)
assert d.is_file(follow_symlinks=False)
assert not d.is_dir(follow_symlinks=False)
def test_dir2(self):
posix = self.posix
d = next(posix.scandir(self.dir2))
assert d.name == 'subdir2'
assert not d.is_file()
assert d.is_dir()
assert not d.is_symlink()
assert not d.is_file(follow_symlinks=False)
assert d.is_dir(follow_symlinks=False)
@py.test.mark.skipif(sys.platform == "win32", reason="no symlink support so far")
def test_dir3(self):
posix = self.posix
d = next(posix.scandir(self.dir3))
assert d.name == 'sfile3'
assert d.is_file()
assert not d.is_dir()
assert d.is_symlink()
assert d.is_file(follow_symlinks=True)
assert not d.is_file(follow_symlinks=False)
@py.test.mark.skipif(sys.platform == "win32", reason="no symlink support so far")
def test_dir4(self):
posix = self.posix
d = next(posix.scandir(self.dir4))
assert d.name == 'sdir4'
assert not d.is_file()
assert d.is_dir()
assert d.is_symlink()
assert d.is_dir(follow_symlinks=True)
assert not d.is_dir(follow_symlinks=False)
@py.test.mark.skipif(sys.platform == "win32", reason="no symlink support so far")
def test_dir5(self):
posix = self.posix
d = next(posix.scandir(self.dir5))
assert d.name == 'sbrok5'
assert not d.is_file()
assert not d.is_dir()
assert d.is_symlink()
raises(OSError, d.stat)
@py.test.mark.skipif(sys.platform == "win32", reason="no symlink support so far")
def test_dir6(self):
posix = self.posix
d = next(posix.scandir(self.dir6))
assert d.name == 'serr6'
raises(OSError, d.is_file)
raises(OSError, d.is_dir)
assert d.is_symlink()
def test_fdopendir_unsupported(self):
posix = self.posix
raises(TypeError, posix.scandir, 1234)
def test_inode(self):
posix = self.posix
d = next(posix.scandir(self.dir1))
assert d.name == 'file1'
ino = d.inode()
assert ino == d.stat().st_ino
def test_repr(self):
posix = self.posix
d = next(posix.scandir(self.dir1))
assert repr(d) == "<DirEntry 'file1'>"
|
