Added unit test to verify that threading.local doesn't cause ref leaks. It seems that the thread local storage always keeps the storage of the last stopped thread alive. Can anybody comment on it, please?

This commit is contained in:
Christian Heimes 2008-01-19 13:46:06 +00:00
parent f8f4eb69ce
commit 71a0451f67
1 changed files with 27 additions and 1 deletions

View File

@ -1,9 +1,35 @@
import unittest import unittest
from doctest import DocTestSuite from doctest import DocTestSuite
from test import test_support from test import test_support
import threading
import weakref
class Weak(object):
pass
def target(local, weaklist):
weak = Weak()
local.weak = weak
weaklist.append(weakref.ref(weak))
class ThreadingLocalTest(unittest.TestCase):
def test_local_refs(self):
local = threading.local()
weaklist = []
n = 20
for i in range(n):
t = threading.Thread(target=target, args=(local, weaklist))
t.start()
t.join()
self.assertEqual(len(weaklist), n)
deadlist = [weak for weak in weaklist if weak() is None]
# XXX threading.local keeps the local of the last stopped thread alive
self.assertEqual(len(deadlist), n-1)
def test_main(): def test_main():
suite = DocTestSuite('_threading_local') suite = unittest.TestSuite()
suite.addTest(DocTestSuite('_threading_local'))
suite.addTest(unittest.makeSuite(ThreadingLocalTest))
try: try:
from thread import _local from thread import _local