tests/extmod: Add uasyncio tests.

All .exp files are included because they require CPython 3.8 which may not
always be available.

1 files changed, 49 insertions, 0 deletions

diff --git a/tests/extmod/uasyncio_gather.py b/tests/extmod/uasyncio_gather.py
new file mode 100644
index 0000000000..2697a6278b
--- /dev/null
+++ b/tests/extmod/uasyncio_gather.py
@@ -0,0 +1,49 @@
+# test uasyncio.gather() function
+
+try:
+    import uasyncio as asyncio
+except ImportError:
+    try:
+        import asyncio
+    except ImportError:
+        print("SKIP")
+        raise SystemExit
+
+
+async def factorial(name, number):
+    f = 1
+    for i in range(2, number + 1):
+        print("Task {}: Compute factorial({})...".format(name, i))
+        await asyncio.sleep(0.01)
+        f *= i
+    print("Task {}: factorial({}) = {}".format(name, number, f))
+    return f
+
+
+async def task(id):
+    print("start", id)
+    await asyncio.sleep(0.2)
+    print("end", id)
+
+
+async def gather_task():
+    print("gather_task")
+    await asyncio.gather(task(1), task(2))
+    print("gather_task2")
+
+
+async def main():
+    # Simple gather with return values
+    print(await asyncio.gather(factorial("A", 2), factorial("B", 3), factorial("C", 4),))
+
+    # Cancel a multi gather
+    # TODO doesn't work, Task should not forward cancellation from gather to sub-task
+    # but rather CancelledError should cancel the gather directly, which will then cancel
+    # all sub-tasks explicitly
+    # t = asyncio.create_task(gather_task())
+    # await asyncio.sleep(0.1)
+    # t.cancel()
+    # await asyncio.sleep(0.01)
+
+
+asyncio.run(main())