80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
# Test ilistdir __del__ for VfsPosix.
|
|
import gc
|
|
|
|
try:
|
|
import os
|
|
|
|
os.VfsPosix
|
|
except (ImportError, AttributeError):
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
|
|
def test(testdir):
|
|
curdir = os.getcwd()
|
|
vfs = os.VfsPosix(testdir)
|
|
# When VfsPosix is used the intended way via os.mount(), it can only be called
|
|
# with relative paths when the CWD is inside or at its root, so simulate that.
|
|
# (Although perhaps calling with a relative path was an oversight in this case
|
|
# and the respective line below was meant to read `vfs.rmdir("/" + dname)`.)
|
|
os.chdir(testdir)
|
|
vfs.mkdir("/test_d1")
|
|
vfs.mkdir("/test_d2")
|
|
vfs.mkdir("/test_d3")
|
|
|
|
for i in range(10):
|
|
print(i)
|
|
|
|
# We want to partially iterate the ilistdir iterator to leave it in an
|
|
# open state, which will then test the finaliser when it's garbage collected.
|
|
idir = vfs.ilistdir("/")
|
|
print(any(idir))
|
|
|
|
# Alternate way of partially iterating the ilistdir object, modifying the
|
|
# filesystem while it's open.
|
|
for dname, *_ in vfs.ilistdir("/"):
|
|
vfs.rmdir(dname)
|
|
break
|
|
vfs.mkdir(dname)
|
|
|
|
# Also create a fully drained iterator and ensure trying to re-use it
|
|
# throws the correct exception.
|
|
idir_emptied = vfs.ilistdir("/")
|
|
l = list(idir_emptied)
|
|
print(len(l))
|
|
try:
|
|
next(idir_emptied)
|
|
except StopIteration:
|
|
pass
|
|
|
|
gc.collect()
|
|
|
|
# Create and delete a file, try to flush out any filesystem
|
|
# corruption that may be caused over the loops.
|
|
vfs.open("/test", "w").close()
|
|
vfs.remove("/test")
|
|
|
|
# Done with vfs, restore CWD.
|
|
os.chdir(curdir)
|
|
|
|
|
|
# We need an empty directory for testing.
|
|
# Skip the test if it already exists.
|
|
temp_dir = "vfs_posix_ilistdir_del_test_dir"
|
|
try:
|
|
os.stat(temp_dir)
|
|
print("SKIP")
|
|
raise SystemExit
|
|
except OSError:
|
|
pass
|
|
|
|
os.mkdir(temp_dir)
|
|
|
|
test(temp_dir)
|
|
|
|
# Remove tempdir.
|
|
for td in os.listdir(temp_dir):
|
|
os.rmdir("/".join((temp_dir, td)))
|
|
|
|
os.rmdir(temp_dir)
|