U c6{i@sBdZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZGdd d ejZe e jGd d d e e jd Zd dZddZGdddeZGdddeZGdddeZe eeje je eeje jeeddZere eee jeejejfee jdddZdS)z- Infrastructure for test running and suites. N)IteratorUnion) implementer) components)itrialreporter) _logObserverc@seZdZdZddZdS) TestSuitezh Extend the standard library's C{TestSuite} with a consistently overrideable C{run} method. cCs"|jD]}|jrq||q|S)z; Call C{run} on every member of the suite. )_testsZ shouldStop)selfresulttestr>/tmp/pip-unpacked-wheel-hfbvu0bz/twisted/trial/_asyncrunner.pyruns   z TestSuite.runN__name__ __module__ __qualname____doc__rrrrrr sr c@s eZdZdZddZddZdS) TestDecoratorz Decorator for test cases. @param _originalTest: The wrapped instance of test. @type _originalTest: A provider of L{itrial.ITestCase} cCs ||SzQ Run the unit test. @param result: A TestResult object. )rr r rrr__call__4szTestDecorator.__call__cCs|jt||jSr) _originalTestrrZ_AdaptedReporter __class__rrrrr<szTestDecorator.runN)rrrrrrrrrrr'srrcCs g|_dS)z Clear all tests from C{suite}. This messes with the internals of C{suite}. In particular, it assumes that the suite keeps all of its tests in a list in an instance variable called C{_tests}. N)r )suiterrr _clearSuiteEsrcCsPz t|}Wntk r(||YSXt||D]}|t||q6|S)a2 Decorate all test cases in C{test} with C{decorator}. C{test} can be a test case or a test suite. If it is a test suite, then the structure of the suite is preserved. L{decorate} tries to preserve the class of the test suites it finds, but assumes the presence of the C{_tests} attribute on the suite. @param test: The C{TestCase} or C{TestSuite} to decorate. @param decorator: A unary callable used to decorate C{TestCase}s. @return: A decorated C{TestCase} or a C{TestSuite} containing decorated C{TestCase}s. )iter TypeErrorrZaddTestdecorate)r decoratortestsZcaserrrr Ps r c@seZdZdZdS)_PyUnitTestCaseAdapterz2 Adapt from pyunit.TestCase to ITestCase. N)rrrrrrrrr#osr#c@seZdZdZddZdS)_BrokenIDTestCaseAdapterz Adapter for pyunit-style C{TestCase} subclasses that have undesirable id() methods. That is C{unittest.FunctionTestCase} and C{unittest.DocTestCase}. cCs |j}|dk r|S|jS)zH Return the fully-qualified Python name of the doctest. N)rZshortDescriptionid)r ZtestIDrrrr%{s z_BrokenIDTestCaseAdapter.idN)rrrrr%rrrrr$usr$c@seZdZdZddZdS) _ForceGarbageCollectionDecoratorz Forces garbage collection to be run before and after the test. Any errors logged during the post-test collection are added to the test result as errors. cCsRtt||tttD]}|||q,tt dS)N) gcZcollectrrrZ_addZ getErrorsZaddErrorZ flushErrors_remove)r r errorrrrrs  z$_ForceGarbageCollectionDecorator.runNrrrrrr&sr&Z DocTestCase)testSuiteOrCasereturnccsDz t|}Wntk r&|VYnX|D]}t|EdHq,dS)zF Iterate through all of the test cases in C{testSuiteOrCase}. N)rr _iterateTests)r*rr rrrr,s   r,) rdoctestr'ZunittestZpyunittypingrrZzope.interfacerZtwisted.pythonrZ twisted.trialrrZtwisted.trial._synctestrr Z ITestCaseZproxyForInterfacerrr r#r$r&ZregisterAdapterZTestCaseZFunctionTestCasegetattrZ _docTestCaser,rrrrsB