U -Wj&@sdZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z dd lmZmZmZmZGd d d eed ZGd dde ZdS)z& Tests for L{twisted._threads._team}. ) annotations)Callable)proxyForInterface)callget)Failure)SynchronousTestCase) AlreadyQuitIWorkerTeamcreateMemoryWorkercs:eZdZdZddddddZddd fd d ZZS) ContextualWorkerz: A worker implementation that supplies a context. r objectNone) realWorkerctxreturncKs||_||_dS)z: Create with a real worker and a context. N) _realWorker_context)selfrrrjc:\users\psant\downloads\hotstarbackend\school\venv38\Lib\site-packages\twisted/_threads/test/test_team.py__init__szContextualWorker.__init__zCallable[[], object])workrcstfdddS)z Perform the given work with the context given to __init__. @param work: the work to pass on to the real worker. cs tjSN)rrrrrrr%z%ContextualWorker.do..N)superdor __class__rrr szContextualWorker.do)__name__ __module__ __qualname____doc__rr __classcell__rrr!rrsrrc@seZdZdZddddZddddZddd d Zddd d Zddd dZddddZ ddddZ ddddZ ddddZ ddddZ ddddZddddZddddZdddd Zddd!d"Zddd#d$Zd%S)& TeamTestsz Tests for L{Team} rrcsxt\}_t|dd_g_g_g_g_dd_ddfdd }g_ d dfd d }t |||_ d S) zl Set up a L{Team} with inspectable, synchronous workers that can be single-stepped. coordinatorworkercSsdS)NFrrrrrr8rz!TeamTests.setUp..zContextualWorker | Noner)csr dSt\}jjt|tjdjjj ddfdd }|_ S)Nr+rr)cs"jjdSr)allUnquitWorkersremoveactivePerformersr)cw performerrealQuitrrr quitAndRemoveEs z.createWorker..quitAndRemove) noMoreWorkersr workerPerformersappendr/rlenallWorkersEverr-quit)r,r3r)r0r1r2r createWorker:s     z%TeamTests.setUp..createWorkerrcsjtdSr)failuresr6rrr:rr logExceptionOsz%TeamTests.setUp..logExceptionN) r coordinateOncerr*r5r8r-r/r4r<r team)rr*r;r=rr:rsetUp-s  zTeamTests.setUpboolcCsd}|rd}q|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)r>)rZdidrrr coordinateTszTeamTests.coordinatecCs>d}|r:|}|jD]}||jkr|q|p6|}qdS)zj Perform all work on the coordinator and worker performers that needs to be done. TN)rBr5r/)rZ continuingr1rrrperformAllOutstandingWorkas  z#TeamTests.performAllOutstandingWorkcsjdddfdd }|j||||jjd||d||jjddS)zd L{Team.do} does the work in a worker created by the createWorker callable. Nrr)cs tddS)Nr,)rrZwhorr somethingusz4TeamTests.test_doDoesWorkInWorker..somethingr)r?r rB assertEqual statisticsbusyWorkerCountrC)rrErrDrtest_doDoesWorkInWorkerns  z!TeamTests.test_doDoesWorkInWorkercCs8|j}||jd||jd||jddS)z L{Team.statistics} returns an object with idleWorkerCount, busyWorkerCount, and backloggedWorkCount integer attributes. rN)r?rHrGidleWorkerCountrIbackloggedWorkCount)rstatsrrrtest_initialStatisticss z TeamTests.test_initialStatisticscCs*|jd||t|jddS)zN L{Team.grow} increases the number of available idle workers. N)r?growrCrGr7r5r:rrrtest_growCreatesIdleWorkerss z%TeamTests.test_growCreatesIdleWorkerscsLfdd_jdtjdjjddS)z L{Team.grow} increases the number of available idle workers until the C{createWorker} callable starts returning None. cstjdkS)N)r7r8rr:rrrrz0TeamTests.test_growCreateLimit..rOrRN) r4r?rPrCrGr7r8rHrKr:rr:rtest_growCreateLimits  zTeamTests.test_growCreateLimitcCs>|jd||jd||t|jddS)zG L{Team.shrink} will quit the given number of workers. rOrRr N)r?rPrCshrinkrGr7r-r:rrrtest_shrinkQuitsWorkerss   z!TeamTests.test_shrinkQuitsWorkerscCs`|jd||t|jd|j|t|jd||t|jddS)zU L{Team.shrink} with no arguments will stop all outstanding workers. rN)r?rPrCrGr7r-rTr:rrrtest_shrinkToZeros  zTeamTests.test_shrinkToZerocs|jd|dddfdd }||jjdtdD]}|j|qF|||jjddd|_|j||||jjd||jj d | ||jj d|d d S) z When no additional workers are available, the given work is backlogged, and then performed later when the work was. rRrrr)cs d7dS)NrFrrtimesrrrEsz@TeamTests.test_moreWorkWhenNoWorkersAvailable..somethingcSsdS)NTrrrrrrrz?TeamTests.test_moreWorkWhenNoWorkersAvailable..rFN) r?rPrBrGrHrKranger r4rLrC)rrEirrXr#test_moreWorkWhenNoWorkersAvailables"    z-TeamTests.test_moreWorkWhenNoWorkersAvailablecCsB|jdd||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. cSsddS)NrFrrrrrrrrz0TeamTests.test_exceptionInTask..rFrN)r?r rCrGr7r<typeZeroDivisionErrorr:rrrtest_exceptionInTaskszTeamTests.test_exceptionInTaskcCs0|j|t|jj|t|jjtdS)zx L{Team.quit} causes future invocations of L{Team.do} and L{Team.quit} to raise L{AlreadyQuit}. N)r?r9 assertRaisesr r listr:rrr test_quits zTeamTests.test_quitcCsZtdD]}|jtq||j||t|jd| t |j jdS)zk L{Team.quit} causes all idle workers, as well as the coordinator worker, to quit. rVrN) r[r?r rbrCr9rGr7r-rar r*rxrrrtest_quitQuitss  zTeamTests.test_quitQuitscCs|jdtdD]}|jtq||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. rVrOrN)r?rPr[r rbrBr9rGr7r-rCrar r*rdrrrtest_quitQuitsLaterWhenBusys   z%TeamTests.test_quitQuitsLaterWhenBusycsdjtjjjddfdd }|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. rr)csdSr)rCrZ originalSetrrrperformWorkConcurrently szOTeamTests.test_quitConcurrentWithWorkHappening..performWorkConcurrentlyN)r?r rbZ_quitsetr9rar )rrirrhr$test_quitConcurrentWithWorkHappenings    z.TeamTests.test_quitConcurrentWithWorkHappeningcCs^tdD]}|jtq||t|jd|jd| |t|jddS)zl L{Team.shrink} will wait for busy workers to finish being busy and then quit them. rVrRN) r[r?r rbrBrGr7r-rTrCrdrrrtest_shrinkWhenBusys  zTeamTests.test_shrinkWhenBusyN)r#r$r%r&r@rBrCrJrNrQrSrUrWr]r`rcrfrgrkrmrrrrr((s"'         r(N)r& __future__rtypingrZtwisted.python.componentsrZtwisted.python.contextrrZtwisted.python.failurerZtwisted.trial.unittestrr r r r rr(rrrrs