U  W[@s(dZddlmZmZddlZddlZddlmZddlm Z m Z ddl m Z e dZddlmZGd d d ejZee jGd d d ee jd ZddZddZGdddeZGdddeZGdddeZeeeje jeeeje jeeddZereeee jddZdS)z- Infrastructure for test running and suites. )divisionabsolute_importN) components)itrialreporter) _logObserverZunittest) implementerc@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)selfresulttestrs zTestDecorator.runN)rrrrrrrrrrr+s rrcCs 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 _clearSuiteIsrcCsPz 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 Z decoratorZtestsZcaserrrr Us r c@seZdZdZdS)_PyUnitTestCaseAdapterz2 Adapt from pyunit.TestCase to ITestCase. N)rrrrrrrrr!usr!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"|sr"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 DocTestCaseccsJz t|}Wntk r&|VYn X|D]}t|D] }|Vq8q,dS)zF Iterate through all of the test cases in C{testSuiteOrCase}. N)rr _iterateTests)ZtestSuiteOrCaserr Zsubtestrrrr(s   r() rZ __future__rrZdoctestr%Ztwisted.pythonrZ twisted.trialrrZtwisted.trial._synctestr __import__ZpyunitZzope.interfacerr Z ITestCaseZproxyForInterfacerrr r!r"r$ZregisterAdapterZTestCaseZFunctionTestCasegetattrZ _docTestCaser(rrrrsH