extmod/uasyncio: Add asyncio.wait_for_ms function.

Fixes issue #6107.

1 files changed, 37 insertions, 0 deletions

diff --git a/tests/extmod/uasyncio_micropython.py b/tests/extmod/uasyncio_micropython.py
new file mode 100644
index 0000000000..69e5fa3224
--- /dev/null
+++ b/tests/extmod/uasyncio_micropython.py
@@ -0,0 +1,37 @@
+# Test MicroPython extensions on CPython asyncio:
+# - sleep_ms
+# - wait_for_ms
+
+try:
+    import utime, uasyncio
+except ImportError:
+    print("SKIP")
+    raise SystemExit
+
+
+async def task(id, t):
+    print("task start", id)
+    await uasyncio.sleep_ms(t)
+    print("task end", id)
+    return id * 2
+
+
+async def main():
+    # Simple sleep_ms
+    t0 = utime.ticks_ms()
+    await uasyncio.sleep_ms(1)
+    print(utime.ticks_diff(utime.ticks_ms(), t0) < 100)
+
+    # When task finished before the timeout
+    print(await uasyncio.wait_for_ms(task(1, 5), 50))
+
+    # When timeout passes and task is cancelled
+    try:
+        print(await uasyncio.wait_for_ms(task(2, 50), 5))
+    except uasyncio.TimeoutError:
+        print("timeout")
+
+    print("finish")
+
+
+uasyncio.run(main())