1 files changed, 24 insertions, 0 deletions
diff --git a/tests/test_async_generator.py b/tests/test_async_generator.py
new file mode 100644
index 0000000..fd2c109
--- /dev/null
+++ b/tests/test_async_generator.py
@@ -0,0 +1,24 @@
+from prompt_toolkit.eventloop import generator_to_async_generator, get_event_loop
+
+
+def _sync_generator():
+    yield 1
+    yield 10
+
+
+def test_generator_to_async_generator():
+    """
+    Test conversion of sync to asycn generator.
+    This should run the synchronous parts in a background thread.
+    """
+    async_gen = generator_to_async_generator(_sync_generator)
+
+    items = []
+
+    async def consume_async_generator():
+        async for item in async_gen:
+            items.append(item)
+
+    # Run the event loop until all items are collected.
+    get_event_loop().run_until_complete(consume_async_generator())
+    assert items == [1, 10]