a )(b@sdZddlmZddlmZmZmZddlZddl m Z ddl m Z m Z ddlmZddlmZd 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)LogLeveljsonFileLogObserver) MemoryReactor)SignalCapturingMemoryReactor) ExitStatus)Runner) DummyExit)IService MultiService)_twist) TwistOptions)Twistc@seZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ ddddZ ddddZ ddddZ dS) TwistTestsz Tests for L{Twist}. NreturncCs |dSN)patchInstallReactorselfrHlib/python3.9/site-packages/twisted/application/twist/test/test_twist.pysetUpszTwistTests.setUpcCst|_|td|jdS)zX Patch L{_twist.exit} so we can capture usage and prevent actual exits. exitN)rrpatchrrrrr patchExit!szTwistTests.patchExitcs.i_tttdfdd }td|dS)zo Patch C{_options.installReactor} so we can capture usage and prevent actual installs. )_namercst}|j|<|Sr)r installedReactors)r"r#reactorrrrinstallReactor/s z6TwistTests.patchInstallReactor..installReactorr&N)r$rstrrr )rr&rrrr(szTwistTests.patchInstallReactorcs,g_tddfdd }td|dS)zo Patch L{MultiService.startService} so we can capture usage and prevent actual starts. N)servicercsj|dSr) serviceStartsappend)r(rrr startService=sz2TwistTests.patchStartService..startServicer+)r)rr r)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/ZassertIdenticalrZstatusr EX_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 subOptionsr3rZ providedBy)rr/r(rrr test_service^szTwistTests.test_servicecCstddg}|d}|j}|dus(Jtj|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/ZbeforeZshutdownr) rr/ subCommandr(r7r8r,r+ assertEqualr)ZtriggersZ 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*)rargsZargsSeenrrz%TwistTests.test_run..runcSsdSrrrrrrr@rAr-z--reactor=defaultz--log-format=jsonr.rdefaultr%ZdefaultLogLevelZlogFilefileLogObserverFactoryN) r r rr/rBr;lendictr$rinforr)rZ twistOptionsrr?rtest_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.eZdZeddfdd ZddddZdS)z$TwistTests.test_main..RunnerN)kwargsrcs||_d|_|dS)Nr)r>runsr*)rrLZrunnersrrr=sz-TwistTests.test_main..Runner.__init__rcSs|jd7_dS)NrD)rMrrrrrBsz(TwistTests.test_main..Runner.run)__name__ __module__ __qualname__rr=rBrrNrrr sr rCrDrrErFN)r,r rrmainr;rHr)r>rIr$rrJrrrM)rr rrNr test_mains"  zTwistTests.test_main)rOrPrQ__doc__rr!rr,r0r6r9r<rKrSrrrrrs rc@sHeZdZdZddddZddddZdddd Zddd d ZdS) TwistExitTestszr Tests to verify that the Twist script takes the expected actions related to signals and the reactor. NrcsJd_tddfdd }td|tdddd}td |dS) NF)sigrcs d_dS)z Fake to capture whether L{twisted.application._exitWithSignal was called. @param sig: Signal value @type sig: C{int} TN)exitWithSignalCalled)rVrrrfakeExitWithSignalsz0TwistExitTests.setUp..fakeExitWithSignalZ_exitWithSignal)r"rcSsdS)z Prevent Runner from adding new log observers or other tests outside this module will fail. @param _: Unused self param Nr)r"rrr startLoggingsz*TwistExitTests.setUp..startLoggingrY)rWintr rr )rrXrYrrrrs  zTwistExitTests.setUpcCs<t}d|_t}||d<t|d<t|||jdS)zg _exitWithSignal is not called if the reactor's _exitSignal attribute is zero. Nr%rG)r _exitSignalrrrrB assertFalserWrr%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%rGN)r rrrrBr\rWr]rrr$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%rGN)r r[rrrrBr3rWr]rrr test_twistReactorExitsWithSignals z/TwistExitTests.test_twistReactorExitsWithSignal)rOrPrQrTrr^r_r`rrrrrUs  rU)%rTsysrtypingrrrZtwisted.trial.unittestZtwistedZtwisted.internet.interfacesrZtwisted.loggerrrZtwisted.test.proto_helpersr Ztwisted.test.test_twistdr Z runner._exitr Zrunner._runnerr Zrunner.test.test_runnerrr(rrr-rZ_optionsrrZtrialZunittestZTestCaserrUrrrrs"          $