File: //usr/local/lib/python3.7/test/libregrtest/__pycache__/main.cpython-37.opt-2.pyc
B
��g]_ � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z d dl m!Z! d dl"m#Z#m$Z$m%Z%m&Z& d dl'm(Z( dZ)G dd � d �Z*d d
d �Z+dS )
� N)� _parse_args)� findtests�runtest�get_abs_module�STDTESTS�NOTTESTS�PASSED�FAILED� ENV_CHANGED�SKIPPED�RESOURCE_DENIED� INTERRUPTED� CHILD_ERROR�TEST_DID_NOT_RUN�TIMEOUT�PROGRESS_MIN_TIME�format_test_result� is_failed)� setup_tests)�removepy�count�format_duration� printlist)�supportg ^@c @ s� e Zd Zdd� Zdd� Zd6dd�Zd7d d
�Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd8d0d1�Zd2d3� Zd4d5� Zd/S )9�Regrtestc C s� d | _ g | _g | _g | _g | _g | _g | _g | _g | _g | _ d | _
d| _ g | _ d | _
t�� | _d| _d| _d | _d | _d | _d | _d | _d | _d S )NF� � )�ns�tests�selected�good�bad�skipped�resource_denieds�environment_changed� run_no_tests�rerun� first_result� interrupted�
test_times�tracer�time� monotonic�
start_time�
test_count�test_count_width�next_single_test�next_single_filename�
testsuite_xml�win_load_tracker�tmp_dir�worker_test_name)�self� r7 �1/usr/local/lib/python3.7/test/libregrtest/main.py�__init__5 s.
zRegrtest.__init__c C s< t | j�t | j�B t | j�B t | j�B t | j�B t | j�B S )N)�setr r! r" r# r$ r% )r6 r7 r7 r8 � get_executed_ s zRegrtest.get_executedFc C s� |j }|j}|ttfkr.|s.| j�|j|f� |tkrD| j�|� n�|t tfkrb|s�| j
�|� n�|t krx| j �|� n�|t
kr�| j�|� nj|tkr�| j�|� | j�|� nH|tkr�| j�|� n2|tkr�d| _n"|tkr�| j
�|� n td| ��|�r|t tthk�r| j
�|� |j}|�r�dd lm m} xN|D ]F}y| j�|� |�� W n( |j k
�r� t |t j d� � Y nX �q>W d S )NTzinvalid test result: %rr )�file)!� test_name�resultr r
r) �append� test_timer r r r! r
r$ r r" r r# r r% r( r �
ValueError�remove�xml_data�xml.etree.ElementTree�etree� ElementTreer2 Z
fromstringZ
ParseError�print�sys�
__stderr__)r6 r> r& r= �okrC �ET�er7 r7 r8 �accumulate_resultd sD
zRegrtest.accumulate_resultr c C st | }| � � }|d k r(d|d�d|� �}t�� | j }tjt|�d�}|� d|� �}|rd|d d� }t|dd� d S )Nz
load avg: z.2f� )Zseconds���T)�flush)�
getloadavgr+ r, r- �datetimeZ timedelta�intrG )r6 �line�emptyZload_avgr@ r7 r7 r8 �log� s z Regrtest.logc C sh | j jr d S || j� �| j� �}t| j�t| j� }|rN| j jsN|� d|� �}| �d|� d|� �� d S )N�/�[z] ) r �quietr/ r. �lenr! r$ �pgorV )r6 �
test_index�textrT Zfailsr7 r7 r8 �display_progress� s zRegrtest.display_progressc
C s� t tjdd � f|�}|jr(g t_| _|j}|d k r\ddlm } ||j�\}}||_|| _
t |j � |j
r�|j
\}}}|dk s�|dk r�d} t| tjdd� t�d� |jr�tj�|j�|_|| _d S )Nr r )�parse_worker_argszwInvalid values for the --huntrleaks/-R parameters. The number of warmups and repetitions must be at least 1 each (1:1).T)r< rP � )r rH �argv�xmlpathr Zjunit_xml_listr2 � worker_args�test.libregrtest.runtest_mpr_ r5 r �args�
huntrleaksrG �stderr�exit�tempdir�os�path�
expanduserr )
r6 �kwargsr rc r_ r= �warmup� repetitions�_�msgr7 r7 r8 �
parse_args� s&
zRegrtest.parse_argsc C s� || _ | jjrftj�| jd�| _y0t| jd��}|� � �
� }|g| _ W d Q R X W n t k
rd Y nX | jj r�g | _ t
�d�}ttj�tj| jj ���N}xF|D ]>}|�dd�d }|�
� }|�|�}|d k r�| j �|�� � q�W W d Q R X t| j � td d � }t�� }| jj�rLx0| jjD ]$} | |k�r4|�| � |� | � �q W g | j_| jj �rlt | jj t � t � �}
nt | jj ||�}
| jj �s�| j �p�| jj�p�|
| _!n| j | _!| jj�r�| j!d d� | _!y"|
�"| j!d �} |
| d | _#W n t$k
�r� Y nX | jj%�rPy | j!d | j!�"| jj%��= W n, t&k
�rN t'd| jj% t(j)d� Y nX | jj*�r�| jj+d k�rvt,�-d �| j_+t,�.| jj+� t,�/| j!� d S )
NZ
pynexttest�rz\btest_[a-zA-Z0-9_]+\b�#r r z1Couldn't find starting test (%s), using all tests)r< i��� )0r r �singlerj rk �joinr4 r1 �open�read�strip�OSErrorZfromfile�re�compiler �SAVEDCWD�split�searchr? �groupr r r �copyZexcludere rB �addZtestdirr �listr: r �indexr0 �
IndexError�startrA rG rH rg � randomize� random_seed�randomZ randrangeZseedZshuffle) r6 r �fpZ next_testZregexrT �matchZstdtestsZnottests�argZalltests�posr7 r7 r8 �
find_tests� sh
zRegrtest.find_testsc C s x| j D ] }t|� qW d S )N)r rG )r6 �namer7 r7 r8 �
list_tests
s zRegrtest.list_testsc C s\ xV|D ]N}t |tjj�rqt |tj�r2| �|� qt |tj�rt�|�rt |�
� � qW d S )N) �
isinstance�unittest�loaderZ _FailedTestZ TestSuite� _list_casesZTestCaser Z
match_testrG �id)r6 �suite�testr7 r7 r8 r� s
zRegrtest._list_casesc C s� dt _t �| jj| jj� xX| jD ]N}t| j|�}ytj �
|�}| � |� W q" tj k
rn | j
�|� Y q"X q"W | j
r�ttjd� ttt| j
�d�dtjd� t| j
tjd� d S )NF)r< r� zskipped:)r �verboseZset_match_testsr Z match_testsZ ignore_testsr r r� ZdefaultTestLoaderZloadTestsFromNamer� ZSkipTestr" r? rG rH rg r rZ r )r6 r= Zabstestr� r7 r7 r8 �
list_cases s zRegrtest.list_casesc C s� d| j _d| j _d| j _| �� | _| �� | �d� | jd d � | _xN| jD ]D}| �d|� d�� d| j _t | j |�}| j
|dd� |j t krLP qLW | jr�t
tt| j�d�d� t| j� | �� d S ) NTFz'Re-running failed tests in verbose modez Re-running z in verbose mode)r& r� z
failed again:)r r� �failfastZverbose3�get_tests_resultr' rV r! r& r rM r> r
rG r rZ r �display_result)r6 r= r> r7 r7 r8 �rerun_failed_tests, s$
zRegrtest.rerun_failed_testsc C s | j jr d S t� td| �� � | jr0td� t| j�| �� }|rht� ttt |�d�d� t
|� | j r�| j j s�t� | j
s�| js�| js�t | j �dkr�tddd� ttt | j �d�d � | j j�r| jjd
d � t� td � x.| jd d
� D ] \}}td|t|�f � q�W | j
�rBt� ttt | j
�d�d� t
| j
� | j�rtt� td�tt | j�d��� t
| j� | j�r�| j j �s�t� ttt | j�d�d� t
| j� | j�r�t� tdtt | j�d� � t
| j� | j�r
t� ttt | j�d�d� t
| j� d S )Nz== Tests result: %s ==z(Test suite interrupted by signal SIGINT.r� zomitted:r ZAllrN )�endzOK.T)�reversez10 slowest tests:�
z- %s: %szfailed:z%{} altered the execution environment:zskipped:z%s:z re-run testz
run no tests:)r r[ rG r� r( r: r r; r rZ r r rY r! r" Z
print_slowr) �sortr r$ �formatr& r% )r6 Zomittedr@ r� r7 r7 r8 r� F s\
zRegrtest.display_resultc C s� | j jr dd l}|jddd�| _tj�� }d}| j jrJ|dt| j j� 7 }| � |� d }�x"t
| j d�D �]\}}t �
� }|}|r�d||f }| �||� | jr�d } tt� �}
| jj| t� |
d
� |
d } nt| j |�} | �| � | jtkr�P t| �}t �
� | } | tk�r d |t| �f }n| jtk�r.d }x4tj�� D ]&}
|
|k�r:|
�d
��r:t� |
� �q:W | j j rht | | j �rhP qhW |�r�t |� d S )Nr FT)�tracer zRun tests sequentiallyz (timeout: %s)r z%s -- %szDresult = runtest(self.ns, test_name); self.accumulate_result(result))�globals�localsr> z%s in %sztest.) r r� ZTracer* rH �modules�keysZtimeoutr rV � enumerater r+ r, r^ �dictr� Zrunctxr� r rM r> r
r r r �
startswithr Zunloadr� r rG )r6 r� Z save_modulesrq Z
previous_testr\ r= r- r] �cmdr r> r@ �moduler7 r7 r8 �run_tests_sequential� sJ
z Regrtest.run_tests_sequentialc c s: x4x.|D ]&}|V | j r d S | jjr| jrd S qW qW d S )N)r! r �fail_env_changedr$ )r6 r r= r7 r7 r8 �
_test_forever� s
zRegrtest._test_foreverc C sv t dt�� ftj�� �� t dtjdd�dtj � t dt�� � t� � }|rXt d|� t dt
� d�t� � f � d S ) Nz==T)Zaliasedz %s-endianz== cwd:z
== CPU count:z == encodings: locale=%s, FS=%sF)
rG �platformZpython_implementationrH �versionr~ � byteorderrj �getcwd� cpu_count�locale�getpreferredencoding�getfilesystemencoding)r6 r� r7 r7 r8 �display_header� s
zRegrtest.display_headerc C s� g }| j r|�d� n@| jjr0| jr0|�d� n&t| j| j | j| j| jf�sV|�d� | jrf|�d� |st|�d� d� |�}| j
r�d| j
|f }|S )N�FAILUREz ENV CHANGEDz NO TEST RUNr
�SUCCESSz, z
%s then %s) r! r? r r� r$ �anyr r" r( rv r' )r6 r> r7 r7 r8 r� � s
zRegrtest.get_tests_resultc C s� | j js.| j js6| j js6| j js6| js6| j js6| �� | j jrh| j j\}}}|dk rhd}t |t
j dd� | j j r~t d| j j
� | j jr�| �t| j��| _d| _d| _n.t| j�| _d�t| j��| _t| j�d | _| j jr�d d
lm} || � n| �� d S ) N� zhWARNING: Running tests with --huntrleaks/-R and less than 3 warmup repetitions can give false positives!T)r< rP zUsing random seedr z/{}r r )�run_tests_multiprocess)r �headerr[ rY ru r re r� rf rG rH �stdoutr� r� Zforeverr� r� r r. r/ �iterr� rZ Zuse_mprd r� r� )r6 rn ro rp rq r� r7 r7 r8 � run_tests� s,
zRegrtest.run_testsc C s� | j rB| jr6t| j d��}|�| jd � W d Q R X n t�| j � | jrf| j�� }|jdd| j j
d� t � t �
� | j }t dt|� � t d| �� � | j jr�t�dt�� � d S )N�w�
T)Z show_missingZsummary�coverdirzTotal duration: %szTests result: %szleaks %d)r1 r0 rw �writerj �unlinkr* �resultsZ
write_resultsr r� rG r+ r, r- r r� Zrunleaks�system�getpid)r6 r� rs Zdurationr7 r7 r8 �finalize s
zRegrtest.finalizec
C s
| j js| jsd S dd lm m} |�d�}dddd�}x\| jD ]R}|�|� xB|D ]:}y || t|� |d��7 < W qV t
k
r� Y qVX qVW qBW x$|� � D ]\}}|� |t
|�� q�W tj�tj| j j�}t|d��$}x |�|�D ]} |�| � q�W W d Q R X d S )Nr Z
testsuites)r �errorsZfailures�wb)r rb r2 rD rE rF ZElementr? rS �getrA �itemsr: �strrj rk rv r r} rw Z tostringlistr� )
r6 rK �rootZtotalsr� �k�vrb �f�sr7 r7 r8 �save_xml_result s$
zRegrtest.save_xml_resultc C st | j jr| j j| _| js`t�� rVt�d�| _| jd krBt�d�| _tj�| jd�| _n
t �
� | _tj� | j�| _d S )NZ abs_builddirZsrcdirZbuild) r ri r4 � sysconfigZis_python_buildZget_config_varrj rk rv �tempfileZ
gettempdir�abspath)r6 r7 r7 r8 � set_temp_dir1 s
zRegrtest.set_temp_dirc C sL t j| jdd� t �� }| jd k r.d�|�}n
d�|�}t j�| j|�}|S )NT)�exist_okztest_python_worker_{}ztest_python_{})rj �makedirsr4 r� r5 r� rk rv )r6 �pid�test_cwdr7 r7 r8 �create_temp_dirG s
zRegrtest.create_temp_dirc C sx dd l }tj�| jd�}td| j � xL|� |�D ]>}tj�|�rZtd| � t�|� q2td| � t� |� q2W d S )Nr z
test_python_*zCleanup %s directoryzRemove directory: %szRemove file: %s)
�globrj rk rv r4 rG �isdirr Zrmtreer� )r6 r� rk r� r7 r7 r8 �cleanupU s zRegrtest.cleanupNc
K s� | � |� | �� | jjr,| �� t�d� | �� }y2tj|dd�� || j_ | �
||� W d Q R X W n< t k
r� } z t j
tdd� t�|j� W d d }~X Y nX d S )Nr T)rY )rh )rr r� r r� rH rh r� r Ztemp_cwdri �_main�
SystemExit� faulthandlerZdump_traceback_later� EXIT_TIMEOUT�code)r6 r rm r� �excr7 r7 r8 �mainb s
z
Regrtest.mainc C s. | j d k r| j �� S ttd�r*t�� d S d S )NrQ r )r3 rQ �hasattrrj )r6 r7 r7 r8 rQ � s
zRegrtest.getloadavgc
C s� | j d k r$ddlm} || j| j � | jjr4td� | jjt_t | j� | �
|� | jj rl| � � t �
d� | jjr�| �� t �
d� t jdkr�| j d kr�ddlm} y |� | _W n0 tk
r� } ztd|� �� W d d }~X Y nX z.| �� | �� | jj�r| j�r| �� W d | jd k �r0| j�� d | _X | �� | � � | j�rTt �
d� | j �rft �
d� | jj �r�| j �r�t �
d � t �
d� d S )
Nr )�run_tests_workerz Press any key to continue...�win32)�WindowsLoadTrackerz%Failed to create WindowsLoadTracker: r` � r� ) r5 rd r� r �wait�inputr[ r ZPGOr r� r� rH rh r� r� Ztest.libregrtest.win_utilsr� r3 �FileNotFoundErrorrG r� r� Zverbose2r! r� �closer� r� r( r� r$ )r6 r rm r� r� �errorr7 r7 r8 r� � sJ
zRegrtest._main)F)r )N) �__name__�
__module__� __qualname__r9 r; rM rV r^ rr r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rQ r� r7 r7 r7 r8 r s2 *
)
E
;7
r c K s t � jf d| i|�� d S )Nr )r r� )r rm r7 r7 r8 r� � s r� )N),rR r� r� rj r� r� r{ rH r� r� r+ r� Ztest.libregrtest.cmdliner Ztest.libregrtest.runtestr r r r r r r r
r r r
r r r r r r Ztest.libregrtest.setupr Ztest.libregrtest.utilsr r r r r� r r� r r� r7 r7 r7 r8 �<module> s0 L *