circuitpython/tests/extmod/vfs_lfs_mount.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

115 lines
2.4 KiB
Python
Raw Normal View History

2019-10-18 02:25:48 -04:00
# Test for VfsLittle using a RAM device, with mount/umount
try:
2023-08-22 11:15:46 -04:00
import os
2023-08-22 11:15:46 -04:00
os.VfsLfs1
os.VfsLfs2
2019-10-18 02:25:48 -04:00
except (ImportError, AttributeError):
print("SKIP")
raise SystemExit
2019-10-18 02:25:48 -04:00
class RAMBlockDevice:
ERASE_BLOCK_SIZE = 1024
def __init__(self, blocks):
self.data = bytearray(blocks * self.ERASE_BLOCK_SIZE)
def readblocks(self, block, buf, off=0):
2019-10-18 02:25:48 -04:00
addr = block * self.ERASE_BLOCK_SIZE + off
for i in range(len(buf)):
buf[i] = self.data[addr + i]
def writeblocks(self, block, buf, off=0):
2019-10-18 02:25:48 -04:00
addr = block * self.ERASE_BLOCK_SIZE + off
for i in range(len(buf)):
self.data[addr + i] = buf[i]
def ioctl(self, op, arg):
if op == 4: # block count
return len(self.data) // self.ERASE_BLOCK_SIZE
if op == 5: # block size
return self.ERASE_BLOCK_SIZE
if op == 6: # erase block
return 0
def test(vfs_class):
2019-10-18 02:25:48 -04:00
print("test", vfs_class)
bdev = RAMBlockDevice(30)
# mount bdev unformatted
try:
2023-08-22 11:15:46 -04:00
os.mount(bdev, "/lfs")
except Exception as er:
print(repr(er))
2019-10-18 02:25:48 -04:00
# mkfs
vfs_class.mkfs(bdev)
# construction
vfs = vfs_class(bdev)
# mount
2023-08-22 11:15:46 -04:00
os.mount(vfs, "/lfs")
2019-10-18 02:25:48 -04:00
# import
with open("/lfs/lfsmod.py", "w") as f:
f.write('print("hello from lfs")\n')
import lfsmod
# import package
2023-08-22 11:15:46 -04:00
os.mkdir("/lfs/lfspkg")
2019-10-18 02:25:48 -04:00
with open("/lfs/lfspkg/__init__.py", "w") as f:
f.write('print("package")\n')
import lfspkg
# chdir and import module from current directory (needs "" in sys.path)
2023-08-22 11:15:46 -04:00
os.mkdir("/lfs/subdir")
os.chdir("/lfs/subdir")
os.rename("/lfs/lfsmod.py", "/lfs/subdir/lfsmod2.py")
import lfsmod2
2019-10-18 02:25:48 -04:00
# umount
2023-08-22 11:15:46 -04:00
os.umount("/lfs")
2019-10-18 02:25:48 -04:00
# mount read-only
vfs = vfs_class(bdev)
2023-08-22 11:15:46 -04:00
os.mount(vfs, "/lfs", readonly=True)
# test reading works
with open("/lfs/subdir/lfsmod2.py") as f:
print("lfsmod2.py:", f.read())
# test writing fails
try:
open("/lfs/test_write", "w")
except OSError as er:
print(repr(er))
# umount
2023-08-22 11:15:46 -04:00
os.umount("/lfs")
# mount bdev again
2023-08-22 11:15:46 -04:00
os.mount(bdev, "/lfs")
# umount
2023-08-22 11:15:46 -04:00
os.umount("/lfs")
2019-10-18 02:25:48 -04:00
# clear imported modules
sys.modules.clear()
2019-10-18 02:25:48 -04:00
# initialise path
import sys
2019-10-18 02:25:48 -04:00
sys.path.clear()
sys.path.append("/lfs")
sys.path.append("")
2019-10-18 02:25:48 -04:00
# run tests
2023-08-22 11:15:46 -04:00
test(os.VfsLfs1)
test(os.VfsLfs2)