o >h8@sUdZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZdd lmZmZmZe d ZgZeded <eejGd d d eZdS)zP Things likely to be used by writers of unit tests. Maintainer: Jonathan Lange N)CallableList) implementer) ParamSpec)deferutils)failure)itrialutil)FailTestSkipTestSynchronousTestCase_P_wait_is_runningcseZdZdZd4fdd ZddZeZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZejddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'eeefd(ejd)ejd*d+ffd,d- Zd.d/Zd0d1Z e!fd2d3Z"Z#S)5TestCasea= 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. runTestcst|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. N)super__init__)self methodName __class__x/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/_asynctest.pyr1s zTestCase.__init__cs&fdd}fdd}|||S)z Fail if C{deferred} does not errback with one of C{expectedFailures}. Returns the original Deferred with callbacks added. You will need to return this Deferred from your test case. csd|)Nz$did not catch an error, instead got )failureException)ignorerrr_cbDsz#TestCase.assertFailure.._cbcs*|jr|jSdt|}|)Nz Expected: {!r} Got: {})checkvalueformatstrr)routputexpectedFailuresrrr_ebIs  z#TestCase.assertFailure.._eb) addCallbacks)rdeferredr$rr%rr#r assertFailure=s  zTestCase.assertFailurecsddlmfdd}t|tjtd}t}t |r4t d |}t |St tj|}|||fdd|S) Nrreactorcstddd}t|}z||WdStjyJd_}|durA| |rA ||YdS |YdSw)Nz (z) still running at z secsT) r TimeoutErrorrFailureerrbackAlreadyCalledErrorcrash _timedOutgetTodoexpectedaddExpectedFailureaddError)deftodo)rr*resultrtimeoutrr onTimeout[s z TestCase._run..onTimeoutcategoryz9{!r} is a generator function and therefore will never runcsrp |SN)activecancel)x)callrrszTestCase._run..)twisted.internetr* getTimeoutrsuppressWarningsr suppressDeprecationWarninggetattrinspectisgeneratorfunction TypeErrorr rfail maybeDeferredrunWithWarningsSuppressed _getSuppress callLateraddBoth)rrr9r;methodexcr5r)rBrr*r9rr:r_runVs(      z TestCase._runcOs|j|i|Sr>)run)rargskwargsrrr__call__szTestCase.__call__cCs*|d|}|j|j|j|f|fd|S)NsetUp callbackArgs errbackArgs)rUr&deferTestMethod _ebDeferSetUprignoredr9r5rrr deferSetUps zTestCase.deferSetUpcCsN|tr||||j|jn||||tr!|| d|Sr>) rr addSkip_getSkipReasonrZrr4KeyboardInterruptstopdeferRunCleanupsrrr9rrrr_s    zTestCase._ebDeferSetUpcCsH||j|}|j|j|j|f|fd||j|||j||S)Nr[)rU_testMethodNamer&_cbDeferTestMethod_ebDeferTestMethodrRrg deferTearDownr`rrrr^szTestCase.deferTestMethodcCs*|dur||||Sd|_|S)NT)r1addUnexpectedSuccess_passed)rrar9rrrrjs zTestCase._cbDeferTestMethodcCs|}|dur||r||||dS||jtr%|||dS|tr6|||| dS|t rL| || t ||j|jdS|||dSr>)r1r2r3rrr addFailurerer4rfr rcrdrIrir)rr7r9r8rrrrks    zTestCase._ebDeferTestMethodcCs|d|}||j||S)NtearDown)rU addErrback_ebDeferTearDownr`rrrrls zTestCase.deferTearDowncCs(||||tr|d|_dSNF)r4rrerfrnrhrrrrrs   zTestCase._ebDeferTearDownccsg}t|jdkr4|j\}}}z ||i|VWnty,|tYnwt|jdks |D] }|||d|_q6dS)zf Run any scheduled cleanups and report errors (if any) to the result. object. rFN) len _cleanupspop Exceptionappendrr,r4rn)rrar9failuresfuncrWrXr7rrrrgs  zTestCase.deferRunCleanupscCszt||}|sd|_Wnty#||td|_Ynw|j D] }|||d|_q)| | |jrG| |dSdSrs) r _JanitorpostCaseCleanuprn BaseExceptionr4rr, _observer getErrorsflushLoggedErrors_removeObserver addSuccess)rr9cleanerrorrrr_cleanUps"   zTestCase._cleanUpcCs<z t||WdSty||tYdSwr>)r r{postClassCleanupr}r4rr,)rr9rrr _classCleanUps  zTestCase._classCleanUpcsfdd}|S)z Create a method which wraps the reactor method C{name}. The new method issues a deprecation warning and calls the original. cs,tjdfdtdj|i|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.) stacklevelr=)warningswarnrH_reactorMethods)akwnamerrr_sz&TestCase._makeReactorMethod.._r)rrrrrr_makeReactorMethods zTestCase._makeReactorMethodcCs6i|_dD]}t|||j|<t||||qdS)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. )r/iteraterfN)rrIsetattrr)rr*rrrr_deprecateReactors zTestCase._deprecateReactorcCs*|jD] \}}t|||qi|_dS)z Restore the deprecated reactor methods. Undoes what L{_deprecateReactor} did. @param reactor: The Twisted reactor. N)ritemsr)rr*rrSrrr_undeprecateReactors zTestCase._undeprecateReactorc Csddlm}||d|_z+|d|}z||W||||n ||||wW||dS||w)z Really run C{setUp}, the test method, and C{tearDown}. Any of these may return L{defer.Deferred}s. After they complete, do some reactor cleanup. @param result: A L{TestResult} object. rr)FN) rDr*rr0rb_waitrrr)rr9r*r5rrr_runFixturesAndTests       zTestCase._runFixturesAndTestr7rWrXreturnNcstj|g|Ri|S)a  Extend the base cleanup feature with support for cleanup functions which return Deferreds. If the function C{f} returns a Deferred, C{TestCase} will wait until the Deferred has fired before proceeding to the next function. )r addCleanup)rr7rWrXrrrr0s zTestCase.addCleanupcCs|Sr>)rPrrrr getSuppress<szTestCase.getSuppressc CsHt|jdtj}zt|WSttfy#tjdt dtjYSw)ae Returns the timeout value set on this test. Checks on the instance first, then the class, then the module, then packages. As soon as it finds something with a C{timeout} attribute, returns that. Returns L{util.DEFAULT_TIMEOUT_DURATION} if it cannot find anything. See L{TestCase} docstring for more details. r:z)'timeout' attribute needs to be a number.r<) r acquireAttribute_parentsDEFAULT_TIMEOUT_DURATIONfloat ValueErrorrLrrrH)rr:rrrrE?s   zTestCase.getTimeoutcs|rtdddlmgfdd}fdd}t|tjdtd }fd d }t|tjdtd }|d z5| |rMWd | d S| ||_ z W` n` wsf|j roWd | d Std | w) zATake a Deferred that only ever callbacks. Block until it happens.z_wait is not reentrantrr)csdur |dSdSr>)rx)any)resultsrrrx_szTestCase._wait..appendcsdur dSdSr>r/)ignr*rrrr/cs zTestCase._wait..crashzreactor\.crash cannot be used.*)messager=cs dSr>rrr)rrrfns zTestCase._wait..stopN) RuntimeErrorrDr*rrFr rGrHrxrRrvrfrVr0re)rr5runningrxr/rfrrrrVsL            zTestCase._wait)r)$__name__ __module__ __qualname____doc__rr(failUnlessFailurerUrYrbr_r^rjrkrlrrrinlineCallbacksrgrrrrrrrrobjectrWrXrrrErr __classcell__rrrrr!sD -        r)rrJrtypingrrzope.interfacertyping_extensionsrrDrrtwisted.pythonr twisted.trialr r twisted.trial._synctestr r r rr__annotations__ ITestCaserrrrrs