o 9Bin<@sUdZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZmZdd lmZmZmZe d ZgZd ed<e ejGdddeZdS)z4 Things likely to be used by writers of unit tests. ) annotationsN)CallableList) implementer) ParamSpec)deferutils)_T)failure)itrialutil)FailTestSkipTestSynchronousTestCase_Pz List[None]_wait_is_runningcseZdZdZd5fdd Zd6d d ZeZd d ZddZddZ ddZ ddZ ddZ e jddZddZddZddZd d!Zd"d#Zd$d%Zd7fd-d. Zd/d0Zd1d2Zefd3d4ZZS)8TestCasea= 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. rreactorFN)super__init__twisted.internetr_twistedPrivateScheduler_twistedPrivateNeedsTearDown)self methodNamer __class__X/var/www/html/Trade-python/venv/lib/python3.10/site-packages/twisted/trial/_asynctest.pyr1s  zTestCase.__init__deferreddefer.Deferred[_T]expectedFailurestype[BaseException]returncs&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 _cbRsz#TestCase.assertFailure.._cbcs*|jr|jSdt|}|)Nz Expected: {!r} Got: {})checkvalueformatstrr&)r outputr#rrr _ebWs  z#TestCase.assertFailure.._eb) addCallbacks)rr!r#r)r0rr/r assertFailureIs  zTestCase.assertFailurecsfdd}t|tjtd}t|r(td |}t |St tj |}j|||fdd|S)Ncstddd}t|}z||WdStjyKjd_ }|durB| |rB ||YdS |YdSw)Nz (z) still running at z secsT) r TimeoutErrorr FailureerrbackAlreadyCalledErrorrcrash _timedOutgetTodoexpectedaddExpectedFailureaddError)deftodo)funcDescriptionresultrtimeoutrr onTimeoutgs  z TestCase._run..onTimeoutcategoryz9{!r} is a generator function and therefore will never runcsrp |SN)activecancel)x)callrr szTestCase._run..) getTimeoutrsuppressWarningsr suppressDeprecationWarninginspectisgeneratorfunction TypeErrorr,rfail maybeDeferredrunWithWarningsSuppressed _getSuppressr callLateraddBoth)rfuncrArBrDexcr=r)rKrArBrrCr _runds     z TestCase._runcOs|j|i|SrG)run)rargskwargsrrr __call__szTestCase.__call__c svz,z||IdHW||IdHn ||IdHwW|jr,||IdHdSdS|jr:||IdHwwrG)_deferSetUpAndRun_deferRunCleanupsr_deferTearDownrrBrrr _deferSetUps& zTestCase._deferSetUpc sNz ||jd|IdHWnXty,}z||||j|WYd}~dSd}~wtyK}z||t|| WYd}~dSd}~wt yf}z||t|WYd}~dSd}~wwd|_ z&|t ||j |j |IdH|dur|||WdSd|_WdSt y}z |t||d}~ww)zc Execute the setUp and run part of a test. Teardown and cleanups are not executed. setUpNT)r\rfraddSkip_getSkipReasonKeyboardInterruptr<r r4stop BaseExceptionrgetattr_testMethodNamer9addUnexpectedSuccess_passed_ebDeferTestMethodrrBr>rrr ras<  zTestCase._deferSetUpAndRuncCs|}|dur||r||||dS||jtr%|||dS|tr6|||| dS|t rL| || t ||j|jdS|||dSrG)r9r:r;r*r&r addFailurerir<rjrrgrhrlrmr+)rr?rBr@rrr rps    zTestCase._ebDeferTestMethodc sz||jd|IdHWdSty2}z||t||d|_WYd}~dSd}~wtyP}z||t|d|_WYd}~dSd}~ww)NtearDownF) r\rsrir<r r4rjrorkrqrrr rcszTestCase._deferTearDownc#sg}t|jdkr>|j\z|fdddj|VWnty6|tYnwt|jdks |D] }| ||d|_ q@dS)zf Run any scheduled cleanups and report errors (if any) to the result. object. rcsiSrGrrr^rZr_rr rLsz,TestCase._deferRunCleanups..zcleanup function FN) len _cleanupspopr\__name__ Exceptionappendr r4r<ro)rrBfailuresr?rrtr rbs$   zTestCase._deferRunCleanupscCszt||}|sd|_Wnty#||td|_Ynw|j D] }|||d|_q)| | |jrG| |dSdS)NF) r _JanitorpostCaseCleanuprorkr<r r4 _observer getErrorsflushLoggedErrors_removeObserver addSuccess)rrBcleanerrorrrr _cleanUps"   zTestCase._cleanUpcCs<z t||WdSty||tYdSwrG)r r|postClassCleanuprkr<r r4rdrrr _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.) stacklevelrF)warningswarnrP_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. )r7iteraterjN)rrlsetattrr)rrrrrr _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)rrrmethodrrr _undeprecateReactors zTestCase._undeprecateReactorc Csddlm}||d|_z.tj||}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. rrFN) rrrr8rDeferred fromCoroutinere_waitrrr)rrBrr=rrr _runFixturesAndTest)s      zTestCase._runFixturesAndTestr?Callable[_P, object]r^_P.argsr_ _P.kwargsNonecstj|g|Ri|S)ar 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. If the function takes more than C{timeout} settings, then the test will raise an error. )r addCleanup)rr?r^r_rrr r@s zTestCase.addCleanupcCs|SrG)rWr(rrr getSuppressOszTestCase.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. rCz)'timeout' attribute needs to be a number.rE) r acquireAttribute_parentsDEFAULT_TIMEOUT_DURATIONfloat ValueErrorrSrrrP)rrCrrr rMRs   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 reentrantrrcsdur |dSdSrG)rz)any)resultsrr rzrszTestCase._wait..appendcsdur dSdSrGr7)ignrrrr r7vs zTestCase._wait..crashzreactor\.crash cannot be used.*)messagerFcs dSrGrrrrr rjs zTestCase._wait..stopN) RuntimeErrorrrrrNr rOrPrzrYrwrjr]r8ri)rr=runningrzr7rjrrr risL            zTestCase._wait)r)r!r"r#r$r%r")r?rr^rr_rr%r)rx __module__ __qualname____doc__rr2failUnlessFailurer\r`rerarprcrinlineCallbacksrbrrrrrrrrrMrr __classcell__rrrr r!s. (    r)r __future__rrQrtypingrrzope.interfacertyping_extensionsrrrrtwisted.internet.deferr twisted.pythonr twisted.trialr r twisted.trial._synctestr rrrr__annotations__ ITestCaserrrrr s