tfdZddlZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZGdd eZGd d eZdS) z* Tests for C{await} support in Deferreds. N)NoReturn)DeferredensureDeferredfail maybeDeferredsucceed)Clock)Failure)TestCaseceZdZdZdS)SampleExceptionz2 A specific sample exception for testing. N)__name__ __module__ __qualname____doc__e/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/internet/test/test_defer_await.pyr r srr cbeZdZdZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z dS) AwaitTestsz@ Tests for using Deferreds in conjunction with PEP-492. returnNct}|}||t|dS)z< C{Deferred.__await__} returns an iterable. N)r __await__ assertEqualiter)selfdawaitedDeferreds rtest_awaitReturnsIterablez$AwaitTests.test_awaitReturnsIterable$s>%JJ++-- $*?*?@@@@@rc@dtffd }dtfd|}||tjt j|}||t||}||ddS)zU L{Deferred.fromCoroutine} will turn a coroutine into a L{Deferred}. rcZKtd}|d{Vd{V}|S)Nbarr)rresrun2s rrunz2AwaitTests.test_deferredFromCoroutine..run1s?AGGGGGGG,,,,,,CJrc8Ktd}|d{V}|SNfoor#rr$s rr%z3AwaitTests.test_deferredFromCoroutine..run27'A''''''CJrr)N)strassertIsInstancetypes CoroutineTyper fromCoroutinesuccessResultOfr)rr&rrr$r%s @rtest_deferredFromCoroutinez%AwaitTests.test_deferredFromCoroutine,s  3        C     CEE a!4555  "1 % % a***""1%% e$$$$$rcdtfd}tj|}||}||ddS)zc L{Deferred.fromCoroutine} allows a function to C{await} on a L{Deferred}. rc8Ktd}|d{V}|Sr(r#r*s rr&z"AwaitTests.test_basic..runNr+rr)N)r,rr0r1rrr&rr$s r test_basiczAwaitTests.test_basicHsd  3      "3355 ) )""1%% e$$$$$rcdtfd}t|}||}||ddS)zS L{ensureDeferred} allows a function to C{await} on a L{Deferred}. rc8Ktd}|d{V}|Sr(r#r*s rr&z0AwaitTests.test_basicEnsureDeferred..run\r+rr)N)r,rr1rr6s rtest_basicEnsureDeferredz#AwaitTests.test_basicEnsureDeferredWsa  3     3355 ! !""1%% e$$$$$rcdtfd}tj|}||}|t |jt||jjddS)z An exception in a coroutine scheduled with L{Deferred.fromCoroutine} will cause the returned L{Deferred} to fire with a failure. rcRKtd}|d{Vtd)Nr)Oh no!)r ValueError)rs rr&z&AwaitTests.test_exception..runks0AGGGGGGGX&& &r)r=N) rrr0failureResultOfrtypevaluer>argsr6s rtest_exceptionzAwaitTests.test_exceptiones  '8 ' ' ' '  "3355 ) )""1%% ci*555 55555rc8dd}t|dfd }|tj|}|d||d|dS)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. rNctN)r rrrraiseszCAwaitTests.test_synchronousDeferredFailureTraceback..raises|s!## #rcKd{VSrFritsrdoomedzCAwaitTests.test_synchronousDeferredFailureTraceback..doomed888888Or , in doomed z , in raises rN)rr?rr0assertIn getTraceback)rrGrKfailurerJs @r(test_synchronousDeferredFailureTracebackz3AwaitTests.test_synchronousDeferredFailureTracebackus $ $ $ $6 " "      &&x'=ffhh'G'GHH ow';';'='=>>> ow';';'='=>>>>>rcdtfd}tdfd }tj|}|||||}|d||d|dS)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. rc\ t#t$rtcYSwxYwrF)r r rrrreturnsFailurezEAwaitTests.test_asyncDeferredFailureTraceback..returnsFailures> !%'''" ! ! !yy    !s ++NcKd{VSrFrrIsrrKz=AwaitTests.test_asyncDeferredFailureTraceback..doomedrLrrMz, in returnsFailure rN)r rr0assertNoResulterrbackr?rOrP)rrUrKstartedrQrJs @r"test_asyncDeferredFailureTracebackz-AwaitTests.test_asyncDeferredFailureTracebacks ! ! ! ! ! &ZZ      (22 G$$$ >>##$$$&&w// ow';';'='=>>> -w/C/C/E/EFFFFFrc\tgdtffd dtffd }tj|}d|ddgd|gdd|gdd|gd ||}||d d S) z A coroutine scheduled with L{Deferred.fromCoroutine} that awaits a L{Deferred} suspends its execution until the inner L{Deferred} fires. rcKdt}d|jd|d{VddS)NYay!appendr callLatercallback)rreactorsectionss rrunonez'AwaitTests.test_twoDeep..runonesd OOA   'zzA   aQ / / /GGGGGGG OOA   6rcKdd{V}dt}d|jd|d{Vd|S)Nr^ra)resultrrergrfs rr&z$AwaitTests.test_twoDeep..runs OOA   !688^^^^^^F OOA   'zzA   aQ / / /GGGGGGG OOA   Mrg?r^r]皙?)r^r]r_ri)r^r]r_rirjr`N)r r,rr0advancerr1)rr&rr$rergrfs @@@r test_twoDeepzAwaitTests.test_twoDeepse '' c        3          "3355 ) ) Aq6*** <<<000 <<<000 ???333""1%% f%%%%%rcdtffd }tj|}|ddS)zO Awaiting an already failed Deferred will raise the exception. rcK ttdd{Vn3#t$r&}|jdYd}~dSd}~wwxYwdS)NBoom)rqr^r)rr>rrB)ers rtestz%AwaitTests.test_reraise..tests :f--..........     333qqqqq 1s"( AAAr^N)intr1rr0r)rrsr$s` r test_reraisezAwaitTests.test_reraisesj  C      ""8#9$$&&#A#ABB a     rctdfd }tj|}d||}||ddS)zd Awaiting a paused & chained Deferred will give the result when it has one. rNcKt}t|fd|ddjd|d{VS)NcSrFr)ignoredd2s rz7AwaitTests.test_chained..test..s"rrbye)r addCallbackrdrc)rrzres @rrsz%AwaitTests.test_chained..testss ( A!)B MM,,,, - - - JJt      ae 4 4 4777777Nrrlr|rN)r rr0rmr1r)rrsrr$res @r test_chainedzAwaitTests.test_chaineds ''        "4466 * *""1%% e$$$$$rrN)rrrrrr3r7r:rCrRrZrnrur~rrrrrsAAAA%%%%8 % % % % % % % %6666 ????(GGGG2)&)&)&)&V!!!! %%%%%%rr)rr.typing_extensionsrtwisted.internet.deferrrrrrtwisted.internet.taskr twisted.python.failurer twisted.trial.unittestr Exceptionr rrrrrs  &&&&&&('''''******++++++i R%R%R%R%R%R%R%R%R%R%r