o ¯b÷$ã@szdZddlmZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZGdd „d ee d ƒƒZGd d „d e ƒZd S)z& Tests for L{twisted._threads._team}. é)ÚproxyForInterface)ÚcallÚget)ÚFailure)ÚSynchronousTestCaseé)Ú AlreadyQuitÚIWorkerÚTeamÚcreateMemoryWorkercs(eZdZdZdd„Z‡fdd„Z‡ZS)ÚContextualWorkerz: A worker implementation that supplies a context. cKs||_||_dS)z: Create with a real worker and a context. N)Ú _realWorkerÚ_context)ÚselfÚ realWorkerÚctx©rúA/usr/lib/python3/dist-packages/twisted/_threads/test/test_team.pyÚ__init__s zContextualWorker.__init__cstƒ ‡‡fdd„¡dS)zŠ Perform the given work with the context given to __init__. @param work: the work to pass on to the real worker. cs tˆjˆƒS©N)rrr©rÚworkrrÚ"s z%ContextualWorker.do..N)ÚsuperÚdor©Ú __class__rrrszContextualWorker.do)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚ __classcell__rrrrr sr r c@seZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"S)#Ú TeamTestsz Tests for L{Team} csltƒ\}ˆ_t|ddˆ_gˆ_gˆ_gˆ_gˆ_dd„ˆ_‡fdd„}gˆ_ ‡fdd„}t |||ƒˆ_ d S) zl Set up a L{Team} with inspectable, synchronous workers that can be single-stepped. Ú coordinator©ÚworkercSódS)NFrrrrrr5óz!TeamTests.setUp..cszˆ ¡rdStƒ\}‰ˆj ˆ¡ˆj ˆ¡t|tˆjƒd‰ˆj ˆ¡ˆj ˆ¡ˆj ‰‡‡‡‡fdd„}|ˆ_ ˆS)Nr$cs"ˆƒˆj ˆ¡ˆj ˆ¡dSr)ÚallUnquitWorkersÚremoveÚactivePerformersr)ÚcwÚ performerÚrealQuitrrrÚ quitAndRemoveBs z.createWorker..quitAndRemove) Ú noMoreWorkersr ÚworkerPerformersÚappendr*r ÚlenÚallWorkersEverr(Úquit)r%r.©r)r+r,r-rÚ createWorker7s     z%TeamTests.setUp..createWorkercsˆj tƒ¡dSr)Úfailuresr1rrr5rrÚ logExceptionLóz%TeamTests.setUp..logExceptionN) r ÚcoordinateOncer r#r0r3r(r*r/r7r Úteam)rr#r6r8rr5rÚsetUp*s    zTeamTests.setUpcCsd}| ¡r d}| ¡s|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)r:)rÚdidrrrÚ coordinateQs  ÿzTeamTests.coordinatecCsDd}|r | ¡}|jD] }||jvr|ƒq |p| ¡}|sdSdS)zj Perform all work on the coordinator and worker performers that needs to be done. TN)r>r0r*)rÚ continuingr,rrrÚperformAllOutstandingWork_s  €  ûz#TeamTests.performAllOutstandingWorkcsb‡fdd„‰|j ˆ¡| ¡| |j ¡jd¡| ¡| ˆjd¡| |j ¡jd¡dS)zd L{Team.do} does the work in a worker created by the createWorker callable. cstdƒˆ_dS)Nr%)rÚwhor©Ú somethingrrrCrsz4TeamTests.test_doDoesWorkInWorker..somethingérN)r;rr>Ú assertEqualÚ statisticsÚbusyWorkerCountr@rAr5rrBrÚtest_doDoesWorkInWorkerls  z!TeamTests.test_doDoesWorkInWorkercCs8|j ¡}| |jd¡| |jd¡| |jd¡dS)z‘ L{Team.statistics} returns an object with idleWorkerCount, busyWorkerCount, and backloggedWorkCount integer attributes. rN)r;rFrEÚidleWorkerCountrGÚbackloggedWorkCount)rÚstatsrrrÚtest_initialStatistics|s z TeamTests.test_initialStatisticscCs*|j d¡| ¡| t|jƒd¡dS)zN L{Team.grow} increases the number of available idle workers. éN)r;Úgrowr@rEr2r0r5rrrÚtest_growCreatesIdleWorkers†s z%TeamTests.test_growCreatesIdleWorkerscsL‡fdd„ˆ_ˆj d¡ˆ ¡ˆ tˆjƒd¡ˆ ˆj ¡jd¡dS)z L{Team.grow} increases the number of available idle workers until the C{createWorker} callable starts returning None. cstˆjƒdkS)Né)r2r3rr5rrr“sz0TeamTests.test_growCreateLimit..rMrPN) r/r;rNr@rEr2r3rFrIr5rr5rÚtest_growCreateLimitŽs  zTeamTests.test_growCreateLimitcCs>|j d¡| ¡|j d¡| ¡| t|jƒd¡dS)zG L{Team.shrink} will quit the given number of workers. rMrPrN)r;rNr@ÚshrinkrEr2r(r5rrrÚtest_shrinkQuitsWorkers™s  z!TeamTests.test_shrinkQuitsWorkerscCs`|j d¡| ¡| t|jƒd¡|j ¡| t|jƒd¡| ¡| t|jƒd¡dS)zU L{Team.shrink} with no arguments will stop all outstanding workers. é rN)r;rNr@rEr2r(rRr5rrrÚtest_shrinkToZero£s  zTeamTests.test_shrinkToZerocsä|j d¡| ¡‡fdd„‰dˆ_| |j ¡jd¡tdƒD]}|j ˆ¡q!| ¡| |j ¡jd¡dd„|_ |j ˆ¡| ¡| |j ¡jd¡| |j ¡j d¡|  ¡| |j ¡j d¡| ˆjd¡d S) z When no additional workers are available, the given work is backlogged, and then performed later when the work was. rPcsˆjd7_dS)NrD)ÚtimesrrBrrrC·r9z@TeamTests.test_moreWorkWhenNoWorkersAvailable..somethingrcSr&)NTrrrrrrÂr'z?TeamTests.test_moreWorkWhenNoWorkersAvailable..rDéN) r;rNr>rVrErFrIÚrangerr/rJr@)rÚirrBrÚ#test_moreWorkWhenNoWorkersAvailable¯s"     z-TeamTests.test_moreWorkWhenNoWorkersAvailablecCsB|j dd„¡| ¡| t|jƒd¡| |jdjt¡dS)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. cSsddS)NrDrrrrrrrÑsz0TeamTests.test_exceptionInTask..rDrN)r;rr@rEr2r7ÚtypeÚZeroDivisionErrorr5rrrÚtest_exceptionInTaskËszTeamTests.test_exceptionInTaskcCs0|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)r;r4Ú assertRaisesrrÚlistr5rrrÚ test_quitÖs zTeamTests.test_quitcCsZtdƒD]}|j t¡q| ¡|j ¡| ¡| t|jƒd¡|  t |j j¡dS)zk L{Team.quit} causes all idle workers, as well as the coordinator worker, to quit. rTrN) rXr;rr_r@r4rEr2r(r^rr#©rÚxrrrÚtest_quitQuitsßs  zTeamTests.test_quitQuitscCs€|j d¡tdƒD]}|j t¡q | ¡|j ¡| ¡| t|j ƒd¡|  ¡| t|j ƒd¡|  t |j j¡dS)z| L{Team.quit} causes all busy workers to be quit once they've finished the work they've been given. rTrMrN)r;rNrXrr_r>r4rEr2r(r@r^rr#rarrrÚtest_quitQuitsLaterWhenBusyìs   z%TeamTests.test_quitQuitsLaterWhenBusycs^ˆj t¡ˆjjj‰‡‡fdd„}|ˆjj_ˆj ¡ˆ tˆjj¡ˆ tˆjjt¡dS)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. csˆƒˆ ¡dSr)r@r©Ú originalSetrrrÚperformWorkConcurrentlys zOTeamTests.test_quitConcurrentWithWorkHappening..performWorkConcurrentlyN)r;rr_Ú_quitÚsetr4r^r)rrgrrerÚ$test_quitConcurrentWithWorkHappeningüs    z.TeamTests.test_quitConcurrentWithWorkHappeningcCs^tdƒD]}|j t¡q| ¡| t|jƒd¡|j d¡|  ¡| t|jƒd¡dS)zl L{Team.shrink} will wait for busy workers to finish being busy and then quit them. rTérPN) rXr;rr_r>rEr2r(rRr@rarrrÚtest_shrinkWhenBusy s  zTeamTests.test_shrinkWhenBusyN)rrrr r<r>r@rHrLrOrQrSrUrZr]r`rcrdrjrlrrrrr"%s$'         r"N)r Útwisted.python.componentsrÚtwisted.python.contextrrÚtwisted.python.failurerÚtwisted.trial.unittestrÚrr r r r r"rrrrÚs