From a4c96fb3b0c5e3bf83238a0edd0fbcbfd96208c8 Mon Sep 17 00:00:00 2001
From: Damien George
Date: Fri, 5 Jun 2020 21:26:27 +1000
Subject: extmod/uasyncio: Add asyncio.wait_for_ms function.

Fixes issue #6107.
---
 tests/extmod/uasyncio_micropython.py     | 37 ++++++++++++++++++++++++++++++++
 tests/extmod/uasyncio_micropython.py.exp |  7 ++++++
 2 files changed, 44 insertions(+)
 create mode 100644 tests/extmod/uasyncio_micropython.py
 create mode 100644 tests/extmod/uasyncio_micropython.py.exp

(limited to 'tests/extmod')

diff --git a/tests/extmod/uasyncio_micropython.py b/tests/extmod/uasyncio_micropython.py
new file mode 100644
index 000000000..69e5fa322
--- /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())
diff --git a/tests/extmod/uasyncio_micropython.py.exp b/tests/extmod/uasyncio_micropython.py.exp
new file mode 100644
index 000000000..f5be1dc75
--- /dev/null
+++ b/tests/extmod/uasyncio_micropython.py.exp
@@ -0,0 +1,7 @@
+True
+task start 1
+task end 1
+2
+task start 2
+timeout
+finish
-- 
cgit v1.2.3