�
S�Ljc @ s� d Z d d l Z d d l Z y d d l Z Wn e k
rM d d l Z n Xy d d l Z Wn e k
r} d d l Z n Xd e j f d � � YZ d e j
f d � � YZ e e d � r� e Z
e j Z n e j
Z
d � Z d S( s[ A subclass of unittest.TestCase which checks for reference leaks.
To use:
- Use testing_refleak.BaseTestCase instead of unittest.TestCase
- Configure and compile Python with --with-pydebug
If sys.gettotalrefcount() is not available (because Python was built without
the Py_DEBUG option), then this module is a no-op and tests will run normally.
i����Nt LocalTestResultc B s2 e Z d Z d � Z d � Z d � Z d � Z RS( sH A TestResult which forwards events to a parent object, except for Skips.c C s t j j | � | | _ d S( N( t unittestt
TestResultt __init__t
parent_result( t selfR ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR : s c C s | j j | | � d S( N( R t addError( R t testt error( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR > s c C s | j j | | � d S( N( R t
addFailure( R R R ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR A s c C s d S( N( ( R R t reason( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyt addSkipD s ( t __name__t
__module__t __doc__R R R R ( ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR 7 s
t ReferenceLeakCheckerTestCasec B s) e Z d Z d Z d d � Z d � Z RS( sH A TestCase which runs tests multiple times, collecting reference counts.i c C s t j j � | _ t t | � j d | � t t | � j d | � d } t | � } g } xY t | j � D]H } | j
� } t t | � j d | � | j
� } | j | | � ql W| | f GHy | j | d g | j � Wn' t
k
r
| j | t j � � n Xd S( Nt resulti ( t copyregt dispatch_tablet copyt _saved_pickle_registryt superR t runR t ranget NB_RUNSt
_getRefcountst appendt assertEqualt ExceptionR t syst exc_info( R R t oldrefcountt local_resultt refcount_deltast _t newrefcount( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR M s
c C sH t j j � t j j | j � t j � t j � t j � t j � S( N( R R t cleart updateR t gct collectR t gettotalrefcount( R ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR g s
N( R R
R R t NoneR R ( ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyR H s R( c C s ~ d � } | S( Nc S s | S( N( ( t func( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyt Same| s ( ( R
R+ ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyt SkipReferenceLeakCheckerz s ( R R& R t copy_regR t ImportErrort unittest2R R R t TestCaseR t hasattrt BaseTestCaset skipR, ( ( ( sM /usr/lib/python2.7/site-packages/google/protobuf/internal/testing_refleaks.pyt ' s"
*