3 ��h��@s�dZddlZddlZyddlZWnek r<ddlZYnXyddlZWnek rfddlZYnXGdd�dej�ZGdd�dej �Z eed�r�e Zej Znej Zdd �ZdS) a[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. �Nc@s0eZdZdZdd�Zdd�Zdd�Zdd �Zd S)�LocalTestResultzHA TestResult which forwards events to a parent object, except for Skips.cCstjj|�||_dS)N)�unittest� TestResult�__init__� parent_result)�selfr�r�&/usr/lib/python3.6/testing_refleaks.pyr:szLocalTestResult.__init__cCs|jj||�dS)N)r�addError)r�test�errorrrr r >szLocalTestResult.addErrorcCs|jj||�dS)N)r� addFailure)rrrrrr r AszLocalTestResult.addFailurecCsdS)Nr)rr�reasonrrr �addSkipDszLocalTestResult.addSkipN)�__name__� __module__�__qualname__�__doc__rr r rrrrr r7s rcs.eZdZdZdZd�fdd� Zdd�Z�ZS) �ReferenceLeakCheckerTestCasezHA TestCase which runs tests multiple times, collecting reference counts.�Ncs�tjj�|_tt|�j|d�tt|�j|d�d}t|�}g}xBt|j �D]4}|j �}tt|�j|d�|j �}|j||�qLWt||f�y|j |dg|j �Wn$tk r�|j|tj��YnXdS)N)�resultr)�copyreg�dispatch_table�copy�_saved_pickle_registry�superr�runr�range�NB_RUNS� _getRefcounts�append�printZassertEqual� Exceptionr �sys�exc_info)rrZoldrefcountZlocal_resultZrefcount_deltas�_Znewrefcount)� __class__rr rMs z ReferenceLeakCheckerTestCase.runcCs8tjj�tjj|j�tj�tj�tj�tj�S)N) rr�clear�updater�gcZcollectr#�gettotalrefcount)rrrr rgs z*ReferenceLeakCheckerTestCase._getRefcounts)N)rrrrrrr� __classcell__rr)r&r rHsrr*cCs~dd�}|S)NcSs|S)Nr)�funcrrr �Same|sz&SkipReferenceLeakChecker..Samer)rr-rrr �SkipReferenceLeakCheckerzsr.)rr)r#r�ImportErrorZ unittest2rrrZTestCaser�hasattrZBaseTestCase�skipr.rrrr �'s"*