o ¼>hÃ&ã@s’dZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z dd lmZmZmZmZGd d „d eed ƒƒZGd d„de ƒZdS)z& Tests for L{twisted._threads._team}. é)Ú annotations)ÚCallable)ÚproxyForInterface)ÚcallÚget)ÚFailure)ÚSynchronousTestCaseé)Ú AlreadyQuitÚIWorkerÚTeamÚcreateMemoryWorkercs,eZdZdZddd „Zd‡fd d „ Z‡ZS)ÚContextualWorkerz: A worker implementation that supplies a context. Ú realWorkerr ÚctxÚobjectÚreturnÚNonecKs||_||_dS)z: Create with a real worker and a context. N)Ú _realWorkerÚ_context)Úselfrr©rú/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/_threads/test/test_team.pyÚ__init__s zContextualWorker.__init__ÚworkúCallable[[], object]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©rrrrÚ%s z%ContextualWorker.do..N)ÚsuperÚdor©Ú __class__rrr szContextualWorker.do)rr rrrr)rrrr)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr Ú __classcell__rrr!rrs rrc@s°eZdZdZd&dd„Zd'dd„Zd&d d „Zd&d d „Zd&d d„Zd&dd„Z d&dd„Z d&dd„Z d&dd„Z d&dd„Z d&dd„Zd&dd„Zd&dd„Zd&dd „Zd&d!d"„Zd&d#d$„Zd%S)(Ú TeamTestsz Tests for L{Team} rrcsptƒ\}ˆ_t|ddˆ_gˆ_gˆ_gˆ_gˆ_dd„ˆ_d ‡fdd„ }gˆ_ d‡fd d „ }t |||ƒˆ_ d S)zl Set up a L{Team} with inspectable, synchronous workers that can be single-stepped. Ú coordinator©ÚworkercSódS)NFrrrrrr8óz!TeamTests.setUp..rúContextualWorker | Nonecs|ˆ ¡rdStƒ\}‰ˆj ˆ¡ˆj ˆ¡t|tˆjƒd‰ˆj ˆ¡ˆj ˆ¡ˆj ‰d‡‡‡‡fdd„ }|ˆ_ ˆS)Nr*rrcs"ˆƒˆj ˆ¡ˆj ˆ¡dSr)ÚallUnquitWorkersÚremoveÚactivePerformersr)ÚcwÚ performerÚrealQuitrrrÚ quitAndRemoveEs z.createWorker..quitAndRemove©rr) Ú noMoreWorkersr ÚworkerPerformersÚappendr1rÚlenÚallWorkersEverr/Úquit)r+r5©r)r2r3r4rÚ createWorker:s     z%TeamTests.setUp..createWorkerrcsˆj tƒ¡dSr)Úfailuresr9rrr=rrÚ logExceptionOsz%TeamTests.setUp..logExceptionN)rr.r6) r ÚcoordinateOncerr)r8r;r/r1r7r?r Úteam)rr)r>r@rr=rÚsetUp-s  zTeamTests.setUpÚboolcCsd}| ¡r d}| ¡s|S)zø 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}). FT)rA)rÚdidrrrÚ coordinateTs ÿzTeamTests.coordinatecCsDd}|r | ¡}|jD] }||jvr|ƒq |p| ¡}|sdSdS)zj Perform all work on the coordinator and worker performers that needs to be done. TN)rFr8r1)rÚ continuingr3rrrÚperformAllOutstandingWorkas  €  ûz#TeamTests.performAllOutstandingWorkcsfd‰d‡fdd„ }|j |¡| ¡| |j ¡jd¡| ¡| ˆd¡| |j ¡jd¡dS) zd L{Team.do} does the work in a worker created by the createWorker callable. Nrrcs tdƒ‰dS)Nr+)rr©ÚwhorrÚ somethinguó z4TeamTests.test_doDoesWorkInWorker..somethingérr6)rBr rFÚ assertEqualÚ statisticsÚbusyWorkerCountrH)rrKrrIrÚtest_doDoesWorkInWorkerns  z!TeamTests.test_doDoesWorkInWorkercCs8|j ¡}| |jd¡| |jd¡| |jd¡dS)z‘ L{Team.statistics} returns an object with idleWorkerCount, busyWorkerCount, and backloggedWorkCount integer attributes. rN)rBrOrNÚidleWorkerCountrPÚ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)rBÚgrowrHrNr:r8r=rrrÚ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é)r:r;rr=rrr—sz0TeamTests.test_growCreateLimit..rVrYN) r7rBrWrHrNr:r;rOrRr=rr=rÚ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. rVrYr N)rBrWrHÚshrinkrNr:r/r=rrrÚtest_shrinkQuitsWorkerss  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)rBrWrHrNr:r/r[r=rrrÚtest_shrinkToZero§s  zTeamTests.test_shrinkToZerocsâ|j d¡| ¡d‰d ‡fdd„ }| |j ¡jd¡tdƒD]}|j |¡q!| ¡| |j ¡jd¡dd„|_|j |¡| ¡| |j ¡jd¡| |j ¡j d ¡|  ¡| |j ¡j d¡| ˆd ¡d S) z When no additional workers are available, the given work is backlogged, and then performed later when the work was. rYrrrcs ˆd7‰dS)NrMrr©ÚtimesrrrK¼rLz@TeamTests.test_moreWorkWhenNoWorkersAvailable..somethingcSr,)NTrrrrrrÇr-z?TeamTests.test_moreWorkWhenNoWorkersAvailable..rMéNr6) rBrWrFrNrOrRÚranger r7rSrH)rrKÚirr_rÚ#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)NrMrrrrrrrÖsz0TeamTests.test_exceptionInTask..rMrN)rBr rHrNr:r?ÚtypeÚZeroDivisionErrorr=rrrÚ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)rBr<Ú assertRaisesr r Úlistr=rrrÚ 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. r]rN) rbrBr rirHr<rNr:r/rhr r)©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. r]rVrN)rBrWrbr rirFr<rNr:r/rHrhr r)rkrrrÚtest_quitQuitsLaterWhenBusyñs   z%TeamTests.test_quitQuitsLaterWhenBusycs`ˆj t¡ˆjjj‰d‡‡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. rrcsˆƒˆ ¡dSr)rHr©Ú originalSetrrrÚperformWorkConcurrently s zOTeamTests.test_quitConcurrentWithWorkHappening..performWorkConcurrentlyNr6)rBr riÚ_quitÚsetr<rhr )rrqrrorÚ$test_quitConcurrentWithWorkHappenings    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. r]érYN) rbrBr rirFrNr:r/r[rHrkrrrÚtest_shrinkWhenBusys  zTeamTests.test_shrinkWhenBusyNr6)rrD)r#r$r%r&rCrFrHrQrUrXrZr\r^rdrgrjrmrnrtrvrrrrr((s$  '    r(N)r&Ú __future__rÚtypingrÚtwisted.python.componentsrÚtwisted.python.contextrrÚtwisted.python.failurerÚtwisted.trial.unittestrÚr r r r rr(rrrrÚs