File: //snap/core18/current/usr/lib/python3.6/test/libregrtest/__pycache__/runtest_mp.cpython-36.pyc
3
�ähg � @ 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 ) a) Run the given test in a subprocess with --worker-args.
ns is the option Namespace parsed from command-line arguments. regrtest
is invoked in a subprocess with the --worker-args argument; when the
subprocess exits, its return code, stdout and stderr are returned as a
3-tuple.
r )�Popen�PIPEz-uz-mz
test.regrtestz
--worker-argsz--pgoT�nt)�stdout�stderrZuniversal_newlinesZ close_fds�cwdN)�
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" �1/usr/lib/python3.6/test/libregrtest/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 ZdZdd� Zdd� Zdd� ZdS ) �MultiprocessIteratorz8A thread-safe iterator over tests for multiprocess mode.c 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__�__doc__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 rE rF 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|�}t|�dks�td|����| jj||j � |j � |f� dS )
NTr z Exit code %sF�
� zInvalid result tuple: )NNNN)NNNN)r= rE r< rF �put�time� monotonicrH rG r$ r r �rstrip�strip�
rpartitionr r' �len�AssertionError)r9 �testr! r r r1 �_r" r" r# �_runtestq s0
zMultiprocessThread._runtestc C sB yd}x|s| j � }qW W n" tk
r< | jjd� � Y nX d S )NF)NNNN)rV r+ rF rL )r9 �stopr" r" r# �run� s zMultiprocessThread.run)r? r@ rA r: rV rX �
__classcell__r" r" )rI r# rC h s rC 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" )rC r )�.0�i)rF rE �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))rG rM rN rH r r
�append)�workers�running�workerrG �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" )rG )rZ ra r" r" r# r] � s zWaiting for %s (%s tests)z%s since %.0f sec)%�queueZQueuer4 r8 r rf �rangeZuse_mpr/ rR �start�max�PROGRESS_UPDATEr � faulthandlerZdump_traceback_later�getZEmptyr �joinZaccumulate_resultr r r r
Zdisplay_progressr r r* r5 Zcancel_dump_traceback_laterrM rN �list�filter�bool�
WAIT_PROGRESS)r\ Z test_timeoutZ use_timeoutr_ ra rd ZfinishedZ
test_indexZ get_timeout�itemr` rT r r r1 �okZ test_timeZxml_datarc Z
wait_startrb �liner" )rF rE r\ r# �run_tests_multiprocess� s�
rw ) rm r r rh r rM r, r( rT r r6 � ImportErrorr/ r0 Ztest.libregrtest.runtestr r r r r Ztest.libregrtest.setupr Ztest.libregrtest.utilsr
rl rs r$ r3 r4 ZThreadrC rw r" r" r" r# �<module> s. #3