Your IP : 216.73.216.60
U
i�f = � @ sN 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 dlmZmZmZmZmZmZmZmZ d dlmZ d dlmZmZ dZ dZ dZ e ed�o�e ed �Z d
d � Z!d d
� Z"dd� Z#dd� Z$G dd� d�Z%e �&dd�Z'G dd� de(�Z)G dd� dej*�Z+dd� Z,G d d � d �Z-d d � Z.dS ) � N)�support)�runtest� INTERRUPTED� CHILD_ERROR�PROGRESS_MIN_TIME�format_test_result�
TestResult� is_failed�TIMEOUT)� setup_tests)�format_duration�
print_warningg >@g �r@�setsid�killpgc C s& | j tkrdS |jr"t| |�r"dS dS )NTF)�resultr Zfailfastr )r �ns� r �@/opt/alt/python38/lib64/python3.8/test/libregrtest/runtest_mp.py� must_stop& s
r c C s" t �| �\}}tjf |�}||fS �N)�json�loads�types�SimpleNamespace)� worker_args�ns_dict� test_namer r r r �parse_worker_args. s r c C sp t |�}|| f}t�|�}tjft�� dddd|f�}i }trFd|d< tj |ftj
tj
dt j dktj
d�|��S ) Nz-uz-mz
test.regrtestz
--worker-argsTZstart_new_session�nt)�stdout�stderrZuniversal_newlinesZ close_fds�cwd)�varsr �dumps�sys�
executabler Zargs_from_interpreter_flags�USE_PROCESS_GROUP�
subprocess�Popen�PIPE�os�nameZSAVEDCWD)Ztestnamer r r �cmd�kwr r r �run_test_in_subprocess4 s,
���r. c C s< t | � t| |�}t� tt�t|��dd� t�d� d S )NT��flushr )r r �printr r# �listr$ �exit)r r r r r r �run_tests_workerM s
r4 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�MultiprocessIteratorc C s t �� | _|| _d S r )� threadingZLock�lock�
tests_iter)�selfr8 r r r �__init__^ s
z MultiprocessIterator.__init__c C s | S r r �r9 r r r �__iter__b s z MultiprocessIterator.__iter__c
C s6 | j �& | jd krt�t| j�W 5 Q R � S Q R X d S r )r7 r8 �
StopIteration�nextr; r r r �__next__e s
z MultiprocessIterator.__next__c C s | j � d | _W 5 Q R X d S r )r7 r8 r; r r r �stopk s zMultiprocessIterator.stopN)�__name__�
__module__� __qualname__r: r< r? r@ r r r r r5 Z s r5 �MultiprocessResultz result stdout stderr error_msgc @ s e Zd ZdS )�
ExitThreadN)rA rB rC r r r r rE s s rE c sf e Zd Z� fdd�Zdd� Zdd� Zdd� Zdd d �Zd
d� Zdd� Z dd� Z
dd� Z dd� Z � Z
S )�TestWorkerProcessc sZ t � �� || _|j| _|j| _|j| _|j| _|j| _d | _ d | _
d | _ d| _ d| _
d S )NF)�superr: � worker_id�pending�outputr �worker_timeout�timeout�regrtest�current_test_name�
start_time�_popen�_killed�_stopped)r9 rH Zrunner�� __class__r r r: x s
zTestWorkerProcess.__init__c C s� d| j � �g}| �� r"|�d� n
|�d� | j}|rF|�d|� �� | j}|d k r�t�� | j }|�d| jj � �dt
|�� �f� dd� |� S ) NzTestWorkerProcess #�runningZstoppedztest=zpid=ztime=z<%s>� ) rH �is_alive�appendrN rP �time� monotonicrO �extend�pidr �join)r9 �info�test�popen�dtr r r �__repr__� s
�zTestWorkerProcess.__repr__c
C s� | j }|d krd S | jr d S d| _tr2| � d�}n| � }td|� �tjdd� z"trft�|jt j
� n|� � W nH t k
r� Y n6 t
k
r� } ztd|� d|��� W 5 d }~X Y nX d S )NTz process groupzKill ��filer0 zFailed to kill z: )rP rQ r& r1 r$ r r* r r\ �signal�SIGKILL�kill�ProcessLookupError�OSErrorr
)r9 r` Zwhat�excr r r �_kill� s$ zTestWorkerProcess._killc C s d| _ | �� d S )NT)rR rk r; r r r r@ � s zTestWorkerProcess.stop� Nc C s* t �� | j }t|||d �}t||||�S r )rY rZ rO r rD )r9 r Z
error_typer r �err_msg� test_timer r r r �mp_result_error� s z!TestWorkerProcess.mp_result_errorc C s t �� | _|| _z t|| j�}d| _|| _W n d | _� Y nX z�z�| j rX| �
� t �z |j | j
d�\}}|j}W nR tjk
r� | j r�t �| �
� d }d }}Y n0 tk
r� | j r�t �� Y nX |�� }|�� }|||fW W � S | �
� � Y nX W 5 | �� d | _d | _X d S )NF�rL rl )rY rZ rO rN r. r rQ rP �_wait_completedrR rk rE Z communicaterL �
returncoder' �TimeoutExpiredri �strip�rstrip)r9 r r` r r �retcoder r r � _run_process� sH
z TestWorkerProcess._run_processc
C s� | � |�\}}}|d kr(| �|t||�S d }|dkr>d| }nd|�d�\}}}|�� }|s`d}nBzt�|�}t|� }W n* tk
r� } z d| }W 5 d }~X Y nX |d k r�| �|t |||�S t
||||�S )Nr z Exit code %s�
z Failed to parse worker stdoutz Failed to parse worker JSON: %s) rw ro r
�
rpartitionru r r r � Exceptionr rD ) r9 r rv r r rm �_r rj r r r �_runtest� s,
�zTestWorkerProcess._runtestc C s� | j s�zZzt| j�}W n tk
r0 Y W q�Y nX | �|�}| j�d|f� t|j| j �r^W q�W q t
k
rx Y q�Y q t k
r� | j�dt �
� f� Y q�Y q X q d S )NFT)rR r> rI r= r| rJ Zputr r r rE �
BaseException� traceback�
format_exc)r9 r � mp_resultr r r �run s
zTestWorkerProcess.runc
C st | j }|j�� |j�� z|�t� W nF tjtfk
rn } z"t d| � dt
t�� d|��� W 5 d }~X Y nX d S )NzFailed to wait for z completion (timeout=z): ) rP r �closer �wait� JOIN_TIMEOUTr' rs ri r
r )r9 r` rj r r r rq ' s
z!TestWorkerProcess._wait_completedc C sd | � d� | �� sq`t�� | }| j�d| � dt|�� �� |tkr td| � dt|�� �� q`q d S )Ng �?z Waiting for z thread for zFailed to join z in ) r] rW rY rZ rM �logr r� r
)r9 rO ra r r r � wait_stopped6 s
z TestWorkerProcess.wait_stopped)rl rl N)rA rB rC r: rb rk r@ ro rw r| r� rq r� �
__classcell__r r rS r rF w s �
; rF c C sN g }| D ]@}|j }|sqt�� |j }|tkrd|t|�f }|�|� q|S )Nz%s (%s))rN rY rZ rO r r rX )�workersrU �workerrN ra �textr r r � get_runningN s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�MultiprocessTestRunnerc C sj || _ | j j| _|j| _t�� | _t| j j�| _| jj d k rZt
| jj d | jj d �| _ nd | _ d | _ d S )Ng �?i, )
rM r� r �queueZQueuerJ r5 ZtestsrI rL �minrK r� )r9 rM r r r r: \ s
�z MultiprocessTestRunner.__init__c s| � fdd�t d� jjd �D �� _dt� j�� d�}� jjrZ|dt� jj�t� j�f 7 }� �|� � jD ] }|� � qjd S )Nc s g | ]}t |� ��qS r )rF )�.0�indexr; r r �
<listcomp>m s �z8MultiprocessTestRunner.start_workers.<locals>.<listcomp>� z Run tests in parallel using z child processesz" (timeout: %s, worker timeout: %s))
�ranger Zuse_mpr� �lenrL r rK r� �start)r9 �msgr� r r; r �
start_workersl s
�
��
z$MultiprocessTestRunner.start_workersc C s6 t �� }| jD ] }|�� q| jD ]}|�|� q"d S r )rY rZ r� r@ r� )r9 rO r� r r r � stop_workersx s
z#MultiprocessTestRunner.stop_workersc C s� t dd� | jD ��s>z| jjdd�W S tjk
r< Y d S X | jjd k }t}|r`t j
t dd� z| jj|d�W S tjk
r� Y nX t | j�}|rN| jj
sN| �dd�|� � qNd S ) Nc s s | ]}|� � V qd S r )rW )r� r� r r r � <genexpr>� s z5MultiprocessTestRunner._get_result.<locals>.<genexpr>r rp T)r3 z running: %s�, )�anyr� rJ �getr� ZEmptyr rL �PROGRESS_UPDATE� faulthandlerZdump_traceback_later�MAIN_PROCESS_TIMEOUTr� �pgor� r] )r9 Zuse_faulthandlerrL rU r r r � _get_result s$ �
z"MultiprocessTestRunner._get_resultc C s� |j }t|�}|jd k r(|d|j 7 }n$|jtkrL| jjsL|dt|j� 7 }t| j �}|rt| jjst|dd�
|� 7 }| j � | j
|� d S )Nz (%s)z -- running: %sr� )r r Z error_msgrn r r r� r r� r� r] rM Zdisplay_progress�
test_index)r9 r� r r� rU r r r �display_result� s
z%MultiprocessTestRunner.display_resultc C s� |d r"|d }t d|� �� dS | jd7 _|d }| j�|j� | �|� |jrdt|jdd� |jr�| j j
s�t|jt jdd� t |j| j �r�dS dS )Nr r� z regrtest worker thread failed: Tr/ rc F)
r
r� rM Zaccumulate_resultr r� r r1 r r r� r$ r )r9 �itemr r� r r r �_process_result� s
z&MultiprocessTestRunner._process_resultc C s� | � � d| _zPz(| � � }|d kr$q6| �
|�}|rq6qW n" t k
rZ t � d| j
_Y nX W 5 | jjd k rtt�� | j�� | �� X d S )Nr T)r� r� r rL r� Zcancel_dump_traceback_laterrI r@ r� r� r� �KeyboardInterruptr1 rM Z interrupted)r9 r� r@ r r r � run_tests� s
z MultiprocessTestRunner.run_testsN)
rA rB rC r: r� r� r� r� r� r� r r r r r� [ s r� c C s t | ��� d S r )r� r� )rM r r r �run_tests_multiprocess� s r� )/� collectionsr� r r* r� re r' r$ r6 rY r~ r r_ r Ztest.libregrtest.runtestr r r r r r r r
Ztest.libregrtest.setupr Ztest.libregrtest.utilsr r
r� r� r� �hasattrr& r r r. r4 r5 �
namedtuplerD rz rE ZThreadrF r� r� r� r r r r �<module> sB (
� X
|