File: //snap/core20/2599/lib/python3.8/test/libregrtest/__pycache__/runtest_mp.cpython-38.pyc
U
���g = � @ sf 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 e eks�t �dZ e e ks�t �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 �1/usr/lib/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 @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �MultiprocessIteratorz8A thread-safe iterator over tests for multiprocess mode.c 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__�__doc__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 rF s s rF 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 rI 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>� ) rI �is_alive�appendrO rQ �time� monotonicrP �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: )rQ rR r& r1 r$ r r* r r] �signal�SIGKILL�kill�ProcessLookupError�OSErrorr
)r9 ra Zwhat�excr r r �_kill� s$ zTestWorkerProcess._killc C s d| _ | �� d S )NT)rS rl r; r r r r@ � s zTestWorkerProcess.stop� Nc C s* t �� | j }t|||d �}t||||�S r )rZ r[ rP r rE )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}|d k s~t�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�rM rm )rZ r[ rP rO r. r rR rQ �_wait_completedrS rl rF Z communicaterM �
returncode�AssertionErrorr' �TimeoutExpiredrj �strip�rstrip)r9 r ra 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) ry rp r
�
rpartitionrw r r r � Exceptionr rE ) r9 r rx r r rn �_r rk 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)rS r> rJ r= r~ rK Zputr r r rF �
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): ) rQ r �closer �wait� JOIN_TIMEOUTr' ru rj r
r )r9 ra rk r r r rr ' 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^ rX rZ r[ rN �logr r� r
)r9 rP rb r r r � wait_stopped6 s
z TestWorkerProcess.wait_stopped)rm rm N)rA rB rC r: rc rl r@ rp ry r~ r� rr r� �
__classcell__r r rT r rG w s �
; rG c C sN g }| D ]@}|j }|sqt�� |j }|tkrd|t|�f }|�|� q|S )Nz%s (%s))rO rZ r[ rP r r rY )�workersrV �workerrO rb �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, )
rN r� r �queueZQueuerK r5 ZtestsrJ rM �minrL r� )r9 rN 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 )rG )�.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� �lenrM r rL r� �start)r9 �msgr� r r; r �
start_workersl s
�
��
z$MultiprocessTestRunner.start_workersc C s6 t �� }| jD ] }|�� q| jD ]}|�|� q"d S r )rZ r[ r� r@ r� )r9 rP 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 )rX )r� r� r r r � <genexpr>� s z5MultiprocessTestRunner._get_result.<locals>.<genexpr>r rq T)r3 z running: %s�, )�anyr� rK �getr� ZEmptyr rM �PROGRESS_UPDATE� faulthandlerZdump_traceback_later�MAIN_PROCESS_TIMEOUTr� �pgor� r^ )r9 Zuse_faulthandlerrM rV 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_msgro r r r� r r� r� r^ rN Zdisplay_progress�
test_index)r9 r� r r� rV 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/ rd F)
r
r� rN 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 rM r� Zcancel_dump_traceback_laterrJ r@ r� r� r� �KeyboardInterruptr1 rN 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� )rN r r r �run_tests_multiprocess� s r� )0� collectionsr� r r* r� rf r' r$ r6 rZ r� r r` r Ztest.libregrtest.runtestr r r r r r r r
Ztest.libregrtest.setupr Ztest.libregrtest.utilsr r
r� rt r� r� �hasattrr& r r r. r4 r5 �
namedtuplerE r| rF ZThreadrG r� r� r� r r r r �<module> sF (
� X
|