o 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@steZdZdZdddZdddZddd Zdd d Zdd d ZdddZ dddZ dddZ dddZ dddZ dS) TwistTestsz Tests for L{Twist}. returnNcCs |dSN)patchInstallReactorselfrK/usr/lib/python3/dist-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)r installedReactors)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)roptionsassertIsInstancer)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.assertIdenticalrstatusr EX_USAGEassertIsNotNonemessage 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 subOptionsr6r 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)pluginr.beforeshutdownr) rr. subCommandr'r:r;r+r* assertEqualr(triggers stopService)rr.r$rAr'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))rargsargsSeenrrs z%TwistTests.test_run..runcSsdSrrrrrrrJr,z--reactor=defaultz--log-format=jsonr-rdefaultr$defaultLogLevellogFilefileLogObserverFactoryN) rr rr.rKrBlendictr#rinforr)r twistOptionsrrHrtest_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)rGrunsr))rrYrunnersrrrFsz-TwistTests.test_main..Runner.__init__cSs|jd7_dS)NrN)rZrrrrrKsz(TwistTests.test_main..Runner.runrN)__name__ __module__ __qualname__rrFrKrr[rrr sr rMrNrrOrPN)r+rrrmainrBrTr(rGrUr#rrVrrrZ)rr rr[r test_mains"  zTwistTests.test_mainr])r^r_r`__doc__rr rr+r0r9r=rErXrbrrrrrs        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)rerrrfakeExitWithSignals z0TwistExitTests.setUp..fakeExitWithSignal_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 startLoggingrLz*TwistExitTests.setUp..startLoggingri)rfintrrr )rrgrirrrrs  zTwistExitTests.setUpcCs<t}d|_t}||d<t|d<t|||jdS)zg _exitWithSignal is not called if the reactor's _exitSignal attribute is zero. Nr$rS)r _exitSignalrrrrK assertFalserfrr$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$rSN)r rrrrKrlrfrmrrr$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$rSN)r rkrrrrKr6rfrmrrr test_twistReactorExitsWithSignals z/TwistExitTests.test_twistReactorExitsWithSignalr])r^r_r`rcrrnrorprrrrrds     rd)%rcsysrtypingrrrtwisted.trial.unittesttwistedtwisted.internet.interfacesrtwisted.loggerrrtwisted.test.proto_helpersr twisted.test.test_twistdr runner._exitr runner._runnerr runner.test.test_runnerrr'rrr,r_optionsrrtrialunittestTestCaserrdrrrrs$          $