Your IP : 3.22.42.153
3
\�( � @ sx d dl Z ddlmZ ddlmZ dZdd� ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�Z dd� Z
G dd� de�Z dS )� N� )�case)�utilTc C s t | |dd� �}|� d S )Nc S s d S )N� r r r �&/usr/lib64/python3.6/unittest/suite.py�<lambda> s z!_call_if_exists.<locals>.<lambda>)�getattr)�parent�attr�funcr r r �_call_if_exists s r c @ sl e Zd ZdZf fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
BaseTestSuiteTc C s g | _ d| _| j|� d S )Nr )�_tests�_removed_tests�addTests)�self�testsr r r �__init__ s zBaseTestSuite.__init__c C s dt j| 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 )N)�
isinstancer �NotImplementedr )r �otherr r r �__eq__ s zBaseTestSuite.__eq__c C s
t | j�S )N)�iterr )r r r r �__iter__" s zBaseTestSuite.__iter__c C s( | j }x | D ]}|r ||j� 7 }q W |S )N)r �countTestCases)r Zcases�testr r r r % s
z BaseTestSuite.countTestCasesc C sL t |�stdjt|����t|t�r<t|tjt f�r<td��| j
j |� d S )Nz{} is not callablezNTestCases and TestSuites must be instantiated before passing them to addTest()) �callable� TypeError�format�reprr �type�
issubclassr ZTestCase� TestSuiter �append)r r r r r �addTest, s zBaseTestSuite.addTestc C s. t |t�rtd��x|D ]}| j|� qW d 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 s: x4t | �D ](\}}|jrP ||� | jr
| j|� q
W |S )N)� enumerate�
shouldStop�_cleanup�_removeTestAtIndex)r �result�indexr r r r �run< s zBaseTestSuite.runc C sN y| j | }W n tk
r" Y n(X t|d�r@| j|j� 7 _d | j |< d S )Nr )r r! �hasattrr r )r r/ r r r r r- E s
z BaseTestSuite._removeTestAtIndexc O s | j ||�S )N)r0 )r �args�kwdsr r r �__call__S s zBaseTestSuite.__call__c C s x| D ] }|j � qW d S )N)�debug)r r r r r r5 V s
zBaseTestSuite.debugN)�__name__�
__module__� __qualname__r, r r r r r r( r r0 r- r4 r5 r r r r r
s
r
c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )r& Fc C s� d}t |dd�dkr d |_}x�t| �D ]�\}}|jr8P t|�r�| j||� | j||� | j||� |j|_ t |jdd�s(t |dd�r�q(|s�||� n|j
� | j r(| j |� q(W |r�| jd |� | j
|� d|_|S )NF�_testRunEnteredT�_classSetupFailed�_moduleSetUpFailed)r r9 r* r+ � _isnotsuite�_tearDownPreviousClass�_handleModuleFixture�_handleClassSetUpr �_previousTestClassr5 r, r- �_handleModuleTearDown)r r. r5 ZtopLevelr/ r r r r r0 f s0
z
TestSuite.runc C s t � }| j|d� d S )NT)� _DebugResultr0 )r r5 r r r r5 � s zTestSuite.debugc C s� t |dd �}|j}||kr d S |jr(d S t |dd�r8d S y
d|_W n tk
rV Y nX t |dd �}|d k r�t|d� zdy
|� W nT tk
r� } z8t|t�r�� d|_t j
|�}d| }| j |||� W Y d d }~X nX W d t|d� X d S ) Nr@ �__unittest_skip__F�
setUpClass� _setupStdoutTzsetUpClass (%s)�_restoreStdout) r r r; r: r! r � Exceptionr rB r r � _addClassOrModuleLevelException) r r r. �
previousClass� currentClassrD �e� className� errorNamer r r r? � s2
$zTestSuite._handleClassSetUpc C s" d }t |dd �}|d k r |j}|S )Nr@ )r r7 )r r. �previousModulerI r r r �_get_previous_module� s
z TestSuite._get_previous_modulec C s� | j |�}|jj}||kr d S | j|� d|_ytj| }W n tk
rP d S X t|dd �}|d k r�t |d� zZy
|� W nJ t
k
r� } z.t |t �r�� d|_d| }| j
|||� W Y d d }~X nX W d t |d� X d S )NF� setUpModulerE TzsetUpModule (%s)rF )rO r r7 rA r; �sys�modules�KeyErrorr r rG r rB rH ) r r r. rN Z
currentModule�modulerP rK rM r r r r> � s,
$z TestSuite._handleModuleFixturec C sL t |�}t|dd �}|d k r8t|tj�r8||t|�� n|j|tj� � d S )N�addSkip) � _ErrorHolderr r r ZSkipTestr) ZaddErrorrQ �exc_info)r r. Z exceptionrM �errorrU r r r rH � s
z)TestSuite._addClassOrModuleLevelExceptionc C s� | j |�}|d krd S |jr d S ytj| }W n tk
rB d S X t|dd �}|d k r�t|d� zTy
|� W nD tk
r� } z(t|t �r�� d| }| j
|||� W Y d d }~X nX W d t|d� X d S )N�tearDownModulerE ztearDownModule (%s)rF ) rO r; rQ rR rS r r rG r rB rH )r r. rN rT rY rK rM r r r rA � 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 r�t|d� z^y
|� W nN tk
r� } z2t|t�r�� tj|�}d| }| j|||� W Y d d }~X nX W d t|d � X d S )
Nr@ r: Fr; rC �
tearDownClassrE ztearDownClass (%s)rF ) r r r rG r rB r r rH ) r r r. rI rJ rZ rK rL rM r r r r= � s,
$z TestSuite._tearDownPreviousClassN)F) r6 r7 r8 r0 r5 r? rO r> rH rA r= r r r r r& \ s
! r& c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )rV Nc C s
|| _ d S )N)� description)r r[ r r r r s z_ErrorHolder.__init__c C s | j S )N)r[ )r r r r �id s z_ErrorHolder.idc C s d S )Nr )r r r r �shortDescription s z _ErrorHolder.shortDescriptionc C s d| j f S )Nz <ErrorHolder description=%r>)r[ )r r r r r # s z_ErrorHolder.__repr__c C s | j � S )N)r\ )r r r r �__str__& s z_ErrorHolder.__str__c C s d S )Nr )r r. r r r r0 ) s z_ErrorHolder.runc C s
| j |�S )N)r0 )r r. r r r r4 . s z_ErrorHolder.__call__c C s dS )Nr r )r r r r r 1 s z_ErrorHolder.countTestCases) r6 r7 r8 ZfailureExceptionr r\ r] r r^ r0 r4 r r r r r rV s
rV c C s& y t | � W n tk
r dS X dS )NTF)r r! )r r r r r< 4 s
r< c @ s e Zd ZdZdZdZdS )rB NF)r6 r7 r8 r@ r; r+ r r r r rB = s rB ) rQ � r r Z
__unittestr �objectr
r& rV r< rB r r r r �<module> s L 3&