3
2a8) @ sh d dl Z d dlZd dlZddlmZmZ ddlmZ dZdZ dZ
dd Z d
d Z G d d
d
e
ZeZdS ) 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 sx t jj| rt| j jdrtt jj| rXt jj| t j }t jj|sP|jt j rT| S |} | d d j
ddj
ddS | S )Nz.py \./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirreplace)nameZrel_path r )/opt/alt/python36/lib64/python3.6/main.py
_convert_name s r c C s dd | D S )Nc S s g | ] }t |qS r )r ).0r r r r
. s z"_convert_names..r )namesr r r _convert_names- s r c @ s e Zd ZdZdZd Z Z Z ZZ dZ
dddde j ddddddf ddddZ
d d d
Zd d Zd
d Zdd Zdd Zdd Zdd Zdd Zd ddZdd ZdS ) TestProgramNr __main__TF) tb_localsc C s t |tr@t|| _x0|jddd D ]}
t| j|
| _q(W n|| _|d krTtj}|| _|| _ | | _
|| _ |
| _ | | _
| d krtj rd| _n| | _|| _|| _|| _tjj|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 s4 |r t | | jd kr | j | j tjd d S )N )print_discovery_parser_initArgParsers _print_helpr( r* )r8 msgr r r usageExita s
zTestProgram.usageExitc O sZ | j d kr6t| jj ttd| ji | jj n t| jj ttd| ji 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 s | j | jd krpt|dkrD|d j dkrD| j|dd d S | jj|dd | | js| jg d S n| jj|dd | | jrt| j| _ t
dkrd | _n6| j d krd | _ n$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 s4 | j d kr | jj| j| _n| jj| j | j| _d S )N)rO r3 ZloadTestsFromModuler% testZloadTestsFromNames)r8 r r r rR s
zTestProgram.createTestsc C s$ | j }| j|| _| j|| _d S )N)_getParentArgParser_getMainArgParserrC _getDiscoveryArgParserr= )r8 Z
parent_parserr r r r> s zTestProgram._initArgParsersc C s t jdd}|jddddddd |jd
d ddd d
d |jddddd | jd krn|jdddddd d| _| jd kr|jdddddd d| _| jd kr|jdd d dd d d| _|S ) NF)add_helpz-vz --verboser- store_constr; zVerbose output)destactionconsthelpz-qz--quietr z Quiet outputz--localsr
store_truez"Show local variables in tracebacks)rY rZ r\ z-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 s2 t j|gd}| j|_| j|_|jdddd |S )N)parentsrN *z?a list of any number of test modules, classes and test methods.)nargsr\ )r^ r_ r5 rB r? rF r` )r8 parentra r r r rU s z TestProgram._getMainArgParserc C s~ t j|gd}d| j |_d|_|jddddd |jd d
d d d |jd
dddd x"dD ]}|j|dt jt jd q\W |S )N)rb z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directorystartz*Directory to start discovery ('.' default))rY r\ 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 | jd kr tj| _t| jtryVy"| j| j| j| j | j
| j d}W n. t k
r| | j| j| j| j | j
d}Y nX W q t k
r | j }Y qX n| j}|j
| j| _| jrtj| jj d S )N)r- r+ r. r0 r )r- r+ r. r0 )r, r r2 r ZTextTestRunnerr" typer- r+ r. r0 r TypeErrorZrunrS resultr* r( Z
wasSuccessful)r8 r2 r r r r7 s.
zTestProgram.runTests)N)N)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 T