ó [³XMc@sxdZeZddlmZddlZddlmZddlm Z defd„ƒYZ e ƒj e j ƒƒdS(s2 Tests for implementations of L{IReactorThreads}. iÿÿÿÿ(trefN(tReactorBuilder(t ThreadPooltThreadTestsBuildercBsDeZdZd„Zd„Zd„Zd„Zd„Zd„ZRS(sD Builder for defining tests relating to L{IReactorThreads}. cs²g‰|jƒ‰ˆjƒ‰|jˆtƒ|jˆjdƒ‡‡‡fd†}ˆj|ƒ|jˆdƒ|jˆddƒ|jˆddƒ|jˆj dƒd S( s£ C{reactor.getThreadPool()} returns an instance of L{ThreadPool} which starts when C{reactor.run()} is called and stops before it returns. s(Pool should not start before reactor.runcs.ˆjˆjƒˆjˆjƒˆjƒdS(N(tappendtstartedtjoinedtstop((tstatetreactortpool(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pytf"siis#Pool should start after reactor.runis-Pool should not be joined before reactor.stops0Pool should be stopped after reactor.run returnsN( t buildReactort getThreadPooltassertIsInstanceRt assertFalseRtcallWhenRunningt runReactort assertTrueR(tselfR ((R RR sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyttest_getThreadPools     cCs<|jƒ}|jdƒ|jƒ}|j|jdƒdS(sm C{reactor.suggestThreadPoolSize()} sets the maximum size of the reactor threadpool. iN(R tsuggestThreadPoolSizeR t assertEqualtmax(RR R ((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyttest_suggestThreadPoolSize4s   cst|jƒ‰‡fd†}ˆjdˆj|ƒˆjƒ}|jˆdƒˆjƒ}|j||dkƒdS(sQ A function scheduled with L{IReactorThreads.callFromThread} invoked from a delayed call is run immediately in the next reactor iteration. When invoked from the reactor thread, previous implementations of L{IReactorThreads.callFromThread} would skip the pipe/socket based wake up step, assuming the reactor would wake up on its own. However, this resulted in the reactor not noticing a insert into the thread queue at the right time (in this case, after the thread queue has been processed for that reactor iteration). csˆjƒdS(N(R((R (sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyt threadCallMsii<iN(R t callLatertcallFromThreadtsecondsRR(RRtbeforetafter((R sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyttest_delayedCallFromThread?s   cCs\|jƒ}t|jƒƒ}|j|jƒ|j|ƒtjƒ|j|ƒdƒdS(s0 When the reactor stops, L{ReactorBase._stopThreadPool} drops the reactor's direct reference to its internal threadpool and removes the associated startup and shutdown triggers. This is the case of the thread pool being created before the reactor is run. N( R RR RRRtgctcollecttassertIdenticaltNone(RR t threadpool((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyttest_stopThreadPooljs   csc|jƒ‰g‰‡‡fd†}ˆj|ƒ|jˆƒtjƒ|jˆdƒdƒdS(s¶ We must handle the case of shutting down the thread pool when it was started after the reactor was run in a special way. Some implementation background: The thread pool is started with callWhenRunning, which only returns a system trigger ID when it is invoked before the reactor is started. This is the case of the thread pool being created after the reactor is started. cs'ˆjtˆjƒƒƒˆjƒdS(N(RRR R((tthreadPoolRefsR (sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pytacquireThreadPool‰siN(R RRR R!R"R#(RR'((R R&sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyt-test_stopThreadPoolWhenStartedAfterReactorRan{s    cCsL|jƒ}t|jƒƒ}|jdƒtjƒ|j|ƒdƒdS(s[ When the reactor has its shutdown event fired before it is run, the thread pool is completely destroyed. For what it's worth, the reason we support this behavior at all is because Trial does this. This is the case of the thread pool being created without the reactor being started at al. tshutdownN(R RR tfireSystemEventR R!R"R#(RR t threadPoolRef((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyt,test_cleanUpThreadPoolEvenBeforeReactorIsRun’s   ( t__name__t __module__t__doc__RRRR%R(R,(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyRs  +  (R/ttypet __metaclass__tweakrefRR t#twisted.internet.test.reactormixinsRttwisted.python.threadpoolRRtglobalstupdatetmakeTestCaseClasses(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_threads.pyts “