o
0i+ @ sr d dl Z d dlZd dlZddlmZmZ ddlmZ dZdZ dZ
dd Z d
d Z d d
Z
G dd deZeZdS ) 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| r>| dr>t j| r,t j| t }t j|s(|t j r*| S |} t j
| d d dd ddS | S )Nz.py\./) ospathisfilelowerendswithisabsrelpathgetcwd
startswithpardirnormpathreplace)nameZrel_path r 4/opt/alt/python310/lib64/python3.10/unittest/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 C s d| vrd| } | S )N*z*%s*r )patternr r r _convert_select_pattern1 s r c @ s e Zd Z dZdZd Z Z Z Z Z Z
dZ dddde j
ddddddf ddddZd d d
Zd d Zd
d Zd ddZdd Zdd Zdd Zdd Zd ddZdd ZdS ) TestProgramNr __main__TF) tb_localsc C s t |tr t|| _|ddd D ] }
t| j|
| _qn|| _|d u r(tj}|| _|| _ | | _
|| _ |
| _ | | _
| d u rEtjsEd| _n| | _|| _|| _|| _tj|d | _| | | d S )Nr r defaultr )
isinstancestr
__import__modulesplitgetattrsysargvexitfailfast
catchbreak verbositybufferr# warnoptionswarnings defaultTest
testRunner
testLoaderr
r basenameprogName parseArgsrunTests)selfr( r4 r, r5 r6 r- r0 r. r/ r1 r3 r# partr r r __init__A s.
zTestProgram.__init__c C s4 |rt | | jd u r| | td d S )N )print_discovery_parser_initArgParsers _print_helpr+ r- )r; msgr r r usageExitg s
zTestProgram.usageExitc O s\ | j d u r t| j ttd| ji | j d S t| j ttd| ji d S )Nprog) r( r? _main_parserZ format_help
MAIN_EXAMPLESr8 r@
print_helpMODULE_EXAMPLES)r; argskwargsr r r rB o s
zTestProgram._print_helpc C s | | jd u r8t|dkr"|d dkr"| |dd d S | j|dd | | js7| g d S n | j|dd | | jrTt| j| _ t
dkrSd | _n| j d u r]d | _ nt | j t
ri| j f| _ nt| j | _ | d S )Nr discoverr> r" )rA r( lenr
_do_discoveryrF
parse_argstestsr testNames__name__r4 r% r& list createTests)r; r, r r r r9 x s,
zTestProgram.parseArgsc C sx | j r| j | j_ |r!|d u r| jn| }|| j| j| j| _d S | jd u r0| j| j | _d S | j
| j| j | _d S N) testNamePatternsr6 rL startr toptestrQ ZloadTestsFromModuler( ZloadTestsFromNames)r; from_discoveryLoaderr r r r rT s
zTestProgram.createTestsc C s$ | }| || _| || _d S rU )_getParentArgParser_getMainArgParserrF _getDiscoveryArgParserr@ )r; Z
parent_parserr r r rA s zTestProgram._initArgParsersc C s t jdd}|jddddddd |jd
d ddd d
d |jddddd | jd u r7|jdddddd d| _| jd u rI|jdddddd d| _| jd u r[|jdd d dd d d| _| jd u rm|jd d d!td"d# g | _|S )$NF)Zadd_helpz-vz --verboser0 Z store_constr> zVerbose output)destactionZconsthelpz-qz--quietr z Quiet outputz--localsr#
store_truez"Show local variables in tracebacks)r_ r` ra z-fz
--failfastr. zStop on first fail or errorz-cz--catchr/ z'Catch Ctrl-C and display results so farz-bz--bufferr1 z%Buffer stdout and stderr during testsz-krV appendz.Only run tests which match the given substring)r_ r` typera )argparseArgumentParser add_argumentr. r/ r1 rV r )r; parserr r r r\ sL
z TestProgram._getParentArgParserc C s2 t j|gd}| j|_| j|_|jdddd |S )NparentsrP r z?a list of any number of test modules, classes and test methods.)nargsra )re rf r8 rE rB rH rg )r; parentrh r r r r] s z TestProgram._getMainArgParserc C sz t j|gd}d| j |_d|_|jddddd |jd d
d d d |jd
dddd dD ]
}|j|dt jt jd q-|S )Nri z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryrW z*Directory to start discovery ('.' default))r_ ra z-pz --patternr z+Pattern to match tests ('test*.py' default)z-tz--top-level-directoryrX z s ]