fix thread safety

python · kumaraditya303 · Jan 4, 2025 · Dec 21, 2024 · Dec 21, 2024 · Dec 21, 2024
commit f3642ae527734d5bc41ea6f09358562eff5a1fc1
@@ -0,0 +1,26 @@
+import asyncio
+from test.support import threading_helper
+from unittest import TestCase
+from threading import Thread
+
+threading_helper.requires_working_threading(module=True)
+
+class TestFreeThreading(TestCase):
+    def test_all_tasks_race(self) -> None:
+        async def task():
+            await asyncio.sleep(0)
+
+        async def main():
+            async with asyncio.TaskGroup() as tg:
+                for _ in range(100):
+                    tg.create_task(task())
+                    asyncio.all_tasks()
+
+        threads = []
+        for _ in range(10):
+            thread = Thread(target=lambda: asyncio.run(main()))
+            threads.append(thread)
+
+        with threading_helper.start_threads(threads):
+            pass
+