B
    ^ ÇcW  ã               @   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   ú lib/python3.7/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 )Nr   )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	   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!   Úerr)r   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   r   r	   r.   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!   Úreason)r   r   r	   r0   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   r   r	   r4   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	   r6   b   s    z#TextTestResult.addUnexpectedSuccessc             C   s6   | j s| jr| j ¡  |  d| j¡ |  d| j¡ d S )Nr(   r,   )r   r   r   r   ÚprintErrorListÚerrorsÚfailures)r   r   r   r	   ÚprintErrorsj   s    
zTextTestResult.printErrorsc             C   s\   xV|D ]N\}}| j  | j¡ | j  d||  |¡f ¡ | j  | j¡ | j  d| ¡ qW d S )Nz%s: %sz%s)r   r   Ú
separator1r"   Ú
separator2)r   Zflavourr8   r!   r+   r   r   r	   r7   p   s
    zTextTestResult.printErrorList)r   r   r   r   r;   r<   r
   r"   r#   r'   r*   r.   r0   r4   r6   r:   r7   Ú__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Úbufferr?   ÚwarningsÚresultclass)	r   r   r   r   rB   rC   rE   rD   r?   r   r   r	   r
   €   s    
zTextTestRunner.__init__c             C   s   |   | j| j| j¡S )N)rE   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 dt|ddƒ}|dk	r¨|ƒ  X t
 ¡ }W dQ R X || }| ¡  t|dƒrä| j |j¡ |j}| j d	||d
koþdpd|f ¡ | j ¡  d }	 }
}y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ÚstopTestRunr<   zRan %d test%s in %.3fsr   r/   Ú r   ZFAILEDzfailures=%dz	errors=%dZOKz
skipped=%dzexpected failures=%dzunexpected successes=%dz (%s)z, r   )rF   r   rB   rC   r?   rD   Úcatch_warningsÚsimplefilterÚfilterwarningsÚDeprecationWarningÚtimeZperf_counterr   r:   Úhasattrr   r   r<   ZtestsRunÚmapÚlenZexpectedFailuresÚunexpectedSuccessesÚskippedr   ZwasSuccessfulr   r9   r8   Úappendr   )r   r!   r   Z	startTimerL   rM   ZstopTimeZ	timeTakenÚrunZexpectedFailsrW   rX   ZresultsZinfosZfailedZerroredr   r   r	   rZ   —   sr    





zTextTestRunner.run)NTr   FFNN)	r   r   r   r   r   rE   r
   rF   rZ   r   r   r   r	   r>   x   s    r>   )r   r@   rS   rD   rN   r   Zsignalsr   Z
__unittestÚobjectr   Z
TestResultr   r>   r   r   r   r	   Ú<module>   s   [