Vhn<UdZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZmZdd lmZmZmZe d ZgZd ed<e ej:GddeZy)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_runningceZdZdZdfd Z ddZeZdZdZdZ dZ dZ d Z e jd Zd Zd Zd ZdZdZdZ dfd ZdZdZefdZxZS)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. cLt||ddlm}||_d|_y)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__s H/home/dcms/DCMS/lib/python3.12/site-packages/twisted/trial/_asynctest.pyrzTestCase.__init__1s+ $ -)0%-2)c@fd}fd}|j||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. c,jd|)Nz$did not catch an error, instead got )failureException)ignorers r _cbz#TestCase.assertFailure.._cbRs ''6vjA r!c|jr |jSdjt|}j |)Nz Expected: {!r} Got: {})checkvalueformatstrr$)r outputexpectedFailuresrs r _ebz#TestCase.assertFailure.._ebWsHw}}./}}$5<<$c'l++F33r!) addCallbacks)rdeferredr-r&r.s` ` r assertFailurezTestCase.assertFailureIs"   4$$S#..r!cjfd}tj|tjt }t j|r/tdj|}tj|Stjtjj|}jj!|||j#fd|S)Nctjddd}tj|} |j |y#tj $rnj jd_j}|&|j|rj||Yyj|YywxYw)Nz (z) still running at z secsT) r TimeoutErrorr FailureerrbackAlreadyCalledErrorrcrash _timedOutgetTodoexpectedaddExpectedFailureaddError)deftodofuncDescriptionresultrtimeouts r onTimeoutz TestCase._run..onTimeoutgs""("_--@ OA"A - ! ++ ---335!%||~# a(8--dAt<OOD!, -sAA*C 4C C categoryz9{!r} is a generator function and therefore will never runcPjxrjxs|SN)activecancel)xcalls r zTestCase._run..sDKKM;dkkm@qr!) getTimeoutr suppressWarningsr suppressDeprecationWarninginspectisgeneratorfunction TypeErrorr*rfail maybeDeferredrunWithWarningsSuppressed _getSuppressr callLateraddBoth) rfuncrBrCrEexcr>rMrDs ` `` @@r _runz TestCase._runds//# -,** t}}.@A   & &t ,KRRSWXC::c? "     + +T->->-@$ ,,66w 1M @Ar!c&|j|i|SrI)run)rargskwargss r __call__zTestCase.__call__stxx(((r!cdK |j|d{|j|d{ |jr|j|d{yy7E7.#|j|d{7wxYw7)#|jr|j|d{7wwxYwwrI)_deferSetUpAndRun_deferRunCleanupsr_deferTearDownrrCs r _deferSetUpzTestCase._deferSetUps 2 5,,V444,,V44400))&1111 54d,,V444200))&1111svB0A#AA#BA!B B0BB0A#!B#A?8A;9A??BB0!B-%B(&B--B0c<K |j|jd|d{d|_ |jt||j|j|d{|j!|j||jyd|_y7~#t$r6}|j||j |j|Yd}~yd}~wt $r?}|j |tj||jYd}~yd}~wt$r/}|j |tj|Yd}~yd}~wwxYw7#t$r+}|j#tj||d}~wwxYww)zc Execute the setUp and run part of a test. Teardown and cleanups are not executed. setUpNT)r^rkraddSkip_getSkipReasonKeyboardInterruptr=r r5stop BaseExceptionrgetattr_testMethodNamer:addUnexpectedSuccess_passed_ebDeferTestMethodrrCr?s r rezTestCase._deferSetUpAndRunsG ))DJJ8 8 8-1) ))d223T5I5I6  ||~)++D$,,.A# + 9  NN4!4!4TZZ!C D    OOD'//!"4 5 KKM   OOD'//!"4 5      # #GOOA$6 ?  sF B%B#B%F4E%%E#&4E%FE%"F#B%% E .,CF E +5D% F% E 1%EFE  F#E%% F.&FFFc |j}|%|j|r|j|||y|j|jt r|j ||y|jtr#|j|||jy|jtrA|j||jt||j|jy|j||yrI)r:r;r<r(r$r addFailurernr=rorrlrmrqrrr))rr@rCrAs r ruzTestCase._ebDeferTestMethods||~   a 0  % %dAt 4 WWT**H 5   dA & WW& ' OOD! $ KKM WWX  NNd))'$8L8L*MqwwW  OOD! $r!czK |j|jd|d{y7#t$rF}|j|t j ||j d|_Yd}~yd}~wt$r6}|j|t j |d|_Yd}~yd}~wwxYww)NtearDownF) r^rzrnr=r r5rortrprvs r rgzTestCase._deferTearDowns !))DMM:v> > >  ! OOD'//!"4 5 KKM DLL ! OOD'//!"4 5 DLL !sHB; ,*,B;, B8.sD$1&1r!zcleanup function FN) len _cleanupspopr^__name__ Exceptionappendr r5r=rt)rrCfailuresr@rar\rbs @@@r rfzTestCase._deferRunCleanupss $..!A%!%!3!3!5 D$ 3ii1' 7$..!A% !A OOD! $ DL ! 3 12 3s.9C'B &C?!C ,C CCCc tj||j}|sd|_|jjD]}|j ||d|_|j|j|jr|j|yy#t$r.|j |t jd|_YwxYw)NF) r _JanitorpostCaseCleanuprtrpr=r r5 _observer getErrorsflushLoggedErrors_removeObserver addSuccess)rrCcleanerrors r _cleanUpzTestCase._cleanUps !MM$/??AE$ ^^--/ !E OOD% ( DL !    <<   d #  ! OOD'//"3 4 DL !s-B&&4CCc tj||jy#t$r'|j |t j YywxYwrI)r rpostClassCleanuprpr=r r5rhs r _classCleanUpzTestCase._classCleanUpsB 5 MM$ ' 8 8 : 5 OOD'//"3 4 5s$'-AAcfd}|S)z Create a method which wraps the reactor method C{name}. The new method issues a deprecation warning and calls the original. cttjddddtj|i|S)Nzreactor.z8 cannot be used inside unit tests. In the future, using z7 will fail the test and may crash or hang the test run.) stacklevelrG)warningswarnrR_reactorMethods)akwnamers r _z&TestCase._makeReactorMethod.._sB MM26t=+  .4''-q7B7 7r!r})rrrs`` r _makeReactorMethodzTestCase._makeReactorMethods  8r!ci|_dD]7}t|||j|<t|||j|9y)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. )r8iterateroN)rrqsetattrr)rrrs r _deprecateReactorzTestCase._deprecateReactorsK "0 BD)0$)?D  & GT4#:#:4#@ A Br!cp|jjD]\}}t|||i|_y)z Restore the deprecated reactor methods. Undoes what L{_deprecateReactor} did. @param reactor: The Twisted reactor. N)ritemsr)rrrmethods r _undeprecateReactorzTestCase._undeprecateReactors:!00668 +LD& GT6 * +!r!cddlm}|j|d|_ tj j |j|} |j||j||j| |j|y#|j||j|wxYw#|j|wxYw)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) rrrr9rDeferred fromCoroutineri_waitrrr)rrCrr>s r _runFixturesAndTestzTestCase._runFixturesAndTest)s - w' .,,T-=-=f-EFA + 1  f%""6*  $ $W - f%""6*  $ $W -s#.B<B "B<$B99B<<Cc*t||g|i|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@rarbrs r rzTestCase.addCleanup@sw!!5d5f55r!c"|jSrI)rY)rs r getSuppresszTestCase.getSuppressOs  ""r!ctj|jdtj} t |S#t t f$r.tjdttjcYSwxYw)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. rDz)'timeout' attribute needs to be a number.rF) r acquireAttribute_parentsDEFAULT_TIMEOUT_DURATIONfloat ValueErrorrUrrrR)rrDs r rOzTestCase.getTimeoutRsj'' MM9d&C&C  1> !I& 1 MM;FX 00 0 1s <:A98A9ch|r tdddlmgfd}fd}tj|t j dt}fd}tj|t j dt}|jd  |j|r d |jy |j||_ j` s |jr d |jy t#` wxYw#d |jwxYw) zATake a Deferred that only ever callbacks. Block until it happens.z_wait is not reentrantrrc.j|yyrI)r)anyresultss r rzTestCase._wait..appendrs"s##r!c,jyyrIr8)ignrrs r r8zTestCase._wait..crashvs" #r!zreactor\.crash cannot be used.*)messagerGc&jyrIrrsr rozTestCase._wait..stops MMOr!N) RuntimeErrorrrr rPr rQrRrr[rror`r9rn)rr>runningrr8rorrs @@r rzTestCase._waitis 78 8, $ &&  MM:EW   %%  MM:EW   t$  IIf <G KKM= IIe GL ! L $.."G KKM$% %-L0G KKMs0D?DD(D DDDD1)runTest)r0defer.Deferred[_T]r-ztype[BaseException]returnr)r@zCallable[_P, object]raz_P.argsrbz _P.kwargsrNone)r __module__ __qualname____doc__rr1failUnlessFailurer^rcrirerurgrinlineCallbacksrfrrrrrrrrrOrr __classcell__)rs@r rr!s 20/*/?R/ /2&&P)2>% ! !!*$ 5 $ B ".. 6% 6.5 6AJ 6  6#1. 0Gr!r)r __future__rrSrtypingrrzope.interfacertyping_extensionsrrrr twisted.internet.deferr twisted.pythonr twisted.trialr r twisted.trial._synctestrrrrr__annotations__ ITestCaserr}r!r rsp #!&' *%"&KKt_!*! V  N"NNr!