U `[:@sdZddlmZmZddddddd d d d d ddddgZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z mZmZmZmZddlmZmZddlmZddlmZmZddlmZddlmZmZddlm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'e(dZ)dd Z*dd Z+d dZ,d!d Z-d6d"d#Z.d$d%Z/d&d'Z0Gd(dde%Z1d)Z2Gd*dde%Z3Gd+d d e%Z4d,dZ5d-dZ6e'eGd.dde7Z8Gd/dde8Z9Gd0dde7Z:Gd1d2d2e:Z;d3d4ZGiven an object return True if the object looks like a packageFr__init__) isinstancetypes ModuleTypeospathsplitextbasename__file__)moduler+r.6/usr/lib/python3/dist-packages/twisted/trial/runner.pyr 2s cCs>dd}|D]*}d|}tjtj||r|SqdS)z Is the directory at path 'dirname' a Python package directory? Returns the name of the __init__ file (it may have a weird extension) if dirname is a package directory. Otherwise, returns False cSs6trddl}|jSddl}tt|dSdSNr)r importlib machinery all_suffixesimplistzipZ get_suffixes)r1r4r.r.r/ _getSuffixes@s  z(isPackageDirectory.._getSuffixesr$F)r(r)existsjoin)dirnamer7ZextinitFiler.r.r/r:s   cCstj|tj|kS)z A hacky implementation of C{os.path.samefile}. Used by L{filenameToModule} when the platform doesn't provide C{os.path.samefile}. Do not use this. )r(r)abspath)Z filename1 filename2r.r.r/rPsc Cstj|std|fztt|}Wn ttfk rNt|YSXt |dddkrht|Stj |j dd}t tjdt }tj |r|||stj|j=t|}|S)aV Given a filename, do whatever possible to return a module object matching that file. If the file in question is a module in Python path, properly import and return that module. Otherwise, load the source manually. @param fn: A filename. @return: A module object. @raise ValueError: If C{fn} does not exist. z%r doesn't existr,Nrz.pyr)r(r)r8 ValueErrorrnamedAnyfilenameToModuleNameAttributeError_importFromFilegetattrr*r,risfilesysr__name__)fnZretZretFileZsamer.r.r/r Xs  c Cst|}|s(tjtj|dd}|tjkr)rGr;r.r.r/rJs  rJcCsHt|j|jt|krBt|jD] }t|j|t|kr |Sq |jS)z Find the attribute name on the method's class which refers to the method. For some methods, notably decorators which have not had __name__ set correctly: getattr(method.im_class, method.__name__) != method )rCim_classrFobjectdir)methodaliasr.r.r/_getMethodNameInClasss  rZc@seZdZdZddZdS)rzQ A test suite which remove the tests once run, to minimize memory usage. cCs(|jr$|jrq$|jd}||q|S)zn Almost the same as L{TestSuite.run}, but with C{self._tests} being empty at the end. r)_testsZ shouldStoppop)selfresulttestr.r.r/runs   zDestructiveTestSuite.runN)rF __module__ __qualname____doc__r`r.r.r.r/rsz cs eZdZdZfddZZS)rz[ Any errors logged in this suite will be reported to the L{TestResult} object. csNt}|tt||||D]}|tt |q,| dS)z Run the suite, storing all errors in C{result}. If an error is logged while no tests are running, then it will be added as an error to C{result}. @param result: A L{TestResult} object. N) r"Z_addsuperrr`_removeZ getErrorsaddErrorrrZ flushErrors)r]r^Zobservererror __class__r.r/r`s zLoggedSuite.run)rFrarbrcr` __classcell__r.r.rhr/rscs2eZdZdZd fdd ZddZdd ZZS) r z Suite to wrap around every single test in a C{trial} run. Used internally by Trial to set up things necessary for Trial tests to work, regardless of what context they are run in. r.FcsJ|r,g}|D]}t|t}||q |}t|}tt||gdSN)rdecorater appendrrdr r$)r]testsforceGarbageCollectionZnewTestsr_suiterhr.r/r$s zTrialSuite.__init__csHddlm}t|ddfdd|dtddS)Nr)reactorZafterZshutdowncs dSrk)callbackr.dr.r/z"TrialSuite._bail..Zmktemp) twisted.internetrqrZDeferredZaddSystemEventTriggerZfireSystemEventrTestCaseZ_wait)r]rqr.rsr/_bails   zTrialSuite._bailcCs zt||W5|XdSrk)ryrr`r]r^r.r.r/r`szTrialSuite.run)r.F)rFrarbrcr$ryr`rjr.r.rhr/r s  cCs@t|rt|}n(z |}Wntk r:|j}YnX|S)z @param thing: an object from modules (instance of PythonModule, PythonAttribute), a TestCase subclass, or an instance of a TestCase. )rrqualidrAr)thingZtheNamer.r.r/rs   cCs*zt|tjWStk r$YdSXdS)z @return: C{True} if C{obj} is a class that contains test cases, C{False} otherwise. Used to find all the tests in a module. FN) issubclasspyunitrx TypeError)objr.r.r/rsc@sDeZdZdZdZddZddZddZd d Zd d Z d dZ dS)rz Placeholder for a L{TestCase} inside a reporter. As far as a L{TestResult} is concerned, this looks exactly like a unit test. NcCs ||_dS)zM @param description: A string to be displayed L{TestResult}. N description)r]rr.r.r/r$*szTestHolder.__init__cCs ||Srkr`rzr.r.r/__call__1szTestHolder.__call__cCs|jSrkrr]r.r.r/r|5sz TestHolder.idcCsdSr0r.rr.r.r/countTestCases9szTestHolder.countTestCasescCs"||||||dS)z This test is just a placeholder. Run the test successfully. @param result: The C{TestResult} to store the results in. @type result: L{twisted.trial.itrial.IReporter}. N) startTest addSuccessstopTestrzr.r.r/r`=s  zTestHolder.runcCs|jSrkrrr.r.r/shortDescriptionIszTestHolder.shortDescription) rFrarbrcZfailureExceptionr$rr|rr`rr.r.r.r/r!s cs0eZdZdZfddZddZddZZS)ra Used to insert arbitrary errors into a test suite run. Provides enough methods to look like a C{TestCase}, however, when it is run, it simply adds an error to the C{TestResult}. The most common use-case is for when a module fails to import. cs tt||t||_dS)a+ @param description: A string used by C{TestResult}s to identify this error. Generally, this is the name of a module that failed to import. @param error: The error to be added to the result. Can be an `exc_info` tuple or a L{twisted.python.failure.Failure}. N)rdrr$rZexcInfoOrFailureToExcInforg)r]rrgrhr.r/r$VszErrorHolder.__init__cCsd|j|jdfS)Nz%)rrgrr.r.r/__repr__bszErrorHolder.__repr__cCs&|||||j||dS)z Run the test, reporting the error. @param result: The C{TestResult} to store the results in. @type result: L{twisted.trial.itrial.IReporter}. N)rrfrgrrzr.r.r/r`gs zErrorHolder.run)rFrarbrcr$rr`rjr.r.rhr/rNs c@seZdZdZdZdZddZddZdd Zd d Z d d Z e Z ddZ e Z ddZddZddZd$ddZddZd%ddZd&ddZeZd'd d!Zd"d#ZdS)(r a4 I find tests inside function, modules, files -- whatever -- then return them wrapped inside a Test (either a L{TestSuite} or a L{TestCase}). @ivar methodPrefix: A string prefix. C{TestLoader} will assume that all the methods in a class that begin with C{methodPrefix} are test cases. @ivar modulePrefix: A string prefix. Every module in a package that begins with C{modulePrefix} is considered a module full of tests. @ivar forceGarbageCollection: A flag applied to each C{TestCase} loaded. See L{unittest.TestCase} for more information. @ivar sorter: A key function used to sort C{TestCase}s, test classes, modules and packages. @ivar suiteFactory: A callable which is passed a list of tests (which themselves may be suites of tests). Must return a test suite. r_Ztest_cCst|_t|_g|_dSrk)r suiteFactoryrsorterZ _importErrorsrr.r.r/r$szTestLoader.__init__cCst||jdS)zt Sort the given things using L{sorter}. @param xs: A list of test cases, class or modules. )key)sortedr)r]Zxsr.r.r/sortszTestLoader.sortcCs4g}t|D]\}}t|r||q||S)z-Given a module, return all Trial test classes)inspectZ getmembersrrmr)r]r-classesrvalr.r.r/findTestClassess  zTestLoader.findTestClassescCstj|rt|St|S)aB Return a Python object given a string describing it. @param name: a string which may be either a filename or a fully-qualified Python name. @return: If C{name} is a filename, return the module. If C{name} is a fully-qualified Python name, return the object it refers to. )r(r)r8r rr?)r]rr.r.r/ findByNames zTestLoader.findByNamecCst|tjstd|ft|dr,|St|dr>|S|}||D]}| | |qPt|dst|S|}|j D]}| | |q|||gS)a Return a test suite with all the tests from a module. Included are TestCase subclasses and doctests listed in the module's __doctests__ module. If that's not good for you, put a function named either C{testSuite} or C{test_suite} in your module that returns a TestSuite, and I'll use the results of that instead. If C{testSuite} and C{test_suite} are both present, then I'll use C{testSuite}. z%r is not a module testSuite test_suite __doctests__) r%r&r'rhasattrrrrraddTest loadClassr loadDoctests)r]r-rpZ testClassZdocSuiteZdocTestr.r.r/ loadModules     zTestLoader.loadModulecshtts$ttjs$tdfts:tdf}fdd|D} |S)zq Given a class which contains test cases, return a sorted list of C{TestCase} instances. %r is not a class%r is not a test casecsg|]}j|qSr. _makeCase methodPrefix.0rklassr]r.r/ sz(TestLoader.loadClass..) r%typer& ClassTyperrr>getTestCaseNamesrrr]rnamesrnr.rr/rs zTestLoader.loadClasscCst||jS)z Given a class that contains C{TestCase}s, return a list of names of methods that probably contain tests. )rZprefixedMethodNamesr)r]rr.r.r/rszTestLoader.getTestCaseNamescCs,t|tjstd|f||jt|S)z~ Given a method of a C{TestCase} that represents a test, return a C{TestCase} instance for that test. z%r not a method)r%r& MethodTyperrrUrZr]rXr.r.r/ loadMethods zTestLoader.loadMethodcCs||Srkr.)r]rZ methodNamer.r.r/rszTestLoader._makeCaseFc Cst|std|ft|j}|r0|}n|}g}|D]&}|jdd |j r@| |q@| }| |D]B}z |} Wnt|jt} Yn X|| } || qz|S)a Load tests from a module object representing a package, and return a TestSuite containing those tests. Tests are only loaded from modules whose name begins with 'test_' (or whatever C{modulePrefix} is set to). @param package: a types.ModuleType object (or reasonable facsimile obtained by importing) which may contain tests. @param recurse: A boolean. If True, inspect modules within packages within the given package (and so on), otherwise, only inspect modules in the package itself. @raise: TypeError if 'package' is not a package. @return: a TestSuite created with my suiteFactory, containing all the tests. z%r is not a package.rH)r rrZ getModulerFZ walkModulesZ iterModulesrrK startswith modulePrefixrmrrloadrrFailurerr) r]packagerecurseZpkgobjZ discoveryZ discoveredZdiscorpZmodinfor-Z thingToAddr.r.r/ loadPackages&      zTestLoader.loadPackagecCsxt|tr4zt|}Wnt|tYSXt|sLt ddSi}t j dkrjdd}||d<t j|f|S)z Return a suite of tests for all the doctests defined in C{module}. @param module: A module object or a module name. z&trial only supports doctesting modulesN)cSs t|d|j|_|j|_dS)z Save C{test.globs} and replace it with a copy so that if necessary, the original will be available for the next test run. _savedGlobalsN)rCZglobsrcopy)r_r.r.r/ saveGlobals,sz,TestLoader.loadDoctests..saveGlobalsZsetUp)r%strrr?rrrrZismodulewarningswarnrE version_infodoctestZ DocTestSuite)r]r-Z extraArgsrr.r.r/rs    zTestLoader.loadDoctestsNcCs|t|tjr*t|r |||S||St|tjr@||St|trT||St|tj rj| |St d|fdS)a Given a Python object, return whatever tests that are in it. Whatever 'in' might mean. @param thing: A Python object. A module, method, class or package. @param recurse: Whether or not to look in subpackages of packages. Defaults to False. @param parent: For compatibility with the Python 3 loader, does nothing. @param qualname: For compatibility with the Python 3 loader, does nothing. @return: A C{TestCase} or C{TestSuite}. z#No loader for %r. Unrecognized typeN) r%r&r'r rrrrrrrr)r]r}rparentqualNamer.r.r/ loadAnything7s         zTestLoader.loadAnythingcCs6z||}Wnt|tYSX|||S)a` Given a string representing a Python object, return whatever tests are in that object. If C{name} is somehow inaccessible (e.g. the module can't be imported, there is no Python object with that name etc) then return an L{ErrorHolder}. @param name: The fully-qualified name of a Python object. )rrrrr)r]rrr}r.r.r/ loadByNameSs zTestLoader.loadByNamec svg}g}|D]:}z||Wq |t|tYq Xq fdd|D}|||S)a  Construct a TestSuite containing all the tests found in 'names', where names is a list of fully qualified python names and/or filenames. The suite returned will have no duplicate tests, even if the same object is named twice. csg|]}|qSr.)r)rr}rr]r.r/rtsz*TestLoader.loadByNames..)rmrrrr _uniqueTestsextendr)r]rrthingserrorsrZsuitesr.rr/ loadByNamesfs  zTestLoader.loadByNamesccsNt}|D]>}t|tjr&||jf}n|f}||kr |dV||q dS)  Gather unique suite objects from loaded things. This will guarantee uniqueness of inherited methods on TestCases which would otherwise hash to same value and collapse to one test unexpectedly if using simpler means: e.g. set(). rN)setr%r&rrUadd)r]rseenr}r.r.r/rzs   zTestLoader._uniqueTests)F)FNN)F)F)rFrarbrcrrr$rrrrloadTestsFromModulerloadTestsFromTestCaserrrrrrrZloadTestsFromNamerrr.r.r.r/r ts*   *   c@sZeZdZdZdddZdddZddd Zdd d Zdd d ZddZ ddZ ddZ dS) Py3TestLoaderz A test loader finds tests from the functions, modules, and files that is asked to and loads them into a L{TestSuite} or L{TestCase}. See L{TestLoader} for further details. FcCs\ddlm}t|}z|||}|j||dWStk rVtd|YnXdS)a1 Load a file, and then the tests in that file. @param fileName: The file name to load. @param recurse: A boolean. If True, inspect modules within packages within the given package (and so on), otherwise, only inspect modules in the package itself. r)SourceFileLoaderrz{} is not a Python file.N) importlib.machineryrrr@ load_modulerOSErrorr>format)r]ZfileNamerrrr-r.r.r/loadFiles  zPy3TestLoader.loadFilec Cs8tj|krFt|}z t|WqJtk rB|j||dYSXn|}d}}}t|D]N\}}zt|}|}WqWq^tk r|dkrt d |Yq^Xq^|dkrt |}| dt d |jdd}z|D]} |t|| }}qWn$tk r$td |YnX|j||||dS) a Find and load tests, given C{name}. This partially duplicates the logic in C{unittest.loader.TestLoader}. @param name: The qualified name of the thing to load. @param recurse: A boolean. If True, inspect modules within packages within the given package (and so on), otherwise, only inspect modules in the package itself. rNzThe module {} does not exist.rrz{} does not exist.)rrr)r(seprr@ __import__ ImportErrorr_qualNameWalkerZ namedModuleZModuleNotFoundrr?rKlenrFrCrAr) r]_namerrrrZ remainingZ searchNameZ remainingNamepartr.r.r/rs6       " zPy3TestLoader.findByNameNcCst|tjr,t|r"|j||dS||St|trLt|tj rL| |St|tj rt|trt|tj r|d}||}t ||j j|kst|St|tr|Std|fdS)a Load absolutely anything (as long as that anything is a module, package, class, or method (with associated parent class and qualname). @param obj: The object to load. @param recurse: A boolean. If True, inspect modules within packages within the given package (and so on), otherwise, only inspect modules in the package itself. @param parent: If C{obj} is a method, this is the parent class of the method. C{qualName} is also required. @param qualName: If C{obj} is a method, this a list containing is the qualified name of the method. C{parent} is also required. rrHz$don't know how to make test from: %sN)r%r&r'r rrrr~rrxr FunctionTyperCZ_testMethodName__func__AssertionErrorrr)r]rrrrrinstr.r.r/rs$      zPy3TestLoader.loadAnythingc Cs@z||j||dgWS|t|tgYSXdS)a* Load some tests by name. @param name: The qualified name for the test to load. @param recurse: A boolean. If True, inspect modules within packages within the given package (and so on), otherwise, only inspect modules in the package itself. rN)rrrrr)r]rrr.r.r/rs zPy3TestLoader.loadByNamec Csfg}g}|D]>}z||j||dWq |t|tYq Xq |||||S)a+ Load some tests by a list of names. @param names: A L{list} of qualified names. @param recurse: A boolean. If True, inspect modules within packages within the given package (and so on), otherwise, only inspect modules in the package itself. r)rmrrrrrrr)r]rrrrrr.r.r/r)s  zPy3TestLoader.loadByNamescs\ttstdfts.tdf}fdd|D}|S)z Given a class which contains test cases, return a list of L{TestCase}s. @param klass: The class to load tests from. rrcsg|]}j|qSr.rrrr.r/rHsz+Py3TestLoader.loadClass..)r%rrrr>rrrrr.rr/r=s  zPy3TestLoader.loadClasscCs tddS)NzCan't happen on Py3)NotImplementedErrorrr.r.r/rMszPy3TestLoader.loadMethodccsDt}|D]4}|j}|D]$}t||kr|V|t|qq dS)rN)rr[rr)r]rrZ testthingZ testthingsr}r.r.r/rQs zPy3TestLoader._uniqueTests)F)F)FNN)F)F) rFrarbrcrrrrrrrrr.r.r.r/rs  D -  rccsP|gfV|d}tdt|D](}d|d| || dfVq"dS)a3 Given a Python qualified name, this function yields a 2-tuple of the most specific qualified name first, followed by the next-most-specific qualified name, and so on, paired with the remainder of the qualified name. @param qualName: A Python qualified name. @type qualName: L{str} rrN)rKrangerr9)rZ qualPartsindexr.r.r/rbs  rc @seZdZdZdZdZddZddZeZ dd Z d d e j d d d d d d d d f ddZ ddZddZddZdddZddZd S)r z= A specialised runner that the trial front end uses. debugzdry-runcCs<|t}t|j}t|\}|_t |j |Srk) _tearDownLogFiler(getcwdrZFilePathworkingDirectoryrZ_unusedTestDirectory _testDirLockchdirr))r]Z currentDirbaseZtestdirr.r.r/ _setUpTestdirs   zTrialRunner._setUpTestdircCst||jdSrk)r(rrZunlock)r]oldDirr.r.r/_tearDownTestdirs zTrialRunner._tearDownTestdircCs8||j|j|j|j}|jr&t|}|jr4t|}|Srk) reporterFactorystreamtbformatrterrors_log _exitFirstruncleanWarningsr)r]Zreporterr.r.r/ _makeResults zTrialRunner._makeResultNztest.logFdefaultc Csp||_||_||_||_||_||_||_d|_| p6d|_d|_ d|_ | |_ | |_ | |_ |rlt|jd|_dS)NZ _trial_tempz profile.data)rlogfilemoderrrrZ_resultr_logFileObserver_logFileObject_forceGarbageCollectiondebuggerrrZprofiledr`) r]rrrrZprofileZtracebackFormatZrealTimeErrorsrrrorZ exitFirstr.r.r/r$s  zTrialRunner.__init__cCs<|jdk rt|jjd|_|jdk r8|jd|_dSrk)rrZremoveObserveremitrcloserr.r.r/rs    zTrialRunner._tearDownLogFilecCsL||jdkrtj}n t|jd}||_t||_t |jj ddS)N-ar) rrrEstdoutrQrrZFileLogObserverrZstartLoggingWithObserverr)r]ZlogFiler.r.r/ _setUpLogFiles   zTrialRunner._setUpLogFilecCst|t}|||jS)zC Run the test or suite and return a result object. )rrlr_runWithoutDecorationr)r]r_r.r.r/r`s zTrialRunner.runc s4t|g|t}jjkrVtD]"}|||q0n\jj krtfdd}nfdd} }z |W5  |Xt}tdd}|dkr(tjdtjtddjd j||d nS) zR Private helper that runs the given test but doesn't decorate it. csjjSrk)rZruncallr`r.r^r]rpr.r/rurvz3TrialRunner._runWithoutDecoration..cs Srkrr.)r^rpr.r/rurvdoneNzR%s should implement done() but doesn't. Falling back to printErrors() and friends.)category stacklevelzRan %d tests in %.3fs )rr timerDRY_RUNr!rrrDEBUGrrrr rCrrrr{riDeprecationWarningZ printErrorsZwritelnZ separatortestsRunwriteZ printSummaryr ) r]r_roZ startTimeZsingler`rZendTimer r.r r/r sH               z!TrialRunner._runWithoutDecorationcCsXd}|d7}|jd|f|dkr2||}n ||}|jdkrHqT|sqTq|S)z8 Repeatedly run C{test} until it fails. rrz Test Pass %d )rrr`r rZ wasSuccessful)r]r_countr^r.r.r/runUntilFailures   zTrialRunner.runUntilFailure)F)rFrarbrcrrrrrrrrErr$rr r`r rr.r.r.r/r ~s0     /)N)>rcZ __future__rr__all__rrr(rErr&rZtwisted.pythonrrrrrZtwisted.python.compatrrrwrZ twisted.trialrrZtwisted.trial.itrialrZtwisted.trial.reporterrrZtwisted.trial._asyncrunnerr r!Ztwisted.trial._synctestr"Ztwisted.trial.unittestrZzope.interfacer#rrr rrr rBrJrZrrrr rrrVrrr rrr r.r.r.r/sz      #  & ,&U