U
���gt � @ sv d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ z d dlZW n e k
r` dZY nX dd� Z
dd� Z dS )� N)�supportc C s� zt j�� }W n ttfk
r* d }Y nZX tjd|d� g }ttd�rT|� tj
� ttd�rj|� tj � |D ]}tj |d|d� qnt
� t�t j� | jr�t j�dtj�| j�� t j�� D ]P}t|d�r�t|j�D ]\}}tj�|�|j|< q�t|dd �r�tj�|j�|_q�| j �r d t j _ | j d k �r8t�!| j � | j"d k �rPt#�$| j"� t�%| j&�od| j&d
k� | j't_'tt d ��r�d d
� }t �(|� d S )NT)Z all_threads�file�SIGALRM�SIGUSR1)�chainr r �__path__�__file__F� � addaudithookc S s d S �N� )�name�argsr r �,/usr/lib/python3.8/test/libregrtest/setup.py�_test_audit_hookL s z%setup_tests.<locals>._test_audit_hook))�sys�
__stderr__�fileno�
ValueError�AttributeError� faulthandlerZenable�hasattr�signal�appendr r �register�replace_stdoutr Zrecord_original_stdout�stdoutZtestdir�path�insert�os�abspath�modules�values� enumerater �getattrr Z
huntrleaks�unittestZ
BaseTestSuiteZ_cleanupZmemlimitZ set_memlimitZ threshold�gcZ
set_thresholdZsuppress_msvcrt_asserts�verboseZ
use_resourcesr
)�nsZ stderr_fdZsignalsZsignum�module�indexr r r r r � setup_tests sB
r+ c s\ t j� z � �� } W n tk
r( Y dS X t| d� jdddd�t _� fdd�}t�|� dS ) z�Set stdout encoder error handler to backslashreplace (as stderr error
handler) to avoid UnicodeEncodeError when printing a tracebackN�w�backslashreplaceF�
)�encoding�errors�closefd�newlinec s t j�� � t _d S r )r r �closer �r r r �restore_stdoute s
z&replace_stdout.<locals>.restore_stdout)r r r r �openr/ �atexitr )�fdr5 r r4 r r Q s � r ) r7 r r r r r% Ztestr r& � ImportErrorr+ r r r r r �<module> s
C |