o br@szddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZmZmZmZdd lmZed Zd d ZGdddejZGdddejZGdddejZ GdddejZ!GdddejZ"Gdddej#Z$Gdddej#Z%Gdddej#Z&Gdddej#Z'Gd d!d!ej#Z(Gd"d#d#ej#Z)dS)$N)StringIO)util)FilePath) UsageError)trial)unittest)DistTrialRunner)DestructiveTestSuite TestLoader TestSuite TrialRunner) testNamesrcCs tt|S)z1 For finding files in twisted/trial/test )rsibpath__file__)filenamer@/usr/lib/python3/dist-packages/twisted/trial/test/test_script.pyrs rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)ForceGarbageCollectionTestsz* Tests for the --force-gc option. cCs>t|_g|_|td|jt|j }t ||g|_ dS)Ncollect) rOptionsconfiglogpatchgcrpyunitFunctionTestCase simpleTestr test)selfrrrrsetUp)s  z!ForceGarbageCollectionTests.setUpcC|jddS)zD A simple test method that records that it was run. rNrappendrrrrr0z&ForceGarbageCollectionTests.simpleTestcCr )zI A replacement for gc.collect that logs calls to itself. rNr!r#rrrr6r$z#ForceGarbageCollectionTests.collectcCst|j}t|_|S)zN Return a L{TrialRunner} object that is safe to use in tests. )r _makeRunnerrrstreamrrunnerrrr makeRunner<s z&ForceGarbageCollectionTests.makeRunnercCs>d|jd<|j|}||j||jgddS)z Passing the --force-gc option to the trial script forces the garbage collector to run before and after each test. Tzforce-gc)rrrrrrNr postOptionsr)runr assertEqualrr'rrr test_forceGcDs    z(ForceGarbageCollectionTests.test_forceGccCs4|j|}||j||jddgdS)z> By default, no garbage collection is forced. rNr*r'rrrtest_unforceGcQs  z*ForceGarbageCollectionTests.test_unforceGcN) __name__ __module__ __qualname____doc__rrrr)r.r/rrrrr$s rc@s(eZdZdZddZddZddZdS) SuiteUsedTestsz? Check the category of tests suite used by the loader. cCt|_dS)z6 Create a trial configuration object. Nrrrr#rrrr`zSuiteUsedTests.setUpcCst|j}||jtdS)zK By default, the loader should use L{DestructiveTestSuite} N)r _getLoaderrr- suiteFactoryr rloaderrrrtest_defaultSuitefs z SuiteUsedTests.test_defaultSuitecCs(d|jd<t|j}||jtdS)zw The C{until-failure} configuration uses the L{TestSuite} to keep instances alive across runs. Tz until-failureN)rrr8r-r9r r:rrrtest_untilFailureSuitems  z%SuiteUsedTests.test_untilFailureSuiteN)r0r1r2r3rr<r=rrrrr4[s  r4c@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5S)6TestModuleTestscCt|_dSNr6r#rrrrxzTestModuleTests.setUpcCs d|_dSr@)rr#rrrtearDown{ zTestModuleTests.tearDowncCs|t||gdS)zq Check that the testNames helper method accurately collects the names of tests in suite. N)r-r idr#rrrtest_testNames~szTestModuleTests.test_testNamescCsBt}t|}ttt|j|}|||||dSr@)r r r map loadByNamesortr-)rtest1namesr;names1names2rrrassertSuitesEquals z!TestModuleTests.assertSuitesEqualcCs|dt|jddS)Nrtests)r-lenrr#rrrtest_baseStateszTestModuleTests.test_baseStatecC*|jtd|t|jdgdS)z Check that --testmodule loads a suite which contains the tests referred to in test-case-name inside its parameter. moduletest.pytwisted.trial.test.test_logNropt_testmodulerrMr _getSuiter#rrrtest_testmoduleOnModulez'TestModuleTests.test_testmoduleOnModulecCs:|jtd|jtd|t|jdgdS)z When the same module is specified with two --testmodule flags, it should only appear once in the suite. rRrSNrTr#rrrtest_testmoduleTwices z$TestModuleTests.test_testmoduleTwicecCs:|jtd|jtd|t|jdgdS)z If --testmodule is specified twice, once for module A and once for a module which refers to module A, then make sure module A is only added once. rRz test_log.pyrSNrTr#rrr test_testmoduleOnSourceAndTargets z0TestModuleTests.test_testmoduleOnSourceAndTargetcCrQ)z When given a module that refers to *itself* in the test-case-name variable, check that --testmodule only adds the tests once. z moduleself.pyztwisted.trial.test.moduleselfNrTr#rrrtest_testmoduleOnSelfModulerXz+TestModuleTests.test_testmoduleOnSelfModulecCs,|jtd|t|jddgdS)zm Check that --testmodule loads tests referred to in test-case-name buffer variables. scripttest.pyrStwisted.trial.test.test_runnerNrTr#rrrtest_testmoduleOnScripts  z'TestModuleTests.test_testmoduleOnScriptcCht}tj|}t_d}z#|j||dt|jd|d|d|W|t_dS|t_w)zz Check that --testmodule displays a meaningful error message when passed a non-existent filename. ztest_thisbetternoteverexist.pyrrNFile  doesn't exist NrsysstderrrrUr-rOgetvalue)rbuffyrdrrrr test_testmoduleOnNonexistentFile z0TestModuleTests.test_testmoduleOnNonexistentFilecCs*|jtd|dt|jddS)z Check that --testmodule adds no tests to the suite for modules which lack test-case-name buffer variables. novars.pyrrNN)rrUrr-rOr#rrrtest_testmoduleOnEmptyVarssz*TestModuleTests.test_testmoduleOnEmptyVarscCr_)z Check that --testmodule does *not* support module names as arguments and that it displays a meaningful error message. ztwisted.trial.test.test_scriptrrNr`raNrb)rrfrd moduleNamerrrtest_testmoduleOnModuleNamerhz+TestModuleTests.test_testmoduleOnModuleNamecC"d}t|}|ddi|dS)Nz5-*- test-case-name: twisted.trial.test.test_tests -*-test-case-nametwisted.trial.test.test_testsr_parseLocalVariablesr-r declaration localVarsrrrtest_parseLocalVariable z'TestModuleTests.test_parseLocalVariablecCrm)Nz6-*- test-case-name: twisted.trial.test.test_tests; -*-rnrorprrrrrtest_trailingSemicolonrvz&TestModuleTests.test_trailingSemicoloncCs$d}t|}|ddd|dS)Nz?-*- test-case-name: twisted.trial.test.test_tests; foo: bar -*-robar)rnfoorprrrrrtest_parseLocalVariabless   z(TestModuleTests.test_parseLocalVariablescCrm)Nz:## -*- test-case-name: twisted.trial.test.test_tests -*- #rnrorprrrrrtest_surroundingGuffrvz$TestModuleTests.test_surroundingGuffcCs|ttjddSNry assertRaises ValueErrorrrqr#rrrtest_invalidLinesz TestModuleTests.test_invalidLinecCs4|ttjd|ttjd|ttjddS)Nz -*- foo -*-z-*- foo: bar; qux -*-z-*- foo: bar: baz; qux: qax -*-r}r#rrrtest_invalidDeclarationsz'TestModuleTests.test_invalidDeclarationcC"ttd}|ddi|dS)NrRrnrSrloadLocalVariablesrr-rrtrrrtest_variablesFromFilez&TestModuleTests.test_variablesFromFilecCsttd}|i|dS)Nrirrrrrtest_noVariablesInFilesz&TestModuleTests.test_noVariablesInFilecCr)Nr\rnz:twisted.trial.test.test_log,twisted.trial.test.test_runnerrrrrrtest_variablesFromScriptsz(TestModuleTests.test_variablesFromScriptcCs ttd}||dgdS)NrRrS)rgetTestModulesrr-rmodulesrrrtest_getTestModules#sz#TestModuleTests.test_getTestModulescCs"ttd}|t|ddS)Nrir)rrrr-rOrrrrtest_getTestModules_noVars'rz*TestModuleTests.test_getTestModules_noVarscCs&ttd}|t|ddhdS)Nr\rSr])rrrr-setrrrrtest_getTestModules_multiple+s z,TestModuleTests.test_getTestModules_multiplecCsVdD]}|t||dqdtdtdfD]}|t||dqdS)N)ztest_script.pyz!twisted/trial/test/test_script.pyz should be a test filez twisted/trial/test/moduletest.pyr\z test_foo.batz should *not* be a test file) assertTruer isTestFiler assertFalse)rrrrrtest_looksLikeTestModule2sz(TestModuleTests.test_looksLikeTestModuleN)r0r1r2rrBrErMrPrWrYrZr[r^rgrjrlrurwrzr{rrrrrrrrrrrrrr>ws6         r>c@HeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)WithoutModuleTestsz* Test the C{without-module} flag. cCst|_ttj|_dS)zl Create a L{trial.Options} object to be used in the tests, and save C{sys.modules}. N)rrrdictrcr savedModulesr#rrrrHs zWithoutModuleTests.setUpcCs8dD]}||jvr|j|tj|<qtj|dqdS)z) Restore C{sys.modules}. )imaplibsmtplibN)rrcrpop)rmodulerrrrBPs  zWithoutModuleTests.tearDowncC ddl}|S)zE Try to import the C{smtplib} module, and return it. rN)r)rrrrr _checkSMTPZzWithoutModuleTests._checkSMTPcCr)zE Try to import the C{imaplib} module, and return it. rN)r)rrrrr _checkIMAPbrzWithoutModuleTests._checkIMAPcCs<|jddg|t|jtjd=||tj dS)zT Check that after disabling a module, it can't be imported anymore. --without-modulerN) r parseOptionsr~ ImportErrorrrcrassertIsInstancetypes ModuleTyper#rrrtest_disableOneModulejsz(WithoutModuleTests.test_disableOneModulecCsd|jddg|t|j|t|jtjd=tjd=||t j ||t j dS)zE Check that several modules can be disabled at once. rzsmtplib,imaplibrrN) rrr~rrrrcrrrrr#rrrtest_disableMultipleModulesusz.WithoutModuleTests.test_disableMultipleModulescCs@||tj|tdtj|jj ddg| t |jdS)zP Disabling an already imported module should produce a warning. z4Module 'smtplib' already imported, disabling anyway.rrN) rrrr assertWarnsRuntimeWarningrrrrr~rr#rrr!test_disableAlreadyImportedModulesz4WithoutModuleTests.test_disableAlreadyImportedModuleN) r0r1r2r3rrBrrrrrrrrrrCs  rc@sDeZdZdZeedddurdZddZddZd d Z d d Z dS) CoverageTestsz+ Tests for the I{coverage} option. gettraceNz;Cannot test trace hook installation without inspection API.cCs|tjtdS)zf Arrange for the current trace hook to be restored when the test is complete. N) addCleanuprcsettracerr#rrrrszCoverageTests.setUpcCs,t}|dg|t|jjdS)z} L{trial.Options} handles C{"--coverage"} by installing a trace hook to record coverage information. z --coverageN)rrrr-rcrtracer globaltraceroptionsrrrtest_tracerInstalleds z"CoverageTests.test_tracerInstalledcCs.t}||td|ddgdS)z L{trial.Options.coverdir} returns a L{FilePath} based on the default for the I{temp-directory} option if that option is not specified. .ztemp-directorycoverageN)rrr-coverdirr descendantrrrrtest_coverdirDefaults z"CoverageTests.test_coverdirDefaultcCs<|}t}|d|g||t|ddS)z If a value is specified for the I{temp-directory} option, L{trial.Options.coverdir} returns a child of that path. z--temp-directoryrN)mktemprrrr-rrchild)rpathrrrrtest_coverdirOverriddensz%CoverageTests.test_coverdirOverridden) r0r1r2r3getattrrcskiprrrrrrrrrs rc@r) OptionsTestsz% Tests for L{trial.Options}. cCr5)zE Build an L{Options} object to be used in the tests. Nrrrr#rrrrr7zOptionsTests.setUpcCsj|tjttr|tj|jgd|j }| d|| d| ddg|dS)z C{_getWorkerArguments} discards options like C{random} as they only matter in the manager, and forwards options like C{recursionlimit} or C{disablegc}. )--recursionlimit2000--random4 --disablegcrrrN) rrcsetrecursionlimitgetrecursionlimitr isenabledenablerr_getWorkerArgumentsassertInremover-)rargsrrrtest_getWorkerArgumentss    z$OptionsTests.test_getWorkerArgumentscC*|t|jjgd}|dt|dS)z C{parseOptions} raises a C{UsageError} when C{--debug} is passed along C{--jobs} as it's not supported yet. @see: U{http://twistedmatrix.com/trac/ticket/5825} )--jobsr--debugz+You can't specify --debug when using --jobsNr~rrrr-strrerrorrrrtest_jobsConflictWithDebugz'OptionsTests.test_jobsConflictWithDebugcCr)z C{parseOptions} raises a C{UsageError} when C{--profile} is passed along C{--jobs} as it's not supported yet. @see: U{http://twistedmatrix.com/trac/ticket/5827} )rrz --profilez-You can't specify --profile when using --jobsNrrrrrtest_jobsConflictWithProfilerz)OptionsTests.test_jobsConflictWithProfilecCr)z C{parseOptions} raises a C{UsageError} when C{--debug-stacktraces} is passed along C{--jobs} as it's not supported yet. @see: U{http://twistedmatrix.com/trac/ticket/5826} )rrz--debug-stacktracesz7You can't specify --debug-stacktraces when using --jobsNrrrrr%test_jobsConflictWithDebugStackTracessz2OptionsTests.test_jobsConflictWithDebugStackTracescCr)z C{parseOptions} raises a C{UsageError} when C{--exitfirst} is passed along C{--jobs} as it's not supported yet. @see: U{http://twistedmatrix.com/trac/ticket/6436} )rr --exitfirstz/You can't specify --exitfirst when using --jobsNrrrrrtest_jobsConflictWithExitFirstrz+OptionsTests.test_jobsConflictWithExitFirstcCr)zr C{parseOptions} raises a C{UsageError} when C{--order} is passed along with C{--random}. )--order alphabeticalr1234z-You can't specify --random when using --orderNrrrrrtest_orderConflictWithRandoms z)OptionsTests.test_orderConflictWithRandomN) r0r1r2r3rrrrrrrrrrrrs   rc@s8eZdZdZddZddZddZdd Zd d Zd S) MakeRunnerTestsz. Tests for the L{_makeRunner} helper. cCr?r@rr#rrrr#rAzMakeRunnerTests.setUpcCsJ|jgdt|j}||t|d|j|dg|jdS)z L{_makeRunner} returns a L{DistTrialRunner} instance when the C{--jobs} option is passed, and passes the C{workerNumber} and C{workerArguments} parameters to it. )rr --force-gcrN) rrrr%rrr- _workerNumber_workerArgumentsr'rrr test_jobs&s   zMakeRunnerTests.test_jobscCs<|jgdt|j}||t|tj|jdS)z L{_makeRunner} returns a L{TrialRunner} instance in C{DRY_RUN} mode when the C{--dry-run} option is passed, even if C{--jobs} is set. )rrz --dry-runN) rrrr%rr r-DRY_RUNmoder'rrrtest_dryRunWithJobs2s  z#MakeRunnerTests.test_dryRunWithJobscsPtjjfdd}|tjd|t}|gd|tjtj|dS)Ncs|dkr tj||S)N doNotFind)rreflectModuleNotFound)fqnnamedAnyrrnamedAnyExceptdoNotFind?s zFMakeRunnerTests.test_DebuggerNotFound..namedAnyExceptdoNotFindr)rz --debuggerr) rrrrrrr~_DebuggerNotFoundr%)rrrrrrtest_DebuggerNotFound<s  z%MakeRunnerTests.test_DebuggerNotFoundcCs*|jdgt|j}||jdS)z Passing C{--exitfirst} wraps the reporter with a L{reporter._ExitWrapper} that stops on any non-success. rN)rrrr%r _exitFirstr'rrrtest_exitfirstKs zMakeRunnerTests.test_exitfirstN) r0r1r2r3rrrrrrrrrrs  rc@ eZdZdZddZddZdS)RunTestsz( Tests for the L{run} function. cCs|tjddddS)NrcSsdSr@rr#rrr]sz RunTests.setUp..)rrrr#rrrrZszRunTests.setUpc Cshdd}|td|ztWnty,}z|dt|WYd}~dSd}~ww|ddS)zY When a debugger is not found, an error message is printed to the user. c_s tdr|)rr)rkwargsrrrr%erCz3RunTests.test_debuggerNotFound.._makeRunnerr%ryNz0Should have exited due to non-existent debugger!)rrr, SystemExitrrfail)rr%errrtest_debuggerNotFound_s zRunTests.test_debuggerNotFoundN)r0r1r2r3rrrrrrrUs rc@r)TestArgumentOrderTestszQ Tests for the order-preserving behavior on provided command-line tests. cCst|_t|_dSr@)rrrr r;r#rrrrws  zTestArgumentOrderTests.setUpcCsRgd}|j|t|j}t|}tt|jj|}t|}| ||dS)zN Multiple tests passed on the command line are not reordered. )roz"twisted.trial.test.test_assertionsz twisted.trial.test.test_deferredN) rrrrVr r rFr;rGr-)rrNsuiterJ expectedSuite expectedNamesrrrtest_preserveArgumentOrder{s  z1TestArgumentOrderTests.test_preserveArgumentOrderN)r0r1r2r3rrrrrrrrs rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS) OrderTestsz' Tests for the --order option. cCr?r@r6r#rrrrrAzOrderTests.setUpcCD|jgdt|j}||jd}|t|gddS)zn --order=alphabetical causes trial to run tests alphabetically within each test case. )rr%twisted.trial.test.ordertests.FooTestrN)0twisted.trial.test.ordertests.FooTest.test_first1twisted.trial.test.ordertests.FooTest.test_fourth1twisted.trial.test.ordertests.FooTest.test_second0twisted.trial.test.ordertests.FooTest.test_thirdNrrrr8 loadByNamesr-r rr;rrrrtest_alphabeticals zOrderTests.test_alphabeticalcCr)zu --order=alphabetical causes trial to run test classes within a given module alphabetically. )rrtwisted.trial.test.ordertestsrN).twisted.trial.test.ordertests.BarTest.test_bar.twisted.trial.test.ordertests.BazTest.test_bazrrrrNr r rrrtest_alphabeticalModule z"OrderTests.test_alphabeticalModulecCsV|jgdt|j}||jd}t|}|j|dd||t|dS)z --order=alphabetical causes trial to run test modules within a given package alphabetically, with tests within each module alphabetized. )rrtwisted.trial.testrNzFailed to load any tests!msgN) rrrr8r r rr-sortedrr;rrJrrrtest_alphabeticalPackages  z#OrderTests.test_alphabeticalPackagecCr)z --order=toptobottom causes trial to run test methods within a given test case from top to bottom as they are defined in the body of the class. )r toptobottomrrN)rrrrNr r rrrtest_toptobottoms zOrderTests.test_toptobottomcCr)z --order=toptobottom causes trial to run test classes within a given module from top to bottom as they are defined in the module's source. rrr rN)rrrrrrNr r rrrtest_toptobottomModulerz!OrderTests.test_toptobottomModulecCsP|jgdt|j}||jd}t|}||t|ddddS)z --order=toptobottom causes trial to run test modules within a given package alphabetically, with tests within each module run top to bottom. )rrrrNcSs|dddS)Nrr)split)namerrrrsz4OrderTests.test_toptobottomPackage..)keyN)rrrr8r r r-rrrrrtest_toptobottomPackages z"OrderTests.test_toptobottomPackagecCs|}t|d}||dd|dtdd|j }t j d|| t j j |ddlm}| t jjd| t jj|j|d |jgd t|j}||}|t|gd d S) z{ --order=toptobottom detects the source line of methods from modules whose source file is missing. twisted_toptobottom_tempz __init__.pyztest_missing.pya from twisted.trial.unittest import TestCase class TestMissing(TestCase): def test_second(self): pass def test_third(self): pass def test_fourth(self): pass def test_first(self): pass utf8r) test_missingr)z=twisted_toptobottom_temp.test_missing.TestMissing.test_secondzrrTestCaserrrrrr1rrrrs:        7MO.^7;