forked from python/cpython
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Ensure objects found by gc.get_objects() are kept alive.
After the `_PyEval_StartTheWorld()` call, other threads may be running and mutating objects. Ensure that the objects are kept alive by incref'ing them when they're added to the `_PyObjectStack`.
- Loading branch information
Showing
3 changed files
with
105 additions
and
40 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,61 @@ | ||
import unittest | ||
|
||
import threading | ||
from threading import Thread | ||
from unittest import TestCase | ||
import gc | ||
|
||
from test.support import threading_helper | ||
|
||
|
||
class MyObj: | ||
pass | ||
|
||
|
||
@threading_helper.requires_working_threading() | ||
class TestGC(TestCase): | ||
def test_get_objects(self): | ||
event = threading.Event() | ||
|
||
def gc_thread(): | ||
for i in range(100): | ||
o = gc.get_objects() | ||
event.set() | ||
|
||
def mutator_thread(): | ||
while not event.is_set(): | ||
o1 = MyObj() | ||
o2 = MyObj() | ||
o3 = MyObj() | ||
o4 = MyObj() | ||
|
||
gcs = [Thread(target=gc_thread)] | ||
mutators = [Thread(target=mutator_thread) for _ in range(4)] | ||
with threading_helper.start_threads(gcs + mutators): | ||
pass | ||
|
||
def test_get_referrers(self): | ||
event = threading.Event() | ||
|
||
obj = MyObj() | ||
|
||
def gc_thread(): | ||
for i in range(100): | ||
o = gc.get_referrers(obj) | ||
event.set() | ||
|
||
def mutator_thread(): | ||
while not event.is_set(): | ||
d1 = { "key": obj } | ||
d2 = { "key": obj } | ||
d3 = { "key": obj } | ||
d4 = { "key": obj } | ||
|
||
gcs = [Thread(target=gc_thread) for _ in range(2)] | ||
mutators = [Thread(target=mutator_thread) for _ in range(4)] | ||
with threading_helper.start_threads(gcs + mutators): | ||
pass | ||
|
||
|
||
if __name__ == "__main__": | ||
unittest.main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters