U n5{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@seZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ ddddZ ddddZ ddddZ dS) TwistTestsz Tests for L{Twist}. NreturncCs |dSN)patchInstallReactorselfrM/tmp/pip-unpacked-wheel-mnhg7r0q/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)rinstalledReactors)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/ZassertIdenticalrstatusr 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 subOptionsr4rZ providedBy)rr/r(rrr test_service^szTwistTests.test_servicecCstddg}|d}|j}|dk s(ttj|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/ subCommandAssertionErrorr(r8r9r,r+ assertEqualr)ZtriggersZ stopService)rr/r%r=r(rrrtest_startServicefs   zTwistTests.test_startServicec sg|tdfdd|tdddtdddd g}t||td |d t|jd t j t t d dS)zi L{Twist.run} runs the runner with arguments corresponding to the given options. __init__cs |Sr)r*)rargsZargsSeenrrz%TwistTests.test_run..runcSsdSrrrrrrrDrEr---reactor=default--log-format=jsonr.rdefaultr%ZdefaultLogLevelZlogFilefileLogObserverFactoryN) r r rr/rFr?lendictr$rinforr )rZ twistOptionsrrCrtest_run~s"  zTwistTests.test_runc s|gGfddd}|td|tddddg|t|jd|td|djt |j d t j t td |djdd S) 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)rBrunsr*)rrQZrunnersrrrAsz-TwistTests.test_main..Runner.__init__rcSs|jd7_dS)NrI)rRrrrrrFsz(TwistTests.test_main..Runner.run)__name__ __module__ __qualname__rrArFrrSrrr sr r-rGrHr.rIrrJrKN)r,r rrmainr?rMr)rBrNr$rrOrr rR)rr rrSr test_mains"  zTwistTests.test_main)rTrUrV__doc__rr!rr,r0r7r:r@rPrXrrrrrs 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)r[rrrfakeExitWithSignalsz0TwistExitTests.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..startLoggingr^)r\intr rr )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%rL)r _exitSignalrr rrF 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%rLN)rrr rrFrar\rbrrr$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%rLN)r r`rr rrFr4r\rbrrr test_twistReactorExitsWithSignals z/TwistExitTests.test_twistReactorExitsWithSignal)rTrUrVrYrrcrdrerrrrrZs  rZ)%rYsysrtypingrrrZtwisted.trial.unittestZtwistedZtwisted.internet.interfacesrZtwisted.internet.testingrZtwisted.loggerrr Ztwisted.test.test_twistdr Z runner._exitr Zrunner._runnerr Zrunner.test.test_runnerrr(rrr-r_optionsrrZtrialZunittestZTestCaserrZrrrrs"          $