U
�AT^�+ c @ st d Z ddlZddlZddlZddlmZmZ ddlmZ dZ dZ
dZ d d
� Z d d � Z
d
d� ZG dd� 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 C sx t j�| �rt| �� �d�rtt j�| �rXt j�| t �� �}t j�|�sP|�t j �rT| S |} | d d� �
dd��
dd�S | S )Nz.pyi����z\�.z/) �os�pathZisfile�lowerZendswithZisabsZrelpathZgetcwdZ
startswithZpardirZreplace)�nameZrel_path� r
�%/usr/lib64/python3.8/unittest/main.py�
_convert_name s r c C s dd� | D �S )Nc S s g | ] }t |��qS r
)r )Z.0r r
r
r Z
<listcomp>. s z"_convert_names.<locals>.<listcomp>r
)Znamesr
r
r �_convert_names- s r
c C s d| krd| } | S )N�*z*%s*r
)�patternr
r
r �_convert_select_pattern1 s r c @ s� e Zd ZdZdZdZd Z Z Z Z Z
Z dZ dddde
jddddddf dd�dd �Zd d
d �Zd d
� Zdd� Zd dd�Zdd� Zdd� Zdd� Zdd� Zd dd�Zd d � 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 C s� t |t�r<t|�| _|�d�dd � D ]}
t| j|
�| _q&n|| _|d krPtj}|| _|| _ | | _
|| _ |
| _ | | _
| d kr�tjs�d| _n| | _|| _|| _|| _tj�|d �| _| �|� | �� d S )Nr r �defaultr )�
isinstance�strZ
__import__�moduleZsplitZgetattr�sys�argv�exit�failfast�
catchbreak� verbosity�bufferr Z warnoptions�warnings� defaultTest�
testRunner�
testLoaderr r Zbasename�progName� parseArgs�runTests)�selfr r r r! r" r r r r r r r Zpartr
r
r �__init__A s,
zTestProgram.__init__c C s4 |r t |� | jd kr | �� | �� t�d� d S )N� )�print�_discovery_parser�_initArgParsers� _print_helpr r )r& Zmsgr
r
r � usageExitg s
zTestProgram.usageExitc O sZ | j d kr6t| j�� � ttd| ji � | j�� n t| j�� � ttd| ji � d S )N�prog) r r) � _main_parserZ format_help�
MAIN_EXAMPLESr# r* �
print_help�MODULE_EXAMPLES)r& ZargsZkwargsr
r
r r, o s
zTestProgram._print_helpc C s� | � � | jd krpt|�dkrD|d �� dkrD| �|dd � � d S | j�|dd � | � | js�| �g � d S n| j�|dd � | � | jr�t| j�| _ t
dkr�d | _n6| j d kr�d | _ n$t | j t
�r�| j f| _ n t| j �| _ | �� d S )Nr �discoverr( r )r+ r Zlenr �
_do_discoveryr/ �
parse_args�testsr
� testNames�__name__r r r Zlist� createTests)r& r r
r
r r$ x s(
zTestProgram.parseArgsc C st | j r| j | j_ |r@|d kr"| jn|� }|�| j| j| j�| _n0| jd kr\| j�| j �| _n| j�
| j| j �| _d S �N) �testNamePatternsr" r3 �startr �top�testr7 ZloadTestsFromModuler ZloadTestsFromNames)r& �from_discovery�Loaderr r
r
r r9 � s
�zTestProgram.createTestsc C s$ | � � }| �|�| _| �|�| _d S r: )�_getParentArgParser�_getMainArgParserr/ �_getDiscoveryArgParserr* )r& 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| _| jd kr�|jd d d!td"d#� g | _|S )$NF)Zadd_helpz-vz --verboser Z store_constr( zVerbose output)�dest�actionZconst�helpz-qz--quietr z Quiet outputz--localsr Z
store_truez"Show local variables in tracebacks)rD rE rF 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 testsz-kr; Zappendz.Only run tests which match the given substring)rD rE �typerF )�argparse�ArgumentParser� add_argumentr r r r; r )r& �parserr
r
r rA � sR
�
��
�
�
�
�z TestProgram._getParentArgParserc C s2 t j|gd�}| j|_| j|_|jdddd� |S )N�Zparentsr6 r z?a list of any number of test modules, classes and test methods.)�nargsrF )rH rI r# r. r, r1 rJ )r& �parentrK r
r
r rB � 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� qZ|S )NrL z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryr< z*Directory to start discovery ('.' default))rD rF z-pz --patternr z+Pattern to match tests ('test*.py' default)z-tz--top-level-directoryr= z<Top level directory of project (defaults to start directory))r< r r= z?)rM r rF )rH rI r# r. ZepilogrJ ZSUPPRESS)r& rN rK Zargr
r
r rC � s$
�
�
��z"TestProgram._getDiscoveryArgParserc C sL d| _ d| _d | _|d k r:| jd kr,| �� | j�|| � | jd|d� d S )Nr ztest*.pyT)r? r@ )r< r r= r* r+ r5 r9 )r& r r@ r
r
r r4 � s
zTestProgram._do_discoveryc C s� | j r t� | jd kr tj| _t| jt�r�zVz"| 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� | �� }Y q�X n| j}|�
| j�| _| jr�t�| j�� � d S )N)r r r r r )r r r r )r r r! r ZTextTestRunnerr rG r r r r r Z TypeErrorZrunr> Zresultr r Z
wasSuccessful)r& r! r
r
r r% � s2
�
�zTestProgram.runTestsr: )FNr: )r8 Z
__module__Z __qualname__�__doc__r r r r r r# r r; r* r ZdefaultTestLoaderr' r- r, r$ r9 r+ rA rB rC r4 r% r
r
r
r r 7 s6 �� &
#
r )rO r rH r Z r r Zsignalsr Z
__unittestr0 r2 r r
r Zobjectr Zmainr
r
r
r Z<module> s ]