mirror of
https://github.com/python/cpython.git
synced 2025-11-03 23:21:29 +00:00
219b8266db
This approach eliminates the originally reported race. It also gets rid of the deadlock reported in gh-96071, so we can remove the workaround added then.
This is mostly a cherry-pick of 1c0a104 (AKA gh-126989). The difference is we add PyInterpreterState.threads_preallocated at the end of PyInterpreterState, instead of adding PyInterpreterState.threads.preallocated. That avoids ABI disruption.
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
import threading
|
|
import unittest
|
|
|
|
from test import support
|
|
from test.support import import_helper
|
|
from test.support import threading_helper
|
|
# Raise SkipTest if subinterpreters not supported.
|
|
import_helper.import_module('_interpreters')
|
|
from test.support import interpreters
|
|
from .utils import TestBase
|
|
|
|
|
|
class StressTests(TestBase):
|
|
|
|
# In these tests we generally want a lot of interpreters,
|
|
# but not so many that any test takes too long.
|
|
|
|
@support.requires_resource('cpu')
|
|
def test_create_many_sequential(self):
|
|
alive = []
|
|
for _ in range(100):
|
|
interp = interpreters.create()
|
|
alive.append(interp)
|
|
|
|
@support.requires_resource('cpu')
|
|
@threading_helper.requires_working_threading()
|
|
def test_create_many_threaded(self):
|
|
alive = []
|
|
def task():
|
|
interp = interpreters.create()
|
|
alive.append(interp)
|
|
threads = (threading.Thread(target=task) for _ in range(200))
|
|
with threading_helper.start_threads(threads):
|
|
pass
|
|
|
|
@support.requires_resource('cpu')
|
|
@threading_helper.requires_working_threading()
|
|
def test_many_threads_running_interp_in_other_interp(self):
|
|
interp = interpreters.create()
|
|
|
|
script = f"""if True:
|
|
import _interpreters
|
|
_interpreters.run_string({interp.id}, '1')
|
|
"""
|
|
|
|
def run():
|
|
interp = interpreters.create()
|
|
alreadyrunning = (f'{interpreters.InterpreterError}: '
|
|
'interpreter already running')
|
|
success = False
|
|
while not success:
|
|
try:
|
|
interp.exec(script)
|
|
except interpreters.ExecutionFailed as exc:
|
|
if exc.excinfo.msg != 'interpreter already running':
|
|
raise # re-raise
|
|
assert exc.excinfo.type.__name__ == 'InterpreterError'
|
|
else:
|
|
success = True
|
|
|
|
threads = (threading.Thread(target=run) for _ in range(200))
|
|
with threading_helper.start_threads(threads):
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Test needs to be a package, so we can do relative imports.
|
|
unittest.main()
|