U
    kÖ]eW  ã                   @   sn   d Z ddlZddlZddlZddlmZ ddlmZ dZG dd„ de	ƒZ
G d	d
„ d
ejƒZG dd„ de	ƒZdS )zRunning testsé    Né   )Úresult)ÚregisterResultTc                   @   s*   e Zd ZdZdd„ Zdd„ Zd	dd„ZdS )
Ú_WritelnDecoratorz@Used to decorate file-like objects with a handy 'writeln' methodc                 C   s
   || _ d S ©N)Ústream)Úselfr   © r	   ú%/usr/lib/python3.8/unittest/runner.pyÚ__init__   s    z_WritelnDecorator.__init__c                 C   s   |dkrt |ƒ‚t| j|ƒS )N)r   Ú__getstate__)ÚAttributeErrorÚgetattrr   )r   Úattrr	   r	   r
   Ú__getattr__   s    z_WritelnDecorator.__getattr__Nc                 C   s   |r|   |¡ |   d¡ d S ©NÚ
)Úwrite)r   Úargr	   r	   r
   Úwriteln   s    
z_WritelnDecorator.writeln)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r	   r	   r	   r
   r      s   r   c                       s”   e Zd ZdZdZdZ‡ fdd„Zdd„ Z‡ fdd	„Z‡ fd
d„Z	‡ fdd„Z
‡ fdd„Z‡ fdd„Z‡ fdd„Z‡ fdd„Zdd„ Zdd„ Z‡  ZS )ÚTextTestResultzhA test result class that can print formatted text results to a stream.

    Used by TextTestRunner.
    zF======================================================================zF----------------------------------------------------------------------c                    s8   t t| ƒ |||¡ || _|dk| _|dk| _|| _d S )Nr   )Úsuperr   r   r   ÚshowAllÚdotsÚdescriptions)r   r   r   Ú	verbosity©Ú	__class__r	   r
   r   %   s
    

zTextTestResult.__init__c                 C   s0   |  ¡ }| jr$|r$d t|ƒ|f¡S t|ƒS d S r   )ZshortDescriptionr   ÚjoinÚstr)r   ÚtestZdoc_first_liner	   r	   r
   ÚgetDescription,   s    
zTextTestResult.getDescriptionc                    sB   t t| ƒ |¡ | jr>| j |  |¡¡ | j d¡ | j ¡  d S )Nz ... )r   r   Ú	startTestr   r   r   r%   Úflush©r   r$   r    r	   r
   r&   3   s
    zTextTestResult.startTestc                    sD   t t| ƒ |¡ | jr$| j d¡ n| jr@| j d¡ | j ¡  d S )NÚokÚ.)	r   r   Ú
addSuccessr   r   r   r   r   r'   r(   r    r	   r
   r+   :   s    zTextTestResult.addSuccessc                    sF   t t| ƒ ||¡ | jr&| j d¡ n| jrB| j d¡ | j ¡  d S )NÚERRORÚE)	r   r   ÚaddErrorr   r   r   r   r   r'   ©r   r$   Úerrr    r	   r
   r.   B   s    zTextTestResult.addErrorc                    sF   t t| ƒ ||¡ | jr&| j d¡ n| jrB| j d¡ | j ¡  d S )NÚFAILÚF)	r   r   Ú
addFailurer   r   r   r   r   r'   r/   r    r	   r
   r3   J   s    zTextTestResult.addFailurec                    sL   t t| ƒ ||¡ | jr,| j d |¡¡ n| jrH| j d¡ | j 	¡  d S )Nzskipped {0!r}Ús)
r   r   ÚaddSkipr   r   r   Úformatr   r   r'   )r   r$   Úreasonr    r	   r
   r5   R   s    zTextTestResult.addSkipc                    sF   t t| ƒ ||¡ | jr&| j d¡ n| jrB| j d¡ | j ¡  d S )Nzexpected failureÚx)	r   r   ÚaddExpectedFailurer   r   r   r   r   r'   r/   r    r	   r
   r9   Z   s    z!TextTestResult.addExpectedFailurec                    sD   t t| ƒ |¡ | jr$| j d¡ n| jr@| j d¡ | j ¡  d S )Nzunexpected successÚu)	r   r   ÚaddUnexpectedSuccessr   r   r   r   r   r'   r(   r    r	   r
   r;   b   s    z#TextTestResult.addUnexpectedSuccessc                 C   s6   | j s| jr| j ¡  |  d| j¡ |  d| j¡ d S )Nr,   r1   )r   r   r   r   ÚprintErrorListÚerrorsÚfailures©r   r	   r	   r
   ÚprintErrorsj   s    
zTextTestResult.printErrorsc                 C   sX   |D ]N\}}| j  | j¡ | j  d||  |¡f ¡ | j  | j¡ | j  d| ¡ qd S )Nz%s: %sz%s)r   r   Ú
separator1r%   Ú
separator2)r   Zflavourr=   r$   r0   r	   r	   r
   r<   p   s
    zTextTestResult.printErrorList)r   r   r   r   rA   rB   r   r%   r&   r+   r.   r3   r5   r9   r;   r@   r<   Ú__classcell__r	   r	   r    r
   r      s   r   c                   @   s4   e Zd ZdZeZdddœdd„Zd	d
„ Zdd„ ZdS )ÚTextTestRunnerzÉA test runner class that displays results in textual form.

    It prints out the names of tests as they are run, errors as they
    occur, and a summary of the results at the end of the test run.
    NTr   F)Ú	tb_localsc          	      C   sN   |dkrt j}t|ƒ| _|| _|| _|| _|| _|| _|| _	|dk	rJ|| _
dS )z‰Construct a TextTestRunner.

        Subclasses should accept **kwargs to ensure compatibility as the
        interface changes.
        N)ÚsysÚstderrr   r   r   r   ÚfailfastÚbufferrE   ÚwarningsÚresultclass)	r   r   r   r   rH   rI   rK   rJ   rE   r	   r	   r
   r   €   s    
zTextTestRunner.__init__c                 C   s   |   | j| j| j¡S r   )rK   r   r   r   r?   r	   r	   r
   Ú_makeResult”   s    zTextTestRunner._makeResultc                 C   s2  |   ¡ }t|ƒ | j|_| j|_| j|_t ¡ † | jr^t | j¡ | jdkr^tjdt	dd t
 ¡ }t|ddƒ}|dk	r€|ƒ  z||ƒ W 5 t|ddƒ}|dk	r¨|ƒ  X t
 ¡ }W 5 Q R X || }| ¡  t|dƒrä| j |j¡ |j}| j d	||d
koþdpd|f ¡ | j ¡  d }	 }
}ztt|j|j|jfƒ}W n tk
rT   Y nX |\}	}
}g }| ¡ sº| j d¡ t|jƒt|jƒ }}|r¤| d| ¡ |rÆ| d| ¡ n| j d¡ |rÚ| d| ¡ |	rî| d|	 ¡ |
r| d|
 ¡ |r"| j dd |¡f ¡ n| j d¡ |S )z&Run the given test case or test suite.)ÚdefaultÚalwaysÚmodulezPlease use assert\w+ instead.)ÚcategoryÚmessageÚstartTestRunNÚstopTestRunrB   zRan %d test%s in %.3fsr   r4   Ú r   ZFAILEDzfailures=%dz	errors=%dZOKz
skipped=%dzexpected failures=%dzunexpected successes=%dz (%s)z, r   )rL   r   rH   rI   rE   rJ   Úcatch_warningsÚsimplefilterÚfilterwarningsÚDeprecationWarningÚtimeÚperf_counterr   r@   Úhasattrr   r   rB   ZtestsRunÚmapÚlenZexpectedFailuresÚunexpectedSuccessesÚskippedr   ZwasSuccessfulr   r>   r=   Úappendr"   )r   r$   r   Z	startTimerR   rS   ZstopTimeZ	timeTakenÚrunZexpectedFailsr^   r_   ZresultsZinfosZfailedZerroredr	   r	   r
   ra   —   sx    

þ
ÿ
þ


zTextTestRunner.run)NTr   FFNN)	r   r   r   r   r   rK   r   rL   ra   r	   r	   r	   r
   rD   x   s           ÿþrD   )r   rF   rY   rJ   rT   r   Zsignalsr   Z
__unittestÚobjectr   Z
TestResultr   rD   r	   r	   r	   r
   Ú<module>   s   [