Yf8) @ s d Z 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)zUnittest main program 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 2/opt/alt/python35/lib64/python3.5/unittest/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
<listcomp>. s z"_convert_names.<locals>.<listcomp>r )namesr r r _convert_names- s r c @ s e Z d 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) TestProgramzA command-line program that runs a set of tests; this is primarily
for making test modules conveniently executable.
Nr __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_parserZ 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)rN r3 ZloadTestsFromModuler% testZloadTestsFromNames)r8 r r r rQ 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)!NZadd_helpFz-vz --verbosedestr- actionZ store_constZconstr; 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 rS s4 z TestProgram._getParentArgParserc C sJ t j d | g } | j | _ | j | _ | j d d d d d | S)NparentsrM nargs*rX z?a list of any number of test modules, classes and test methods.)rZ r[ r5 rB r? rE r\ )r8 parentr] r r r rT 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)Nr^ z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryrV startrX z*Directory to start discovery ('.' default)z-pz --patternpatternz+Pattern to match tests ('test*.py' default)z-tz--top-level-directorytopz<Top level directory of project (defaults to start directory)r_ ?r! )rb rc rd )rZ r[ r5 rB Zepilogr\ ZSUPPRESS)r8 ra r] argr r r rU s
z"TestProgram._getDiscoveryArgParserc C s d | _ d | _ d | _ | d k rS | j d k r@ | j | j j | | | d k rh | j n | } | j | j | j | j | _ d S)Nr ztest*.py) rb rc rd r= r> rL r3 rI rR )r8 r) Loaderr r r r rK 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 TypeErrorZrunrR resultr* r( Z
wasSuccessful)r8 r2 r r r r7 s.
zTestProgram.runTests)rO
__module__ __qualname____doc__r% r- r+ r, r. r5 r0 r= r ZdefaultTestLoaderr: rA r? r6 rQ r> rS rT rU rK r7 r r r r r 1 s& # r )rm r( rZ r r r Zsignalsr Z
__unittestrD rF r r objectr mainr r r r <module> s