3
\�( � @ 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.6/unittest/suite.py�<lambda> s z!_call_if_exists.<locals>.<lambda>)�getattr)�parent�attr�funcr r r �_call_if_exists s r
c @ sp e Zd ZdZdZf fdd�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�
BaseTestSuitezNA simple test suite that doesn't provide class or module shared fixtures.
Tc 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 ZTestCaser r �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 |< dS )z2Stop holding a reference to the TestCase at index.r N)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 dS )z7Run the tests without collecting errors in a TestResultN)�debug)r r r r r r5 V s
zBaseTestSuite.debugN)�__name__�
__module__� __qualname__�__doc__r, r r r r r r( r r0 r- r4 r5 r r r r r s
r c @ sR e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )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 |_}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 r: 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� dS )z7Run the tests without collecting errors in a TestResultTN)� _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 ) NrA �__unittest_skip__F�
setUpClass� _setupStdoutTzsetUpClass (%s)�_restoreStdout) r r r< r; r" r
� Exceptionr rC r r � _addClassOrModuleLevelException) r r r. �
previousClass� currentClassrE �e� className� errorNamer r r r@ � s2
$zTestSuite._handleClassSetUpc C s" d }t |dd �}|d k r |j}|S )NrA )r r7 )r r. �previousModulerJ 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� setUpModulerF TzsetUpModule (%s)rG )rP r r7 rB r< �sys�modules�KeyErrorr r
rH r rC rI ) r r r. rO Z
currentModule�modulerQ rL rN 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) ZaddErrorrR �exc_info)r r. Z exceptionrN �errorrV r r r rI � 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�tearDownModulerF ztearDownModule (%s)rG ) rP r< rR rS rT r r
rH r rC rI )r r. rO rU rZ rL rN r r r rB � 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 )
NrA r; Fr< rD �
tearDownClassrF ztearDownClass (%s)rG ) r r r
rH r rC r r rI ) r r r. rJ rK r[ rL rM rN r r r r> � s,
$z TestSuite._tearDownPreviousClassN)F) r6 r7 r8 r9 r0 r5 r@ rP r? rI rB r> 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 )rW 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 )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 r9 ZfailureExceptionr r] r^ r r_ r0 r4 r r r r r rW s rW c C s& y t | � W n tk
r dS X dS )z?A crude way to tell apart testcases and suites with duck-typingTF)r r" )r r r r r= 4 s
r= c @ s e Zd ZdZdZdZdZdS )rC zCUsed by the TestSuite to hold previous class when running in debug.NF)r6 r7 r8 r9 rA r< r+ r r r r rC = s rC )
r9 rR � r r Z
__unittestr
�objectr r rW r= rC r r r r �<module> s L 3&