U  W[R@sddlmZmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZddlmZmZmZddlmZddlmZddlmZdd lmZdd lmZmZmZd d d d d d dZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&dej'fde&fdZ(ddZ)Gd d!d!e*Z+Gd"d#d#e+ej,ej-Z,d$d%Z.d&d'Z/d(d)Z0d*d+Z1Gd,d-d-e2Z3d.d/Z4d0d1Z5dS)2)absolute_importdivisionprint_functionN)defer)app)usagereflectfailure)FilePath) namedModule)long)plugin)runneritrialreporterdefaultbriefverbose)plainremacsrcgitbrcCsd}||t|}||}|dks0|dkr>td|f|||d}i}|D]T}t|dkrnqX|d}t|dkrtd||f|d ||d<qX|S) a  Accepts a single line in Emacs local variable declaration format and returns a dict of all the variables {name: value}. Raises ValueError if 'line' is in the wrong format. See http://www.gnu.org/software/emacs/manual/html_node/File-Variables.html z-*-z)%r not a valid local variable declaration;r:z"%r contains invalid declaration %r)findlenrfind ValueErrorsplitstrip)lineZparenstartenditemsZ localVarsitemr r'7/usr/lib/python3/dist-packages/twisted/scripts/trial.py_parseLocalVariables&s"   r)c CsXt|d}||g}W5QRX|D](}zt|WStk rPYq*Xq*iS)z Accepts a filename and attempts to load the Emacs variable declarations from that file, simulating what Emacs does. See http://www.gnu.org/software/emacs/manual/html_node/File-Variables.html r)openreadliner)r)filenameflinesr"r'r'r(loadLocalVariables@s r0cCs&t|dd}|dkrgS|dS)Nztest-case-name,)r0getr )r-Z testCaseVarr'r'r(getTestModulesQsr3cCs*tj|}|do(tj|ddkS)z Returns true if 'filename' looks like a file containing unit tests. False otherwise. Doesn't care whether filename exists. Ztest_rz.py)ospathbasename startswithsplitext)r-r6r'r'r( isTestFileXs  r9cCstddttjDS)NcSsg|] }|jqSr')longOpt).0pr'r'r( csz#_reporterAction..)r CompleteListr getPluginsr IReporterr'r'r'r(_reporterActionbs  rAc Csht|dd}|dk rt||}t|dd}|dk r8|jSzt|dWSttfk rbYdSXdS)aH Try to find the source line of the given test thing. @param testThing: the test item to attempt to inspect @type testThing: an L{TestCase}, test method, or module, though only the former two have a chance to succeed @rtype: int @return: the starting source line, or -1 if one couldn't be found Z_testMethodNameN__code__rr)getattrco_firstlinenoinspectZgetsourcelinesIOError TypeError)Z testThingmethodcoder'r'r(_maybeFindSourceLinegs   rJzCalphabetical order for test methods, arbitrary order for test caseszDattempt to run test cases and methods in the order they were defined)Z alphabeticalZ toptobottomcCs,|tkr(tddddtDf|S)ai Check that the given order is a known test running order. Does nothing else, since looking up the appropriate callable to sort the tests should be done when it actually will be used, as the default argument will not be coerced by this function. @param order: one of the known orders in C{_runOrders} @return: the order unmodified z9--order must be one of: %s. See --help-orders for detailsz, css|]}t|VqdSN)repr)r;orderr'r'r( sz&_checkKnownRunOrder..) _runOrdersr UsageErrorjoin)rMr'r'r(_checkKnownRunOrders rRc @sJeZdZdZdZddgdddgdd d gd d d gd ddgdd dgdd dgdddggZddd degddd dgdd ddgd d d!d"ggZej e e e ej d#d$ejd%d$d&ej d'd(d)d*gd+ZejZd Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Zd@dAZdBdCZdDdEZ dFdGZ!dHdIZ"dJdKZ#d S)L _BasicOptionszC Basic options shared between trial and its local workers. zutrial loads and executes a suite of unit tests, obtained from modules, packages and files listed on the command line.helph no-recurseNzDon't recurse into packagesz help-ordersNz%Help on available test running orderszhelp-reportersz,Help on available output plugins (reporters)rterrorsez;realtime errors, print out tracebacks as soon as they occurunclean-warningsz'Turn dirty reactor errors into warningsforce-gcz.   r)rvsortedrOr%rnexit)rlsynopsisname description_r'r'r(opt_help_orders s z_BasicOptions.opt_help_orderscCs>d}t|ttjD]}td|jd|jqtddS)NzTrial's output can be customized using plugins called Reporters. You can select any of the following reporters using --reporter= rrr) rvr r?rr@r:rrnr)rlrr<r'r'r(opt_help_reporters+s z _BasicOptions.opt_help_reporterscCsd|d<tdS)z/ Disable the garbage collector T disablegcN)gcdisablerkr'r'r( opt_disablegc5sz_BasicOptions.opt_disablegccCs4zt||d<Wntk r.tdYnXdS)z Specify the format to display tracebacks with. Valid formats are 'plain', 'emacs', and 'cgitb' which uses the nicely verbose stdlib cgitb.text function tbformatz.tbformat must be 'plain', 'emacs', or 'cgitb'.N) TBFORMAT_MAPKeyErrorrrP)rlZoptr'r'r( opt_tbformat=s z_BasicOptions.opt_tbformatc CsFztt|Wn"ttfk r4tdYnXt||d<dS)z- see sys.setrecursionlimit() z-argument to recursionlimit must be an integerrecursionlimitN)rnsetrecursionlimitintrGrrrP)rlargr'r'r(opt_recursionlimitJs z _BasicOptions.opt_recursionlimitcCslzt||d<Wntk r.tdYn:X|ddkrHtdn |ddkrhttd|d<dS)Nr_z/Argument to --random must be a positive integerrd)r rrrPtime)rloptionr'r'r( opt_randomWs   z_BasicOptions.opt_randomcCsD||d<|dD],}|tjkr4tjd|ftddtj|<qdS)zP Fake the lack of the specified modules, separated with commas. without-moduler1z/Module '%s' already imported, disabling anyway.)categoryN)r rnmoduleswarningswarnRuntimeWarning)rlrmoduler'r'r(opt_without_modulees z _BasicOptions.opt_without_modulecGs|d|dSrg)r)rlargsr'r'r( parseArgsrsz_BasicOptions.parseArgscCsHttjD],}d|j|jf}|j|kr t|Sq t ddS)Nz%s.%szVOnly pass names of Reporter plugins to --reporter. See --help-reporters for more info.) r r?rr@rklassr:rnamedAnyrrP)rlrr<Zqualr'r'r(_loadReporterByNamevs  z!_BasicOptions._loadReporterByNamecCsH||d|d<d|kr"d|d<|ddk rD|ddk rDtddS)NrrrrMr_z-You can't specify --random when using --order)rrrPrkr'r'r( postOptionssz_BasicOptions.postOptions)$__name__ __module__ __qualname____doc__ZlongdescoptFlagsrR optParametersr Completionsr>rOrAZ CompleteFilesZ CompletercompDatar TreeReporterfallbackReporterr{rjrtrwr}rrrrrrrrrrrrr'r'r'r(rSsz             rSc@seZdZdZdddgdddgdd d gd d d gdd dgdddggZdd ddgddddgddd dggZejedddge d d!Z d"d#d$gZ d%d&d'gZ e jZd Zd Zd(d)Zd*d+Zd,d-Zd S).ria Options to the trial command line tool. @ivar _workerFlags: List of flags which are accepted by trial distributed workers. This is used by C{_getWorkerArguments} to build the command line arguments. @type _workerFlags: C{list} @ivar _workerParameters: List of parameter which are accepted by trial distributed workers. This is used by C{_getWorkerArguments} to build the command line arguments. @type _workerParameters: C{list} debugbziRun tests in a debugger. If that debugger is pdb, will load '.pdbrc' from current directory if it exists.debug-stacktracesBz2Report Deferred creation and callback stack tracesnopmNzFdon't automatically jump into debugger for postmorteming of exceptionsdry-runnzdo everything but run the testsprofilez#Run tests under the Python profiler until-failureuzRepeat test until it failsdebuggerpdbzBthe fully qualified name of a debugger to use if --debug is passedrdlztest.logrbjobsjzNumber of local workers to runrrr)rr)rfrr[rurZreactorrcCsNz t|}Wn"tk r.td|YnX|dkrBtd||d<dS)zN Number of local workers to run, a strictly positive integer. z,Expecting integer argument to jobs, got '%s'rz4Argument to jobs must be a strictly positive integerrN)rrrrP)rlZnumberr'r'r(opt_jobss  zOptions.opt_jobscCspg}|jD]*}||dk r ||r |d|fq |jD].}||dk r<|d|ft||gq<|S)zJ Return a list of options to pass to distributed workers. Nz--%s) _workerFlagsr2r_workerParametersrstr)rlrrr'r'r(_getWorkerArgumentss  zOptions._getWorkerArgumentscCsbt||dr>ddddg}|D]}||r"td|q"|dr^|dsXtdd t_dS) Nrrrrr\z(You can't specify --%s when using --jobsrz+You must specify --debug when using --nopm F)rSrrrPr ZDO_POST_MORTEM)rlZ conflictsrr'r'r(rs   zOptions.postOptions)rrrrrrrrr>rArrrrrrZextrar{rrrr'r'r'r(ris2     ricCs.|drt|ds |dr*tddS)NrrT)r ZstartDebugModerZ setDebugging)configr'r'r(_initialDebugSetupsrcCs$t|}|d }|j|d|dS)NrVrh)recurse) _getLoaderZ loadByNames)rloaderrr'r'r( _getSuites rcsxt}|drFt|dfdd|_td|dn|drdt|d\}}||_|dsttj|_ |S)Nr_csSrK)r_)r]Zrandomerr'r(z_getLoader..z$Running tests shuffled with seed %d rMr) rZ TestLoaderr_ZRandomZseedsorterrvrOZDestructiveTestSuiteZ suiteFactory)rrrrr'rr(rsrc Cst}z tdWntk r0tdYnXdD]T}tj|r6zt|d}Wnt k rhYq6X||j | W5QRXq6|S)zR Wrap an instance of C{pdb.Pdb} with readline support and load any .rcs. r,zreadline module not available)z.pdbrcZpdbrcr*) rZPdbr ImportErrorrvr4r5existsr+rFZrcLinesr readlines)Zdbgr5ZrcFiler'r'r( _wrappedPdb s  rc@seZdZdZdS)_DebuggerNotFoundzk A debugger import failed. Used to allow translating these errors into usage error messages. N)rrrrr'r'r'r(r#srcCstj}|d|d|d|d|d|dd}|drFtjj|d <n|d rxd d lm}|}|d |d <||d<n|drtjj|d <|d}|dkrzt||d<Wqtj k rt d|fYqXn t |d<|d|d<|d|d<|d|d<|f|S)z Return a trial runner class set up with the parameters extracted from C{config}. @return: A trial runner instance. @rtype: L{runner.TrialRunner} or C{DistTrialRunner} depending on the configuration. rrrXrZrdra)ZreporterFactoryZtracebackFormatZrealTimeErrorsZuncleanWarningsrdZworkingDirectoryrmoderr)DistTrialRunnerZ workerNumberZworkerArgumentsrrrz%r debugger could not be found.r\Z exitFirstrr[ZforceGarbageCollection) rZ TrialRunnerZDRY_RUNZtwisted.trial._dist.disttrialrrDEBUGrrZModuleNotFoundrr)rclsrrrr'r'r( _makeRunner-s<         rc Cs&ttjdkrtjdt}z |Wn:tjk rf}ztdtjd|fW5d}~XYnXt |z t |}Wn<t k r}ztdtjdt |fW5d}~XYnXt |}|dr||}n ||}|jrtd|j}|jdd|jdt| dS)Nrz--helpz%s: %srrF)Z show_missingZsummaryrw)rrnrorriZ parseOptionsrerror SystemExitrrrrrZrunUntilFailurerunr{r|resultsZ write_resultsrwr5rZ wasSuccessful)rZueZ trialRunnerrYZsuiteZ test_resultrr'r'r(rZs.  ( ,    r)6Z __future__rrrrrEr4rr_rnrrZtwisted.internetrZtwisted.applicationrZtwisted.pythonrrr Ztwisted.python.filepathr Ztwisted.python.reflectr Ztwisted.python.compatr Ztwistedr Z twisted.trialrrrrr)r0r3r9rArJrrOrRobjectrSriZReactorSelectionMixinrrrr Exceptionrrrr'r'r'r(s^        ! k`  -