CasperSecurity
o
`�wi� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z m
Z
zd dl Z W n e y7 dZ Y nw d dl
mZmZ dZdd� Zdd � Zd
d � ZdS ) � N)�support)�TESTFN_UNDECODABLE� FS_NONASCII)�setup_unraisable_hook�setup_threading_excepthook� PYTHONREGRTEST_UNICODE_GUARDc C sj zt j�� }W n
ttfy d }Y n-w tjd|d� g }ttd�r)|� tj
� ttd�r4|� tj � |D ]
}tj |d|d� q6t
� t� t�t j� | jr\t j�dtj�| j�� t j�� D ](}t|d�r{t|j�D ]
\}}tj�|�|j|< qmt|dd �r�tj�|j �|_ qa| j r�d t j _ t �!� | j"d ur�t�#| j"� | j$d ur�t%�&| j$� t�'| j(o�| j(d
k� | j)t_)tt d �r�d d
� }t �*|� t+� t,� | j-d u�rt.tj/| j-d �t_/t.tj0| j-d �t_0t1tj2| j-�t_2t1tj3| j-�t_3t1tj/| j-�t_/t1tj0| j-�t_0| j4�rddl5m6} d|_7t8�r'tj9�r'tj:�;t<�=� t8� d S t>�r3tj?�;t<t>� d S d S )NT)� all_threads�file�SIGALRM�SIGUSR1)�chainr r �__path__�__file__F� � addaudithookc S s d S �N� )�name�argsr r �-/usr/lib/python3.10/test/libregrtest/setup.py�_test_audit_hookU s z%setup_tests.<locals>._test_audit_hook�( � )�RegressionTestResult)@�sys�
__stderr__�fileno�
ValueError�AttributeError� faulthandler�enable�hasattr�signal�appendr
r �register�_adjust_resource_limits�replace_stdoutr �record_original_stdout�stdout�testdir�path�insert�os�abspath�modules�values� enumerater
�getattrr �
huntrleaks�unittest�
BaseTestSuite�_cleanup�_deactivate_opcache�memlimit� set_memlimit� threshold�gc�
set_threshold�suppress_msvcrt_asserts�verbose�
use_resourcesr r r �timeout�max�
SHORT_TIMEOUT� LONG_TIMEOUT�min�LOOPBACK_TIMEOUT�INTERNET_TIMEOUT�xmlpath�test.support.testresultr �USE_XMLr �supports_bytes_environ�environb�
setdefault�UNICODE_GUARD_ENV�encoder �environ) �ns� stderr_fd�signals�signum�module�indexr* r r r r r � setup_tests sl �
�
�rU c sZ t j� z� �� } W n
ty Y dS w 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_stdout� s
z&replace_stdout.<locals>.restore_stdout)r r( r r �openrY �atexitr$ )�fdr_ r r^ r r& s s �� r& c C s� zddl } ddl m}m} W n
ty Y dS w | �|�\}}d}||k rh||k rjt||�}z| �|||f� td|� d|� �� W dS tt fyg } ztd|� d|� d |� d
�� W Y d}~dS d}~ww dS 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: �.)
�resourcerc rd � ImportError� getrlimitrC � setrlimit�printr �OSError)rf rc rd �fd_limit�max_fds� desired_fds� new_fd_limit�errr r r r% � s, �
�����r% )ra r r, r" r r3 �testr �test.support.os_helperr r r: rg �test.libregrtest.utilsr r rL rU r& r% r r r r �<module> s$ �^