Fixed problem with regrtest caused by the additional of objects to _abcoll.

This commit is contained in:
Christian Heimes 2007-11-30 21:53:03 +00:00
parent d59c64c49f
commit 941973a328
1 changed files with 6 additions and 1 deletions

View File

@ -689,6 +689,7 @@ def cleanup_test_droppings(testname, verbose):
def dash_R(the_module, test, indirect_test, huntrleaks):
# This code is hackish and inelegant, but it seems to do the job.
import copy_reg, _abcoll
from abc import _Abstract
if not hasattr(sys, 'gettotalrefcount'):
raise Exception("Tracking reference leaks requires a debug build "
@ -699,7 +700,8 @@ def dash_R(the_module, test, indirect_test, huntrleaks):
ps = copy_reg.dispatch_table.copy()
pic = sys.path_importer_cache.copy()
abcs = {obj: obj._abc_registry.copy()
for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]
for abc in [getattr(_abcoll, a) for a in _abcoll.__all__
if isinstance(getattr(_abcoll, a), _Abstract)]
for obj in abc.__subclasses__() + [abc]}
if indirect_test:
@ -737,6 +739,7 @@ def dash_R_cleanup(fs, ps, pic, abcs):
import _strptime, linecache, dircache
import urlparse, urllib, urllib2, mimetypes, doctest
import struct, filecmp, _abcoll
from abc import _Abstract
from distutils.dir_util import _path_created
# Restore some original values.
@ -748,6 +751,8 @@ def dash_R_cleanup(fs, ps, pic, abcs):
# Clear ABC registries, restoring previously saved ABC registries.
for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
if not isinstance(abc, _Abstract):
continue
for obj in abc.__subclasses__() + [abc]:
obj._abc_registry = abcs.get(obj, {}).copy()
obj._abc_cache.clear()