o in<@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__^C:\Users\psant\Downloads\PAF backend\school\venv\Lib\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.._ebN)Z 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 FailureZerrbackZAlreadyCalledErrorrcrash _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,rZfailZ maybeDeferredZrunWithWarningsSuppressed _getSuppressrZ callLateraddBoth)rfuncr>r?rAexcr:r)rHr>r?rr@r _runds     z TestCase._runcOs|j|i|SrD)run)rargskwargsrrr __call__szTestCase.__call__c svz,z||IdHW||IdHn ||IdHwW|jr,||IdHdSdS|jr:||IdHwwrD)_deferSetUpAndRun_deferRunCleanupsr_deferTearDownrr?rrr _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)rUr_raddSkip_getSkipReasonKeyboardInterruptr9r r3stop BaseExceptionrgetattr_testMethodNamer6ZaddUnexpectedSuccess_passed_ebDeferTestMethodrr?r;rrr rZs<  zTestCase._deferSetUpAndRuncCs|}|dur||r||||dS||jtr%|||dS|tr6|||| dS|t rL| || t ||j|jdS|||dSrD)r6r7r8r*r&r Z addFailurerbr9rcrr`rarerfr+)rr<r?r=rrr rhs    zTestCase._ebDeferTestMethodc sz||jd|IdHWdSty2}z||t||d|_WYd}~dSd}~wtyP}z||t|d|_WYd}~dSd}~ww)NtearDownF) rUrjrbr9r r3rcrgrdrirrr r\szTestCase._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. rcsiSrDrrrWrSrXrr rIsz,TestCase._deferRunCleanups..zcleanup function FN) len _cleanupspoprU__name__ Exceptionappendr r3r9rg)rr?Zfailuresr<rrkr r[s$   zTestCase._deferRunCleanupscCszt||}|sd|_Wnty#||td|_Ynw|j D] }|||d|_q)| | |jrG| |dSdS)NF) r _JanitorZpostCaseCleanuprgrdr9r r3Z _observerZ getErrorsZflushLoggedErrorsZ_removeObserverZ addSuccess)rr?cleanerrorrrr _cleanUps"   zTestCase._cleanUpcCs<z t||WdSty||tYdSwrD)r rrZpostClassCleanuprdr9r r3r]rrr _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.) stacklevelrC)warningswarnrM_reactorMethods)akwnamerrr _sz&TestCase._makeReactorMethod.._Nr)rrrrr~r _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. )r4ZiteratercN)r{resetattrr)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)r{itemsr)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) rrrr5rZDeferredZ fromCoroutiner^_waitrurvr)rr?rr:rrr _runFixturesAndTest)s      zTestCase._runFixturesAndTestr<Callable[_P, object]rW_P.argsrX _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. N)r addCleanup)rr<rWrXrrr r@s zTestCase.addCleanupcCs|SrD)rQr(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. r@z)'timeout' attribute needs to be a number.rBN) r ZacquireAttribute_parentsZDEFAULT_TIMEOUT_DURATIONfloat ValueErrorrPryrzrM)rr@rrr rJRs   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 |dSdSrD)rq)any)resultsrr rqrszTestCase._wait..appendcsdur dSdSrDr4)Zignrrrr r4vs zTestCase._wait..crashzreactor\.crash cannot be used.*)messagerCcs dSrDrrrrr rcs zTestCase._wait..stopN) RuntimeErrorrrrrKr rLrMrqrRrnrcrVr5rb)rr:runningrqr4rcrrr risL            zTestCase._wait)r)r!r"r#r$r%r")r<rrWrrXrr%r)ro __module__ __qualname____doc__rr1ZfailUnlessFailurerUrYr^rZrhr\rZinlineCallbacksr[rurvrrrrrrrJrr __classcell__rrrr r!s. (    r)r __future__rrNrytypingrrZzope.interfacerZtyping_extensionsrrrrZtwisted.internet.deferr Ztwisted.pythonr Z twisted.trialr r Ztwisted.trial._synctestr rrrr__annotations__Z ITestCaserrrrr s