3
\ @ sX d dl Z d dlZd dlZddlmZ d dlmZ dZdd ZdZ d Z
G d
d d e Z dS ) N )util)wrapsTc s t fdd}|S )Nc s$ t | ddr| j | f||S )NfailfastF)getattrstop)selfargskw)method '/usr/lib64/python3.6/unittest/result.pyinner
s zfailfast.<locals>.inner)r )r r r )r r
r s r z
Stdout:
%sz
Stderr:
%sc @ s e Zd ZdZdZdZd-ddZdd Zdd Zd d
Z d d Z
d
d Z dd Z dd Z
edd Zedd Zdd Zdd Zdd Zd d Zed d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, ZdS ).
TestResultNFc C sb d| _ g | _g | _d| _g | _g | _g | _d| _d| _d| _ d | _
d | _ t j
| _t j| _d| _d S )NFr )r failureserrorstestsRunskippedexpectedFailuresunexpectedSuccesses
shouldStopbuffer tb_locals_stdout_buffer_stderr_buffersysstdout_original_stdoutstderr_original_stderr
_mirrorOutput)r streamZ descriptions verbosityr r r
__init__& s zTestResult.__init__c C s d S )Nr )r r r r
printErrors7 s zTestResult.printErrorsc C s | j d7 _ d| _| j d S )Nr F)r r _setupStdout)r testr r r
startTest: s zTestResult.startTestc C s8 | j r4| jd kr$tj | _tj | _| jt_| jt_d S )N)r r ioStringIOr r r r )r r r r
r% @ s
zTestResult._setupStdoutc C s d S )Nr )r r r r
startTestRunH s zTestResult.startTestRunc C s | j d| _d S )NF)_restoreStdoutr )r r& r r r
stopTestN s zTestResult.stopTestc C s | j r| jrltjj }tjj }|rF|jds6|d7 }| jjt | |rl|jds\|d7 }| j
jt | | jt_| j
t_| j j
d | j j | jj
d | jj d S )N
r )r r r r getvaluer endswithr write STDOUT_LINEr STDERR_LINEr seektruncater )r outputerrorr r r
r+ S s$
zTestResult._restoreStdoutc C s d S )Nr )r r r r
stopTestRunh s zTestResult.stopTestRunc C s" | j j|| j||f d| _d S )NT)r append_exc_info_to_stringr )r r& errr r r
addErrorn s zTestResult.addErrorc C s" | j j|| j||f d| _d S )NT)r r8 r9 r )r r& r: r r r
addFailurev s zTestResult.addFailurec C sZ |d k rVt | ddr | j t|d |jr4| j}n| j}|j|| j||f d| _d S )Nr Fr T) r r
issubclassfailureExceptionr r r8 r9 r )r r& Zsubtestr: r r r r
addSubTest} s zTestResult.addSubTestc C s d S )Nr )r r& r r r
addSuccess s zTestResult.addSuccessc C s | j j||f d S )N)r r8 )r r& reasonr r r
addSkip s zTestResult.addSkipc C s | j j|| j||f d S )N)r r8 r9 )r r& r: r r r
addExpectedFailure s z TestResult.addExpectedFailurec C s | j j| d S )N)r r8 )r r& r r r
addUnexpectedSuccess s z TestResult.addUnexpectedSuccessc C s> t | jt | j ko dkn o<t| d p<t | jdkS )Nr r )lenr r hasattrr )r r r r
wasSuccessful s $ zTestResult.wasSuccessfulc C s
d| _ d S )NT)r )r r r r
r s zTestResult.stopc C s |\}}}x|r"| j |r"|j}q W ||jkr:| j|}nd }tj||||| jd}t|j }| j rt
j j } t
j
j }
| r| jds| d7 } |jt| |
r|
jds|
d7 }
|jt|
dj|S )N)limitcapture_localsr- )_is_relevant_tb_leveltb_nextr> _count_relevant_tb_levels tracebackTracebackExceptionr listformatr r r r. r r/ r8 r1 r2 join) r r: r& exctypevaluetblengthZtb_eZmsgLinesr5 r6 r r r
r9 s*
z TestResult._exc_info_to_stringc C s d|j jkS )N
__unittest)tb_frame f_globals)r rU r r r
rK s z TestResult._is_relevant_tb_levelc C s, d}x"|r&| j | r&|d7 }|j}qW |S )Nr r )rK rL )r rU rV r r r
rM s
z$TestResult._count_relevant_tb_levelsc C s& dt j| j| jt| jt| jf S )Nz!<%s run=%i errors=%i failures=%i>)r Zstrclass __class__r rE r r )r r r r
__repr__ s zTestResult.__repr__)NNN) __name__
__module__ __qualname__Z_previousTestClassZ_testRunEnteredZ_moduleSetUpFailedr# r$ r' r% r* r, r+ r7 r r; r< r? r@ rB rC rD rG r r9 rK rM r[ r r r r
r s0
r )
r( r rN rJ r functoolsr rW r r1 r2 objectr r r r r
<module> s