j f & @ 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)installHandlerTzExamples:
%(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
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| } n | d d j
d d j
d d S| S)Nz.py \./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirreplace)nameZrel_path r 2/opt/alt/python34/lib64/python3.4/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
- s z"_convert_names..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 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__Tc
C s" t | t rZ t | | _ xB | j d d d D] } t | j | | _ q8 Wn | | _ | d k r{ t j } n | | _ | | _ | | _
| | _ |
| _ | d k r t j
r d | _ n | | _ | | _ | | _ | | _ t j j | d | _ | j | | j d S)Nr r defaultr )
isinstancestr
__import__modulesplitgetattrsysargvexitfailfast
catchbreak verbositybuffer warnoptionswarnings defaultTest
testRunner
testLoaderr r basenameprogName parseArgsrunTests)
selfr$ r0 r( r1 r2 r) r, r* r+ r- r/ partr r r __init__: s*
zTestProgram.__init__c C sJ | r t | n | j d k r/ | j n | j t j d d S)N )print_discovery_parser_initArgParsers _print_helpr' r) )r7 msgr r r usageExit_ s
zTestProgram.usageExitc O sy | j d k rJ t | j j t t i | j d 6 | j j n+ t | j j t t i | j d 6 d S)Nprog) r$ r; _main_parserZ format_help
MAIN_EXAMPLESr4 r<
print_helpMODULE_EXAMPLES)r7 argskwargsr r r r> g s zTestProgram._print_helpc C sH | 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 r:d | _ q: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_discoveryrB
parse_argstestsr testNames__name__r0 r! r" list createTests)r7 r( r r r r5 p s(
(
zTestProgram.parseArgsc C sL | j d k r* | j j | j | _ n | j j | j | j | _ d S)N)rM r2 ZloadTestsFromModuler$ testZloadTestsFromNames)r7 r r r rP s zTestProgram.createTestsc C s4 | j } | j | | _ | j | | _ d S)N)_getParentArgParser_getMainArgParserrB _getDiscoveryArgParserr< )r7 Z
parent_parserr r r r= s zTestProgram._initArgParsersc C s 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 k r | j d d d d d d d d d | _ n | j d k r | j d d d d d d d d d | _ n | j d k r| j d d d d d d d d d | _ n | S) NZadd_helpFz-vz --verbosedestr, actionZ store_constZconstr: helpzVerbose outputz-qz--quietr z Quiet outputz-fz
--failfastr*
store_truezStop 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- )r7 parserr r r rR s. z TestProgram._getParentArgParserc C sJ t j d | g } | j | _ | j | _ | j d d d d d | S)NparentsrL nargs*rW z?a list of any number of test modules, classes and test methods.)rY rZ r4 rA r> rD r[ )r7 parentr\ r r r rS 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-directoryrU startrW z*Directory to start discovery ('.' default)z-pz --patternpatternz+Pattern to match tests ('test*.py' default)z-tz--top-level-directorytopz r5 rP r= rR rS rT rJ r6 r r r r r 0 s& " r )rk r' rY r r r Zsignalsr Z
__unittestrC rE r r objectr mainr r r r s