OwlCyberSecurity - MANAGER
Edit File: 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 |