o i@sdZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZddlmZd d lmZd d lmZd d lmZd d lmZmZd dlmZddlmZddlmZGdddej j!j"Z#Gdddej j!j"Z$dS)z0 Tests for L{twisted.application.twist._twist}. )stdout)AnyDictListN) IReactorCore) MemoryReactor)LogLeveljsonFileLogObserver)SignalCapturingMemoryReactor) ExitStatus)Runner) DummyExit)IService MultiService)_twist) TwistOptions)Twistc@steZdZdZdddZdddZddd Zdd d Zdd d ZdddZ dddZ dddZ dddZ dddZ dS) TwistTestsz Tests for L{Twist}. returnNcCs |dSN)patchInstallReactorselfroC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/application/twist/test/test_twist.pysetUps zTwistTests.setUpcCst|_|td|jdS)zX Patch L{_twist.exit} so we can capture usage and prevent actual exits. exitN)rrpatchrrrrr patchExit!szTwistTests.patchExitcs2i_dtdtdtffdd }td|dS)zo Patch C{_options.installReactor} so we can capture usage and prevent actual installs. _namercst}|j|<|Sr)rinstalledReactors)r!r"reactorrrrinstallReactor/s z6TwistTests.patchInstallReactor..installReactorr%N)r#rstrrr)rr%rrrr(szTwistTests.patchInstallReactorcs.g_dtddffdd }td|dS)zo Patch L{MultiService.startService} so we can capture usage and prevent actual starts. servicerNcsj|dSr) serviceStartsappend)r'rrr startService=sz2TwistTests.patchStartService..startServicer*)r(rrr)rr*rrrpatchStartService6szTwistTests.patchStartServicecCstddg}||tdS)zI L{Twist.options} given valid arguments returns options. twistwebN)roptionsZassertIsInstancer)rr.rrrtest_optionsValidArgumentsBsz%TwistTests.test_optionsValidArgumentscCsj|tddg||jjtj||jj | |jj d| |jj dt dS)z L{Twist.options} given invalid arguments exits with L{ExitStatus.EX_USAGE} and an error/usage message. r,z--bogus-bagelszError: z N)r rr.ZassertIdenticalrstatusr ZEX_USAGEZassertIsNotNonemessage assertTrue startswithendswithrrrrrtest_optionsInvalidArgumentsJs  z'TwistTests.test_optionsInvalidArgumentscCs6tddg}t|jd|j}|t|dS)z: L{Twist.service} returns an L{IService}. r,r-N)rr.r'plugins subOptionsr2rZ providedBy)rr.r'rrr test_service^szTwistTests.test_servicecCstddg}|d}|j}|dusJtj|j||jd}|t||||j |g||j dd|j difgdS) z L{Twist.startService} starts the service and registers a trigger to stop the service when the reactor shuts down. r,r-r$N)Zpluginr.beforeshutdownr) rr. subCommandr'r6r7r+r* assertEqualr(triggersZ stopService)rr.r$r;r'rrrtest_startServicefs  zTwistTests.test_startServicec s|g|tdfdd|tdddtgd}t||td|dt|jd t j t t d d S) zi L{Twist.run} runs the runner with arguments corresponding to the given options. __init__cs |Sr)r))rargsZargsSeenrrs z%TwistTests.test_run..runcSsdSrrrrrrrBr,z--reactor=defaultz--log-format=jsonr-rdefaultr$ZdefaultLogLevelZlogFilefileLogObserverFactoryN) rr rr.rCr<lendictr#rinforr )rZ twistOptionsrrArtest_run~s" zTwistTests.test_runc s|gGfddd}|td|tgd|t|jd|td|djt |j dt j t td|djddS) zy L{Twist.main} runs the runner with arguments corresponding to the given command line arguments. cs,eZdZdeddffdd ZdddZdS) z$TwistTests.test_main..RunnerkwargsrNcs||_d|_|dS)Nr)r@runsr))rrNrunnersrrr?sz-TwistTests.test_main..Runner.__init__cSs|jd7_dS)NrF)rOrrrrrCsz(TwistTests.test_main..Runner.runrN)__name__ __module__ __qualname__rr?rCrrPrrr sr rErFrrGrHN)r+rrrmainr<rJr(r@rKr#rrLrr rO)rr rrPr test_mains"  zTwistTests.test_mainrR)rSrTrU__doc__rr rr+r/r5r8r>rMrWrrrrrs        rc@s8eZdZdZd ddZd ddZd dd Zd d d ZdS) TwistExitTestszr Tests to verify that the Twist script takes the expected actions related to signals and the reactor. rNcsNd_dtddffdd }td|dtddfdd }td |dS) NFsigrcs d_dS)z Fake to capture whether L{twisted.application._exitWithSignal was called. @param sig: Signal value @type sig: C{int} TN)exitWithSignalCalled)rZrrrfakeExitWithSignals z0TwistExitTests.setUp..fakeExitWithSignalZ_exitWithSignalr!cSsdS)z Prevent Runner from adding new log observers or other tests outside this module will fail. @param _: Unused self param Nr)r!rrr startLoggingrDz*TwistExitTests.setUp..startLoggingr])r[intrrr )rr\r]rrrrs  zTwistExitTests.setUpcCs<t}d|_t}||d<t|d<t|||jdS)zg _exitWithSignal is not called if the reactor's _exitSignal attribute is zero. Nr$rI)r _exitSignalrr rrC assertFalser[rr$r.rrr%test_twistReactorDoesntExitWithSignals z4TwistExitTests.test_twistReactorDoesntExitWithSignalcCs6t}t}||d<t|d<t|||jdS)z _exitWithSignal is not called if the runner's reactor does not implement L{twisted.internet.interfaces._ISupportsExitSignalCapturing} r$rIN)rrr rrCr`r[rarrr$test_twistReactorHasNoExitSignalAttrs  z3TwistExitTests.test_twistReactorHasNoExitSignalAttrcCs<t}d|_t}||d<t|d<t|||jdS)zb _exitWithSignal is called if the runner's reactor exits due to a signal. rr$rIN)r r_rr rrCr2r[rarrr test_twistReactorExitsWithSignals z/TwistExitTests.test_twistReactorExitsWithSignalrR)rSrTrUrXrrbrcrdrrrrrYs     rY)%rXsysrtypingrrrZtwisted.trial.unittestZtwistedZtwisted.internet.interfacesrZtwisted.internet.testingrZtwisted.loggerrr Ztwisted.test.test_twistdr Z runner._exitr Zrunner._runnerr Zrunner.test.test_runnerrr'rrr,r_optionsrrZtrialZunittestZTestCaserrYrrrrs$          $