test: add tests for new task result and exception behavior

imnotjames · imnotjames · commit 58ed2e1680b5 · 2023-11-10T17:46:31.000-05:00
diff --git a/tests/extmod/asyncio_cancel_task.py b/tests/extmod/asyncio_cancel_task.py
@@ -33,13 +33,17 @@ async def main():
     t = asyncio.create_task(task(2, True))
     print(t.cancel())
 
+    print("main checking task is cancelled", t.cancelled())
+
     # Cancel task after it has started
     t = asyncio.create_task(task(2, True))
     await asyncio.sleep(0.01)
     print(t.cancel())
     print("main sleep")
     await asyncio.sleep(0.01)
 
+    print("main checking task is cancelled", t.cancelled())
+
     # Cancel task multiple times after it has started
     t = asyncio.create_task(task(2, True))
     await asyncio.sleep(0.01)
@@ -48,6 +52,8 @@ async def main():
     print("main sleep")
     await asyncio.sleep(0.01)
 
+    print("main checking task is cancelled", t.cancelled())
+
     # Await on a cancelled task
     print("main wait")
     try:
@@ -60,6 +66,8 @@ async def main():
     await asyncio.sleep(0.05)
     print(t.cancel())
 
+    print("main checking task is cancelled", t.cancelled())
+
     # Nested: task2 waits on task, task2 is cancelled (should cancel task then task2)
     print("----")
     t = asyncio.create_task(task2(True))
@@ -69,6 +77,8 @@ async def main():
     print("main sleep")
     await asyncio.sleep(0.1)
 
+    print("main checking task 2 is cancelled", t.cancelled())
+
     # Nested: task2 waits on task, task2 is cancelled but task doesn't allow it (task2 should continue)
     print("----")
     t = asyncio.create_task(task2(False))
@@ -77,6 +87,7 @@ async def main():
     t.cancel()
     print("main sleep")
     await asyncio.sleep(0.1)
+    print("main checking task 2 is cancelled", t.cancelled())
 
 
 asyncio.run(main())
diff --git a/tests/extmod/asyncio_task_exception.py b/tests/extmod/asyncio_task_exception.py
@@ -0,0 +1,59 @@
+# Test the Task.done() method
+
+try:
+    import asyncio
+except ImportError:
+    print("SKIP")
+    raise SystemExit
+
+
+async def task(t, exc=None):
+    print("task start")
+    if t >= 0:
+        await asyncio.sleep(t)
+    if exc:
+        raise exc
+    print("task done")
+
+
+async def main():
+    # Task that is not done yet raises an InvalidStateError
+    print("=" * 10)
+    t = asyncio.create_task(task(1))
+    await asyncio.sleep(0)
+    try:
+        t.exception()
+        assert False, "Should not get here"
+    except Exception as e:
+        print(repr(e))
+
+    # Task that is cancelled raises CancelledError
+    print("=" * 10)
+    t = asyncio.create_task(task(1))
+    t.cancel()
+    await asyncio.sleep(0)
+    try:
+        t.exception()
+        assert False, "Should not get here"
+    except asyncio.CancelledError as e:
+        print(repr(e))
+
+    # Task that starts, runs and finishes without an exception should return None
+    print("=" * 10)
+    t = asyncio.create_task(task(0.01))
+    await t
+    print(t.exception())
+
+    # Task that raises immediately should return that exception
+    print("=" * 10)
+    t = asyncio.create_task(task(-1, ValueError))
+    print(t.done())
+    try:
+        await t
+        assert False, "Should not get here"
+    except ValueError as e:
+        print(repr(e))
+    print(repr(t.exception()))
+
+
+asyncio.run(main())
diff --git a/tests/extmod/asyncio_task_exception.py.exp b/tests/extmod/asyncio_task_exception.py.exp
@@ -0,0 +1,24 @@
+==========
+False
+task start
+task done
+True
+True
+==========
+False
+task start
+False
+task done
+True
+==========
+False
+task start
+True
+ValueError()
+True
+==========
+False
+task start
+False
+ValueError()
+True
diff --git a/tests/extmod/asyncio_task_result.py b/tests/extmod/asyncio_task_result.py
@@ -0,0 +1,71 @@
+# Test the Task.done() method
+
+try:
+    import asyncio
+except ImportError:
+    print("SKIP")
+    raise SystemExit
+
+
+async def task(t, exc=None, ret=None):
+    print("task start")
+    if t >= 0:
+        await asyncio.sleep(t)
+    if exc:
+        raise exc
+    print("task done")
+    return ret
+
+
+async def main():
+    # Task that is not done yet raises an InvalidStateError
+    print("=" * 10)
+    t = asyncio.create_task(task(1))
+    await asyncio.sleep(0)
+    try:
+        t.result()
+        assert False, "Should not get here"
+    except Exception as e:
+        print(repr(e))
+
+    # Task that is cancelled raises CancelledError
+    print("=" * 10)
+    t = asyncio.create_task(task(1))
+    t.cancel()
+    await asyncio.sleep(0)
+    try:
+        t.result()
+        assert False, "Should not get here"
+    except asyncio.CancelledError as e:
+        print(repr(e))
+
+    # Task that starts, runs and finishes without an exception or value should return None
+    print("=" * 10)
+    t = asyncio.create_task(task(0.01))
+    await t
+    print(t.result())
+    assert t.result() is None
+
+    # Task that raises immediately should raise that exception when calling result
+    print("=" * 10)
+    t = asyncio.create_task(task(-1, ValueError))
+    try:
+        await t
+        assert False, "Should not get here"
+    except ValueError as e:
+        print(repr(e))
+
+    try:
+        t.result()
+        assert False, "Should not get here"
+    except ValueError as e:
+        print(repr(e))
+
+    # Task that starts, runs and finishes without exception should return result
+    print("=" * 10)
+    t = asyncio.create_task(task(0.01, None, "hello world"))
+    await t
+    print(t.result())
+
+
+asyncio.run(main())
diff --git a/tests/extmod/asyncio_task_result.py.exp b/tests/extmod/asyncio_task_result.py.exp
@@ -0,0 +1,24 @@
+==========
+False
+task start
+task done
+True
+True
+==========
+False
+task start
+False
+task done
+True
+==========
+False
+task start
+True
+ValueError()
+True
+==========
+False
+task start
+False
+ValueError()
+True
