a

�DOg�+�@stdZd
dl
Z
d
dlZd
dlZddlmZmZ
ddlmZ
dZ	dZ
dZd	d
�Zdd�Z
d
d�ZGdd�de�ZeZdS)zUnittest main program�N�
)�loader�runner)
�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
Cs�tj
�|�
r||���d
�
r|tj
�|�
rXtj
�|t���}
tj
�|
�
sP|
�tj	�
rT|S|
}tj
�
|�
dd��dd��dd�S|S)Nz.py����
\�
.�
/)�os�path�isfile�lower�endswith�isabs�relpath�getcwd�
startswith�pardir�normpath�replace)�nameZrel_path�r�%/usr/lib64/python3.9/unittest/main.py�
_convert_names




$
rc

Csd
d�|D�
S)Nc
Ssg|]}
t|
�
�qSr)
r)�.0rrrr�
<listcomp>.�z"_convert_names.<locals>.<listcomp>r)
�namesrrr�_convert_names-s
rc

Csd
|v
rd|}|S)N�
*z*%s*r)
�patternrrr�_convert_select_pattern1s


r!c@s�eZ
dZd
ZdZdZdZZZZ	Z
ZdZdddde
jddddddfdd�
dd	�Zdd
d�
Zdd
�Zdd�Zddd�
Zdd�Zdd�Zdd�Zdd�Zd dd�
Zdd�ZdS)!�TestProgramzA command-line program that runs a set of tests; this is primarily
       for making test modules conveniently executable.
    Nr�__main__TF)
�	tb_localsc
Cs�t|
t
�r<t|
�
|_|
�d
�
dd�D]}
t|j|
�|_q&n|
|_|durPtj}||_||_	|	|_
||_|
|_||_
|dur�tjs�d|_n||_||_||_||_tj�|d�
|_|�|�

|��
dS)Nrr�defaultr
)�
isinstance�str�
__import__�module�split�getattr�sys�argv�exit�failfast�
catchbreak�	verbosity�bufferr$�warnoptions�warnings�defaultTest�
testRunner�
testLoaderr
r�basename�progName�	parseArgs�runTests)�selfr)r5r-r6r7r.r1r/r0r2r4r$�partrrr�__init__As,



















zTestProgram.__init__cCs4|
rt|
�

|j
dur|��
|��
t�d
�

dS)N�)�print�_discovery_parser�_initArgParsers�_print_helpr,r.)r<�msgrrr�	usageExitgs






zTestProgram.usageExitc
OsZ|jdur6t
|j���

t
td
|ji
�

|j��
n t
|j���

t
td
|ji
�

dS)N�prog)	r)r@�_main_parser�format_help�
MAIN_EXAMPLESr9rA�
print_help�MODULE_EXAMPLES)r<�args�kwargsrrrrCos





zTestProgram._print_helpcCs�|��
|j
durpt|
�
d
krD|
d
��dkrD|�|
dd��

dS|j�|
d
d�|�
|js�|�g�

dSn|j�|
d
d�|�
|jr�t|j�
|_	t
dkr�d|_
n6|jdur�d|_	n$t|jt
�r�|jf
|_	nt|j�
|_	|��
dS)Nr�discoverr?r#)rBr)�lenr
�
_do_discoveryrG�
parse_args�testsr�	testNames�__name__r5r&r'�list�createTests)r<r-rrrr:xs(
















zTestProgram.parseArgscCst|jr|j|j
_|
r@|dur"|j
n|�}|�|j|j|j�|_n0|jdur\|j
�|j	�
|_n|j
�
|j|j	�|_dS�
N)�testNamePatternsr7rN�startr �top�testrSZloadTestsFromModuler)ZloadTestsFromNames)r<�from_discovery�LoaderrrrrrV�s










�zTestProgram.createTestsc
Cs$|��}
|�
|
�
|_|�|
�
|_dSrW)�_getParentArgParser�_getMainArgParserrG�_getDiscoveryArgParserrA)r<Z
parent_parserrrrrB�s


zTestProgram._initArgParsersc
Cs�tj
d
d�
}
|
jddddddd	�
|
jd
ddddd
d	�
|
jddddd�
|jdurn|
jdddddd�
d
|_|jdur�|
jdddddd�
d
|_|jdur�|
jdddddd�
d
|_|jdur�|
jdd d!td"d#�
g|_|
S)$NF)
�add_helpz-vz	--verboser1�store_constr?zVerbose output)�dest�action�const�helpz-qz--quietr
zQuiet outputz--localsr$�
store_truez"Show local variables in tracebacks)rcrdrfz-fz
--failfastr/zStop on first fail or errorz-cz--catchr0z'Catch Ctrl-C and display results so farz-bz--bufferr2z%Buffer stdout and stderr during testsz-krX�appendz.Only run tests which match the given substring)rcrd�typerf)�argparse�ArgumentParser�add_argumentr/r0r2rXr!)r<�parserrrrr^�sL



�


�

�




�





�





�




�zTestProgram._getParentArgParsercCs2tj
|
g
d
�
}|j|_|j|_|jdddd�
|S)N�
�parentsrRrz?a list of any number of test modules, classes and test methods.)�nargsrf)rjrkr9rFrCrJrl)r<�parentrmrrrr_�s



�zTestProgram._getMainArgParsercCsztj
|
g
d
�
}d|j|_d|_|jddddd�
|jd	d
ddd�
|jd
dddd�
dD]}|j|dtjtjd�
qZ|S)Nrnz%s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryrYz*Directory to start discovery ('.' default))rcrfz-pz	--patternr z+Pattern to match tests ('test*.py' default)z-tz--top-level-directoryrZz<Top level directory of project (defaults to start directory))rYr rZ�
?)rpr%rf)rjrkr9rF�epilogrl�SUPPRESS)r<rqrm�argrrrr`�s$




�

�

�


�z"TestProgram._getDiscoveryArgParsercCsLd
|_d|_
d|_|
du
r:|jdur,|��
|j�|
|�
|jd|d�
dS)Nrztest*.pyT)r\r])rYr rZrArBrQrV)r<r-r]rrrrP�s





zTestProgram._do_discoveryc
	Cs�|jrt
�
|jdurtj|_t|jt�r�zTz"|j|j|j|j	|j
|jd
�}
Wn,tyz


|j|j|j|j	|j
d�}
Yn0Wq�ty�


|��}
Yq�0n|j}
|
�
|j�
|_|jr�t�|j���

dS)N)r1r/r2r4r$)r1r/r2r4)r0rr6rZTextTestRunnerr&rir1r/r2r4r$�	TypeError�runr[�resultr.r,Z
wasSuccessful)r<r6rrrr;�s2












�



�


zTestProgram.runTests)
N)FN)
N)rT�
__module__�__qualname__�__doc__r)r1r/r0r2r9r4rXrArZdefaultTestLoaderr>rErCr:rVrBr^r_r`rPr;rrrrr"7s,






��&
	
#

r")r{r,rjr
�rrZsignalsrZ
__unittestrIrKrrr!�objectr"�mainrrrr�<module>
s


	]