File: //opt/bitninja-python-dojo/embedded/lib/python3.9/test/libregrtest/__pycache__/setup.cpython-39.pyc
a
�;�i� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ z d dlZW n e y^ dZY n0 d dl
m Z dd� Z dd� Z
dd � ZdS )
� N)�support)�setup_unraisable_hookc C s zt j�� }W n ttfy( d }Y nZ0 tjd|d� g }ttd�rR|� tj
� ttd�rh|� tj � |D ]}tj |d|d� qlt
� t� 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 u�r<t�"| j!� | j#d u�rTt$�%| j#� t�&| j'�oh| j'd
k� | j(t_(tt d ��r�d d
� }t �)|� t*� | j+d u�r t,tj-| j+d �t_-t,tj.| j+d �t_.t/tj0| j+�t_0t/tj1| j+�t_1t/tj-| j+�t_-t/tj.| j+�t_.d S )NT)Z all_threads�file�SIGALRM�SIGUSR1)�chainr r �__path__�__file__F� � addaudithookc S s d S �N� )�name�argsr
r
�J/opt/bitninja-python-dojo/embedded/lib/python3.9/test/libregrtest/setup.py�_test_audit_hookO s z%setup_tests.<locals>._test_audit_hook�( � )2�sys�
__stderr__�fileno�
ValueError�AttributeError� faulthandler�enable�hasattr�signal�appendr r �register�_adjust_resource_limits�replace_stdoutr Zrecord_original_stdout�stdoutZtestdir�path�insert�os�abspath�modules�values� enumerater �getattrr Z
huntrleaks�unittestZ
BaseTestSuite�_cleanup�memlimitZ set_memlimitZ threshold�gcZ
set_thresholdZsuppress_msvcrt_asserts�verboseZ
use_resourcesr r �timeout�maxZ
SHORT_TIMEOUTZ LONG_TIMEOUT�minZLOOPBACK_TIMEOUTZINTERNET_TIMEOUT)�nsZ stderr_fdZsignals�signum�module�indexr" r r
r
r � setup_tests sT
r6 c sZ t j� z � �� } W n ty& Y dS 0 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_stdoutu s
z&replace_stdout.<locals>.restore_stdout)r r! r r �openr: �atexitr )�fdr@ r
r? r r a s � r c C s� z ddl } ddl m}m} W n ty0 Y dS 0 | �|�\}}d}||k r�||k r�t||�}z(| �|||f� td|� d|� �� W nD tt fy� } z(td|� d|� d |� d
�� W Y d}~n
d}~0 0 dS ) z5Adjust the system resource limits (ulimit) if needed.r N)�
RLIMIT_NOFILE�
RLIM_INFINITYi zRaised RLIMIT_NOFILE: z -> z#Unable to raise RLIMIT_NOFILE from z to z: �.)
�resourcerD rE � ImportErrorZ getrlimitr1 Z setrlimit�printr �OSError)rG rD rE Zfd_limitZmax_fdsZ desired_fdsZ new_fd_limit�errr
r
r r { s$
��r )rB r r$ r r r* �testr r- rH Ztest.libregrtest.utilsr r6 r r r
r
r
r �<module> s
Q