Yf2dZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZddlmZd d lmZd d lmZd d lmZd d lmZmZd dlmZddlmZddlmZGddej j!j"Z#Gddej 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)TwistcbeZdZdZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z dS) TwistTestsz Tests for L{Twist}. returnNc.|dSN)patchInstallReactorselfs [/var/www/html/env/lib/python3.11/site-packages/twisted/application/twist/test/test_twist.pysetUpzTwistTests.setUps   """""cnt|_|td|jdS)zX Patch L{_twist.exit} so we can capture usage and prevent actual exits. exitN)rr"patchrrs r patchExitzTwistTests.patchExit!s-KK  6649-----r ci_dtdtdtffd }td|dS)zo Patch C{_options.installReactor} so we can capture usage and prevent actual installs. _namerc8t}|j|<|Sr)rinstalledReactors)r&r'reactorrs rinstallReactorz6TwistTests.patchInstallReactor..installReactor/s#ooG+2D "4 (Nr r+N)r)rstrrr#)rr+s` rrzTwistTests.patchInstallReactor(s_ ;= l # ,       BBBBBr clg_dtddffd }td|dS)zo Patch L{MultiService.startService} so we can capture usage and prevent actual starts. servicerNc<j|dSr) serviceStartsappend)r.rs r startServicez2TwistTests.patchStartService..startService=s   % %g . . . . .r r2)r0rr#r)rr2s` rpatchStartServicezTwistTests.patchStartService6sW .0 /( /t / / / / / / <>>>>>r chtjddg}||tdS)zI L{Twist.options} given valid arguments returns options. twistwebN)roptionsassertIsInstancer)rr7s rtest_optionsValidArgumentsz%TwistTests.test_optionsValidArgumentsBs4-% 011 g|44444r c|tjddg||jjt j||jj | |jj d| |jj dtdS)z L{Twist.options} given invalid arguments exits with L{ExitStatus.EX_USAGE} and an error/usage message. r5z--bogus-bagelszError: z N)r$rr7assertIdenticalr"statusr EX_USAGEassertIsNotNonemessage assertTrue startswithendswithrrs rtest_optionsInvalidArgumentsz'TwistTests.test_optionsInvalidArgumentsJs  w 01222 TY-z/BCCC TY.///  I  ( ( 3 3     I  & &'|~~''       r ctjddg}tj|jd|j}|t j|dS)z: L{Twist.service} returns an L{IService}. r5r6N)rr7r.plugins subOptionsr@r providedBy)rr7r.s r test_servicezTwistTests.test_service^sV-% 011- 68JKK +G4455555r ctjddg}|d}|j}|Jtj|j||j}|tj||||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. r5r6r*N)pluginr7beforeshutdown) rr7 subCommandr.rErFr3r2 assertEqualr0triggers stopService)rr7r*rNr.s rtest_startServicezTwistTests.test_startServicefs -% 011)$' %%%-?:.&        7G,,, +gY777   X &z 2g6I2r5R4S     r c g|tdfd|tddtjgd}tj||t d|dt|jdtj tt d S) zi L{Twist.run} runs the runner with arguments corresponding to the given options. __init__c.|Sr)r1)rargsargsSeens rz%TwistTests.test_run..sHOOD.str r5z--reactor=defaultz--log-format=jsonr6rdefaultr*defaultLogLevellogFilefileLogObserverFactoryN) r#rrr7rYrOlendictr)r inforr )r twistOptionsrWs @rtest_runzTwistTests.test_run~s  6:'Q'Q'Q'QRRR 65"3"3444} F F F    , X***  QK .y9 ( ':         r c 0|gGfdd}|td|tjgd|t |jd|t d|djt|j dtj tt|djddS) zy L{Twist.main} runs the runner with arguments corresponding to the given command line arguments. c,eZdZdeddffd ZddZdS)$TwistTests.test_main..RunnerkwargsrNcN||_d|_|dS)Nr)rVrunsr1)rrjrunnerss rrTz-TwistTests.test_main..Runner.__init__s)"  t$$$$$r c&|xjdz c_dS)Nr\)rlrs rrYz(TwistTests.test_main..Runner.runs Q r rN)__name__ __module__ __qualname__rrTrY)rmsrrrisU % % % % % % % %       r rr[r\rr]r^N)r3r#rrmainrOrbr0rVrcr)r rdrr rl)rrrms @r test_mainzTwistTests.test_mains'               68V,,, MMMNNN T/00!444 Wq)))  AJO .y9 ( ':        !,,,,,r ro)rprqrr__doc__rr$rr3r9rCrHrRrfrtrMr rrrs####.... C C C C ? ? ? ?5555    (6666    0    4!-!-!-!-!-!-r rc2eZdZdZddZddZddZddZdS) TwistExitTestszr Tests to verify that the Twist script takes the expected actions related to signals and the reactor. rNcd_dtddffd }td|dtddfd}td|dS) NFsigrcd_dS)z Fake to capture whether L{twisted.application._exitWithSignal was called. @param sig: Signal value @type sig: C{int} TN)exitWithSignalCalled)ryrs rfakeExitWithSignalz0TwistExitTests.setUp..fakeExitWithSignals)-D % % %r _exitWithSignalr&cdS)z Prevent Runner from adding new log observers or other tests outside this module will fail. @param _: Unused self param NrM)r&s r startLoggingz*TwistExitTests.setUp..startLoggingsr r)r{intr#rr)rr|rs` rrzTwistExitTests.setUps$)! -C -D - - - - - - 6,.@AAA F t     6><88888r ct}d|_t}||d<t|d<t j|||jdS)zg _exitWithSignal is not called if the reactor's _exitSignal attribute is zero. Nr*ra)r _exitSignalrr rrY assertFalser{rr*r7s r%test_twistReactorDoesntExitWithSignalz4TwistExitTests.test_twistReactorDoesntExitWithSignalsb /00"..$ ,?() ' 233333r ct}t}||d<t|d<tj|||jdS)z _exitWithSignal is not called if the runner's reactor does not implement L{twisted.internet.interfaces._ISupportsExitSignalCapturing} r*raN)rrr rrYrr{rs r$test_twistReactorHasNoExitSignalAttrz3TwistExitTests.test_twistReactorHasNoExitSignalAttrsX  //..$ ,?() ' 233333r ct}d|_t}||d<t|d<t j|||jdS)zb _exitWithSignal is called if the runner's reactor exits due to a signal. rr*raN)r rrr rrYr@r{rs r test_twistReactorExitsWithSignalz/TwistExitTests.test_twistReactorExitsWithSignals` /00..$ ,?() ' 122222r ro)rprqrrrurrrrrMr rrwrwsn 99992 4 4 4 4 4 4 4 4 3 3 3 3 3 3r rw)%rusysrtypingrrrtwisted.trial.unittesttwistedtwisted.internet.interfacesrtwisted.internet.testingrtwisted.loggerr r twisted.test.test_twistdr runner._exitr runner._runnerrrunner.test.test_runnerrr.rrr5r_optionsrrtrialunittestTestCaserrwrMr rrs""""""""""44444422222288888888AAAAAA&&&&&&$$$$$$000000--------######`-`-`-`-`-'0`-`-`-FD3D3D3D3D3W]+4D3D3D3D3D3r