3
\
� @ s6 d Z ddlZddlZddlZddlZddlZddlZddljj Z
ddl m Z G dd� dej �Z
G dd� d�Zdd d
�Zdd d �Zed
k�r2G dd� dej�Zej� Zejeje�� ej� Zeedd� ejD ���Z e ej �Z e j e�Z e!dej"� � e!ddd� x(e
j#e j$� �D ]Z%e!e%j&� dd� �qW e!� dS )z=Test runner and result class for the regression test suite.
� N)�datetimec s� e Zd Zdd d Zdd d Z� fdd�Zedd� �Z� fd d
�Zd$d d
�Z dd� Z
edd� �Z � fdd�Z � fdd�Z
� fdd�Z� fdd�Z� fdd�Z� fd d �Zd d � Zd d!� Zd"d#� Z� ZS )%�RegressionTestResult�=�F �
�-c s\ t � j||dd� d| _tjd�| _| jjdtj� j d�� d | _
d | _ g | _ t
|�| _d S )Nr )�stream� descriptions� verbosityTZ testsuite�start� )�super�__init__�buffer�ETZElement� _RegressionTestResult__suite�setr ZutcnowZ isoformat�_RegressionTestResult__e�!_RegressionTestResult__start_timeZ _RegressionTestResult__results�bool� _RegressionTestResult__verbose)�selfr r r
)� __class__� �//usr/lib64/python3.6/test/support/testresult.pyr s z RegressionTestResult.__init__c C sL y
|j }W n tk
r" t|�S X y|� S tk
rB t|�S X t|�S )N)�id�AttributeError�str� TypeError�repr)�cls�testZtest_idr r r Z__getId s
z RegressionTestResult.__getIdc sV t � j|� tj| jd� | _}tj� | _| j rR| j
j | j |�� d�� | j
j
� d S )NZtestcasez ... )r
� startTestr �
SubElementr r �time� perf_counterr r r �write�getDescription�flush)r r! �e)r r r r" + s
z RegressionTestResult.startTestFc K sP | j }d | _ |d krd S |jd|jd| j|��� |jd|jdd�� |jd|jdd�� | jrz|jdtj� | j d�� |r�| jd k r�| jj� j � }|t
j |d�_ | j
d k r�| j
j� j � }|t
j |d �_ x�|j� D ]t\}}| s�| r�q�t
j ||�} t|d
��r>xD|j� D ],\}
} |
�r,| j|
t| �� n
t| �| _ �q W q�t|�| _ q�W d S ) N�nameZstatus�run�resultZ completedr$ z0.6fz
system-outz
system-err�items)r r �pop� _RegressionTestResult__getIdr r$ r% Z_stdout_buffer�getvalue�rstripr r# �textZ_stderr_bufferr- �hasattrr ) r r! Zcapture�argsr) �stdout�stderr�k�vZe2Zk2Zv2r r r � _add_result3 s4
z RegressionTestResult._add_resultc C s | j r| jj|� d�� d S )Nr )r r r& )r �cZwordr r r Z__writeS s z RegressionTestResult.__writec C sl t |t�r0|jdkr |j}q8|j� d|j� �}nt|�}tj||d �}tj|||�}|dj|�dj|�d�S )N�builtins�.� )�type�messager= )�
isinstancer> �
__module__�__name__r � traceback�format_exception�join)r Zerr_typeZ err_valueZerr_tb�typename�msg�tbr r r Z__makeErrorDictW s
z$RegressionTestResult.__makeErrorDictc s4 | j |d| j|� d� t� j||� | jdd� d S )NT)�error�E�ERROR)r9 �$_RegressionTestResult__makeErrorDictr
�addError� _RegressionTestResult__write)r r! �err)r r r rM j s z RegressionTestResult.addErrorc s4 | j |d| j|� d� t� j||� | jdd� d S )NT)�output�xzexpected failure)r9 rL r
�addExpectedFailurerN )r r! rO )r r r rR o s z'RegressionTestResult.addExpectedFailurec s4 | j |d| j|� d� t� j||� | jdd� d S )NT)Zfailure�F�FAIL)r9 rL r
�
addFailurerN )r r! rO )r r r rU t s z RegressionTestResult.addFailurec s2 | j ||d� t� j||� | jdd|��� d S )N)Zskipped�Szskipped )r9 r
�addSkiprN )r r! �reason)r r r rW y s z RegressionTestResult.addSkipc s& | j |� t� j|� | jdd� d S )Nr< �ok)r9 r
�
addSuccessrN )r r! )r r r rZ ~ s
z RegressionTestResult.addSuccessc s* | j |dd� t� j|� | jdd� d S )NZUNEXPECTED_SUCCESS)Zoutcome�uzunexpected success)r9 r
�addUnexpectedSuccessrN )r r! )r r r r\ � s z)RegressionTestResult.addUnexpectedSuccessc C s2 | j r| jjd� | jd| j� | jd| j� d S )Nr rK rT )r r r& �printErrorList�errors�failures)r r r r � printErrors� s z RegressionTestResult.printErrorsc C s` xZ|D ]R\}}| j j| j� | j j|� d| j|�� d�� | j j| j� | j jd| � qW d S )Nz: r z%s
)r r& �
separator1r' �
separator2)r Zflavorr^ r! rO r r r r] � s
z#RegressionTestResult.printErrorListc C sH | j }|jdt| j�� |jdtt| j��� |jdtt| j��� |S )NZtestsr^ r_ )r r r ZtestsRun�lenr^ r_ )r r) r r r �get_xml_element� s
z$RegressionTestResult.get_xml_element)F)rB rA � __qualname__ra rb r � classmethodr/ r" r9 rN rL rM rR rU rW rZ r\ r` r] rd �
__classcell__r r )r r r s"
r c @ s e Zd Zddd�Zdd� ZdS )�QuietRegressionTestRunnerFc C s t |d d�| _|| j_d S )Nr )r r, r )r r r r r r r � s z"QuietRegressionTestRunner.__init__c C s || j � | j S )N)r, )r r! r r r r+ � s
z QuietRegressionTestRunner.runN)F)rB rA re r r+ r r r r rh � s
rh Fc C s& | rt jtjt|| d�S t jt|d�S )N)Z resultclassr r
)r )� functools�partial�unittestZTextTestRunnerr rh )r
r r r r �get_test_runner_class� s rl c C s t ||�| �S )N)rl )r r
Zcapture_outputr r r �get_test_runner� s rm �__main__c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestTestsc C s d S )Nr )r r r r � test_pass� s zTestTests.test_passc C s t jd� d S )Ng �?)r$ Zsleep)r r r r �test_pass_slow� s zTestTests.test_pass_slowc C s* t dtjd� t dtjd� | jd� d S )Nr5 )�filer6 zfailure message)�print�sysr5 r6 Zfail)r r r r � test_fail� s zTestTests.test_failc C s( t dtjd� t dtjd� td��d S )Nr5 )rr r6 z
error message)rs rt r5 r6 � RuntimeError)r r r r �
test_error� s zTestTests.test_errorN)rB rA re rp rq ru rw r r r r ro � s ro c c s | ]}|d kV qdS )z-vNr )�.0�ar r r � � s rz zOutput:zXML: r= )�end)F)F)'�__doc__ri �iort r$ rC rk Zxml.etree.ElementTreeZetreeZ ElementTreer r ZTextTestResultr rh rl rm rB ZTestCasero Z TestSuiteZsuiteZaddTestZ makeSuite�StringIOr �sum�argvZ
runner_clsr5 Zrunnerr+ r, rs r0 Z tostringlistrd �s�decoder r r r � s4