U P|jn<ã@sÈUdZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZmZdd lmZmZmZe d ƒZgZd ed<e ejƒGdd„deƒƒZdS)z4 Things likely to be used by writers of unit tests. é)Ú annotationsN)ÚCallableÚList)Ú implementer)Ú ParamSpec)ÚdeferÚutils)Ú_T)Úfailure)ÚitrialÚutil)ÚFailTestÚSkipTestÚSynchronousTestCaseÚ_Pz List[None]Ú_wait_is_runningcsÚeZdZdZd1‡fdd„ Zddddœdd „ZeZd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z e jdd„ƒZdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%d&d'd(œ‡fd)d*„ Zd+d,„Zd-d.„Zefd/d0„Z‡ZS)2ÚTestCasea= A unit test. The atom of the unit testing universe. This class extends L{SynchronousTestCase} which extends C{unittest.TestCase} from the standard library. The main feature is the ability to return C{Deferred}s from tests and fixture methods and to have the suite wait for those C{Deferred}s to fire. Also provides new assertions such as L{assertFailure}. @ivar timeout: A real number of seconds. If set, the test will raise an error if it takes longer than C{timeout} seconds. If not set, util.DEFAULT_TIMEOUT_DURATION is used. ÚrunTestcs(tƒ |¡ddlm}||_d|_dS)aŸ Construct an asynchronous test case for C{methodName}. @param methodName: The name of a method on C{self}. This method should be a unit test. That is, it should be a short method that calls some of the assert* methods. If C{methodName} is unspecified, L{SynchronousTestCase.runTest} will be used as the test method. This is mostly useful for testing Trial. r©ÚreactorFN)ÚsuperÚ__init__Útwisted.internetrÚ_twistedPrivateSchedulerÚ_twistedPrivateNeedsTearDown)ÚselfÚ methodNamer©Ú __class__©ú._cbcs.|jˆŽr|jSd ˆt|ƒ¡}ˆ |¡‚dS)Nz Expected: {!r} Got: {})ÚcheckÚvalueÚformatÚstrr$)r Úoutput©r"rrr Ú_ebWs ÿz#TestCase.assertFailure.._eb)Z addCallbacks)rr!r"r'r.rr-r Ú assertFailureIs  zTestCase.assertFailurecsŠˆ ¡‰‡‡‡‡fdd„}t |tjtd¡}t |¡rPtd  |¡ƒ}t   |¡St   tj ˆ ¡|¡}ˆj ˆ||¡‰| ‡fdd„¡|S)Ncs–t ˆ›dˆ›dˆ›d¡}t |¡}z| |¡Wn\tjk rˆj ¡dˆ_ˆ  ¡}|dk r€|  |¡r€ˆ  ˆ||¡n ˆ  ˆ|¡YnXdS)Nz (z) still running at z secsT) rÚ TimeoutErrorr ÚFailureZerrbackZAlreadyCalledErrorrÚcrashÚ _timedOutÚgetTodoÚexpectedÚaddExpectedFailureÚaddError)ÚdÚeÚfÚtodo)ÚfuncDescriptionÚresultrÚtimeoutrr Ú onTimeoutgsÿ  z TestCase._run..onTimeout©Úcategoryz9{!r} is a generator function and therefore will never runcsˆ ¡rˆ ¡p|S©N)ÚactiveÚcancel)Úx)Úcallrr Ú‰ózTestCase._run..)Ú getTimeoutrÚsuppressWarningsr ÚsuppressÚDeprecationWarningÚinspectÚisgeneratorfunctionÚ TypeErrorr*rZfailZ maybeDeferredZrunWithWarningsSuppressedÚ _getSuppressrZ callLaterÚaddBoth)rÚfuncr<r=r?Úexcr8r)rFr<r=rr>r Ú_runds& ÿ ÿ ÿz TestCase._runcOs |j||ŽSrB)Úrun)rÚargsÚkwargsrrr Ú__call__ŒszTestCase.__call__c ÃsJz,z| |¡IdHW5| |¡IdHXW5|jrD| |¡IdHXdSrB)rÚ_deferTearDownÚ_deferRunCleanupsÚ_deferSetUpAndRun©rr=rrr Ú _deferSetUps zTestCase._deferSetUpc Ãs^z| |jd|¡IdHWn¶tk rX}z | || |j|¡¡WY¢dSd}~XYnztk r˜}z$| |t |¡¡|  ¡WY¢dSd}~XYn:t k rÐ}z| |t |¡¡WY¢dSd}~XYnXd|_ zH| t ||j ƒ|j |¡IdH| ¡dk r| || ¡¡nd|_Wn8t k rX}z| t |¡|¡‚W5d}~XYnXdS)zc Execute the setUp and run part of a test. Teardown and cleanups are not executed. ÚsetUpNT)rTr^rÚaddSkipÚ_getSkipReasonÚKeyboardInterruptr7r r1ÚstopÚ BaseExceptionrÚgetattrÚ_testMethodNamer4ZaddUnexpectedSuccessÚ_passedÚ_ebDeferTestMethod©rr=r9rrr r[™s2 ÿ  zTestCase._deferSetUpAndRuncCs | ¡}|dk r*| |¡r*| |||¡nr| |jt¡rF| ||¡nV| t¡rf| ||¡|  ¡n6| t ¡r|  ||  t ||jƒ|j¡¡n | ||¡dSrB)r4r5r6r(r$r Z addFailurerar7rbrr_r`rdrer))rr:r=r;rrr rg¸s    ÿzTestCase._ebDeferTestMethodc Ãsšz| |jd|¡IdHWnztk r\}z$| |t |¡¡| ¡d|_W5d}~XYn:tk r”}z| |t |¡¡d|_W5d}~XYnXdS)NÚtearDownF) rTrirar7r r1rbrfrcrhrrr rYÈszTestCase._deferTearDownc#sg}t|jƒdkrp|j ¡\‰‰‰z(| ‡‡‡fdd„dˆj›|¡VWqtk rl| t ¡¡YqXq|D]}|  ||¡d|_ qtdS)zf Run any scheduled cleanups and report errors (if any) to the result. object. rcs ˆˆˆŽSrBrr©rVrRrWrr rGÞrHz,TestCase._deferRunCleanups..zcleanup function FN) ÚlenÚ _cleanupsÚpoprTÚ__name__Ú ExceptionÚappendr r1r7rf)rr=Zfailuresr:rrjr rZÓs ý  zTestCase._deferRunCleanupscCszt ||¡ ¡}|sd|_Wn*tk rH| |t ¡¡d|_YnX|j  ¡D]}| ||¡d|_qT|  ¡|  ¡|jrŒ|  |¡dS)NF) r Ú_JanitorZpostCaseCleanuprfrcr7r r1Z _observerZ getErrorsZflushLoggedErrorsZ_removeObserverZ addSuccess)rr=ÚcleanÚerrorrrr Ú_cleanUpés   zTestCase._cleanUpcCs>zt ||¡ ¡Wn$tk r8| |t ¡¡YnXdSrB)r rqZpostClassCleanuprcr7r r1r\rrr Ú _classCleanUpùszTestCase._classCleanUpcs‡‡fdd„}|S)z• Create a method which wraps the reactor method C{name}. The new method issues a deprecation warning and calls the original. cs(tjdˆˆfdtdˆjˆ||ŽS)Nz{reactor.%s cannot be used inside unit tests. In the future, using %s will fail the test and may crash or hang the test run.é)Ú stacklevelrA)ÚwarningsÚwarnrLÚ_reactorMethods)ÚaÚkw©Únamerrr Ú_sþûz&TestCase._makeReactorMethod.._r)rr~rrr}r Ú_makeReactorMethodÿs zTestCase._makeReactorMethodcCs6i|_dD]&}t||ƒ|j|<t||| |¡ƒq dS)zó Deprecate C{iterate}, C{crash} and C{stop} on C{reactor}. That is, each method is wrapped in a function that issues a deprecation warning, then calls the original. @param reactor: The Twisted reactor. )r2ZiteraterbN)rzrdÚsetattrr€)rrr~rrr Ú_deprecateReactorszTestCase._deprecateReactorcCs*|j ¡D]\}}t|||ƒq i|_dS)z• Restore the deprecated reactor methods. Undoes what L{_deprecateReactor} did. @param reactor: The Twisted reactor. N)rzÚitemsr)rrr~Úmethodrrr Ú_undeprecateReactorszTestCase._undeprecateReactorc Csjddlm}| |¡d|_zz)'timeout' attribute needs to be a number.r@N) r ZacquireAttributeÚ_parentsZDEFAULT_TIMEOUT_DURATIONÚfloatÚ ValueErrorrOrxryrL)rr>rrr rIRsÿ ÿzTestCase.getTimeoutcsâ|r tdƒ‚ddlm‰g‰‡fdd„}‡‡fdd„}t |tjdtd ¡}‡fd d „}t |tjdtd ¡}| d ¡zV|  |¡ˆrW¢Dd S|  |¡|ˆ_ z ˆ  ¡W5ˆ` Xˆs¾|j rÆW¢d Stƒ‚W5d ‰|  ¡Xd S) zATake a Deferred that only ever callbacks. Block until it happens.z_wait is not reentrantrrcsˆdk rˆ |¡dSrB)rp)Úany)Úresultsrr rprszTestCase._wait..appendcsˆdk rˆ ¡dSrB©r2)Zign©rrrr r2vszTestCase._wait..crashzreactor\.crash cannot be used.*)ÚmessagerAcs ˆ ¡dSrBrrrrr rbszTestCase._wait..stopN)Ú RuntimeErrorrrrrJr rKrLrprmrQrbrUr3ra)rr8Zrunningrpr2rbrr‘r r†isH  ÿþ ÿþ      zTestCase._wait)r)rnÚ __module__Ú __qualname__Ú__doc__rr/ZfailUnlessFailurerTrXr]r[rgrYrZinlineCallbacksrZrtrur€r‚r…r‡r‰rŠrIrr†Ú __classcell__rrrr r!s,(     r)r–Ú __future__rrMrxÚtypingrrZzope.interfacerZtyping_extensionsrrrrZtwisted.internet.deferr Ztwisted.pythonr Z twisted.trialr r Ztwisted.trial._synctestr rrrrÚ__annotations__Z ITestCaserrrrr Ús