1 files changed, 46 insertions, 0 deletions
diff --git a/tests/extmod/uasyncio_heaplock.py b/tests/extmod/uasyncio_heaplock.py
new file mode 100644
index 000000000..771d3f0d9
--- /dev/null
+++ b/tests/extmod/uasyncio_heaplock.py
@@ -0,0 +1,46 @@
+# test that basic scheduling of tasks, and uasyncio.sleep_ms, does not use the heap
+
+import micropython
+
+# strict stackless builds can't call functions without allocating a frame on the heap
+try:
+    f = lambda: 0
+    micropython.heap_lock()
+    f()
+    micropython.heap_unlock()
+except RuntimeError:
+    print("SKIP")
+    raise SystemExit
+
+try:
+    import uasyncio as asyncio
+except ImportError:
+    try:
+        import asyncio
+    except ImportError:
+        print("SKIP")
+        raise SystemExit
+
+
+async def task(id, n, t):
+    for i in range(n):
+        print(id, i)
+        await asyncio.sleep_ms(t)
+
+
+async def main():
+    t1 = asyncio.create_task(task(1, 4, 10))
+    t2 = asyncio.create_task(task(2, 4, 25))
+
+    micropython.heap_lock()
+
+    print("start")
+    await asyncio.sleep_ms(1)
+    print("sleep")
+    await asyncio.sleep_ms(100)
+    print("finish")
+
+    micropython.heap_unlock()
+
+
+asyncio.run(main())