def shared_tempdir():
d = TemporaryDirectory()
yield d.name
try:
d.cleanup()
except Exception:
pass
def with_directory_contents(contents, func):
tempd = TemporaryDirectory(prefix="test-")
dirname = tempd.name
try:
for filename, file_content in contents.items():
path = os.path.join(dirname, filename)
if file_content is None:
# make a directory
makedirs_ok_if_exists(path)
else:
makedirs_ok_if_exists(os.path.dirname(path))
with codecs.open(path, 'w', 'utf-8') as f:
f.write(file_content)
result = func(os.path.realpath(dirname))
finally:
# Windows experiences PermissionError exceptions here,
# and Unix sometimes experiences FileNotFound exceptions.
# The reasons are not 100% clear, but they should not be
# allowed to interrupt test passage, either.
try:
tempd.cleanup()
except Exception as exc:
print('Unexpected error cleaning temporary directory:')
print(' ' + dirname)
print(' ' + str(exc))
pass
return result
class TestEglob(object):
def __init__(self):
self.files = None
self.dirs = None
self.tempdir = None
def setup(self):
self.tempdir = TemporaryDirectory()
tempname = self.tempdir.name
foo = path.join(tempname, "foo")
bar = path.join(tempname, "bar")
baz = path.join(tempname, "baz")
quux = path.join(tempname, "quux")
foo_sub = path.join(foo, "foosub1")
bar_sub = path.join(bar, "barsub1")
foo_sub_sub = path.join(foo_sub, "hellothere")
self.dirs = (foo, bar, baz, quux, foo_sub, bar_sub, foo_sub_sub)
self.files = (
path.join(foo, "file1.py"),
path.join(bar, "file2.py"),
path.join(foo_sub, "file3.py"),
path.join(foo_sub_sub, "file4.py"),
path.join(quux, "readme.txt"),
path.join(baz, "license.txt"),
path.join(foo_sub_sub, "readme.txt"),
path.join(bar_sub, "changelog.md"),
path.join(tempname, "foo.py"),
path.join(tempname, "bar.py"),
path.join(tempname, "readme"),
path.join(tempname, "license.txt"),
)
self.py_files = {f for f in self.files if f.endswith(".py")}
self.readmes = {f for f in self.files if f.endswith("readme.txt")}
self.txt_files = {f for f in self.files if f.endswith(".txt")}
self.top_level_files = {
f
for f in self.files if path.dirname(f) == tempname
}
for d in self.dirs:
os.makedirs(d)
for f in self.files:
touch(f)
def teardown(self):
self.tempdir.cleanup()
def testReadMes(self):
files = set(eglob("**/readme.txt", self.tempdir.name))
assert files == self.readmes
def testPyFiles(self):
files = set(eglob("**/*.py", self.tempdir.name))
assert files == self.py_files
def testTextFiles(self):
files = set(eglob("**/*.txt", self.tempdir.name))
assert files == self.txt_files
def testTopLevelFiles(self):
top = set(eglob("*", self.tempdir.name))
files = {f for f in top if os.path.isfile(f)}
assert files == self.top_level_files