]8) @ s d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d Z d Z d Z
d d Z d
d Z Gd d
d
e
Z e Z d S) N )loaderrunner)installHandlerTa Examples:
%(prog)s test_module - run tests from test_module
%(prog)s module.TestClass - run tests from module.TestClass
%(prog)s module.Class.test_method - run specified test method
%(prog)s path/to/test_file.py - run tests from test_file.py
aF Examples:
%(prog)s - run default set of tests
%(prog)s MyTestSuite - run suite 'MyTestSuite'
%(prog)s MyTestCase.testSomething - run MyTestCase.testSomething
%(prog)s MyTestCase - run all 'test*' test methods
in MyTestCase
c C s t j j | r | j j d r t j j | r t j j | t j } t j j | sx | j t j r| | S| } | d d j
d d j
d d S| S)Nz.py \./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirreplace)nameZrel_path r )/opt/alt/python35/lib64/python3.5/main.py
_convert_name s '$&r c C s d d | D S)Nc S s g | ] } t | q Sr )r ).0r r r r
. s z"_convert_names..r )namesr r r _convert_names- s r c @ s e Z d Z d Z d Z d Z Z Z Z Z d Z
d d d d e j d d d d d d d d d d Z
d d d
Z d d Z d
d Z d d Z d d Z d d Z d d Z d d Z d d d Z d d Z d S) TestProgramNr __main__T tb_localsFc C s( t | t rZ t | | _ xB | j d d d D] }
t | j |
| _ q8 Wn | | _ | d k rx t j } | | _ | | _ | | _
| | _ |
| _ | | _
| d k r t j r d | _ n | | _ | | _ | | _ | | _ t j j | d | _ | j | | j d S)Nr r defaultr )
isinstancestr
__import__modulesplitgetattrsysargvexitfailfast
catchbreak verbositybufferr warnoptionswarnings defaultTest
testRunner
testLoaderr r basenameprogName parseArgsrunTests)selfr% r1 r) r2 r3 r* r- r+ r, r. r0 r partr r r __init__; s,
zTestProgram.__init__c C sD | r t | | j d k r) | j | j t j d d S)N )print_discovery_parser_initArgParsers _print_helpr( r* )r8 msgr r r usageExita s
zTestProgram.usageExitc O sw | j d k rI t | j j t t d | j i | j j n* t | j j t t d | j i d S)Nprog) r% r< _main_parser format_help
MAIN_EXAMPLESr5 r=
print_helpMODULE_EXAMPLES)r8 argskwargsr r r r? i s zTestProgram._print_helpc C sE | j | j d k r t | d k r\ | d j d k r\ | j | d d d S| j j | d d | | j s | j g d Sn | j j | d d | | j r t | j | _ t
d k r7d | _ nQ | j d k rd | _ n6 t | j t
r%| j f | _ n t | j | _ | j d S)Nr discoverr; r )r> r% lenr
_do_discoveryrC
parse_argstestsr testNames__name__r1 r" r# list createTests)r8 r) r r r r6 r s(
(
zTestProgram.parseArgsc C sL | j d k r* | j j | j | _ n | j j | j | j | _ d S)N)rO r3 ZloadTestsFromModuler% testZloadTestsFromNames)r8 r r r rR s zTestProgram.createTestsc C s4 | j } | j | | _ | j | | _ d S)N)_getParentArgParser_getMainArgParserrC _getDiscoveryArgParserr= )r8 Z
parent_parserr r r r> s zTestProgram._initArgParsersc C s3 t j d d } | j d d d d d d d d
d d | j d
d d d d d d d d d | j d d d d d d d | j d k r | j d d d d d d d d d | _ | j d k r | j d d d d d d d d d | _ | j d k r/| j d d d d d d d d d | _ | S)!Nadd_helpFz-vz --verbosedestr- action store_constconstr; helpzVerbose outputz-qz--quietr z Quiet outputz--localsr
store_truez"Show local variables in tracebacksz-fz
--failfastr+ zStop on first fail or errorz-cz--catchr, z'Catch Ctrl-C and display results so farz-bz--bufferr. z%Buffer stdout and stderr during tests)argparseArgumentParser add_argumentr+ r, r. )r8 parserr r r rT s4 z TestProgram._getParentArgParserc C sJ t j d | g } | j | _ | j | _ | j d d d d d | S)NparentsrN nargs*r\ z?a list of any number of test modules, classes and test methods.)r^ r_ r5 rB r? rF r` )r8 parentra r r r rU s z TestProgram._getMainArgParserc C s t j d | g } d | j | _ d | _ | j d d d d d d | j d
d d d d d
| j d d d d d d x3 d D]+ } | j | d d d t j d t j q W| S)Nrb z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryrX startr\ z*Directory to start discovery ('.' default)z-pz --patternpatternz+Pattern to match tests ('test*.py' default)z-tz--top-level-directorytopz rM r3 rJ rS )r8 r) Loaderr r r r rL s
zTestProgram._do_discoveryc C s) | j r t | j d k r+ t j | _ t | j t r y y= | j d | j d | j d | j d | j
d | j } WnB t k
r | j d | j d | j d | j d | j
} Yn XWq t k
r | j } Yq Xn | j } | j
| j | _ | j r%t j | j j d S)Nr- r+ r. r0 r )r, r r2 r ZTextTestRunnerr" typer- r+ r. r0 r TypeErrorrunrS resultr* r( Z
wasSuccessful)r8 r2 r r r r7 s.
zTestProgram.runTests)rP
__module__ __qualname__r% r- r+ r, r. r5 r0 r= r ZdefaultTestLoaderr: rA r? r6 rR r> rT rU rV rL r7 r r r r r 1 s$ # r )r( r^ r r r Zsignalsr Z
__unittestrE rG r r objectr mainr r r r s