ruạṛ
a ���d�+ � @ 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�| �rXt j�| t �� �}t j�|�sP|�t j �rT| 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/lib64/python3.9/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>. � 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| �| _q&n|| _|d u rPtj}|| _|| _ | | _ || _ | | _ | | _ | d u r�tjs�d| _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* r6 r. r7 r8 r/ r2 r0 r1 r3 r5 r% �partr r r �__init__A s, zTestProgram.__init__c C s4 |r t |� | jd u r | �� | �� t�d� d S )N� )�print�_discovery_parser�_initArgParsers� _print_helpr- r/ )r= �msgr r r � usageExitg s zTestProgram.usageExitc O sZ | j d u r6t| j�� � ttd| ji � | j�� n t| j�� � ttd| ji � d S )N�prog) r* rA � _main_parser� format_help� MAIN_EXAMPLESr: rB � print_help�MODULE_EXAMPLES)r= �args�kwargsr r r rD o s zTestProgram._print_helpc C s� | � � | jd u rpt|�dkrD|d �� dkrD| �|dd � � d S | j�|dd � | � | js�| �g � d S n| j�|dd � | � | jr�t| j�| _ t dkr�d | _n6| j d u r�d | _ n$t | j t �r�| j f| _ n t| j �| _ | �� d S )Nr �discoverr@ r$ )rC r* �lenr � _do_discoveryrH � parse_args�testsr � testNames�__name__r6 r' r( �list� createTests)r= r. r r r r; x s( zTestProgram.parseArgsc C st | j r| j | j_ |r@|d u r"| jn|� }|�| j| j| j�| _n0| jd u r\| j�| j �| _n| j� | j| j �| _d S �N) �testNamePatternsr8 rO �startr! �top�testrT �loadTestsFromModuler* �loadTestsFromNames)r= �from_discovery�Loaderr r r r rW � s �zTestProgram.createTestsc C s$ | � � }| �|�| _| �|�| _d S rX )�_getParentArgParser�_getMainArgParserrH �_getDiscoveryArgParserrB )r= � parent_parserr r r rC � s zTestProgram._initArgParsersc C s� t jdd�}|jddddddd � |jd d ddd d d � |jddddd� | jd u rn|jdddddd� d| _| jd u r�|jdddddd� d| _| jd u r�|jdd d dd d� d| _| jd u r�|jd d d!td"d#� g | _|S )$NF)�add_helpz-vz --verboser2 � store_constr@ zVerbose output)�dest�action�const�helpz-qz--quietr z Quiet outputz--localsr% � store_truez"Show local variables in tracebacks)rg rh rj z-fz --failfastr0 zStop on first fail or errorz-cz--catchr1 z'Catch Ctrl-C and display results so farz-bz--bufferr3 z%Buffer stdout and stderr during testsz-krY �appendz.Only run tests which match the given substring)rg rh �typerj )�argparse�ArgumentParser� add_argumentr0 r1 r3 rY r" )r= �parserr r r ra � sL � �� � � � �z TestProgram._getParentArgParserc C s2 t j|gd�}| j|_| j|_|jdddd� |S )N��parentsrS r z?a list of any number of test modules, classes and test methods.)�nargsrj )rn ro r: rG rD rK rp )r= �parentrq 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 )Nrr z %s discoverzcFor test discovery all test modules must be importable from the top level directory of the project.z-sz--start-directoryrZ z*Directory to start discovery ('.' default))rg rj 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))rZ r! r[ �?)rt r& rj )rn ro r: rG �epilogrp �SUPPRESS)r= ru rq �argr r r rc � s$ � � ��z"TestProgram._getDiscoveryArgParserc C sL d| _ d| _d | _|d ur:| jd u r,| �� | j�|| � | jd|d� d S )Nr ztest*.pyT)r_ r` )rZ r! r[ rB rC rR rW )r= r. r` r r r rQ � s zTestProgram._do_discoveryc C s� | j r t� | jd u r tj| _t| jt�r�zTz"| j| j| j| j | j | j d�}W n, t yz | j| j| j| j | j d�}Y n0 W q� t y� | �� }Y q�0 n| j}|� | j�| _| jr�t�| j�� � d S )N)r2 r0 r3 r5 r% )r2 r0 r3 r5 )r1 r r7 r �TextTestRunnerr' rm r2 r0 r3 r5 r% � TypeError�runr\ �resultr/ r- � wasSuccessful)r= r7 r r r r< � s2 � � zTestProgram.runTests)N)FN)N)rU � __module__� __qualname__�__doc__r* r2 r0 r1 r3 r: r5 rY rB r �defaultTestLoaderr? rF rD r; rW rC ra rb rc rQ r< r r r r r# 7 s, �� & # r# )r� r- rn r � r r �signalsr � __unittestrJ rL r r r" �objectr# �mainr r r r �<module> s ]
cải xoăn