tf$dZddlmZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZGdd ee d ZGd d e Zd S)z& Tests for L{twisted._threads._team}. )proxyForInterface)callget)Failure)SynchronousTestCase) AlreadyQuitIWorkerTeamcreateMemoryWorkerc(eZdZdZdZfdZxZS)ContextualWorkerz: A worker implementation that supplies a context. c "||_||_dS)z: Create with a real worker and a context. N) _realWorker_context)self realWorkerctxs ^/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/_threads/test/test_team.py__init__zContextualWorker.__init__s& cVtfddS)z Perform the given work with the context given to __init__. @param work: the work to pass on to the real worker. c.tjSN)rr)rworksrz%ContextualWorker.do.."s4 t44rN)superdo)rr __class__s``rrzContextualWorker.dos0  4444455555r)__name__ __module__ __qualname____doc__rr __classcell__)rs@rrrsQ666666666rrrcreZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdS) TeamTestsz Tests for L{Team} ct\}_t|d_g_g_g_g_d_fd}g_ fd}t|||_ dS)zl Set up a L{Team} with inspectable, synchronous workers that can be single-stepped. coordinatorworkercdS)NFr,rrrz!TeamTests.setUp..5sUrcrdSt\}jjt |t jjjj fd}|_ S)Nr)cjjdSr)allUnquitWorkersremoveactivePerformers)cw performerrealQuitrsr quitAndRemovez.createWorker..quitAndRemoveBsB %,,R000%,,Y77777r) noMoreWorkersr workerPerformersappendr1rlenallWorkersEverr/quit)r*r5r2r3r4rs @@@r createWorkerz%TeamTests.setUp..createWorker7s!!## t 2 4 4 FI  ! ( ( 3 3 3  ! ( ( 3 3 3!&T5J1K1KLLLB   & &r * * *  ! ( ( , , ,wH 8 8 8 8 8 8 8 8 $BGIrcTjtdSr)failuresr8rrsr logExceptionz%TeamTests.setUp..logExceptionLs# M  + + + + +rN) r coordinateOncerr(r7r:r/r1r6r>r team)rr(r<r@s` rsetUpzTeamTests.setUp*s ,>+?+?( T(+K NNN "  " "*]     &  , , , , ,lLAA rc^d}|rd}||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)rA)rdids r coordinatezTeamTests.coordinateQs=!!## C!!##  rcd}|rK|}|jD]}||jvr ||p|}|IdSdS)zj Perform all work on the coordinator and worker performers that needs to be done. TN)rFr7r1)r continuingr3s rperformAllOutstandingWorkz#TeamTests.performAllOutstandingWork_s~   9**J!2   555IKKK#8t'8'8J  9 9 9 9 9rcfd|j|||jjd||jd||jjddS)zd L{Team.do} does the work in a worker created by the createWorker callable. c0td_dS)Nr*)rwho somethingsrrNz4TeamTests.test_doDoesWorkInWorker..somethingrsMMIMMMrrN)rBrrF assertEqual statisticsbusyWorkerCountrIrL)rrNs @rtest_doDoesWorkInWorkerz!TeamTests.test_doDoesWorkInWorkerls  * * * * * Y  --//?CCC &&((( *** --//?CCCCCrc|j}||jd||jd||jddS)z L{Team.statistics} returns an object with idleWorkerCount, busyWorkerCount, and backloggedWorkCount integer attributes. rN)rBrQrPidleWorkerCountrRbackloggedWorkCount)rstatss rtest_initialStatisticsz TeamTests.test_initialStatistics|sh  $$&& .222 .222 2A66666rc|jd||t |jddS)zN L{Team.grow} increases the number of available idle workers. N)rBgrowrIrPr9r7r?s rtest_growCreatesIdleWorkersz%TeamTests.test_growCreatesIdleWorkerssQ q &&((( T233Q77777rc,fd_jdt jdjjddS)z L{Team.grow} increases the number of available idle workers until the C{createWorker} callable starts returning None. c4tjdkS)N)r9r:r?srrz0TeamTests.test_growCreateLimit..sS)<%=%=%BrrZr_N) r6rBr[rIrPr9r:rQrUr?s`rtest_growCreateLimitzTeamTests.test_growCreateLimits CBBB q &&((( T0111555 --//?CCCCCrc|jd||jd||t |jddS)zG L{Team.shrink} will quit the given number of workers. rZr_rN)rBr[rIshrinkrPr9r/r?s rtest_shrinkQuitsWorkersz!TeamTests.test_shrinkQuitsWorkerssy q &&(((  &&((( T233Q77777rc|jd||t |jd|j|t |jd||t |jddS)zU L{Team.shrink} with no arguments will stop all outstanding workers. rN)rBr[rIrPr9r/rbr?s rtest_shrinkToZerozTeamTests.test_shrinkToZeros r &&((( T233R888  T233R888 &&((( T233Q77777rc|jd|fdd_||jjdtdD]}|j|||jjdd|_ |j|||jjd||jj d| ||jj d|jddS)z When no additional workers are available, the given work is backlogged, and then performed later when the work was. r_c(xjdz c_dS)NrO)timesrMsrrNz@TeamTests.test_moreWorkWhenNoWorkersAvailable..somethings OOq OOOOrrcdS)NTr,r,rrrz?TeamTests.test_moreWorkWhenNoWorkersAvailable..sTrrON) rBr[rFrirPrQrUrangerr6rVrI)rirNs @r#test_moreWorkWhenNoWorkersAvailablez-TeamTests.test_moreWorkWhenNoWorkersAvailables q  ! ! ! ! !  --//?CCCq $ $A ILL # # # #  --//?CCC)\ Y  --//?CCC --//CQGGG &&((( --//CQGGG !,,,,,rc |jd||t |jd||jdjtdS)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)NrOrr,r,rrrz0TeamTests.test_exceptionInTask..s QUrrOrN)rBrrIrPr9r>typeZeroDivisionErrorr?s rtest_exceptionInTaskzTeamTests.test_exceptionInTaskss ]]### &&((( T]++Q/// q).0ABBBBBrc|j|t|jj|t|jjt dS)zx L{Team.quit} causes future invocations of L{Team.do} and L{Team.quit} to raise L{AlreadyQuit}. N)rBr; assertRaisesr rlistr?s r test_quitzTeamTests.test_quitsR  +ty~666 +ty|T:::::rctdD]!}|jt"||j||t|jd| t|j jdS)zk L{Team.quit} causes all idle workers, as well as the coordinator worker, to quit. rerN) rlrBrrvrIr;rPr9r/rur r(rxs rtest_quitQuitszTeamTests.test_quitQuitss r  A ILL     &&(((  &&((( T233Q777 +t'7'<=====rc0|jdtdD]!}|jt"||j||t|j d| |t|j d| t|j jdS)z| L{Team.quit} causes all busy workers to be quit once they've finished the work they've been given. rerZrN)rBr[rlrrvrFr;rPr9r/rIrur r(rys rtest_quitQuitsLaterWhenBusyz%TeamTests.test_quitQuitsLaterWhenBusys rq  A ILL        T233Q777 &&((( T233Q777 +t'7'<=====rcjjtjjjfd}|jj_jtjjtjjtdS)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. cDdSr)rI) originalSetrsrperformWorkConcurrentlyzOTeamTests.test_quitConcurrentWithWorkHappening..performWorkConcurrentlys& KMMM  * * , , , , ,rN)rBrrv_quitsetr;rur )rrrs` @r$test_quitConcurrentWithWorkHappeningz.TeamTests.test_quitConcurrentWithWorkHappenings Tio)  - - - - - -6   +ty~666 +ty|T:::::rctdD]!}|jt"||t |jd|jd| |t |jddS)zl L{Team.shrink} will wait for busy workers to finish being busy and then quit them. rer_N) rlrBrrvrFrPr9r/rbrIrys rtest_shrinkWhenBusyzTeamTests.test_shrinkWhenBusy s r  A ILL      T233R888  &&((( T233Q77777rN)r r!r"r#rCrFrIrSrXr\r`rcrfrnrsrwr{r}rrr,rrr&r&%s%B%B%BN    9 9 9DDD 777888 D D D888 8 8 8---8 C C C;;; > > >>>> ;;;" 8 8 8 8 8rr&N)r#twisted.python.componentsrtwisted.python.contextrrtwisted.python.failurertwisted.trial.unittestrr r r r rr&r,rrrs 877777,,,,,,,,******666666============66666((-@@666*t8t8t8t8t8#t8t8t8t8t8r