shell bypass 403
o
a bVe � @ 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.*)Zlikeztest.test_supportz!.*You should NOT be seeing this.*� )�len�warnings�filtersr Zignore_deprecations_from�_warnings_helper_token�_test_support_token��clsZorig_filter_len� r �:/usr/local/python-3.10/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 Zclear_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 Z_warn_about_deprecation�warn�DeprecationWarning� assertEqualr
)�selfZ 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 Zimport_fresh_moduler, 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 Z
get_attributer/ r) r* r+ r, r r r r/ ? s �z TestSupport.test_get_attributezfailing buildbotsc C s | � t�� tj� d S �N)r" r Zget_original_stdout�sys�stdoutr, 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)r4 �assertInr1 �modulesr Zunload� assertNotIn)r# r4 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�dZsubdir�verboser )r r= r@ rA �join�mkdir�rmtreer? rB r � swap_attr�chmod�stat�S_IRUSR�S_IXUSR)r# �dirpathZ
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 r9 r: zfoo = 1��filer � __pycache__)r= r< �printr1 rA �insertr@ �curdir� importlib�invalidate_caches�
__import__r5 r6 r Zforgetr7 r r> rI )r# Z mod_filenamerC �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 r0 )r Zfind_unused_portr\ r] r^ r_ )r# �portra r r r �test_find_unused_port� s z!TestSupport.test_find_unused_portc C s& t � � }t�|� |�� |�� d S r0 )r\ r Z bind_port�listenr_ r` 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.ZtempN)
�tempfile�mkdtempr@ rA �realpathrG r? �isdirr �temp_dirr" �
assertTruerI )r# �
parent_dirrA � 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 rk rl r@ rA rj r? )r# rn 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 rk � Exception)rA rn r r r �
call_temp_dir� � �zMTestSupport.test_temp_dir__existing_dir__quiet_default.<locals>.call_temp_dirN) rg rh r@ rA ri rl rj r) �FileExistsError�shutilrI )r# rt rA 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 �: )rg rh r@ rA ri r �check_warningsr rk r" r rl rj rw rI r
�
startswith)r# rA �recorderrn 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 Zassert_python_ok�textwrap�dedentr, 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 r0 )r@ �getcwdr rk �
change_cwdr" )r# � original_cwdrn �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 rq rr )r r� rs )rA r� r r r �call_change_cwd� ru zFTestSupport.test_change_cwd__non_existent_dir.<locals>.call_change_cwd�does_not_existN) r@ r� r rk rA rG r) �FileNotFoundErrorr" )r# r� r� rm Znon_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� Try 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| )r@ r� r rk rA rG r r} r� r" r r
rl r~ )r# r� rm Zbad_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.Z_does_not_existT)rA rz Nc S r r r r r r r r r z>TestSupport.test_change_cwd__chdir_warning.<locals>.<listcomp>r{ r r� r| )
r= r r} r r� r r" r
rl r~ )r# rA 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)
r@ r� r �temp_cwdr= r" rA �basenamer? rB )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� ) r@ r� r r� ZassertNotEqualrl rA rj 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 Zsortdictr, 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 Z make_bad_fdr) �OSErrorr@ �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 Zcheck_syntax_errorr) �AssertionErrorr, 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 Zpprint)rW r Z CleanImportr( )r# rW 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' Zbar)r Z
DirsOnSysPathr5 r1 rA r7 r, 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 Zcaptured_stdoutrT r" �getvalue)r# r2 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� rQ r� )r Zcaptured_stderrrT r1 �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 Zcaptured_stdinr� �seek�inputr" )r# �stdinZcapturedr r r �test_captured_stdinX s
�z TestSupport.test_captured_stdinc C s t �� d S r0 )r �
gc_collectr, r r r �test_gc_collect` s zTestSupport.test_gc_collectc C s | � t�� t� d S r0 )ZassertIsInstancer Zpython_is_optimized�boolr, 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 rJ r" r� � assertIsNoner? �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 Z swap_itemr" r� r7 )r# �Dr� r� r r r �test_swap_itemw s � � �zTestSupport.test_swap_itemc @ � e Zd ZdZdZdZdZdS )zTestSupport.RefClassN)r� r� r� �
attribute1�
attribute2Z_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� Z
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_exportedZTextTestResultZinstallHandler> Z TestProgramZ
load_testsZ
BaseTestSuite)zunittest.resultz
unittest.casezunittest.suitezunittest.loaderz
unittest.mainzunittest.runnerzunittest.signalszunittest.async_case)r Z check__all__rg r* 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 jrKqQt�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_childrenr@ r� �_exit�time� monotonicZ
SHORT_TIMEOUTZenvironment_altered�io�StringIOZfailr1 �
__stderr__�sleepr5 r� rl )r# �pid�t0�deadlineZ 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 )ZPYTHON)r~ )r �key�valuer r r �
<dictcomp>� s �z-TestSupport.check_options.<locals>.<dictcomp>T)r2 r� �universal_newlines�envr )r1 �
executabler@ �environ�items�
subprocess�run�PIPE�DEVNULLr" r2 �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)r r r � faulthandler�
importtime� showrefcount� tracemallocz
tracemalloc=3��optsZargs_from_interpreter_flags)r r r� �ZsubTestr� �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 Z!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 r0 ��test_id)r# r r r r �__init__+ s
z2TestSupport.test_match_test.<locals>.Test.__init__c S s | j S r0 r r, r r r �id. s z,TestSupport.test_match_test.<locals>.Test.idN)r� r� r� r r r r r r �Test* s r z"test.test_os.FileTests.test_accessz'test.test_os.Win32ErrorTests.test_chdirZ_match_test_funcZtest_osztest_*Z filetestsZ FileTestsz*test_os.*.test_*� test_accessZ DONTMATCH)�ignore_patterns)r rJ Zset_match_testsrl Z
match_testr r? )r# r r Z
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 Zfd_countr@ r< �__file__�O_RDONLYr_ r" )r# �startr� Zmorer r r �
test_fd_count� s
zTestSupport.test_fd_countc C sF t �� }tj}z
|t_t�|� W |t_n|t_w | �|�� |� d S r0 )r� r� r1 r� r Z
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
)r r, r r r �test_print_warning� s ��z TestSupport.test_print_warningr0 );r� r� r� � classmethodr r r% r- r. r/ r* �skipr3 r8 rD rP r[ rb rd rf ro rp rx r� Z
skipUnlessr� 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� r
r r r r r r r r r r st
�
/" i r �__main__)r� rW r� r@ rw r\ rL r� r1 rg r� r� r* r �testr Z test.supportr r r r r r= ZTestCaser r� �mainr r r r �<module> s< 3 �