Your IP : 216.73.216.60
3
��2ag � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z y d dl
Z
W n& e k
r~ e d� ej
d� Y nX d dlmZmZmZmZmZ d dlmZ d dlmZ dZd Zd
d � Zd d
� ZG dd� d�Z G dd� de
j �Z dd� Z dS )� N)�supportz+Multiprocess option requires thread support� )�runtest� INTERRUPTED� CHILD_ERROR�PROGRESS_MIN_TIME�format_test_result)� setup_tests)�format_durationg >@g @c
C s� ddl m}m} t|�}|| f}tj|�}tjftj � dddd|f�}|j
rV|dg7 }||||dt j d ktj
d
�}|� |j� \}} |j� }
W d Q R X |
|| fS ) Nr )�Popen�PIPEz-uz-mz
test.regrtestz
--worker-argsz--pgoT�nt)�stdout�stderrZuniversal_newlinesZ close_fds�cwd)�
subprocessr r �vars�json�dumps�sys�
executabler Zargs_from_interpreter_flags�pgo�os�nameZSAVEDCWDZ communicate�wait) �testname�nsr r �ns_dict� worker_args�cmd�popenr r �retcode� r" �//opt/alt/python36/lib64/python3.6/runtest_mp.py�run_test_in_subprocess s&
r$ c C s� t j| �\}}tjf |�}t|� yt||�}W nR tk
rN tdd f}Y n6 tk
r� } zt j
� t t |�f}W Y d d }~X nX t
� t
t j|�dd� tjd� d S )N� T)�flushr )r �loads�types�SimpleNamespacer r �KeyboardInterruptr �
BaseException� traceback� print_excr �str�printr r �exit)r r r r �result�er" r" r# �run_tests_workerA s r3 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�MultiprocessIteratorc C s d| _ tj� | _|| _d S )NF)� interrupted� threadingZLock�lock�tests)�selfr8 r" r" r# �__init__Y s
z MultiprocessIterator.__init__c C s | S )Nr" )r9 r" r" r# �__iter__^ s z MultiprocessIterator.__iter__c C s* | j � | jrtd��t| j�S Q R X d S )Nztests interrupted)r7 r5 �
StopIteration�nextr8 )r9 r" r" r# �__next__a s z MultiprocessIterator.__next__N)�__name__�
__module__� __qualname__r: r; r> r" r" r" r# r4 U s r4 c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�MultiprocessThreadc s, t � j� || _|| _|| _d | _d | _d S )N)�superr: �pending�outputr � current_test�
start_time)r9 rD rE r )� __class__r" r# r: i s
zMultiprocessThread.__init__c C s� yt | j�}W n tk
r. | jjd� dS X z&tj� | _|| _t || j
�\}}}W d d | _X |dkr�t d| d f}| jj||j � |j � |f� dS |j
� jd�\}}}|s�| jjd� dS tj|�}| jj||j � |j � |f� dS )NTr z Exit code %sF�
)NNNN)NNNN)r= rD r< rE �put�time� monotonicrG rF r$ r r �rstrip�strip�
rpartitionr r' )r9 �testr! r r r1 �_r" r" r# �_runtestq s.
zMultiprocessThread._runtestc C sB yd}x|s| j � }qW W n" tk
r< | jjd� � Y nX d S )NF)NNNN)rR r+ rE rJ )r9 �stopr" r" r# �run� s zMultiprocessThread.run)r? r@ rA r: rR rT �
__classcell__r" r" )rH r# rB h s rB c s� t j� � t�j���jj}|d k }� ��fdd�t�jj�D �}tdt |� � x|D ] }|j
� qVW dd� }d}d}t t t
�}�zƐy��x�|�jjk �r|r�tj|dd � y� j|d
�} W nF t jk
�r ||�}
|
r��jj r�td d j|
� dd
� w�Y nX | \} } }
}| d k�r$|d7 }q��j| |� |\}}}t| |�}|ttfk�rz|t
k�rz�jj �rz|dt|� 7 }n|tk�r�d||f }||�}
|
�r��jj �r�|dd j|
� 7 }�j||� | �r�t| dd
� |
�r��jj �r�t|
tjdd� |d tk�rt �|d7 }q�W W n( t k
�rD d�_ d�_ t� Y nX W d |�rXtj � X t j � }x�dd� |D �}
t!t"t#|
��}
|
�s�P t j � | }dd j|
�t |
�f }|t$k�r�d||f }t|dd
� x|D ]}|jt$� �q�W �qdW d S )Nc s g | ]}t �� �j��qS r" )rB r )�.0�i)rE rD �regrtestr" r# �
<listcomp>� s z*run_tests_multiprocess.<locals>.<listcomp>z.Run tests in parallel using %s child processesc S sR g }xH| D ]@}|j }|sq
tj� |j }|tkr
d|t|�f }|j|� q
W |S )Nz%s (%s))rF rK rL rG r r
�append)�workers�running�workerrF �dt�textr" r" r# � get_running� s
z+run_tests_multiprocess.<locals>.get_runningr � T)r0 )�timeoutz running: %sz, )r& z (%s)z%s (%s)z -- running: %s)�filer& c S s g | ]
}|j �qS r" )rF )rV r] r" r" r# rY � s zWaiting for %s (%s tests)z%s since %.0f sec)%�queueZQueuer4 r8 r rb �rangeZuse_mpr/ �len�start�max�PROGRESS_UPDATEr � faulthandlerZdump_traceback_later�getZEmptyr �joinZaccumulate_resultr r r r
Zdisplay_progressr r r* r5 Zcancel_dump_traceback_laterrK rL �list�filter�bool�
WAIT_PROGRESS)rX Z test_timeoutZ use_timeoutr[ r] r` ZfinishedZ
test_indexZ get_timeout�itemr\ rP r r r1 �okZ test_timeZxml_datar_ Z
wait_startr^ �liner" )rE rD rX r# �run_tests_multiprocess� s�
rt ) rj r r rd r rK r, r( rP r r6 � ImportErrorr/ r0 Ztest.libregrtest.runtestr r r r r Ztest.libregrtest.setupr Ztest.libregrtest.utilsr
ri rp r$ r3 r4 ZThreadrB rt r" r" r" r# �<module> s. #3