U ¹êW[*9ã@s’dZddlmZmZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZmZmZgZeejƒGd d „d eƒƒZdS) zP Things likely to be used by writers of unit tests. Maintainer: Jonathan Lange é)ÚdivisionÚabsolute_importN)Ú implementer)ÚdeferÚutils)Úfailure)ÚitrialÚutil)ÚFailTestÚSkipTestÚSynchronousTestCasecsæeZdZdZd3‡fdd„ Zdd„ZeZdd„Zd d „Zd d „Z d d„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Z‡fd+d,„Zd-d.„Zd/d0„Zefd1d2„Z‡ZS)4Ú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. ÚrunTestcstt|ƒ |¡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)Úsuperr Ú__init__)ÚselfÚ methodName©Ú __class__©ú:/usr/lib/python3/dist-packages/twisted/trial/_asynctest.pyr.s 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. csˆ d|f¡‚dS)Nz&did not catch an error, instead got %r)ÚfailureException)Úignore©rrrÚ_cbAsÿz#TestCase.assertFailure.._cbcs.|jˆŽr|jSdˆt|ƒf}ˆ |¡‚dS)Nz Expected: %r Got: %s)ÚcheckÚvalueÚstrr)rÚoutput©ÚexpectedFailuresrrrÚ_ebEs   ÿz#TestCase.assertFailure.._eb)Ú addCallbacks)rZdeferredr rr!rrrÚ assertFailure;s zTestCase.assertFailurecs ddlm‰ˆ ¡‰‡‡‡‡‡fdd„}t |tjtd¡}tˆˆƒ}t   |¡rht d|fƒ}t   |¡St  tjˆ ¡|¡}ˆ ˆ||¡‰| ‡fdd„¡|S) Nr©ÚreactorcsŒt dˆˆˆf¡}t |¡}z| |¡WnZtjk r†ˆ ¡dˆ_ˆ ¡}|dk rv|  |¡rvˆ  ˆ||¡n ˆ  ˆ|¡YnXdS)Nz %r (%s) still running at %s secsT) rÚ TimeoutErrorrÚFailureZerrbackZAlreadyCalledErrorÚcrashÚ _timedOutÚgetTodoÚexpectedÚaddExpectedFailureÚaddError)ÚdÚeÚfÚtodo)rr%ÚresultrÚtimeoutrrÚ onTimeoutSsÿ z TestCase._run..onTimeout©Úcategoryz7%r is a generator function and therefore will never runcsˆ ¡rˆ ¡p|S©N)ZactiveZcancel)Úx)ÚcallrrÚrózTestCase._run..)Útwisted.internetr%Ú getTimeoutrÚsuppressWarningsr ÚsuppressÚDeprecationWarningÚgetattrÚinspectZisgeneratorfunctionÚ TypeErrorrZfailZ maybeDeferredZrunWithWarningsSuppressedÚ _getSuppressZ callLaterÚaddBoth)rrr2r4ÚmethodÚexcr.r)r9rr%r2rr3rÚ_runPs.  ÿ  ÿÿ ÿz TestCase._runcOs |j||ŽSr7)Úrun)rÚargsÚkwargsrrrÚ__call__vszTestCase.__call__cCs*| d|¡}|j|j|j|f|fd|S)NÚsetUp©Z callbackArgsZ errbackArgs)rHr"ÚdeferTestMethodÚ _ebDeferSetUp©rÚignoredr2r.rrrÚ deferSetUpzs   þzTestCase.deferSetUpcCsN| t¡r$| || |j|j¡¡n| ||¡| t¡rB| ¡|  d|¡Sr7) rr ÚaddSkipÚ_getSkipReasonrMrr-ÚKeyboardInterruptÚstopÚdeferRunCleanups©rrr2rrrrP‚s    zTestCase._ebDeferSetUpcCsH| |j|¡}|j|j|j|f|fd| |j|¡| |j|¡|S)NrN)rHÚ_testMethodNamer"Ú_cbDeferTestMethodÚ_ebDeferTestMethodrErXÚ deferTearDownrQrrrrOŒs þzTestCase.deferTestMethodcCs(| ¡dk r| || ¡¡nd|_|S)NT)r*ZaddUnexpectedSuccessÚ_passed)rrRr2rrrr[–s zTestCase._cbDeferTestMethodcCs | ¡}|dk r*| |¡r*| |||¡nr| |jt¡rF| ||¡nV| t¡rf| ||¡|  ¡n6| t ¡r|  ||  t ||jƒ|j¡¡n | ||¡dSr7)r*r+r,rrr Z addFailurerVr-rWr rTrUrArZr)rr0r2r1rrrr\žs    þzTestCase._ebDeferTestMethodcCs| d|¡}| |j|¡|S)NZtearDown)rHZ addErrbackÚ_ebDeferTearDownrQrrrr]¯s zTestCase.deferTearDowncCs(| ||¡| t¡r| ¡d|_dS©NF)r-rrVrWr^rYrrrr_µs  zTestCase._ebDeferTearDowncCs| ¡}| |j|¡|S)zd Run any scheduled cleanups and report errors (if any to the result object. )Ú _runCleanupsZ addCallbackÚ_cbDeferRunCleanupsrQrrrrX¼szTestCase.deferRunCleanupscCs@|D]6\}}|tjkr| ||¡| t¡r4| ¡d|_qdSr`)rÚFAILUREr-rrVrWr^)rZcleanupResultsr2ÚflagZ testFailurerrrrbÆs     zTestCase._cbDeferRunCleanupscCsˆzt ||¡ ¡}|sd|_Wn"| |t ¡¡d|_YnX|j ¡D]}| ||¡d|_qL|  ¡|  ¡|jr„|  |¡dSr`) r Ú_JanitorZpostCaseCleanupr^r-rr'Z _observerZ getErrorsZflushLoggedErrorsZ_removeObserverZ addSuccess)rr2ZcleanÚerrorrrrÚ_cleanUpÏs   zTestCase._cleanUpcCs6zt ||¡ ¡Wn| |t ¡¡YnXdSr7)r reZpostClassCleanupr-rr')rr2rrrÚ _classCleanUpàszTestCase._classCleanUpcs‡‡fdd„}|S)z• Create a method which wraps the reactor method C{name}. The new method issues a deprecation warning and calls the original. cs(tjdˆˆfdtdˆjˆ||Ž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.é)Ú stacklevelr6)ÚwarningsÚwarnr@Ú_reactorMethods)ÚaÚkw©ÚnamerrrÚ_ìsýüz&TestCase._makeReactorMethod.._r)rrqrrrrprÚ_makeReactorMethodçszTestCase._makeReactorMethodcCs6i|_dD]&}t||ƒ|j|<t||| |¡ƒq 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. )r(ZiteraterWN)rmrAÚsetattrrs)rr%rqrrrÚ_deprecateReactoröszTestCase._deprecateReactorcCs*|j ¡D]\}}t|||ƒq i|_dS)z• Restore the deprecated reactor methods. Undoes what L{_deprecateReactor} did. @param reactor: The Twisted reactor. N)rmÚitemsrt)rr%rqrFrrrÚ_undeprecateReactorszTestCase._undeprecateReactorcCsHdd„}g}t|jƒdkr>|j ¡\}}}| ||||ƒ¡q t |¡S)z‹ Run the cleanups added with L{addCleanup} in order. @return: A C{Deferred} that fires when all cleanups are run. cs‡‡‡fdd„S)Ncs ˆˆˆŽSr7rr©rJr0rKrrr:r;z>TestCase._runCleanups.._makeFunction..r)r0rJrKrrxrÚ _makeFunctionsz,TestCase._runCleanups.._makeFunctionr)ÚlenZ _cleanupsÚpopÚappendr Z_runSequentially)rryZ callablesr0rJrKrrrras zTestCase._runCleanupsc Csdddlm}| |¡d|_z6| d|¡}z| |¡W5| |¡| |¡XW5| |¡XdS)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) r<r%rur)rwrSrgrhÚ_wait)rr2r%r.rrrÚ_runFixturesAndTests    zTestCase._runFixturesAndTestcstt|ƒj|f|ž|Ž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. )rr Ú addCleanup)rr0rJrKrrrr4szTestCase.addCleanupcCs| ¡Sr7)rDrrrrÚ getSuppress?szTestCase.getSuppressc CsNt |jdtj¡}z t|ƒWSttfk rHtjdt dtjYSXdS)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. r3z)'timeout' attribute needs to be a number.r5N) r ZacquireAttributeZ_parentsZDEFAULT_TIMEOUT_DURATIONÚfloatÚ ValueErrorrCrkrlr@)rr3rrrr=Cs ÿ ÿzTestCase.getTimeoutcsâ|r tdƒ‚ddlm‰g‰‡fdd„}‡‡fdd„}t |tjdtd ¡}‡fd d „}t |tjdtd ¡}| d ¡zV|  |¡ˆrW¢Dd S|  |¡|ˆ_ z ˆ  ¡W5ˆ` Xˆs¾|j rÆW¢d Stƒ‚W5d ‰|  ¡Xd S) zJTake a Deferred that only ever callbacks. Block until it happens. z_wait is not reentrantrr$csˆdk rˆ |¡dSr7)r|)Úany)Úresultsrrr|aszTestCase._wait..appendcsˆdk rˆ ¡dSr7©r()Zign©r%r„rrr(dszTestCase._wait..crashzreactor\.crash cannot be used.*)Úmessager6cs ˆ ¡dSr7r…rr$rrrWjszTestCase._wait..stopN)Ú RuntimeErrorr<r%rr>r r?r@r|r{rErWrIr)rV)rr.Úrunningr|r(rWrr†rr}YsD  ÿÿ ÿÿ      zTestCase._wait)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr#ZfailUnlessFailurerHrLrSrPrOr[r\r]r_rXrbrgrhrsrurwrar~rr€r=Ú_wait_is_runningr}Ú __classcell__rrrrr s4 &      r )rZ __future__rrrBrkZzope.interfacerr<rrZtwisted.pythonrZ twisted.trialrr Ztwisted.trial._synctestr r r rŽZ ITestCaser rrrrÚs