Your IP : 216.73.216.218
a
ؒd5d � @ 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 d dl m Z d dl m!Z! d dl"m#Z#m$Z$m%Z%m&Z& d dl'm(Z( dZ)G d d
� d
�Z*d
d d �Z+dS )� N)� _parse_args)� findtests�runtest�get_abs_module� is_failed�STDTESTS�NOTTESTS�PROGRESS_MIN_TIME�Passed�Failed�
EnvChanged�Skipped�ResourceDenied� Interrupted�
ChildError� DidNotRun)� setup_tests)�setup_pgo_tests)�removepy�count�format_duration� printlist)�supportg ^@c @ s� e Zd ZdZdd� Zdd� Zd7dd�Zd8d
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-� Zd.d/� Zd9d1d2�Zd3d4� Zd5d6� Z d0S ):�Regrtestal Execute a test suite.
This also parses command-line options and modifies its behavior
accordingly.
tests -- a list of strings containing test names (optional)
testdir -- the directory in which to look for tests (optional)
Users other than the Python test suite will certainly want to
specify testdir; if it's omitted, the directory containing the
Python test suite is searched for.
If the tests argument is omitted, the tests listed on the
command-line will be used. If that's empty, too, then all *.py
files beginning with test_ will be used.
The other default arguments (verbose, quiet, exclude,
single, randomize, findleaks, use_resources, trace, coverdir,
print_slow, and random_seed) allow programmers calling main()
directly to set the values that would normally be set by flags
on the command line.
c C s� d | _ g | _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�
need_rerun�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� r8 �+/usr/lib/python3.9/test/libregrtest/main.py�__init__6 s0
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 r8 r8 r9 � get_executeda s ���zRegrtest.get_executedFc C s� |j }t|ttf�s*|s*| j�|j|f� t|t�rB| j�|� n�t|t �rf| j
�|� | j �|� n�t|t �r~| j
�|� nvt|t
�r�| j�|� n^t|t�r�|s�| j�|� | j�|� n6t|t�r�| j�|� n t|t�r�d| _n td| ��|�rt|ttf��s| j�|� |j}|�r~dd lm m} |D ]D}z| j�|� |�� W n& |j �yx t |t j d� � Y n0 �q8d S )NTzinvalid test result: %rr ��file)!�name�
isinstancer r r) �appendZ duration_secr
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__)r7 �resultr& � test_namerE �ET�er8 r8 r9 �accumulate_resultf s@
zRegrtest.accumulate_resultr c C st | }| � � }|d ur(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�intrI )r7 �line�emptyZload_avg� test_timer8 r8 r9 �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# �pgorZ )r7 �
test_index�textrW Zfailsr8 r8 r9 �display_progress� s zRegrtest.display_progressc
C s� t tjdd � fi |��}|jr,g t_| _|j}|d ur`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? rS � )r rJ �argv�xmlpathr Zjunit_xml_listr2 � worker_args�test.libregrtest.runtest_mprc r5 r �args�
huntrleaksrI �stderr�exit�tempdir�os�path�
expanduserr )
r7 �kwargsr rh rc rM �warmup� repetitions�_�msgr8 r8 r9 �
parse_args� s&
zRegrtest.parse_argsc C s� || _ | jjrxtj�| jd�| _zDt| jd��$}|� � �
� }|g| _ W d � n1 sX0 Y W n t yv Y n0 | jj �rg | _ t
�d�}ttj�tj| jj ���T}|D ]>}|�dd�d }|�
� }|�|�}|d ur�| j �|�� � q�W d � n1 �s0 Y t| j � | jj�r.t| j� td d � }t�� }| jj�r�| jj D ]$} | |v �rl|� | � |� | � �qTg | j_ | jj �r�t | jj t!� t"� �}
nt | jj ||�}
| jj �s�| j �p�| jj �p�|
| _#n| j | _#| jj�r0| j#d d� | _#z"|
�$| j#d �} |
| d | _%W n t&�y. Y n0 | jj'�r�z | j#d | j#�$| jj'��= W n* t(�y� t)d| jj' t*j+d� Y n0 | jj,�r�| jj-d u �r�t.�/d �| j_-t.�0| jj-� t.�1| j#� d S )
NZ
pynexttest�rz\btest_[a-zA-Z0-9_]+\b�#r r z1Couldn't find starting test (%s), using all testsr> i��� )2r r �singlero rp �joinr4 r1 �open�read�strip�OSErrorZfromfile�re�compiler �SAVEDCWD�split�searchrB �groupr r_ r r r �copyZexcluderj rD �addZtestdirr �listr< r �indexr0 �
IndexError�startrC rI rJ rl � randomize� random_seed�randomZ randrangeZseedZshuffle) r7 r �fpZ next_testZregexrW �matchZstdtestsZnottests�argZalltests�posr8 r8 r9 �
find_tests� sr *
2
��
zRegrtest.find_testsc C s | j D ] }t|� qd S r; )r rI )r7 r@ r8 r8 r9 �
list_tests s
zRegrtest.list_testsc C sX |D ]N}t |tjj�rqt |tj�r0| �|� qt |tj�rt�|�rt |�
� � qd S r; ) rA �unittest�loaderZ _FailedTestZ TestSuite� _list_casesZTestCaser Z
match_testrI �id)r7 �suite�testr8 r8 r9 r� s
zRegrtest._list_casesc C s� dt _t �| jj| jj� | jD ]L}t| j|�}ztj �
|�}| � |� W q tj yj | j
�|� Y q 0 q | j
r�ttjd� ttt| j
�d�dtjd� t| j
tjd� d S )NFr> r� �skipped:)r �verboseZset_match_testsr � match_testsZ ignore_testsr r r� ZdefaultTestLoaderZloadTestsFromNamer� ZSkipTestr! rB rI rJ rl r r^ r )r7 rM Zabstestr� r8 r8 r9 �
list_cases s
zRegrtest.list_casesc
C s� d| j _d| j _d| j _| �� | _| �� | �d� t| j�}| j� � |D �] }|j
}| j � |� |j
plg }|jpvg }dd� |D �}dd� |D �}d| j _| j j}|s�|�r
| j jd u r�g | j _| j j�|� | j j�|� dd�| j j� } | �d |� d
| � d �� n| �d |� d �� t| j |�}|| j _| j|dd
� t|t�rL �qPqL| j�rxttt| j�d�d� t| j� | �� d S )NTFz'Re-running failed tests in verbose modec S s g | ]^}}|� d �d �qS �rQ r �r� ��.0Ztest_full_nameru r8 r8 r9 �
<listcomp>A � z/Regrtest.rerun_failed_tests.<locals>.<listcomp>c S s g | ]^}}|� d �d �qS r� r� r� r8 r8 r9 r� B r� z
matching: �, z Re-running z in verbose mode (�)z in verbose mode)r& r� z
failed again:) r r� �failfastZverbose3�get_tests_resultr' rZ r� r% �clearr@ r& rB �errors�failuresr� �extendr{ r rP rA r r rI r r^ r �display_result)
r7 Z
rerun_listrL rM r� r� Z error_namesZ
failure_namesZorig_match_testsZmatchingr8 r8 r9 �rerun_failed_tests0 sD
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 � | jd d
� D ] \}}td|t|�f � q�| j
�r>t� ttt | j
�d�d� t
| j
� | j�rpt� 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�rt� ttt | j�d�d� t
| j� d S )Nz== Tests result: %s ==z(Test suite interrupted by signal SIGINT.r� zomitted:r ZAllrQ )�endzOK.T)�reversez10 slowest tests:�
z- %s: %szfailed:z%{} altered the execution environment:r� z%s:z re-run testz
run no tests:)r r_ rI r� r( r< r r= r r^ r r r] r r! Z
print_slowr) �sortr r# �formatr& r$ )r7 ZomittedrY r� r8 r8 r9 r� \ sd �� �
�
zRegrtest.display_resultc C s� | j jr dd l}|jddd�| _tj�� }d}| j jrJ|dt| j j� 7 }| � |� d }t
| j d�D �]\}}t �
� }|}|r�d||f }| �||� | jr�d } tt� �}
| jj| t� |
d
� |
d } nt| j |�} | �| � t| t�r� �q|t| �}t �
� | } | tk�r d |t| �f }nt| t��r.d }tj�� D ]&}
|
|v�r8|
�d
��r8t� |
� �q8| j j rdt | | j �rd �q|qd|�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�localsrL z%s in %sztest.) r r� ZTracer* rJ �modules�keysZtimeoutr rZ � enumerater r+ r, rb �dictr� Zrunctxr� r rP rA r �strr r
�
startswithr Zunloadr� r rI )r7 r� Z save_modulesrv Z
previous_testr` rM r- ra �cmdr rL rY �moduler8 r8 r9 �run_tests_sequential� sJ
z Regrtest.run_tests_sequentialc c s6 |D ]*}|V | j r d S | jjr| jr d S qq d S r; )r r �fail_env_changedr# )r7 r rM r8 r8 r9 �
_test_forever� s zRegrtest._test_foreverc C sx t dt�� gtj�� �R � t dtjdd�dtj � t dt�� � t� � }|rZt d|� t dt
� d�t� � f � d S ) Nz==T)Zaliasedz %s-endianz== cwd:z
== CPU count:z == encodings: locale=%s, FS=%sF)
rI �platformZpython_implementationrJ �versionr� � byteorderro �getcwd� cpu_count�localeZgetpreferredencoding�getfilesystemencoding)r7 r� r8 r8 r9 �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 RUNZ INTERRUPTED�SUCCESSr� z
%s then %s) r rB r r� r# �anyr r! r( r{ r' )r7 rL r8 r8 r9 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 j�r�d d
lm} t
jd k�rP| jd u �rPd d l m } z |� | _ W n4 t �yN } zt d
|� �� W Y d }~n
d }~0 0 z(|| � W | j d u�r�| j � � d | _ n | j d u�r�| j � � d | _ 0 n| �!� d S )N� zhWARNING: Running tests with --huntrleaks/-R and less than 3 warmup repetitions can give false positives!Trd zUsing random seedr z/{}r r )�run_tests_multiprocessZwin32)�WindowsLoadTrackerz%Failed to create WindowsLoadTracker: )"r �headerr_ r] rz r rj r� rk rI rJ �stdoutr� r� Zforeverr� r� r r. r/ �iterr� r^ Zuse_mpri r� r� r5 Ztest.libregrtest.win_utilsr� r3 �PermissionError�closer� )r7 rs rt ru rv r� r� �errorr8 r8 r9 � run_tests� sV �����
$
�
zRegrtest.run_testsc C s� | j rV| jrJt| j d�� }|�| jd � W d � qV1 s>0 Y n t�| j � | jrz| 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 r| �writero �unlinkr* ZresultsZ
write_resultsr r� rI r+ r, r- r r� Zrunleaks�system�getpid)r7 r� rx Zdurationr8 r8 r9 �finalize* s 0
�zRegrtest.finalizec
C s | j js| jsd S dd lm m} |�d�}dddd�}| jD ]L}|�|� |D ]8}z || t|� |d��7 < W qR t
y� Y qR0 qRq@|� � D ]\}}|� |t
|�� q�tj�tj| j j�}t|d��*}|�|�D ]} |�| � q�W d � n1 s�0 Y d S )Nr Z
testsuites)r r� r� �wb)r rg r2 rF rG rH ZElementrB rV �getrC �itemsr< r� ro rp r{ r r� r| Z tostringlistr� )
r7 rN �rootZtotalsr� �k�vrg �f�sr8 r8 r9 �save_xml_result? s$
zRegrtest.save_xml_resultc C st | j jr| j j| _| js`t�� rVt�d�| _| jd u rBt�d�| _tj�| jd�| _n
t �
� | _tj� | j�| _d S )NZ abs_builddirZsrcdirZbuild) r rn r4 � sysconfigZis_python_buildZget_config_varro rp r{ �tempfileZ
gettempdir�abspathr6 r8 r8 r9 � set_temp_dirX s
zRegrtest.set_temp_dirc C sV t j| jdd� t �� }| jd ur.d�|�}n
d�|�}|tj7 }t j� | j|�}|S )NT)�exist_okztest_python_worker_{}ztest_python_{})
ro �makedirsr4 r� r5 r� r Z FS_NONASCIIrp r{ )r7 �pid�test_cwdr8 r8 r9 �create_temp_dirn s
zRegrtest.create_temp_dirc C sz dd l }tj�|�| j�d�}td| j � |� |�D ]>}tj�|�r^td| � t� |� q6td| � t�
|� q6d S )Nr z
test_python_*zCleanup %s directoryzRemove directory: %szRemove file: %s) �globro rp r{ �escaper4 rI �isdirr Zrmtreer� )r7 r� rp r@ r8 r8 r9 �cleanup} s zRegrtest.cleanupNc
K s� | � |� | �� | jjr,| �� t�d� | �� }zFtj|dd��$ || j_ | �
||� W d � n1 sn0 Y W n> t y� } z&t j
tdd� t�|j� W Y d }~n
d }~0 0 d S )Nr T)r] )rm )rw r� r r� rJ rm r� r Ztemp_cwdrn �_main�
SystemExit� faulthandlerZdump_traceback_later� EXIT_TIMEOUT�code)r7 r rr r� �excr8 r8 r9 �main� s
.z
Regrtest.mainc C s. | j d ur| j �� S ttd�r*t�� d S d S )NrT r )r3 rT �hasattrro r6 r8 r8 r9 rT � s
zRegrtest.getloadavgc C s | j d ur$ddlm} || j| j � | jjr4td� | jjt_| jj t_
t | j� | � |� | jj
rv| �
� t�d� | jjr�| �� t�d� | �� | �� | jjr�| jr�| �� | �� | �� | jr�t�d� | jr�t�d� | jjr�| jr�t�d� t�d� d S )Nr )�run_tests_workerz Press any key to continue...re � r� )r5 ri r� r �wait�inputr_ r ZPGOZ pgo_extendedZ PGO_EXTENDEDr r� r� rJ rm r� r� r� Zverbose2r r� r� r� r( r� r# )r7 r rr r� r8 r8 r9 r� � s8
zRegrtest._main)F)r )N) �__name__�
__module__� __qualname__�__doc__r: r= rP rZ rb rw r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rT r� r8 r8 r8 r9 r s4 +
'
I
,;7
1
r c K s t � jf d| i|�� dS )zRun the Python suite.r N)r r� )r rr r8 r8 r9 r� � s r� )N),rU r� r� ro r� r� r� rJ r� r� r+ r� Ztest.libregrtest.cmdliner Ztest.libregrtest.runtestr r r r r r r r
r r r
r r r r Ztest.libregrtest.setupr Ztest.libregrtest.pgor Ztest.libregrtest.utilsr r r r r� r r� r r� r8 r8 r8 r9 �<module> s2 D A