Ϫf$dZddlmZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZGdd ee d ZGd d e Zy )z& Tests for L{twisted._threads._team}. )proxyForInterface)callget)Failure)SynchronousTestCase) AlreadyQuitIWorkerTeamcreateMemoryWorkerc(eZdZdZdZfdZxZS)ContextualWorkerz: A worker implementation that supplies a context. c ||_||_y)z: Create with a real worker and a context. N) _realWorker_context)self realWorkerctxs A/usr/lib/python3/dist-packages/twisted/_threads/test/test_team.py__init__zContextualWorker.__init__s& c0tfdy)z Perform the given work with the context given to __init__. @param work: the work to pass on to the real worker. c0tjSN)rr)rworksrz%ContextualWorker.do.."s4 t4rN)superdo)rr __class__s``rrzContextualWorker.dos  45r)__name__ __module__ __qualname____doc__rr __classcell__)rs@rrrs66rrrcpeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZy) TeamTestsz Tests for L{Team} ct\}_t|d_g_g_g_g_d_fd}g_ fd}t|||_ y)zl Set up a L{Team} with inspectable, synchronous workers that can be single-stepped. coordinatorworkercy)NFr,rrrz!TeamTests.setUp..5rcjryt\}jjjjt |t jjjjjjfd}|_ S)Nr)cjjjjyr)allUnquitWorkersremoveactivePerformers)cw performerrealQuitrsr quitAndRemovez.createWorker..quitAndRemoveBs0 %%,,R0%%,,Y7r) noMoreWorkersr workerPerformersappendr2rlenallWorkersEverr0quit)r*r6r3r4r5rs @@@r createWorkerz%TeamTests.setUp..createWorker7s!!# 2 4 FI  ! ! ( ( 3  ! ! ( ( 3!&T5J5J1KLB    & &r *  ! ! ( ( ,wwH 8 $BGIrcLjjtyr)failuresr9rrsr logExceptionz%TeamTests.setUp..logExceptionLs MM  +rN) r coordinateOncerr(r8r;r0r2r7r?r team)rr(r=rAs` rsetUpzTeamTests.setUp*sq ,>+?( T(+K N "  " "* &  ,lLA rcPd}|jrd}|jr|S)a Perform all work currently scheduled in the coordinator. @return: whether any coordination work was performed; if the coordinator was idle when this was called, return L{False} (otherwise L{True}). @rtype: L{bool} FT)rB)rdids r coordinatezTeamTests.coordinateQs-!!#C!!# rcd}|rO|j}|jD]}||jvs||xs|j}|rNyy)zj Perform all work on the coordinator and worker performers that needs to be done. TN)rGr8r2)r continuingr4s rperformAllOutstandingWorkz#TeamTests.performAllOutstandingWork_sX  *J!22  5 55K $8t'8J rcfd|jj|j|j|jj j d|j |jjd|j|jj j dy)zd L{Team.do} does the work in a worker created by the createWorker callable. c&td_y)Nr*)rwho somethingsrrOz4TeamTests.test_doDoesWorkInWorker..somethingrsMIMrrN)rCrrG assertEqual statisticsbusyWorkerCountrJrM)rrOs @rtest_doDoesWorkInWorkerz!TeamTests.test_doDoesWorkInWorkerls  * Y  --/??C &&( * --/??Crc|jj}|j|jd|j|jd|j|j dy)z L{Team.statistics} returns an object with idleWorkerCount, busyWorkerCount, and backloggedWorkCount integer attributes. rN)rCrRrQidleWorkerCountrSbackloggedWorkCount)rstatss rtest_initialStatisticsz TeamTests.test_initialStatistics|sX  $$& ..2 ..2 22A6rc|jjd|j|jt |j dy)zN L{Team.grow} increases the number of available idle workers. N)rCgrowrJrQr:r8r@s rtest_growCreatesIdleWorkersz%TeamTests.test_growCreatesIdleWorkerss; q &&( T223Q7rc"fd_jjdjj t j dj jjjdy)z L{Team.grow} increases the number of available idle workers until the C{createWorker} callable starts returning None. c4tjdk\S)N)r:r;r@srrz0TeamTests.test_growCreateLimit..sS)<)<%=%Brr[r`N) r7rCr\rJrQr:r;rRrVr@s`rtest_growCreateLimitzTeamTests.test_growCreateLimitsh C q &&( T00115 --/??Crc|jjd|j|jjd|j|j t |j dy)zG L{Team.shrink} will quit the given number of workers. r[r`rN)rCr\rJshrinkrQr:r0r@s rtest_shrinkQuitsWorkersz!TeamTests.test_shrinkQuitsWorkerssY q &&(  &&( T223Q7rc|jjd|j|jt |j d|jj |jt |j d|j|jt |j dy)zU L{Team.shrink} with no arguments will stop all outstanding workers. rN)rCr\rJrQr:r0rcr@s rtest_shrinkToZerozTeamTests.test_shrinkToZeros r &&( T223R8  T223R8 &&( T223Q7rc|jjd|jfdd_|j |jj j dtdD]}|jj|j|j |jj j dd|_ |jj|j|j |jj j d|j |jj jd|j|j |jj jd|j jdy)z When no additional workers are available, the given work is backlogged, and then performed later when the work was. r`c0xjdz c_y)NrP)timesrNsrrOz@TeamTests.test_moreWorkWhenNoWorkersAvailable..somethings OOq Orrcy)NTr,r,rrrz?TeamTests.test_moreWorkWhenNoWorkersAvailable..r-rrPN) rCr\rGrjrQrRrVrangerr7rWrJ)rirOs @r#test_moreWorkWhenNoWorkersAvailablez-TeamTests.test_moreWorkWhenNoWorkersAvailablesM q  !  --/??Cq $A IILL # $  --/??C) Y  --/??C --/CCQG &&( --/CCQG !,rc|jjd|j|jt |j d|j|j dj ty)z When an exception is raised in a task passed to L{Team.do}, the C{logException} given to the L{Team} at construction is invoked in the exception context. c ddz S)NrPrr,r,rrrz0TeamTests.test_exceptionInTask..s QUrrPrN)rCrrJrQr:r?typeZeroDivisionErrorr@s rtest_exceptionInTaskzTeamTests.test_exceptionInTasksY ]# &&( T]]+Q/ q)..0ABrc|jj|jt|jj|jt|jjt y)zx L{Team.quit} causes future invocations of L{Team.do} and L{Team.quit} to raise L{AlreadyQuit}. N)rCr< assertRaisesr rlistr@s r test_quitzTeamTests.test_quitsD  +tyy~~6 +tyy||T:rcttdD]!}|jjt#|j |jj |j |j t|jd|jt|jj y)zk L{Team.quit} causes all idle workers, as well as the coordinator worker, to quit. rfrN) rmrCrrwrJr<rQr:r0rvr r(rxs rtest_quitQuitszTeamTests.test_quitQuitss r A IILL   &&(  &&( T223Q7 +t'7'7'<'<=rc|jjdtdD]!}|jjt#|j |jj |j |jt|jd|j|jt|jd|jt|jj y)z| L{Team.quit} causes all busy workers to be quit once they've finished the work they've been given. rfr[rN)rCr\rmrrwrGr<rQr:r0rJrvr r(rzs rtest_quitQuitsLaterWhenBusyz%TeamTests.test_quitQuitsLaterWhenBusys rq A IILL      T223Q7 &&( T223Q7 +t'7'7'<'<=rcjjtjjjfd}|jj_jj j tjj j tjjty)z If work happens after L{Team.quit} sets its C{Quit} flag, but before any other work takes place, the L{Team} should still exit gracefully. c4jyr)rJ) originalSetrsrperformWorkConcurrentlyzOTeamTests.test_quitConcurrentWithWorkHappening..performWorkConcurrentlys M  * * ,rN)rCrrw_quitsetr<rvr )rrrs` @r$test_quitConcurrentWithWorkHappeningz.TeamTests.test_quitConcurrentWithWorkHappenings Tiioo))  -6   +tyy~~6 +tyy||T:rcltdD]!}|jjt#|j |j t |jd|jjd|j|j t |jdy)zl L{Team.shrink} will wait for busy workers to finish being busy and then quit them. rfr`N) rmrCrrwrGrQr:r0rcrJrzs rtest_shrinkWhenBusyzTeamTests.test_shrinkWhenBusy s r A IILL    T223R8  &&( T223Q7rN)r r!r"r#rDrGrJrTrYr]rardrgrortrxr|r~rrr,rrr&r&%s\%BN  9D 78 D8 8-8 C; >> ;" 8rr&N)r#twisted.python.componentsrtwisted.python.contextrrtwisted.python.failurertwisted.trial.unittestrr r r r rr&r,rrrsA 8,*6==6(-@6*t8#t8r