mirror of
https://github.com/python/cpython.git
synced 2025-08-27 20:25:18 +00:00
bpo-22087: Fix Policy.get_event_loop() to detect fork (GH-7208)
Original patch by Dan O'Reilly.
This commit is contained in:
parent
e55de2d77f
commit
5d97b7bcc1
3 changed files with 42 additions and 0 deletions
|
@ -13,6 +13,7 @@ import sys
|
|||
import tempfile
|
||||
import threading
|
||||
import unittest
|
||||
import multiprocessing
|
||||
from unittest import mock
|
||||
from test import support
|
||||
|
||||
|
@ -1804,6 +1805,37 @@ class FastChildWatcherTests (ChildWatcherTestsMixin, test_utils.TestCase):
|
|||
return asyncio.FastChildWatcher()
|
||||
|
||||
|
||||
class ForkedProcessTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.parent_loop = asyncio.SelectorEventLoop()
|
||||
asyncio.set_event_loop(self.parent_loop)
|
||||
self.ctx = multiprocessing.get_context("fork")
|
||||
|
||||
def tearDown(self):
|
||||
self.parent_loop.close()
|
||||
|
||||
def _check_loops_not_equal(self, old_loop):
|
||||
loop = asyncio.get_event_loop()
|
||||
if loop is old_loop:
|
||||
raise RuntimeError("Child process inherited parent's event loop")
|
||||
|
||||
try:
|
||||
val = loop.run_until_complete(asyncio.sleep(0.05, result=42))
|
||||
if val != 42:
|
||||
raise RuntimeError("new event loop does not work")
|
||||
finally:
|
||||
loop.close()
|
||||
|
||||
sys.exit(loop is old_loop)
|
||||
|
||||
def test_new_loop_in_child(self):
|
||||
p = self.ctx.Process(target=self._check_loops_not_equal,
|
||||
args=(self.parent_loop,))
|
||||
p.start()
|
||||
p.join()
|
||||
self.assertEqual(p.exitcode, 0)
|
||||
|
||||
|
||||
class PolicyTests(unittest.TestCase):
|
||||
|
||||
def create_policy(self):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue