File: //usr/local/lib/python3.10/test/libregrtest/__pycache__/main.cpython-310.opt-2.pyc
o
�i�d � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z d dl m Z d dl m Z d dl!m"Z"m#Z#m$Z$m%Z% d dl&m'Z' d dl(m)Z) d Z*G d
d � d �Z+dd 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)�support)� os_helperg ^@c @ s� e Zd Z dd� 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 | _ 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� r9 �2/usr/local/lib/python3.10/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% r7 r9 r9 r: � get_executeda s ���zRegrtest.get_executedFc C sv |j }t|ttf�s|s| j�|j|f� t|t�r!| j�|� nYt|t �r3| j
�|� | j �|� nGt|t �r?| j
�|� n;t|t
�rK| j�|� n/t|t�r_|s^| j�|� | j�|� nt|t�rk| j�|� nt|t�rtd| _ntd| ��|r�t|ttf�s�| j�|� |j}|r�dd lm m} |D ] }z | j�|� |�� W q� |j y� t |t j d� � w d S d 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� ElementTreer3 Z
fromstringZ
ParseError�print�sys�
__stderr__)r8 �resultr' � test_namerF �ET�er9 r9 r: �accumulate_resultf sH
�
��zRegrtest.accumulate_resultr c C s� | }| � � }|d urd|d�d|� �}t�� | j }tt|�d�\}}t|d�\}}d|||f }|� d|� �}|rA|d d� }t|dd� d S ) Nz
load avg: z.2f� �< z %d:%02d:%02d���T)�flush)�
getloadavgr, r- r. �divmod�intrJ )r8 �line�emptyZload_avg� test_timeZminsZsecsZhoursr9 r9 r: �log� s z Regrtest.logc C sh | j jrd S || j� �| j� �}t| j�t| j� }|r'| j js'|� d|� �}| �d|� d|� �� d S )N�/�[z] ) r �quietr0 r/ �lenr! r$ �pgor\ )r8 �
test_index�textrY Zfailsr9 r9 r: �display_progress� s zRegrtest.display_progressc
C s� t tjdd � fi |��}|jrg t_| _|j}|d ur0ddlm } ||j�\}}||_|| _
t |j � |j
rU|j
\}}}|dk sF|dk rUd} 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@ rU � )r rK �argv�xmlpathr Zjunit_xml_listr3 � worker_args�test.libregrtest.runtest_mpre r6 r �args�
huntrleaksrJ �stderr�exit�tempdir�os�path�
expanduserr )
r8 �kwargsr rj re rN �warmup� repetitions�_�msgr9 r9 r: �
parse_args� s&
zRegrtest.parse_argsc C s� || _ | jjr<tj�| jd�| _z"t| jd��}|� � �
� }|g| _ W d � n1 s,w Y W n t y; Y nw | jj r�g | _ t
�d�}ttj�tj| jj ���*}|D ] }|�dd�d }|�
� }|�|�}|d urv| j �|�� � qWW d � n1 s�w Y t| j � | jjr�t| j� td d � }t�� }| jjr�| jj D ]} | |v r�|� | � |� | � q�g | j_ | jj r�t | jj t!� t"� �}
nt | jj ||�}
| jj s�| j p�| jj p�|
| _#n| j | _#| jj�r| j#d d� | _#z|
�$| j#d �} |
| d | _%W n
t&�y
Y nw | jj'�r7z| j#d | j#�$| jj'��= W n t(�y6 t)d| jj' t*j+d� Y nw | jj,�rY| jj-d u �rJt.�/d �| j_-t.�0| jj-� t.�1| j#� d S 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 �singlerq rr �joinr5 r2 �open�read�strip�OSError�fromfile�re�compiler �SAVEDCWD�split�searchrC �groupr ra r r r �copyZexcluderl rE �addZtestdirr �listr= r �indexr1 �
IndexError�startrD rJ rK rn � randomize� random_seed�random� randrange�seed�shuffle) r8 r �fpZ next_testZregexrY �matchZstdtestsZnottests�argZalltests�posr9 r9 r: �
find_tests� s�
�� �
�� �
�
�
��
�zRegrtest.find_testsc C s | j D ]}t|� qd S r< )r rJ )r8 rA r9 r9 r: �
list_tests s
�zRegrtest.list_testsc C sX |D ]'}t |tjj�r qt |tj�r| �|� qt |tj�r)t�|�r)t |�
� � qd S r< ) rB �unittest�loaderZ _FailedTestZ TestSuite� _list_casesZTestCaser Z
match_testrJ �id)r8 �suite�testr9 r9 r: r� s
��zRegrtest._list_casesc C s� dt _t �| jj| jj� | jD ]%}t| j|�}z
tj �
|�}| � |� W q tj y5 | j
�|� Y qw | j
rWttjd� ttt| j
�d�dtjd� t| j
tjd� d S d S )NFr? r� �skipped:)r �verboseZset_match_testsr � match_testsZ ignore_testsr r r� ZdefaultTestLoaderZloadTestsFromNamer� ZSkipTestr" rC rJ rK rn r r` r )r8 rN Zabstestr� r9 r9 r: �
list_cases" s
� �zRegrtest.list_casesc
C s| d| j _d| j _d| j _| �� | _| �� | �d� t| j�}| j� � |D ]~}|j
}| j � |� |j
p5g }|jp:g }dd� |D �}dd� |D �}d| j _| j j}|sU|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�r� nq&| jr�ttt| j�d�d� t| j� | �� d S )NTFz'Re-running failed tests in verbose modec S � g | ] ^}}|� d �d �qS �rR r �r� ��.0Ztest_full_namerw r9 r9 r: �
<listcomp>D � z/Regrtest.rerun_failed_tests.<locals>.<listcomp>c S r� r� r� r� r9 r9 r: r� E 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( r\ r� r& �clearrA r' rC �errors�failuresr� �extendr} r rQ rB r r! rJ r r` r �display_result)
r8 Z
rerun_listrM rN r� r� Z error_namesZ
failure_namesZorig_match_testsZmatchingr9 r9 r: �rerun_failed_tests3 sF
�
zRegrtest.rerun_failed_testsc C s | j jrd S t� td| �� � | jrtd� t| j�| �� }|r4t� ttt |�d�d� t
|� | j r_| j j s_t� | j
sT| jsT| jsTt | j �dkrTtddd� ttt | j �d�d � | j jr�| jjd
d � t� td � | jd d
� D ]\}}td|t|�f � qx| j
r�t� ttt | j
�d�d� t
| j
� | jr�t� td�tt | j�d��� t
| j� | jr�| j j s�t� ttt | j�d�d� t
| j� | jr�t� tdtt | j�d� � t
| j� | jr�t� ttt | j�d�d� t
| j� d S d S )Nz== Tests result: %s ==z(Test suite interrupted by signal SIGINT.r� zomitted:r ZAllrR )�endzOK.T)�reversez10 slowest tests:�
z- %s: %szfailed:z%{} altered the execution environment:r� z%s:z re-run testz
run no tests:)r ra rJ r� r) r= r r> r r` r r r_ r! r" Z
print_slowr* �sortr r$ �formatr' r% )r8 Zomittedr[ r� r9 r9 r: r� _ sd ��
�
�zRegrtest.display_resultc C s� | j jrdd l}|jddd�| _tj�� }d}| j jr%|dt| j j� 7 }| � |� d }t
| j d�D ]�\}}t �
� }|}|rDd||f }| �||� | jrcd } tt� �}
| jj| t� |
d
� |
d } n t| j |�} | �| � t| t�ru nAt| �}t �
� | } | tkr�d |t| �f }nt| t�r�d }tj�� D ]}
|
|vr�|
�d
�r�t� |
� q�| j j r�t | | j �r� nq2|r�t |� d S d S )Nr FT)�tracer zRun tests sequentiallyz (timeout: %s)r z%s -- %szDresult = runtest(self.ns, test_name); self.accumulate_result(result))�globals�localsrM z%s in %sztest.) r r� ZTracer+ rK �modules�keys�timeoutr r\ � enumerater r, r- rd �dictr� Zrunctxr� r rQ rB r �strr r
�
startswithr Zunloadr� r rJ )r8 r� Z save_modulesrx Z
previous_testrb rN r. rc �cmdr rM r[ �moduler9 r9 r: �run_tests_sequential� sP
�� �z Regrtest.run_tests_sequentialc c s6 � |D ]}|V | j r d S | jjr| jr d S qqr< )r! r �fail_env_changedr$ )r8 r rN r9 r9 r: �
_test_forever� s ���zRegrtest._test_foreverc C sx t dt�� gtj�� �R � t dtjdd�dtj � t dt�� � t� � }|r-t d|� t dt
� d�t� � f � d S ) Nz==T)Zaliasedz %s-endianz== cwd:z
== CPU count:z == encodings: locale=%s, FS=%sF)
rJ �platformZpython_implementationrK �versionr� � byteorderrq �getcwd� cpu_count�locale�getpreferredencoding�getfilesystemencoding)r8 r� r9 r9 r: �display_header� s �
�
�zRegrtest.display_headerc C s� g }| j r |�d� n | jjr| jr|�d� nt| j| j | j| j| jf�s+|�d� | jr3|�d� |s:|�d� d� |�}| j
rId| j
|f }|S )N�FAILUREz ENV CHANGEDz NO TEST RUNZ INTERRUPTED�SUCCESSr� z
%s then %s) r! rC r r� r$ �anyr r" r) r} r( )r8 rM r9 r9 r: r� � s" �
zRegrtest.get_tests_resultc
C s� | j js| j js| j js| j js| js| j js| �� | j jr4| j j\}}}|dk r4d}t |t
j dd� | j j r?t d| j j
� | j jrS| �t| j��| _d| _d| _nt| j�| _d�t| j��| _t| j�d | _| j jr�d d
lm} t
jd kr�| jd u r�d d l m } z|� | _ W n t y� } z
t d
|� �� W Y d }~nd }~ww z|| � W | j d ur�| j � � d | _ d S d S | j d ur�| j � � d | _ w | �!� d S )N� zhWARNING: Running tests with --huntrleaks/-R and less than 3 warmup repetitions can give false positives!Trf zUsing random seedr z/{}r r )�run_tests_multiprocess�win32)�WindowsLoadTrackerz%Failed to create WindowsLoadTracker: )"r �headerra r_ r| r rl r� rm rJ rK �stdoutr� r� Zforeverr� r� r r/ r0 �iterr� r` Zuse_mprk r� r� r6 Ztest.libregrtest.win_utilsr� r4 �PermissionError�closer� )r8 ru rv rw rx r� r� �errorr9 r9 r: � run_tests� sZ ����� ��
�
zRegrtest.run_testsc C s� | j r+| jr%t| j d��}|�| jd � W d � n1 s w Y nt�| j � | jr=| j�� }|jdd| j j
d� t � t �
� | j }t dt|� � t d| �� � | j jrft�dt�� � d S d S )N�w�
T)Z show_missingZsummary�coverdirzTotal duration: %szTests result: %szleaks %d)r2 r1 r~ �writerq �unlinkr+ �resultsZ
write_resultsr r� rJ r, r- r. r r� Zrunleaks�system�getpid)r8 r� rz Zdurationr9 r9 r: �finalize- s& ��
��zRegrtest.finalizec
C s | j js | js d S dd lm m} |�d�}dddd�}| jD ]%}|�|� |D ]}z|| t|� |d��7 < W q) t
yD Y q)w q |� � D ] \}}|� |t
|�� qJtj�tj| j j�}t|d��}|�|�D ]} |�| � qlW d � d S 1 sw Y d S )Nr Z
testsuites)r r� r� �wb)r ri r3 rG rH rI ZElementrC rX �getrD �itemsr= r� rq rr r} r r� r~ Z tostringlistr� )
r8 rO �rootZtotalsr� �k�vri �f�sr9 r9 r: �save_xml_resultB s,
�� �"�zRegrtest.save_xml_resultc C st | j jr | j j| _| js0t�� r+t�d�| _| jd u r!t�d�| _tj�| jd�| _nt �
� | _tj� | j�| _d S )NZ abs_builddirZsrcdirZbuild) r rp r5 � sysconfigZis_python_buildZget_config_varrq rr r} �tempfile�
gettempdir�abspathr7 r9 r9 r: � set_temp_dir[ s
zRegrtest.set_temp_dirc C sV t j| jdd� t �� }| jd urd�|�}nd�|�}|tj7 }t j� | j|�}|S )NT)�exist_okztest_python_worker_{}ztest_python_{})
rq �makedirsr5 r� r6 r� r Z FS_NONASCIIrr r} )r8 �pid�test_cwdr9 r9 r: �create_temp_dirq s
zRegrtest.create_temp_dirc C sz dd l }tj�|�| j�d�}td| j � |� |�D ] }tj�|�r/td| � t� |� qtd| � t�
|� qd S )Nr z
test_python_*zCleanup %s directoryzRemove directory: %szRemove file: %s) �globrq rr r} �escaper5 rJ �isdirr �rmtreer� )r8 r� rr rA r9 r9 r: �cleanup� s �zRegrtest.cleanupNc
K s� | � |� | �� | jjr| �� t�d� | �� }z&tj|dd�� || j_ | �
||� W d � W d S 1 s9w Y W d S t y` } zt j
tdd� t�|j� W Y d }~d S d }~ww )Nr T)r_ )ro )ry r� r r rK ro r� r Ztemp_cwdrp �_main�
SystemExit� faulthandlerZdump_traceback_later� EXIT_TIMEOUT�code)r8 r rt r� �excr9 r9 r: �main� s
&���z
Regrtest.mainc C s. | j d ur
| j �� S ttd�rt�� d S d S )NrV r )r4 rV �hasattrrq r7 r9 r9 r: rV � s
zRegrtest.getloadavgc C s | j d urddlm} || j| j � | jjrtd� | jjt_| jj t_
t | j� | � |� | jj
r;| �
� t�d� | jjrH| �� t�d� | �� | �� | jjr[| jr[| �� | �� | �� | jrkt�d� | jrst�d� | jjr| jrt�d� t�d� d S )Nr )�run_tests_workerz Press any key to continue...rg � r� )r6 rk r
r �wait�inputra r ZPGOZ pgo_extendedZ PGO_EXTENDEDr r� r� rK ro r� r� r� Zverbose2r! r� r� r� r) r� r$ )r8 r rt r
r9 r9 r: r � s8
zRegrtest._main)F)r r< ) �__name__�
__module__� __qualname__r; r> rQ r\ rd ry r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r rV r r9 r9 r9 r: r s6 +
' I
,;7
1
r c K s t � jdd| i|�� d S )Nr r9 )r r )r rt r9 r9 r: r � s r r< )-r r� rq r� r� r� rK 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 Z test.supportr r r r r9 r9 r9 r: �<module> s4 D D