\de7UdZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZmZmZgZeded <ee jGd d eZdS) zP Things likely to be used by writers of unit tests. Maintainer: Jonathan Lange N)List) implementer)deferutils)failure)itrialutil)FailTestSkipTestSynchronousTestCase_wait_is_runningceZdZdZdfd ZdZeZdZdZdZ dZ d Z d Z d Z d Zd ZejdZdZdZdZdZdZdZfdZdZdZefdZxZS)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. runTestcJt|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__s 8lib/python3.11/site-packages/twisted/trial/_asynctest.pyrzTestCase.__init__-s# $$$$$cHfd}fd}|||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. c4d|)Nz$did not catch an error, instead got )failureException)ignorers r_cbz#TestCase.assertFailure.._cb@s&''AvAA rc|jr|jSdt|}|)Nz Expected: {!r} Got: {})checkvalueformatstrr)routputexpectedFailuresrs r_ebz#TestCase.assertFailure.._ebEsSw}./ 4}$5<<$c'll++F333r) addCallbacks)rdeferredr$rr%s` ` r assertFailurezTestCase.assertFailure9sS       4 4 4 4 4 4$$S#...rc( ddlm  fd}tj|t jt}t}tj |r6td |}tj|Stjtj|} |||fd|S)Nrreactorctjddd}tj|} ||dS#tj$rzd_}|/| |r ||YdS |YdSwxYw)Nz (z) still running at z secsT) r TimeoutErrorrFailureerrbackAlreadyCalledErrorcrash _timedOutgetTodoexpectedaddExpectedFailureaddError) deftodorr+resultrtimeouts r onTimeoutz TestCase._run..onTimeoutWs"JJZJJGJJJA""A - ! + - - - !%||~~# a(8(8#--dAt<<<<<<OOD!,,,,,, -sA A,C:CCcategoryz9{!r} is a generator function and therefore will never runcXrp|SN)activecancel)xcalls rzTestCase._run..|s!DKKMM;dkkmm@qr)twisted.internetr+ getTimeoutrsuppressWarningsr suppressDeprecationWarninggetattrinspectisgeneratorfunction TypeErrorr!rfail maybeDeferredrunWithWarningsSuppressed _getSuppress callLateraddBoth) rrr;r=methodexcr7rEr+r<s ``` @@@r_runz TestCase._runRs8,,,,,,//## - - - - - - - - -,* t}.@AAA  z**  &v . . #KRRC :c?? "    +T->->-@-@&    )Q77 @@@@AAArc|j|i|SrA)run)rargskwargss r__call__zTestCase.__call__stx((((rc||d|}||j|j|f|f|S)NsetUp callbackArgs errbackArgs)rXr&deferTestMethod _ebDeferSetUprignoredr;r7s r deferSetUpzTestCase.deferSetUpsM IIgv & &         rc@|jtr5||||j|jn?||||jtr|| d|SrA) rr addSkip_getSkipReasonr_r r6KeyboardInterruptstopdeferRunCleanupsrrr;s rrdzTestCase._ebDeferSetUps 7= " "  NN4!4!4TZ!O!O P P P P OOD' * * *w}.//  $$T6222rc||j|}||j|j|f|f||j|||j||S)Nr`)rX_testMethodNamer&_cbDeferTestMethod_ebDeferTestMethodrUrm deferTearDownres rrczTestCase.deferTestMethods~ IId*F 3 3   #  #     $'000 $$f---rc|)|||nd|_|S)NT)r3addUnexpectedSuccess_passed)rrfr;s rrqzTestCase._cbDeferTestMethods= <<>> %  ' 'dllnn = = = =DLrc r|}|.||r||||dS||jt r|||dS|tr,|||| dS|trD| || t||j|jdS|||dSrA)r3r4r5rrr addFailurerkr6rlr rirjrLrpr )rr9r;r:s rrrzTestCase._ebDeferTestMethods'||~~   a 0 0   % %dAt 4 4 4 4 4 WWT*H 5 5 %   dA & & & & & WW& ' ' % OOD! $ $ $ KKMMMMM WWX   % NNd))'$8L*M*MqwWW      OOD! $ $ $ $ $rch|d|}||j||S)NtearDown)rX addErrback_ebDeferTearDownres rrszTestCase.deferTearDowns1 IIj& ) ) T*F333rc||||jtr|d|_dSNF)r6rrkrlrvrns rr|zTestCase._ebDeferTearDownsCg&&& 7=* + +  KKMMM rc#pKg}t|jdkrw|j\}}} ||i|Vn6#t$r)|t jYnwxYwt|jdkw|D]}|||d|_ dS)zf Run any scheduled cleanups and report errors (if any) to the result. object. rFN) len _cleanupspop Exceptionappendrr.r6rv)rrfr;failuresfuncr[r\r9s rrmzTestCase.deferRunCleanupss $.!!A%%!%!3!3!5!5 D$ 3dD+F++++++ 3 3 3 1 122222 3 $.!!A%% ! !A OOD! $ $ $ DLL ! !s A0A98A9c tj||}|sd|_n>#t$r1||t jd|_YnwxYw|j D]}|||d|_ | | |jr| |dSdSr~) r _JanitorpostCaseCleanuprv BaseExceptionr6rr. _observer getErrorsflushLoggedErrors_removeObserver addSuccess)rr;cleanerrors r_cleanUpzTestCase._cleanUps !M$//??AAE %$  ! ! ! OOD'/"3"3 4 4 4 DLLL !^--// ! !E OOD% ( ( ( DLL      < $   d # # # # # $ $s038A.-A.c tj||dS#t$r+||t jYdSwxYwrA)r rpostClassCleanuprr6rr.)rr;s r _classCleanUpzTestCase._classCleanUpsk 5 M$ ' ' 8 8 : : : : : 5 5 5 OOD'/"3"3 4 4 4 4 4 4 5s'+1A A cfd}|S)z Create a method which wraps the reactor method C{name}. The new method issues a deprecation warning and calls the original. cptjddddtj|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.) stacklevelr?)warningswarnrK_reactorMethods)akwnamers r_z&TestCase._makeReactorMethod.._sY MM26ttt=+     .4'-q7B77 7r)rrrs`` r_makeReactorMethodzTestCase._makeReactorMethods)  8 8 8 8 8 8rci|_dD]>}t|||j|<t||||?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. )r1iteraterlN)rrLsetattrr)rr+rs r_deprecateReactorzTestCase._deprecateReactorsc "0 B BD)0$)?)?D  & GT4#:#:4#@#@ A A A A B Brct|jD]\}}t|||i|_dS)z Restore the deprecated reactor methods. Undoes what L{_deprecateReactor} did. @param reactor: The Twisted reactor. N)ritemsr)rr+rrVs r_undeprecateReactorzTestCase._undeprecateReactor sI!06688 + +LD& GT6 * * * *!rcddlm}||d|_ |d|} ||||||n/#||||wxYw ||dS#||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. rr*FN) rGr+rr2rg_waitrrr)rr;r+r7s r_runFixturesAndTestzTestCase._runFixturesAndTests -,,,,, w''' .f--A + 1  f%%%""6**** f%%%""6*****  $ $W - - - - -D $ $W - - - -s"CA;+C;,B''CCc>tj|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)rr9r[r\rs rrzTestCase.addCleanup*s,"uww!!5d555f555rc*|SrA)rS)rs r getSuppresszTestCase.getSuppress4s  """rctj|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. r<z)'timeout' attribute needs to be a number.r>) r acquireAttribute_parentsDEFAULT_TIMEOUT_DURATIONfloat ValueErrorrOrrrK)rr<s rrHzTestCase.getTimeout7s' M9d&C   1>> !I& 1 1 1 M;FX    0 0 0 0 1s68A10A1c|rtdddlmgfd}fd}tj|t jdt}fd}tj|t jdt}|d  | |r d | d S| ||_  ` n#` wxYws|j r d | d St#d | wxYw) zATake a Deferred that only ever callbacks. Block until it happens.z_wait is not reentrantrr*c:|dSdSrA)r)anyresultss rrzTestCase._wait..appendWs)"s######"rc8dSdSrAr1)ignr+rs rr1zTestCase._wait..crash[s"" #"rzreactor\.crash cannot be used.*)messager?c0dSrArr*srrlzTestCase._wait..stopfs MMOOOOOrN) RuntimeErrorrGr+rrIr rJrKrrUrrlrZr2rk)rr7runningrr1rlr+rs @@rrzTestCase._waitNs  9788 8,,,,,, $ $ $ $ $      &  M:EW          %  M:EW      t$  IIf    <G KKMMMMM= IIe   GL ! LLGL     $. "G KKMMMMM$%% %G KKMMMMs0D9D9+D?D9D D9+D99E)r)__name__ __module__ __qualname____doc__rr(failUnlessFailurerXr]rgrdrcrqrrrsr|rinlineCallbacksrmrrrrrrrrrHr r __classcell__)rs@rrrs   % % % % % %///.&+++Z)))333   %%%    !!!"$$$ 555 $ B B B " " "...*66666###111. 0GGGGGGGGrr)rrMrtypingrzope.interfacerrGrrtwisted.pythonr twisted.trialrr twisted.trial._synctestr r r r __annotations__ ITestCaserrrrrs &&&&&& *)))))))""""""&&&&&&&&KKKKKKKKKK!$t*!!! V wwwww"wwwwwr