VhdZddlZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZGdd eZGd d eZy) z* Tests for C{await} support in Deferreds. N)NoReturn)DeferredensureDeferredfail maybeDeferredsucceed)Clock)Failure)TestCaseceZdZdZy)SampleExceptionz2 A specific sample exception for testing. N)__name__ __module__ __qualname____doc__V/home/dcms/DCMS/lib/python3.12/site-packages/twisted/internet/test/test_defer_await.pyr r srr c`eZdZdZd dZd dZd dZd dZd dZd dZ d d Z d d Z d d Z d d Z y) AwaitTestsz@ Tests for using Deferreds in conjunction with PEP-492. Ncnt}|j}|j|t|y)z< C{Deferred.__await__} returns an iterable. N)r __await__ assertEqualiter)selfdawaitedDeferreds rtest_awaitReturnsIterablez$AwaitTests.test_awaitReturnsIterable$s+%J++- $*?@rcdtffd }dtfd|}|j|tjt j |}|j|t|j |}|j|dy)zU L{Deferred.fromCoroutine} will turn a coroutine into a L{Deferred}. returnc`Ktd}|d{d{}|S77w)Nbarr)rresrun2s rrunz2AwaitTests.test_deferredFromCoroutine..run1s.AGG,CJ s.*.,..c<Ktd}|d{}|S7wNfoor#rr$s rr%z3AwaitTests.test_deferredFromCoroutine..run27 A'CJ r)N)strassertIsInstancetypes CoroutineTyper fromCoroutinesuccessResultOfr)rr&rrr$r%s @rtest_deferredFromCoroutinez%AwaitTests.test_deferredFromCoroutine,s~  3   C  E a!4!45  " "1 % a*""1% e$rcdtfd}tj|}|j|}|j |dy)zc L{Deferred.fromCoroutine} allows a function to C{await} on a L{Deferred}. r c<Ktd}|d{}|S7wr(r#r*s rr&z"AwaitTests.test_basic..runNr+r,r)N)r-rr1r2rrr&rr$s r test_basiczAwaitTests.test_basicHsB  3   " "35 )""1% e$rc~dtfd}t|}|j|}|j|dy)zS L{ensureDeferred} allows a function to C{await} on a L{Deferred}. r c<Ktd}|d{}|S7wr(r#r*s rr&z0AwaitTests.test_basicEnsureDeferred..run\r+r,r)N)r-rr2rr7s rtest_basicEnsureDeferredz#AwaitTests.test_basicEnsureDeferredWs<  3  35 !""1% e$rc dtfd}tj|}|j|}|j t |j t|j |j jdy)z An exception in a coroutine scheduled with L{Deferred.fromCoroutine} will cause the returned L{Deferred} to fire with a failure. r cNKtd}|d{td7w)Nr)Oh no!)r ValueError)rs rr&z&AwaitTests.test_exception..runks&AGGX& & s %#%)r>N) rrr1failureResultOfrtypevaluer?argsr7s rtest_exceptionzAwaitTests.test_exceptionesb  '8 '  " "35 )""1% cii*5 5rcdd}t|dfd }|jtj|}|j d|j |j d|j y)z When a Deferred is awaited upon that has already failed with a Failure that has a traceback, both the place that the synchronous traceback comes from and the awaiting line are shown in the traceback. NctN)r rrrraiseszCAwaitTests.test_synchronousDeferredFailureTraceback..raises|s !# #rc$Kd{S7wrGritsrdoomedzCAwaitTests.test_synchronousDeferredFailureTraceback..doomed8O8  , in doomed z , in raises r N)rr@rr1assertIn getTraceback)rrHrLfailurerKs @r(test_synchronousDeferredFailureTracebackz3AwaitTests.test_synchronousDeferredFailureTracebackusf $6 " &&x'='=fh'GH ow';';'=> ow';';'=>rc^dtfd}tdfd }tj|}|j|j ||j |}|j d|j|j d|jy)z When a Deferred is awaited upon that later fails with a Failure that has a traceback, both the place that the synchronous traceback comes from and the awaiting line are shown in the traceback. r cJ t#t$r tcYSwxYwrG)r r rrrreturnsFailurezEAwaitTests.test_asyncDeferredFailureTraceback..returnsFailures' !%''" !y  !s ""Nc$Kd{S7wrGrrJsrrLz=AwaitTests.test_asyncDeferredFailureTraceback..doomedrMrNrOz, in returnsFailure rP)r rr1assertNoResulterrbackr@rQrR)rrWrLstartedrSrKs @r"test_asyncDeferredFailureTracebackz-AwaitTests.test_asyncDeferredFailureTracebacks ! ! &Z ((2 G$ >#$&&w/ ow';';'=> -w/C/C/EFrctgdtffd dtffd }tj|}j d|j ddgj d|j gdj d|j gdj d|j gd |j |}|j |d y ) z A coroutine scheduled with L{Deferred.fromCoroutine} that awaits a L{Deferred} suspends its execution until the inner L{Deferred} fires. r cKjdt}jd|jd|d{jdy7w)NYay!appendr callLatercallback)rreactorsectionss rrunonez'AwaitTests.test_twoDeep..runonesK OOA 'zA   aQ /GG OOA  s>AAAcKjdd{}jdt}jd|jd|d{jd|S7Y7w)Nr`rc)resultrrgrirhs rr&z$AwaitTests.test_twoDeep..runsh OOA !8^F OOA 'zA   aQ /GG OOA M $ s"A=A9AA=!A;"A=;A=g?r`r_皙?)r`r_rark)r`r_rarkrlrbN)r r-rr1advancerr2)rr&rr$rgrirhs @@@r test_twoDeepzAwaitTests.test_twoDeeps ' c  3   " "35 ) Aq6* <0 <0 ?3""1% f%rcdtffd }jtj|}j |dy)zO Awaiting an already failed Deferred will raise the exception. r cK ttdd{y7#t$r&}j|jdYd}~yd}~wwxYww)NBoom)rsr`r)rr?rrC)ers rtestz%AwaitTests.test_reraise..testsO :f-... /   3 s6A$"$A$ AA AAAr`N)intr2rr1r)rrur$s` r test_reraisezAwaitTests.test_reraises>  C ""8#9#9$&#AB a rctdfd }tj|}jd|j |}|j |dy)zd Awaiting a paused & chained Deferred will give the result when it has one. NcKt}t|jfd|jdjdjd|d{S7w)NcSrGr)ignoredd2s rz7AwaitTests.test_chained..test..s"rrbye)r addCallbackrfre)rr|rgs @rruz%AwaitTests.test_chained..testsM ( A!)B MM, - JJt    ae 47N7sAA' A%!A'rnr~rP)r rr1ror2r)rrurr$rgs @r test_chainedzAwaitTests.test_chainedsQ '   " "46 *""1% e$rrP)rrrrrr4r8r;rDrTr\rprwrrrrrrs<A%8 % %6 ?(G2)&V! %rr)rr/typing_extensionsrtwisted.internet.deferrrrrrtwisted.internet.taskr twisted.python.failurer twisted.trial.unittestr Exceptionr rrrrrsA  &(*+i R%R%r