CasperSecurity
o
`�wi�+ � @ 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 s� t j�| �r>| �� �d�r>t j�| �r,t j�| t �� �}t j�|�s(|�t j �r*| S |} t j�
| �d d� � dd�� dd�S | S )Nz.py����\�.�/) �os�path�isfile�lower�endswith�isabs�relpath�getcwd�
startswith�pardir�normpath�replace)�name�rel_path� r �$/usr/lib/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 �
<listcomp>. s z"_convert_names.<locals>.<listcomp>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 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|
�| _qn|| _|d u r(tj}|| _|| _ | | _
|| _ |
| _ | | _
| d u rEtjsEd| _n| | _|| _|| _|| _tj�|d �| _| �|� | �� d S )Nr r �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) r5 r- r6 r7 r. r1 r/ r0 r2 r4 r$ �partr r r �__init__A s.
�
zTestProgram.__init__c C s4 |rt |� | jd u r| �� | �� t�d� 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 )N�prog) r) r@ � _main_parser� format_help�
MAIN_EXAMPLESr9 rA �
print_help�MODULE_EXAMPLES)r< �args�kwargsr r r rC 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# )rB r) �lenr
�
_do_discoveryrG �
parse_args�testsr � testNames�__name__r5 r& r' �list� createTests)r< r- r r r r: 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) �testNamePatternsr7 rN �startr �top�testrS �loadTestsFromModuler) �loadTestsFromNames)r< �from_discovery�Loaderr r r r rV � s
�zTestProgram.createTestsc C s$ | � � }| �|�| _| �|�| _d S rW )�_getParentArgParser�_getMainArgParserrG �_getDiscoveryArgParserrA )r< �
parent_parserr r r rB � 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)�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)rf rg ri z-fz
--failfastr/ zStop on first fail or errorz-cz--catchr0 z'Catch Ctrl-C and display results so farz-bz--bufferr2 z%Buffer stdout and stderr during testsz-krX �appendz.Only run tests which match the given substring)rf rg �typeri )�argparse�ArgumentParser� add_argumentr/ r0 r2 rX r! )r< �parserr r r r` � sL
�
��
�
�
�
�z TestProgram._getParentArgParserc C s2 t j|gd�}| j|_| j|_|jdddd� |S )N��parentsrR r z?a list of any number of test modules, classes and test methods.)�nargsri )rm rn r9 rF rC rJ ro )r< �parentrp r r r ra � 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 )Nrq z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryrY z*Directory to start discovery ('.' default))rf ri z-pz --patternr z+Pattern to match tests ('test*.py' default)z-tz--top-level-directoryrZ z<Top level directory of project (defaults to start directory))rY r rZ �?)rs r% ri )rm rn r9 rF �epilogro �SUPPRESS)r< rt rp �argr r r rb � s$
�
�
��z"TestProgram._getDiscoveryArgParserc C sL d| _ d| _d | _|d ur | jd u r| �� | j�|| � | jd|d� d S )Nr ztest*.pyT)r^ r_ )rY r rZ rA rB rQ rV )r< r- r_ r r r rP � s
zTestProgram._do_discoveryc C s� | j rt� | jd u rtj| _t| jt�rMz*z| j| j| j| j | j
| j d�}W n t y= | j| j| j| j | j
d�}Y nw W n t yL | �� }Y nw | j}|�
| j�| _| jret�| j�� � d S d S )N)r1 r/ r2 r4 r$ )r1 r/ r2 r4 )r0 r r6 r �TextTestRunnerr& rl r1 r/ r2 r4 r$ � TypeError�runr[ �resultr. r, �
wasSuccessful)r< r6 r r r r; � s:
�
��� ��zTestProgram.runTestsrW )FN)rT �
__module__� __qualname__�__doc__r) r1 r/ r0 r2 r9 r4 rX rA r �defaultTestLoaderr> rE rC r: rV rB r` ra rb rP r; r r r r r" 7 s. � �
&
#
r" )r� r, rm r
� r r �signalsr �
__unittestrI rK r r r! �objectr" �mainr r r r �<module> s ]