CasperSecurity
o
`�wiVe � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ejZG dd � d e j�Zed
krre �� dS dS ) � N)�support)�
import_helper)� os_helper)�
script_helper)�
socket_helper)�warnings_helperc @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z e
� d
�dd� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Ze
�eed&�d'�d(d)� �Zd*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z!d6d7� Z"d8d9� Z#d:d;� Z$d<d=� Z%d>d?� Z&d@dA� Z'dBdC� Z(dDdE� Z)dFdG� Z*dHdI� Z+dJdK� Z,dLdM� Z-G dNdO� dO�Z.G dPdQ� dQ�Z/dRdS� Z0dTdU� Z1dVdW� Z2e
�eedX�o�eedY�dZ�d[d\� �Z3dld^d_�Z4d`da� Z5dbdc� Z6ddde� Z7dfdg� Z8dhdi� Z9djdk� Z:d]S )m� TestSupportc C sD t tj�}tjddd�| _tjddd�| _t tj�|d ks J �d S )Nz test.support.warnings_helperz.*used in test_support.*)�likeztest.test_supportz!.*You should NOT be seeing this.*� )�len�warnings�filtersr �ignore_deprecations_from�_warnings_helper_token�_test_support_token��cls�orig_filter_len� r �(/usr/lib/python3.10/test/test_support.py�
setUpClass s
��zTestSupport.setUpClassc C s4 t tj�}t�| j| j� t tj�|d ksJ �d S )Nr
)r r r
r �clear_ignored_deprecationsr r r r r r �
tearDownClass& s
�zTestSupport.tearDownClassc C sd t jdd��}t�� t �dt� dd� |D �}W d � n1 s"w Y | �t|�d|� dS )z9Test support.ignore_deprecations_from() silences warningsT)�recordz You should NOT be seeing this.c S � g | ]}t |j��qS r ��str�message��.0�wr r r �
<listcomp>4 � zDTestSupport.test_ignored_deprecations_are_silent.<locals>.<listcomp>Nr )r �catch_warningsr �_warn_about_deprecation�warn�DeprecationWarning� assertEqualr )�self� warning_objs�messagesr r r �$test_ignored_deprecations_are_silent/ s �z0TestSupport.test_ignored_deprecations_are_silentc C s t �d� | �tjt jd� d S )N�ftplib�foo)r �
import_module� assertRaises�unittest�SkipTest�r( r r r �test_import_module7 s
�z TestSupport.test_import_modulec C s t �d� d S )Nr, )r �import_fresh_moduler2 r r r �test_import_fresh_module< s z$TestSupport.test_import_fresh_modulec C s. | � t�| d�| j� | �tjtj| d� d S )N�test_get_attributer- )r' r �
get_attributer6 r/ r0 r1 r2 r r r r6 ? s �z TestSupport.test_get_attributezfailing buildbotsc C s | � t�� tj� d S �N)r' r �get_original_stdout�sys�stdoutr2 r r r �test_get_original_stdoutD s z$TestSupport.test_get_original_stdoutc C s2 dd l }| �dtj� t�d� | �dtj� d S )Nr �sched)r= �assertInr: �modulesr �unload� assertNotIn)r( r= r r r � test_unloadH s
zTestSupport.test_unloadc C sX t tddd��}W d � n1 sw Y t�t� | �tj�t�� t�t� d S )Nr �utf-8��encoding)�open�TESTFNr �unlink� assertFalse�os�path�exists)r( �fr r r � test_unlinkN s �
zTestSupport.test_unlinkc C s` t jd }tj�|d�}t�|� t�|� t �|� | �tj�|�� t �
t dd��
t �|� W d � n1 s;w Y t�|� t�|� t� |t j
t jB � t �
t dd��
t �|� W d � n1 skw Y | �tj�|�� t�|� t�|� t� |d� t �
t dd��
t �|� W d � n1 s�w Y | �tj�|�� d S )N�d�subdir�verboser )r rG rJ rK �join�mkdir�rmtreerI rL r � swap_attr�chmod�stat�S_IRUSR�S_IXUSR)r( �dirpath�
subdirpathr r r � test_rmtreeU s.
�
�
�zTestSupport.test_rmtreec C s� t d }t|ddd��}td|d� W d � n1 s w Y tj�dtj� t� � z(t
t �}| � t tj � t
�t � | �t tj � W tjd= t�|� t�d� d S tjd= t�|� t�d� w ) Nz.pyr rC rD zfoo = 1��filer � __pycache__)rG rF �printr: rK �insertrJ �curdir� importlib�invalidate_caches�
__import__r> r? r �forgetrA r rH rT )r( � mod_filenamerM �modr r r � test_forgetm s" �
�
zTestSupport.test_forgetc C s t �tjdf�}|�� d S )Nr )�socket�
create_serverr �HOST�close�r( �sr r r � test_HOST~ s zTestSupport.test_HOSTc C s$ t �� }t�t j|f�}|�� d S r8 )r �find_unused_portrj rk rl rm )r( �portro r r r �test_find_unused_port� s z!TestSupport.test_find_unused_portc C s& t � � }t�|� |�� |�� d S r8 )rj r � bind_port�listenrm rn r r r �test_bind_port� s
zTestSupport.test_bind_portc C s� t �� }tj�|�}zEtj�|d�}| �tj�|�� t� |��}| �
||� | � tj�|�� W d � n1 s:w Y | �tj�|�� W t� |� dS t� |� w )z8Test that temp_dir() creates and destroys its directory.�tempN)
�tempfile�mkdtemprJ rK �realpathrR rI �isdirr �temp_dirr' �
assertTruerT )r( �
parent_dirrK � temp_pathr r r �
test_temp_dir� s �zTestSupport.test_temp_dirc C sP t �� �}| �tj�|�� W d � n1 sw Y | �tj�|�� dS )zTest passing no path.N)r r| r} rJ rK r{ rI )r( r r r r �test_temp_dir__path_none� s
�z$TestSupport.test_temp_dir__path_nonec C sl dd� }t �� }tj�|�}z!| �tj�|�� | �t||� | �tj�|�� W t �
|� dS t �
|� w )z-Test passing a directory that already exists.c S �( t �| ��}td��1 s
w Y d S �Nzshould not get here)r r| � Exception)rK r r r r �
call_temp_dir� � �zMTestSupport.test_temp_dir__existing_dir__quiet_default.<locals>.call_temp_dirN) rx ry rJ rK rz r} r{ r/ �FileExistsError�shutilrT )r( r� rK r r r �*test_temp_dir__existing_dir__quiet_default� s z6TestSupport.test_temp_dir__existing_dir__quiet_defaultc C s� t �� }tj�|�}zIt�� �-}tj|dd��}| � ||� W d � n1 s(w Y dd� |j
D �}W d � n1 s?w Y | � tj� |�� W t
�|� nt
�|� w | � t|�d|� |d }| � |�d|�d ��|� dS )
z=Test passing a directory that already exists with quiet=True.T��quietNc S r r r r r r r r! � r" zGTestSupport.test_temp_dir__existing_dir__quiet_true.<locals>.<listcomp>� r z5tests may fail, unable to create temporary directory �: )rx ry rJ rK rz r �check_warningsr r| r' r r} r{ r� rT r �
startswith)r( rK �recorderr r r% r r r �'test_temp_dir__existing_dir__quiet_true� s$
� �
��z3TestSupport.test_temp_dir__existing_dir__quiet_true�forkztest requires os.forkc C s t �dt�d�� dS )z?Test that a forked child process does not remove the directory.�-ca
import os
from test import support
from test.support import os_helper
with os_helper.temp_cwd() as temp_path:
pid = os.fork()
if pid != 0:
# parent process
# wait for the child to terminate
support.wait_process(pid, exitcode=0)
# Make sure that temp_path is still present. When the child
# process leaves the 'temp_cwd'-context, the __exit__()-
# method of the context must not remove the temporary
# directory.
if not os.path.isdir(temp_path):
raise AssertionError("Child removed temp_path.")
N)r �assert_python_ok�textwrap�dedentr2 r r r �test_temp_dir__forked_child� s z'TestSupport.test_temp_dir__forked_childc C s� t �� }t�� �+}t�|��}| �||� | �t �� |� W d � n1 s'w Y W d � n1 s6w Y | �t �� |� d S r8 )rJ �getcwdr r| �
change_cwdr' )r( � original_cwdr �new_cwdr r r �test_change_cwd� s
�� �zTestSupport.test_change_cwdc C sh t �� }dd� }t�� �}t j�|d�}| �t||� W d � n1 s%w Y | �t �� |� dS )z&Test passing a non-existent directory.c S r� r� )r r� r� )rK r� r r r �call_change_cwd� r� zFTestSupport.test_change_cwd__non_existent_dir.<locals>.call_change_cwd�does_not_existN) rJ r� r r| rK rR r/ �FileNotFoundErrorr' )r( r� r� r~ �non_existent_dirr r r �!test_change_cwd__non_existent_dir� s
� �z-TestSupport.test_change_cwd__non_existent_dirc
C s� t �� }t�� �P}t j�|d�}t�� �5}tj|dd��}| � ||� | � t �� |� W d � n1 s5w Y dd� |j
D �}W d � n1 sLw Y W d � n1 s[w Y | � t |�d|� |d }| � |�
d |�d
��|� dS ) z6Test passing a non-existent directory with quiet=True.r� Tr� Nc S r r r r r r r r! r" zMTestSupport.test_change_cwd__non_existent_dir__quiet_true.<locals>.<listcomp>r� r �Btests may fail, unable to change the current working directory to r� )rJ r� r r| rK rR r r� r� r' r r r} r� )r( r� r~ �bad_dirr� r� r r% r r r �-test_change_cwd__non_existent_dir__quiet_true� s&
� �� �
��z9TestSupport.test_change_cwd__non_existent_dir__quiet_truec C s� t d }t�� �'}tj|dd�� W d � n1 sw Y dd� |jD �}W d � n1 s2w Y | �t|�d|� |d }| �|� d |�d
��|� dS ) z0Check the warning message when os.chdir() fails.�_does_not_existT)rK r� Nc S r r r r r r r r! r" z>TestSupport.test_change_cwd__chdir_warning.<locals>.<listcomp>r� r r� r� )
rG r r� r r� r r' r r} r� )r( rK r� r* �msgr r r � test_change_cwd__chdir_warning s
� �
��z*TestSupport.test_change_cwd__chdir_warningc C sr t �� }tjtd�� | �t j�t �� �t� W d � n1 s!w Y | �t j� t�� | �t �� |� d S )N��name)
rJ r� r �temp_cwdrG r' rK �basenamerI rL )r( �herer r r �
test_temp_cwd# s �zTestSupport.test_temp_cwdc C sv t �� }tjdd�� }| �||� | �t j�|�� | �t �� |� W d � n1 s,w Y | �t �� |� dS )z Test passing None to temp_cwd().Nr� ) rJ r� r r� �assertNotEqualr} rK r{ r' )r( r� r� r r r �test_temp_cwd__name_none+ s �z$TestSupport.test_temp_cwd__name_nonec C s | � t�dddd��d� d S )N� r
r� )r� r
r� z{1: 1, 2: 2, 3: 3})r' r �sortdictr2 r r r �
test_sortdict4 s zTestSupport.test_sortdictc C sT t �� }| �t��}t�|d� W d � n1 sw Y | �|jjtj � d S )Ns foo)
r � make_bad_fdr/ �OSErrorrJ �writer' � exception�errno�EBADF)r( �fd�cmr r r �test_make_bad_fd7 s
�z TestSupport.test_make_bad_fdc C sN t j| dddd� | �t�� t �| d� W d � d S 1 s w Y d S )Nz def classr� � )�lineno�offsetzx=1)r �check_syntax_errorr/ �AssertionErrorr2 r r r �test_check_syntax_error= s "�z#TestSupport.test_check_syntax_errorc C sB dd l }t�d�� |�d� W d � d S 1 sw Y d S )Nr �pprint)rc r � CleanImportr. )r( rc r r r �test_CleanImportB s "�z TestSupport.test_CleanImportc C sh t �dd�� | �dtj� | �dtj� W d � n1 s w Y | �dtj� | �dtj� d S )Nr- �bar)r �
DirsOnSysPathr> r: rK rA r2 r r r �test_DirsOnSysPathG s �z TestSupport.test_DirsOnSysPathc C sD t �� � }td� W d � n1 sw Y | �|�� d� d S )N�hello�hello
)r �captured_stdoutr` r' �getvalue)r( r; r r r �test_captured_stdoutN s
�z TestSupport.test_captured_stdoutc C sJ t �� �}tdtjd� W d � n1 sw Y | �|�� d� d S )Nr� r] r� )r �captured_stderrr` r: �stderrr' r� )r( r� r r r �test_captured_stderrS s
�z TestSupport.test_captured_stderrc C sR t �� �}|�d� |�d� t� }W d � n1 s w Y | �|d� d S )Nr� r r� )r �captured_stdinr� �seek�inputr' )r( �stdin�capturedr r r �test_captured_stdinX s
�z TestSupport.test_captured_stdinc C s t �� d S r8 )r �
gc_collectr2 r r r �test_gc_collect` s zTestSupport.test_gc_collectc C s | � t�� t� d S r8 )�assertIsInstancer �python_is_optimized�boolr2 r r r �test_python_is_optimizedc s z$TestSupport.test_python_is_optimizedc C s G dd� d�}|� }d|_ t�|dd��}| �|j d� | �|d� W d � n1 s,w Y | �|j d� t�|dd��}| �|jd� | �|� W d � n1 sVw Y | �t|d�� t�|dd��
|`W d � n1 sww Y | �t|d�� d S )Nc @ s e Zd ZdS )z'TestSupport.test_swap_attr.<locals>.ObjN)�__name__�
__module__� __qualname__r r r r �Objg s r� r� �xr� �y)r� r rU r' r� � assertIsNonerI �hasattr)r( r� �objr� r� r r r �test_swap_attrf s" � � �zTestSupport.test_swap_attrc C s� ddi}t �|dd��}| �|d d� | �|d� W d � n1 s$w Y | �|d d� t �|dd��}| �|d d� | �|� W d � n1 sPw Y | �d|� t �|dd�� |d= W d � n1 spw Y | �d|� d S )Nr� r� r� r� )r � swap_itemr' r� rA )r( �Dr� r� r r r �test_swap_itemw s � � �zTestSupport.test_swap_itemc @ � e Zd ZdZdZdZdZdS )zTestSupport.RefClassN)r� r� r� �
attribute1�
attribute2�_hidden_attribute1� __magic_1__r r r r �RefClass� �
r� c @ r� )zTestSupport.OtherClassN)r� r� r� r� �
attribute3r� � __magic_2__r r r r �
OtherClass� r� r� c C sB t �| j| j�}| �dh|� t �| j| j�}| �ddh|� d S )Nr� r� r� )r �detect_api_mismatchr� r� r' )r( �
missing_itemsr r r �test_detect_api_mismatch� s ��z$TestSupport.test_detect_api_mismatchc C sP g d�}t j| j| j|d�}| �t� |� t j| j| j|d�}| �t� |� d S )N)r� r� r� �
not_in_either)�ignore)r r� r� r� r' �set)r( r� r� r r r � test_detect_api_mismatch__ignore� s
�
�z,TestSupport.test_detect_api_mismatch__ignorec C sX dh}dh}t j| t||d� ddh}h d�}t j| td||d� | �tt j| t� d S )N�tempdir�template)�extra� not_exported�TextTestResult�installHandler> �
load_tests� TestProgram�
BaseTestSuite)zunittest.resultz
unittest.casezunittest.suitezunittest.loaderz
unittest.mainzunittest.runnerzunittest.signalszunittest.async_case)r � check__all__rx r0 r/ r� )r( r� r� r r r �test_check__all__� s �� z TestSupport.test_check__all__�waitpid�WNOHANGz need os.waitpid() and os.WNOHANGc C s� t �� t�� }|dkrt�d� t�� }t�� t j }t j}zHdt _t �
� } t�� |kr3| � d� t j
}z |t _
t �� W |t _
n|t _
w t jrKnt�d� q(d| }| �||�� � | �t j� W |t _n|t _w t �� d S )Nr FT�timeoutg�������?z2Warning -- reap_children() reaped child process %s)r �
reap_childrenrJ r� �_exit�time� monotonic�
SHORT_TIMEOUT�environment_altered�io�StringIO�failr: �
__stderr__�sleepr> r� r} )r( �pid�t0�deadline� was_alteredr� �
old_stderrr� r r r �test_reap_children� s6
� z TestSupport.test_reap_childrenNc C s� d|� d|� d�}t jg|�d�|�}dd� tj�� D �}tj|tjtjd|d�}|d u r/|}| � |j
� � t |�� | � |j
d � d S )
Nzfrom test.support import z
; print(repr(z()))r� c S s i | ] \}}|� d �s||�qS )�PYTHON)r� )r �key�valuer r r �
<dictcomp>� s �z-TestSupport.check_options.<locals>.<dictcomp>T)r; r� �universal_newlines�envr )r: �
executablerJ �environ�items�
subprocess�run�PIPE�DEVNULLr' r; �rstrip�repr�
returncode)r( �args�func�expected�code�cmdr �procr r r �
check_options� s �zTestSupport.check_optionsc C s� g dgdgdgdgdgdgdgdgd gd
gd gd d
gg d�d dgd dgd dgd dgd dgfD ] }| j |d�� | �|d� W d � n1 sGw Y q.| �g d�ddg� d S )Nz-B�-sz-S�-Ez-vz-bz-q�-Iz-bbz-vvv�-Wignore�-X�dev)r3 r4 r5 � faulthandler�
importtime� showrefcount� tracemallocz
tracemalloc=3��opts�args_from_interpreter_flags)r2 r1 r0 ��subTestr/ �r( r; r r r � test_args_from_interpreter_flags� s6 � �� �z,TestSupport.test_args_from_interpreter_flagsc C sT g dgdgdgfD ] }| j |d�� | �|d� W d � n1 s"w Y q d S )Nz-Oz-OOz-OOOOr: �!optim_args_from_interpreter_flagsr= r? r r r �&test_optim_args_from_interpreter_flags s � ���z2TestSupport.test_optim_args_from_interpreter_flagsc C s� G dd� d�}|d�}|d�}t �t dd ��� t �g � | �t �|�� | �t �|�� t �d d � | �t �|�� | �t �|�� t �|�� gd � | �t �|�� | �t �|�� t �dgd � | �t �|�� | �t �|�� t �dgd � | �t �|�� | �t �|�� t �dgd � | �t �|�� t �d gd � | �t �|�� t �d
gd � | �t �|�� | �t �|�� t �|�� |�� gd � | �t �|�� | �t �|�� t �d d gd � | �t �|�� | �t �|�� W d � n1 s�w Y t �t dd ��� t jg d
� | �t �|�� | �t �|�� t �d d � | �t �|�� | �t �|�� t �d |�� g� | �t �|�� | �t �|�� t �d dg� | �t �|�� | �t �|�� t �d dg� | �t �|�� | �t �|�� t �d dg� | �t �|�� t �d d g� | �t �|�� t �d d
g� | �t �|�� | �t �|�� t �d |�� |�� g� | �t �|�� | �t �|�� t �d d d g� | �t �|�� | �t �|�� W d � d S 1 �s�w Y d S )Nc @ s e Zd Zdd� Zdd� ZdS )z)TestSupport.test_match_test.<locals>.Testc S s
|| _ d S r8 ��test_id)r( rD r r r �__init__+ s
z2TestSupport.test_match_test.<locals>.Test.__init__c S s | j S r8 rC r2 r r r �id. s z,TestSupport.test_match_test.<locals>.Test.idN)r� r� r� rE rF r r r r �Test* s rG z"test.test_os.FileTests.test_accessz'test.test_os.Win32ErrorTests.test_chdir�_match_test_func�test_osztest_*� filetests� FileTestsz*test_os.*.test_*� test_access� DONTMATCH)�ignore_patterns)r rU �set_match_testsr} �
match_testrF rI )r( rG rL �
test_chdirr r r �test_match_test) s~
�/ $�zTestSupport.test_match_testc C sN t �� }t�ttj�}z t �� }W t�|� nt�|� w | �|| d� d S )Nr� )r �fd_countrJ rF �__file__�O_RDONLYrm r' )r( �startr� �morer r r �
test_fd_count� s
zTestSupport.test_fd_countc C sF t �� }tj}z
|t_t�|� W |t_n|t_w | �|�� |� d S r8 )r r r: r r �
print_warningr' r� )r( r� r+ r� r r r r �check_print_warning� s z TestSupport.check_print_warningc C s | � dd� | � dd� d S )Nr� zWarning -- msg
za
bzWarning -- a
Warning -- b
)rZ r2 r r r �test_print_warning� s ��z TestSupport.test_print_warningr8 );r� r� r� � classmethodr r r+ r3 r5 r6 r0 �skipr< rB rN r\ ri rp rs rv r� r� r� r� �
skipUnlessr� rJ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r/ r@ rB rR rX rZ r[ r r r r r st
�
/" i r �__main__)r� rc r rJ r� rj rW r" r: rx r� r
r0 r �testr � test.supportr r r r r rG �TestCaser r� �mainr r r r �<module> s< 3 �