From 932556d5fc09a196139170fe52436ea8a02020c1 Mon Sep 17 00:00:00 2001
From: Damien George <damien@micropython.org>
Date: Wed, 29 Jun 2022 12:48:42 +1000
Subject: [PATCH] tests/micropython: Add test for builtin execfile() function.

Signed-off-by: Damien George <damien@micropython.org>
---
 tests/micropython/builtin_execfile.py     | 70 +++++++++++++++++++++++
 tests/micropython/builtin_execfile.py.exp |  6 ++
 2 files changed, 76 insertions(+)
 create mode 100644 tests/micropython/builtin_execfile.py
 create mode 100644 tests/micropython/builtin_execfile.py.exp

diff --git a/tests/micropython/builtin_execfile.py b/tests/micropython/builtin_execfile.py
new file mode 100644
index 0000000000..8a8ce79f78
--- /dev/null
+++ b/tests/micropython/builtin_execfile.py
@@ -0,0 +1,70 @@
+# Test builtin execfile function using VFS.
+
+try:
+    import uio, uos
+
+    execfile
+    uio.IOBase
+    uos.mount
+except (ImportError, NameError, AttributeError):
+    print("SKIP")
+    raise SystemExit
+
+
+class File(uio.IOBase):
+    def __init__(self, data):
+        self.data = data
+        self.off = 0
+
+    def ioctl(self, request, arg):
+        return 0
+
+    def readinto(self, buf):
+        buf[:] = memoryview(self.data)[self.off : self.off + len(buf)]
+        self.off += len(buf)
+        return len(buf)
+
+
+class Filesystem:
+    def __init__(self, files):
+        self.files = files
+
+    def mount(self, readonly, mkfs):
+        print("mount", readonly, mkfs)
+
+    def umount(self):
+        print("umount")
+
+    def open(self, file, mode):
+        print("open", file, mode)
+        if file not in self.files:
+            raise OSError(2)  # ENOENT
+        return File(self.files[file])
+
+
+# First umount any existing mount points the target may have.
+try:
+    uos.umount("/")
+except OSError:
+    pass
+for path in uos.listdir("/"):
+    uos.umount("/" + path)
+
+# Create and mount the VFS object.
+files = {
+    "/test.py": "print(123)",
+}
+fs = Filesystem(files)
+uos.mount(fs, "/test_mnt")
+
+# Test execfile with a file that doesn't exist.
+try:
+    execfile("/test_mnt/noexist.py")
+except OSError:
+    print("OSError")
+
+# Test execfile with a file that does exist.
+execfile("/test_mnt/test.py")
+
+# Unmount the VFS object.
+uos.umount(fs)
diff --git a/tests/micropython/builtin_execfile.py.exp b/tests/micropython/builtin_execfile.py.exp
new file mode 100644
index 0000000000..1d5d8ee571
--- /dev/null
+++ b/tests/micropython/builtin_execfile.py.exp
@@ -0,0 +1,6 @@
+mount False False
+open /noexist.py rb
+OSError
+open /test.py rb
+123
+umount