U
e5d2 � @ s| d Z ddlZddlmZ ddlmZ dZdd� ZG d d
� d
e�ZG d d � d e�Z G d d
� d
e�Z
dd� Z G dd� de�Z dS )� TestSuite� N� )�case)�utilTc C s t | |dd� �}|� d S )Nc S s d S �N� r r r �&/usr/lib64/python3.8/unittest/suite.py�<lambda> � z!_call_if_exists.<locals>.<lambda>)�getattr)�parent�attr�funcr r r �_call_if_exists s r c @ sn e Zd ZdZdZddd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �
BaseTestSuitezNA simple test suite that doesn't provide class or module shared fixtures.
Tr c C s g | _ d| _| �|� d S �Nr )�_tests�_removed_tests�addTests)�self�testsr r r �__init__ s zBaseTestSuite.__init__c C s dt �| j�t| �f S )Nz
<%s tests=%s>)r �strclass� __class__�list�r r r r �__repr__ s zBaseTestSuite.__repr__c C s t || j�stS t| �t|�kS r )�
isinstancer �NotImplementedr )r �otherr r r �__eq__ s zBaseTestSuite.__eq__c C s
t | j�S r )�iterr r r r r �__iter__" s zBaseTestSuite.__iter__c C s$ | j }| D ]}|r
||�� 7 }q
|S r )r �countTestCases)r Zcases�testr r r r# % s
z BaseTestSuite.countTestCasesc C sL t |�std�t|����t|t�r<t|tjt f�r<td��| j
� |� d S )Nz{} is not callablezNTestCases and TestSuites must be instantiated before passing them to addTest()) �callable� TypeError�format�reprr �type�
issubclassr ZTestCaser r �append�r r$ r r r �addTest, s �zBaseTestSuite.addTestc C s* t |t�rtd��|D ]}| �|� qd S )Nz0tests must be an iterable of tests, not a string)r �strr& r- )r r r$ r r r r 6 s
zBaseTestSuite.addTestsc C s8 t | �D ]*\}}|jr q4||� | jr| �|� q|S r )� enumerate�
shouldStop�_cleanup�_removeTestAtIndex)r �result�indexr$ r r r �run< s zBaseTestSuite.runc C sN z| j | }W n tk
r" Y n(X t|d�r@| j|�� 7 _d| j |< dS )z2Stop holding a reference to the TestCase at index.r# N)r r& �hasattrr r# )r r4 r$ r r r r2 E s
z BaseTestSuite._removeTestAtIndexc O s | j ||�S r �r5 )r �args�kwdsr r r �__call__S s zBaseTestSuite.__call__c C s | D ] }|� � qdS )�7Run the tests without collecting errors in a TestResultN)�debugr, r r r r< V s zBaseTestSuite.debugN)r )�__name__�
__module__� __qualname__�__doc__r1 r r r r" r# r- r r5 r2 r: r< r r r r r s
r c @ s^ e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zddd�Z ddd�Z
dd� Z dd� Z d
S )r a� A test suite is a composite test consisting of a number of TestCases.
For use, create an instance of TestSuite, then add test case instances.
When all tests have been added, the suite can be passed to a test
runner, such as TextTestRunner. It will run the individual test cases
in the order in which they were added, aggregating the results. When
subclassing, do not forget to call the base class constructor.
Fc C s� d}t |dd�dkr d |_}t| �D ]�\}}|jr8 q�t|�r�| �||� | �||� | �||� |j|_ t |jdd�s&t |dd�r�q&|s�||� n|�
� | j r&| � |� q&|r�| �d |� | �
|� d|_|S )NF�_testRunEnteredT�_classSetupFailed�_moduleSetUpFailed)r rA r/ r0 � _isnotsuite�_tearDownPreviousClass�_handleModuleFixture�_handleClassSetUpr �_previousTestClassr< r1 r2 �_handleModuleTearDown)r r3 r< ZtopLevelr4 r$ r r r r5 f s2
�
z
TestSuite.runc C s t � }| �|d� dS )r; TN)� _DebugResultr5 )r r< r r r r< � s zTestSuite.debugc C s2 t |dd �}|j}||kr d S |jr(d S t |dd�r8d S z
d|_W n tk
rV Y nX t |dd �}|d k �r.t|d� z^z
|� W nN t
k
r� } z0t |t �r�� d|_t
�|�}| � ||d|� W 5 d }~X Y nX W 5 t|d� |jdk�r,|�� t|j�dk�r,|jD ] }| j ||d d||d
� �q X d S ) NrH �__unittest_skip__F�
setUpClass� _setupStdout�_restoreStdoutTr r ��info)r r rC rB r& r �doClassCleanups�len�tearDown_exceptions�"_createClassOrModuleLevelException� Exceptionr rJ r r ) r r$ r3 �
previousClass� currentClassrL �exc� className�er r r rG � sL
�
�zTestSuite._handleClassSetUpc C s" d }t |dd �}|d k r |j}|S )NrH )r r> )r r3 �previousModulerV r r r �_get_previous_module� s
z TestSuite._get_previous_modulec
C s | � |�}|jj}||kr d S | �|� d|_ztj| }W n tk
rR Y d S X t|dd �}|d k �rt |d� z�z
|� W n� t
k
�r } zfz t � � W n2 t
k
r� } z| �
||d|� W 5 d }~X Y nX t|t�r� d|_| �
||d|� W 5 d }~X Y nX W 5 t |d� X d S )NF� setUpModulerM rN T)r\ r r> rI rC �sys�modules�KeyErrorr r rU r �doModuleCleanupsrT r rJ ) r r$ r3 r[ Z
currentModule�moduler] rZ rX r r r rF � s>
�
�z TestSuite._handleModuleFixtureNc C s$ |� d|� d�}| � ||||� d S )Nz (�))� _addClassOrModuleLevelException)r r3 rX Z method_namer rP � errorNamer r r rT � s z,TestSuite._createClassOrModuleLevelExceptionc C s^ t |�}t|dd �}|d k r8t|tj�r8||t|�� n"|sN|�|t�� � n |�||� d S )N�addSkip) � _ErrorHolderr r r ZSkipTestr. ZaddErrorr^ �exc_info)r r3 Z exceptionre rP �errorrf r r r rd � s z)TestSuite._addClassOrModuleLevelExceptionc C s | � |�}|d krd S |jr d S ztj| }W n tk
rD Y d S X t|dd �}|d k �rt|d� zNz
|� W n> t k
r� } z t |t �r�� | �
||d|� W 5 d }~X Y nX W 5 t|d� z t�� W n4 t k
�r } z| �
||d|� W 5 d }~X Y nX X d S )N�tearDownModulerM rN )
r\ rC r^ r_ r` r r r ra rU rT r rJ )r r3 r[ rb rj rZ r r r rI � s:
�
�z TestSuite._handleModuleTearDownc C s t |dd �}|j}||kr d S t |dd�r.d S t |dd�r>d S t |dd�rNd S t |dd �}|d k �rt|d� zXz
|� W nH t k
r� } z*t
|t �r�� t�|�}| �||d|� W 5 d }~X Y nX W 5 t|d� |�� t|j�d k�r|jD ]&}t�|�}| j||d
d||d � q�X d S ) NrH rB FrC rK �
tearDownClassrM rN r r rO ) r r r rQ rR rS r r rT rU r rJ ) r r$ r3 rV rW rk rX rY rZ r r r rE sB
�
�z TestSuite._tearDownPreviousClass)F)N)N)
r= r> r? r@ r5 r< rG r\ rF rT rd rI rE r r r r r \ s
!($ �
�
c @ sT e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )rg z�
Placeholder for a TestCase inside a result. As far as a TestResult
is concerned, this looks exactly like a unit test. Used to insert
arbitrary errors into a test suite run.
Nc C s
|| _ d S r �� description)r rm r r r r B s z_ErrorHolder.__init__c C s | j S r rl r r r r �idE s z_ErrorHolder.idc C s d S r r r r r r �shortDescriptionH s z _ErrorHolder.shortDescriptionc C s d| j f S )Nz <ErrorHolder description=%r>rl r r r r r K s z_ErrorHolder.__repr__c C s | � � S r )rn r r r r �__str__N s z_ErrorHolder.__str__c C s d S r r �r r3 r r r r5 Q s z_ErrorHolder.runc C s
| � |�S r r7 rq r r r r: V s z_ErrorHolder.__call__c C s dS r r r r r r r# Y s z_ErrorHolder.countTestCases)
r= r> r? r@ ZfailureExceptionr rn ro r rp r5 r: r# r r r r rg 6 s rg c C s( z t | � W n tk
r" Y dS X dS )z?A crude way to tell apart testcases and suites with duck-typingTF)r! r& )r$ r r r rD \ s
rD c @ s e Zd ZdZdZdZdZdS )rJ zCUsed by the TestSuite to hold previous class when running in debug.NF)r= r> r? r@ rH rC r0 r r r r rJ e s rJ )
r@ r^ � r r Z
__unittestr �objectr r rg rD rJ r r r r �<module> s L [& 