+ Dfj[SRt^RIHtHtHt^RIHt!RR4t!RR4t!RR]P4t !R R ] 4t !R R ]P4t !R R]P4tR#)z_ This module contains tests for L{twisted.internet.task.Cooperator} and related functionality. )deferreactortask)unittestc0a]tRt^toRtRtRtRtVtR#)FakeDelayedCallz9 Fake delayed call which lets us simulate the scheduler. c WnRVnR#)z A function to run, later. FN)func cancelledselfr s&&>/usr/lib/python3/dist-packages/twisted/test/test_cooperator.py__init__FakeDelayedCall.__init__s c RVnR#)z Don't run my function later. TN)r r s&r cancelFakeDelayedCall.cancels r)r r N) __name__ __module__ __qualname____firstlineno____doc__rr__static_attributes____classdictcell__ __classdict__s@r rrsrrc6a]tRt^!toRtRtRtRtRtVt R#) FakeSchedulerz' A fake scheduler for testing against. c .VnR#)z4 Create a fake scheduler with a list of work to do. Nworkrs&r rFakeScheduler.__init__&s  rc R\V4pVPPV4V#)z+ Schedule a unit of work to be done later. )rr"append)r thunkunits&& r __call__FakeScheduler.__call__,s%u%  rc VP.uqnVF'pVP'dKVP4K) R#)z< Do all of the work that is currently available to be done. N)r"r r )r r"r's& r pumpFakeScheduler.pump4s/))RiD>>> rr!N) rrrrrrr(r+rrrs@r rr!s rrca]tRt^>toRtRtRtRtRtRt Rt Rt R t R t R tR tR tRtRtRtRtRtVtR#)CooperatorTestsdonecXVP\P4VP#N)traprSchedulerStoppedRESULT)r errs&&r ebIterCooperatorTests.ebIterAs &&'{{rc&VP4R#r1)fail)r igns&&r cbIterCooperatorTests.cbIterEs  rc HaaV3RloS!R4PV3Rl4#)zE Test that Cooperators refuse new tasks when they have been stopped. c<\P!4pVP4VP\ R4V4pVP SP 4VPSP4VP V3Rl4#)c<<SPVSP4#r1 assertEqualr4resultr s&r NCooperatorTests.testStoppedRejectsNewTasks..testwith..Ss0@0@0Ur) r Cooperatorstop coiterateiter addCallbackr; addErrbackr6)stuffcdr s& r testwith.testwithMs\!A FFH DHe,A MM$++ & LL %==!UV VrNc:<S!\P!44#r1)rDeferred)r:rPs&r rD.Ushu~~?O6Pr)rK)r rPsf@r testStoppedRejectsNewTasks*CooperatorTests.testStoppedRejectsNewTasksHs!  W~))*PQQrc 0aa\P!4pRoRSnVPS!44pVP SP 4VP SP4VP4VV3RlpVP V4V#)zY Test that a running iterator will not run to completion when the cooperator is stopped. c38"\^4RjxL R#L5i)N)rangerFrr myiter/CooperatorTests.testStopRunning..myiter^sQx  s cx<SPVSP4SPSPR4R#)N)rAr4value)rCr[r s&r doasserts2CooperatorTests.testStopRunning..doassertsgs+   VT[[ 1   V\\2 .rr_) rrGr`rIrKr;rLr6rH)r rNrOrar[sf @r testStopRunningCooperatorTests.testStopRunningWsp OO   KK ! dkk" T[[!  / i rc aaaa\P!4o\P!4oVVV3Rlp\P!4oSP V!44pVV3RlpSP V4VP SP 4VPSP4VP V3Rl4#)z An iterator run with L{Cooperator.coiterate} paused on a L{Deferred} yielded by that iterator will fire its own L{Deferred} (the one returned by C{coiterate}) when L{Cooperator.stop} is called. c3|<"\P!^SPR4SxSP4R#5iN)r callLatercallbackr9) outstandingDr testControlDsr r[3CooperatorTests.testStopOutstanding..myiterws-   a!6!6 =  IIKs9<cJ<SP4SPR4R#) arglebargleN)rHrj)r:rNrks&r stopAndGo6CooperatorTests.testStopOutstanding..stopAndGos FFH  ! !- 0rc<<SPVSP4#r1r@rBs&r rD5CooperatorTests.testStopOutstanding..sD,<,.myiters#n$s rrGrI assertFailurerx)r rNr[rOs& r testUnexpectedError#CooperatorTests.testUnexpectedErrors8 OO  % KK !!!!\22rcRp\P!4pVPV!44pVPV\4#)c3"\P!4p\P!^VP\ 44VxR#5irg)rrSrrierrbackrx)Ds r r[@CooperatorTests.testUnexpectedErrorActuallyLater..myiters. A   aLN ;GsAArzr r[rNrOs& r testUnexpectedErrorActuallyLater0CooperatorTests.testUnexpectedErrorActuallyLaters8  OO  KK !!!!\22rcRp\P!4pVPV!44pVPV\4#)c3N"\P!\44xR#5ir1)rr9rxrFrr r[CCooperatorTests.testUnexpectedErrorNotActuallyLater..myiters**\^, ,s#%rzrs& r #testUnexpectedErrorNotActuallyLater3CooperatorTests.testUnexpectedErrorNotActuallyLaters8 - OO  KK !!!!\22rcaaa.oV3Rlp.ROo\P!4p.pSF)pVPVPV!V444K+ \P !V4P VVV3Rl4#)c3L<"VFpSPV4RxK R#5ir1r%)thingsthLs& r r[/CooperatorTests.testCooperation..myiters   s!$c \<SP\S4\\S!R44#NrF)rAtuplesumzip)r:rgroupsOfThingsr s&r rD1CooperatorTests.testCooperation..s"((q3sN7KR3PQr)abcr^rYdef))rrGr%rIr DeferredListrK)r r[rNtasksrMrrsf @@r testCooperationCooperatorTests.testCooperationsk   > OO #E LLVE]3 4$!!%(44 Q  rcaa.oVV3Rlp!RR4o\P!SR7pVPV!44PVP4VP P 4VP4VP4VPSP4VPS\\^ 444R#)c3z<"\^d4F&pSPV4V^ 8XdRSnVxK( R#5i)dTN)rZr%stopped)i_TPFoutputs r r[6CooperatorTests.testResourceExhaustion..myiters33Z a 6#'DL  s8;c*a]tRt^toRtRtRtVtR#)4CooperatorTests.testResourceExhaustion.._TPFFcVP#r1)rrs&r r(=CooperatorTests.testResourceExhaustion.._TPF.__call__s ||#rrFN)rrrrrr(rrrs@r rrsG $ $rr)terminationPredicateFactoryN)rrGrIrLr6 _delayedCallr_tickrH assertTruerrAlistrZ)r r[rNrrs& @@r testResourceExhaustion&CooperatorTests.testResourceExhaustions  $ $ OO = FH((5      % eBi1rc aaaaa.o!RR4oVVV3Rlp\P!VRR7oSP\R 44p.oVV3RlpVP V4^pS'gXV^, pS'd+SP ^4P 4V^, pK2V^28gKLSPR4K_R#) a If a callback to a deferred returned by coiterate calls coiterate on the same Cooperator, we should make sure to only do the minimal amount of scheduling work. (This test was added to demonstrate a specific bug that was found while writing the scheduler.) c8a]tRt^toRtV3RlRltRtVtR#)9CooperatorTests.testCallbackReCoiterate..FakeCallcWnR#r1r r s&&r rBCooperatorTests.testCallbackReCoiterate..FakeCall.__init__s rc <V^8dQhRS[/#)rreturn)str)formatrs"r __annotate__FCooperatorTests.testCallbackReCoiterate..FakeCall.__annotate__s 3 3# 3rc$RVP: R2#)z rrs&r __repr__BCooperatorTests.testCallbackReCoiterate..FakeCall.__repr__s#DII=22rrN)rrrrrrrrrs@r FakeCallrs ! 3 3rrcz<SPS\S44SPS!V44SR,#)r^r_) assertFalsereprr%)frcallsr s&r sched6CooperatorTests.testCallbackReCoiterate..scheds1   UDK 0 LL! %9 rcR#)cR#TrFrFrr rDKCooperatorTests.testCallbackReCoiterate....srrFrFrr rD9CooperatorTests.testCallbackReCoiterate..sr schedulerrcn<SP\R44PSP4R#r)rIrJaddBothr%)r:rNr/s&r anotherTask.anotherTasks" KKR ! ) )$++ 6rzCooperator took too longNrF)rrGrIrJrKpopr r9) r rrOrr"rrNrr/s f @@@@r testCallbackReCoiterate'CooperatorTests.testCallbackReCoiterates 3 3  OO9M  KKR ! 7 k" AID ! !!# by 45 rc aR.oV3Rlp\P!VR7pVP\^^.44pVP\^^.44pVP S^,P VP 4VP4VPS^,P4VP VPS^,4VP4VPS^,P4VPVP4VP\^^.44VPS^,P4VP VPS^,4R#)z If the last task in a Cooperator is removed, the scheduled call for the next tick is cancelled, since it is no longer necessary. This behavior is useful for tests that want to assert they have left no reactor state behind when they're done. Nc2<\V4S^&S^,#)rh)r)rrs&r rFCooperatorTests.test_removingLastTaskStopsScheduledCall..scheds&q)E!H8Or)r) rrG cooperaterJrAr rrHrr rr assertIsNone)r rcooptask1task2rs& @r 'test_removingLastTaskStopsScheduledCall7CooperatorTests.test_removingLastTaskStopsScheduledCalls% /tQF|,tQF|, q 3   q++, **E!H5   a**+ $++, tQF|$ q++, **E!H5rc f\P!4pVPVP4R#)zU L{Cooperator.running} reports C{True} if the L{Cooperator} was started on creation. N)rrGrrunningr rNs& r test_runningWhenStarted'CooperatorTests.test_runningWhenStarted"s OO   "rc j\P!RR7pVPVP4R#)zS L{Cooperator.running} reports C{False} if the L{Cooperator} has not been started. FstartedN)rrGrrrs& r test_runningWhenNotStarted*CooperatorTests.test_runningWhenNotStarted*s$ OOE * #rc \P!RR7pVP4VPVP4VP VP 4R#)zJ L{Cooperator.running} reports C{True} when the L{Cooperator} is running. FrN)rrGstart addCleanuprHrrrs& r test_runningWhenRunning'CooperatorTests.test_runningWhenRunning2s; OOE *     "rc \P!RR7pVP4VP4VP VP 4R#)zR L{Cooperator.running} reports C{False} after the L{Cooperator} has been stopped. FrN)rrGrrHrrrs& r test_runningWhenStopped'CooperatorTests.test_runningWhenStopped<s6 OOE *    #rrFN)rrrrr4r6r;rUrcrtr|rrrrrrrrrrrrrs@r r.r.>sc F R.S6 333 &24(6T"6H#$#$$rr.c]tRtRtRtRtR#)UnhandledExceptioniGz( An exception that should go unhandled. rFN)rrrrrrrFrr rrGsrrc*a]tRtRtoRtRtRtVtR#) AliasTestsiMz[ Integration test to verify that the global singleton aliases do what they're supposed to. c a\P!4oV3RlpV!4p\P!V4pVP V\P P 4S#)zJ L{twisted.internet.task.cooperate} ought to run the generator that it is c3J<"^x^x^xSPR4R#5i)r^yayN)rj)rOsr doit'AliasTests.test_cooperate..doitYs!GGG JJu s #)rrSrrassertIn_theCooperator_tasks)r rittheTaskrOs& @r test_cooperateAliasTests.test_cooperateSsL NN   V..$ gt2299:rrFN)rrrrrrrrrs@r rrMs rrca]tRtRtoRtRtRtRtRtRt Rt R t R t R t R tR tRtRtRtRtRtRtRtRtRtRtVtR#) RunStateTestsiez Tests to verify the behavior of L{CooperativeTask.pause}, L{CooperativeTask.resume}, L{CooperativeTask.stop}, exhausting the underlying iterator, and their interactions with each other. c <RVnRVnRVn.Vn\ 4Vn\ P!VP RR7VnVPPVP44VnVPP4R#)z Create a cooperator with a fake scheduler and a termination predicate that ensures only one unit of work will take place per tick. FcR#)cR#rrFrFrr rD7RunStateTests.setUp....zsrrFrFrr rD%RunStateTests.setUp..zs rrN) _doDeferNext _doStopNext _doDieNextr"rrrrG cooperatorrworkerrrs&r setUpRunStateTests.setUplsu "  &//nn)=   OO--dkkm<  rc# b"^pV^, pVP'd>RVn\P!4pVPP V4VxKYVP 'dR#VP 'd \4hVPP V4VxK5i)zz This is a sample generator which yields Deferreds when we are testing deferral and an ascending integer count otherwise. FN)rrrSr"r%rrr)r rrOs& r rRunStateTests.workers  FA   $)!NN$   #!!!(**   #sB/AB/0B/+B/c V=V=R#)zw Drop references to interesting parts of the fixture to allow Deferred errors to be noticed when things start failing. N)rrrs&r tearDownRunStateTests.tearDowns I Nrc RVnR#)z0 Defer the next result from my worker iterator. TN)rrs&r deferNextRunStateTests.deferNexts !rc RVnR#)zU Make the next result from my worker iterator be completion (raising StopIteration). TN)rrs&r stopNextRunStateTests.stopNexts  rc RpRVnR#)zS Make the next result from my worker iterator be raising an L{UnhandledException}. c0VP\4R#r1)r2r)failures&r ignoreUnhandled.RunStateTests.dieNext..ignoreUnhandleds LL+ ,rTN)r)r r%s& r dieNextRunStateTests.dieNexts  rc hVPP4VPVP^.4VPP4VPVP^^.4VPP 4VPP4VPVP^^.4VPP 4VPVP^^.4VPP4VPVP.RO4R#)zh Cooperators should stop running their tasks when they're paused, and start again when they're resumed. Nr)rr+rAr"rpauseresumers&r test_pauseResumeRunStateTests.test_pauseResumes  QC(  QF+   QF+  QF+  I.rc >VP\PVPP4VPP 4VPP4VP\PVPP4R#)z L{CooperativeTask.resume} should raise a L{TaskNotPaused} exception if it was not paused; e.g. if L{CooperativeTask.pause} was not invoked more times than L{CooperativeTask.resume} on that object. N) assertRaisesr NotPausedr+r*rs&r test_resumeNotPaused"RunStateTests.test_resumeNotPauseds^ $..$))*:*:;   $..$))*:*:;rc VPP4VPP4VP VP .4VPP4VPP4VP VP .4VPP 4VPP4VP VP .4VPP 4VPP4VP VP ^.4R#)zf Pauses on tasks should behave like a stack. If a task is paused twice, it needs to be resumed twice. N)rr*rr+rAr"r+rs&r test_pauseTwiceRunStateTests.test_pauseTwices   B'   B'   B'   QC(rc VP4VPP4VP\ VP 4^4VP VP ^,\P4VPP4VP\ VP 4^4VPP4VPP4VP\ VP 4^4VPP4VPP4VP\ VP 4^4VP ^,PR4VPP4VP\ VP 4^4VPVP ^,^4R#)z C{pause()}ing a task while it is waiting on an outstanding L{defer.Deferred} should put the task into a state where the outstanding L{defer.Deferred} must be called back I{and} the task is C{resume}d before it will continue processing. zSTUFF!N) rrr+rAlenr"assertIsInstancerrSrr*r+rjrs&r test_pauseWhileDeferred%RunStateTests.test_pauseWhileDeferreds>   TYY+ diilENN;  TYY+   TYY+   TYY+ ! h'  TYY+ 1q)rc aaVPP4pVPP4p.o.o.p.pV3RlpV3RlpVPV4VPV4VPVP4VPVP4VP 4VP P 4VP\S4^4VP\S4^4VPS^,VPP4VPS^,VPP4VPV^.4VPV^.4R#)a L{CooperativeTask.whenDone} returns a Deferred which fires when the Cooperator's iterator is exhausted. It returns a new Deferred each time it is called; callbacks added to other invocations will not modify the value that subsequent invocations will fire with. c*<SPV4^#)r^r)rCresults1s&r callbackOne0RunStateTests.test_whenDone..callbackOne OOF #rc*<SPV4^#)rr)rCresults2s&r callbackTwo0RunStateTests.test_whenDone..callbackTwor@rN) rwhenDonerKr%r rr+rAr7assertIs _iterator) r deferred1 deferred2final1final2r>rCr=rBs & @@r test_whenDoneRunStateTests.test_whenDones(II&&( II&&(    k*k*fmm,fmm,   X* X* hqk499#6#67 hqk499#6#67 !% !%rc ^VPP4p.pVPVP4VP 4VP P 4VP\V4^4VPV^,P\4\4R#)z L{CooperativeTask.whenDone} returns a L{defer.Deferred} that will fail when the iterable's C{next} method raises an exception, with that exception. N) rrErLr%r'rr+rAr7checkr)r rHresultss& r test_whenDoneError RunStateTests.test_whenDoneError/s{ II&&( W^^,   Wq) ))*<=?QRrc fVPP4p.pVPVP4VPP 4VP \ V4^4VP V^,P\P4\P4R#)z L{CooperativeTask.whenDone} returns a L{defer.Deferred} that fails with L{TaskStopped} when the C{stop} method is called on that L{CooperativeTask}. N) rrErLr%rHrAr7rO TaskStopped)r rHerrorss& r test_whenDoneStopRunStateTests.test_whenDoneStop=sx II&&( V]]+  Va( )9)9:D>:rc aa.oVV3RlpSPP4PV4SP4SPP 4SP SSPPR.4R#)a[ If a callback of a L{Deferred} from L{CooperativeTask.whenDone} calls C{Cooperator.stop} on its L{CooperativeTask._cooperator}, the L{Cooperator} will stop, but the L{CooperativeTask} whose callback is calling C{stop} should already be considered 'stopped' by the time the callback is running, and therefore removed from the L{CoooperativeTask}. c<SPV4SPP4SPR4R#)r/N)r%rrH)rCcallbackPhasesr s&r stopit;RunStateTests.test_stopCooperatorReentrancy..stopits1  ! !& ) OO "  ! !& )rr/N)rrErKr rr+rArG)r rprosf @r test_stopCooperatorReentrancy+RunStateTests.test_stopCooperatorReentrancysd * ((0   $))*=*=v)FGr)rrrrrrr"N)rrrrrrrrrr r'r,r1r4r9rLrQrVrYr\r_rcrgrkrrrrrs@r r r esx  &*!   /, <),*0)&V S N 9 ( ((49;HHrr N)rtwisted.internetrrr twisted.trialrrrTestCaser. Exceptionrrr rFrr rxsr 21"&:F$h''F$R ""0MHH%%MHr