tests/extmod: Add littlefs tests.

1 files changed, 117 insertions, 0 deletions

diff --git a/tests/extmod/vfs_lfs_file.py b/tests/extmod/vfs_lfs_file.py
new file mode 100644
index 000000000..477a62e2f
--- /dev/null
+++ b/tests/extmod/vfs_lfs_file.py
@@ -0,0 +1,117 @@
+# Test for VfsLittle using a RAM device, file IO
+
+try:
+    import uos
+    uos.VfsLfs1
+    uos.VfsLfs2
+except (ImportError, AttributeError):
+    print("SKIP")
+    raise SystemExit
+
+class RAMBlockDevice:
+    ERASE_BLOCK_SIZE = 1024
+
+    def __init__(self, blocks):
+        self.data = bytearray(blocks * self.ERASE_BLOCK_SIZE)
+
+    def readblocks(self, block, buf, off):
+        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):
+        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(bdev, vfs_class):
+    print('test', vfs_class)
+
+    # mkfs
+    vfs_class.mkfs(bdev)
+
+    # construction
+    vfs = vfs_class(bdev)
+
+    # create text, print, write, close
+    f = vfs.open('test.txt', 'wt')
+    print(f)
+    f.write('littlefs')
+    f.close()
+
+    # close already-closed file
+    f.close()
+
+    # create binary, print, write, flush, close
+    f = vfs.open('test.bin', 'wb')
+    print(f)
+    f.write('littlefs')
+    f.flush()
+    f.close()
+
+    # create for append
+    f = vfs.open('test.bin', 'ab')
+    f.write('more')
+    f.close()
+
+    # create exclusive
+    f = vfs.open('test2.bin', 'xb')
+    f.close()
+
+    # create exclusive with error
+    try:
+        vfs.open('test2.bin', 'x')
+    except OSError:
+        print('open OSError')
+
+    # read default
+    with vfs.open('test.txt', '') as f:
+        print(f.read())
+
+    # read text
+    with vfs.open('test.txt', 'rt') as f:
+        print(f.read())
+
+    # read binary
+    with vfs.open('test.bin', 'rb') as f:
+        print(f.read())
+
+    # create read and write
+    with vfs.open('test.bin', 'r+b') as f:
+        print(f.read(8))
+        f.write('MORE')
+    with vfs.open('test.bin', 'rb') as f:
+        print(f.read())
+
+    # seek and tell
+    f = vfs.open('test.txt', 'r')
+    print(f.tell())
+    f.seek(3, 0)
+    print(f.tell())
+    f.close()
+
+    # open nonexistent
+    try:
+        vfs.open('noexist', 'r')
+    except OSError:
+        print('open OSError')
+
+    # open multiple files at the same time
+    f1 = vfs.open('test.txt', '')
+    f2 = vfs.open('test.bin', 'b')
+    print(f1.read())
+    print(f2.read())
+    f1.close()
+    f2.close()
+
+bdev = RAMBlockDevice(30)
+test(bdev, uos.VfsLfs1)
+test(bdev, uos.VfsLfs2)