File: //usr/lib/python3.9/test/__pycache__/test_support.cpython-39.pyc
a
kx�g�_ � @ 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
mZ d dlmZ d dlmZ ejZG dd� de j�Zdd� Zed kr�e� dS )
� N)�support)�
script_helper)�
socket_helperc @ s� e Zd Zdd� Zdd� Zdd� Ze�d�dd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze�eed �d!�d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,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)G dHdI� dI�Z*G dJdK� dK�Z+dLdM� Z,dNdO� Z-dPdQ� Z.e�eedR��o�eedS�dT�dUdV� �Z/dfdXdY�Z0dZd[� Z1d\d]� Z2d^d_� Z3d`da� Z4dbdc� Z5ddde� Z6dWS )g� TestSupportc C s t �d� | �tjt jd� d S )N�ftplib�foo)r �
import_module� assertRaises�unittest�SkipTest��self� r �'/usr/lib/python3.9/test/test_support.py�test_import_module 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 )� assertEqualr 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_stdout# s z$TestSupport.test_get_original_stdoutc C s2 dd l }| �dtj� t�d� | �dtj� d S )Nr �sched)r �assertInr �modulesr Zunload� assertNotIn)r
r r r r � test_unload' s
zTestSupport.test_unloadc C sT t td��}W d � n1 s 0 Y t�t� | �tj�t�� t�t� d S )N�w)�open�TESTFNr �unlink� assertFalse�os�path�exists)r
�fr r r � test_unlink- s
zTestSupport.test_unlinkc C sb t jd }tj�|d�}t�|� t�|� t �|� | �tj�|�� t � t dd�� t �|� W d � n1 sv0 Y t�|� t�|� t�
|t j t j
B � t � t dd�� t �|� W d � n1 s�0 Y | �tj�|�� t�|� t�|� t�
|d� t � t dd�� t �|� W d � n1 �sB0 Y | �tj�|�� d S )N�dZsubdir�verboser )r r r" r# �join�mkdir�rmtreer! r$ � swap_attr�chmod�stat�S_IRUSR�S_IXUSR)r
�dirpathZ
subdirpathr r r � test_rmtree4 s(
(
(
*zTestSupport.test_rmtreec C s� t d }t|d�� }td|d� W d � n1 s40 Y tj�dtj� t� � zNt
t �}| � t tj � t
�t � | �t tj � W tjd= t
�|� t
�d� n tjd= t
�|� t
�d� 0 d S )Nz.pyr zfoo = 1��filer � __pycache__)r r �printr r# �insertr" �curdir� importlib�invalidate_caches�
__import__r r r Zforgetr r r+ )r
Z mod_filenamer% �modr r r � test_forgetL 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 r )r Zfind_unused_portr> r? r@ rA )r
ZportrC r r r �test_find_unused_porta s z!TestSupport.test_find_unused_portc C s& t � � }t�|� |�� |�� d S r )r> r Z bind_portZlistenrA rB r r r �test_bind_portf s
zTestSupport.test_bind_portc C s� t �� }tj�|�}z�tj�|d�}| �tj�|�� t� |��.}| �
||� | � tj�|�� W d � n1 st0 Y | �tj�|�� W t� |� n t� |� 0 dS )z8Test that temp_dir() creates and destroys its directory.ZtempN)
�tempfile�mkdtempr" r# �realpathr) r! �isdirr �temp_dirr �
assertTruer+ )r
�
parent_dirr# � temp_pathr r r �
test_temp_dirn s 0zTestSupport.test_temp_dirc C sP t �� �"}| �tj�|�� W d � n1 s00 Y | �tj�|�� dS )zTest passing no path.N)r rK rL r" r# rJ r! )r
rN r r r �test_temp_dir__path_none} s
0z$TestSupport.test_temp_dir__path_nonec C sn dd� }t �� }tj�|�}z@| �tj�|�� | �t||� | �tj�|�� W t �
|� n t �
|� 0 dS )z-Test passing a directory that already exists.c S s6 t �| ��}td��W d � n1 s(0 Y d S �Nzshould not get here)r rK � Exception)r# rN r r r �
call_temp_dir� s zMTestSupport.test_temp_dir__existing_dir__quiet_default.<locals>.call_temp_dirN) rG rH r" r# rI rL rJ r �FileExistsError�shutilr+ )r
rS r# r r r �*test_temp_dir__existing_dir__quiet_default� s z6TestSupport.test_temp_dir__existing_dir__quiet_defaultc C s� t �� }tj�|�}z�t�� �Z}tj|dd�� }| �||� W d � n1 sP0 Y dd� |j D �}W d � n1 s~0 Y | �
tj� |�� W t �
|� n t �
|� 0 | �t|�d|� |d }| �
|�d|�d ��|� dS )
z=Test passing a directory that already exists with quiet=True.T��quietNc S s g | ]}t |j��qS r ��str�message��.0r r r r �
<listcomp>� � zGTestSupport.test_temp_dir__existing_dir__quiet_true.<locals>.<listcomp>� r z5tests may fail, unable to create temporary directory �: )rG rH r" r# rI r �check_warningsrK r �warningsrL rJ rU r+ �len�
startswith)r
r# �recorderrN rc �warnr 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
with support.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�textwrapZdedentr r r r �test_temp_dir__forked_child� s z'TestSupport.test_temp_dir__forked_childc C s� t �� }t�� �V}t�|��,}| �||� | �t �� |� W d � n1 sN0 Y W d � n1 sl0 Y | �t �� |� d S r )r" �getcwdr rK �
change_cwdr )r
� original_cwdrN �new_cwdr r r �test_change_cwd� s
LzTestSupport.test_change_cwdc C sh t �� }dd� }t�� �,}t j�|d�}| �t||� W d � n1 sJ0 Y | �t �� |� dS )z&Test passing a non-existent directory.c S s6 t �| ��}td��W d � n1 s(0 Y d S rQ )r rn rR )r# rp r r r �call_change_cwd� s zFTestSupport.test_change_cwd__non_existent_dir.<locals>.call_change_cwd�does_not_existN) r" rm r rK r# r) r �FileNotFoundErrorr )r
ro rr 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�� ��}t j�|d�}t�� �j}tj|dd��,}| �||� | �t �� |� W d � n1 sj0 Y dd� |j D �}W d � n1 s�0 Y W d � n1 s�0 Y | �t
|�d|� |d }| � |� d |�d
��|� dS ) z6Test passing a non-existent directory with quiet=True.rs TrW Nc S s g | ]}t |j��qS r rY 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 ra )
r" rm r rK r# r) rb rn r rc rd rL re )r
ro rM Zbad_dirrf rp rc rg r r r �-test_change_cwd__non_existent_dir__quiet_true� s
.L
��z9TestSupport.test_change_cwd__non_existent_dir__quiet_truec C s� t d }t�� �N}tj|dd�� W d � n1 s60 Y dd� |jD �}W d � n1 sd0 Y | �t|�d|� |d }| �|�d |�d
��|� dS ) z0Check the warning message when os.chdir() fails.Z_does_not_existT)r# rX Nc S s g | ]}t |j��qS r rY r\ r r r r^ � r_ z>TestSupport.test_change_cwd__chdir_warning.<locals>.<listcomp>r` r rv ra ) r r rb rn rc r rd rL re )r
r# rf Zmessages�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 sB0 Y | �t j� t�� | �t �� |� d S )N��name)
r" rm r �temp_cwdr r r# �basenamer! r$ )r
�herer r r �
test_temp_cwd s
6zTestSupport.test_temp_cwdc C sv t �� }tjdd��>}| �||� | �t j�|�� | �t �� |� W d � n1 sX0 Y | �t �� |� dS )z Test passing None to temp_cwd().Nrz ) r" rm r r| ZassertNotEqualrL r# rJ r )r
ro rp 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` z{1: 1, 2: 2, 3: 3})r r Zsortdictr r r r �
test_sortdict s zTestSupport.test_sortdictc C sT t �� }| �t�� }t�|d� W d � n1 s40 Y | �|jjtj � d S )Ns foo)
r Z make_bad_fdr �OSErrorr" �writer Z exception�errnoZEBADF)r
�fd�cmr r r �test_make_bad_fd s *z TestSupport.test_make_bad_fdc C sL t j| dddd� | �t�� t �| d� W d � n1 s>0 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 s@ dd l }t�d�� |�d� W d � n1 s20 Y d S )Nr Zasyncore)r9 r Z CleanImportr )r
r9 r r r �test_CleanImport s z TestSupport.test_CleanImportc C sh t �dd��, | �dtj� | �dtj� W d � n1 s>0 Y | �dtj� | �dtj� d S )Nr Zbar)r Z
DirsOnSysPathr r r# r r r r r �test_DirsOnSysPath% s
,z TestSupport.test_DirsOnSysPathc C sD t �� �}td� W d � n1 s&0 Y | �|�� d� d S )N�hello�hello
)r Zcaptured_stdoutr6 r �getvalue)r
r r r r �test_captured_stdout, s
&z TestSupport.test_captured_stdoutc C sJ t �� � }tdtjd� W d � n1 s,0 Y | �|�� d� d S )Nr� r3 r� )r Zcaptured_stderrr6 r �stderrr r� )r
r� r r r �test_captured_stderr1 s
,z TestSupport.test_captured_stderrc C sR t �� �*}|�d� |�d� t� }W d � n1 s80 Y | �|d� d S )Nr� r r� )r Zcaptured_stdinr� �seek�inputr )r
�stdinZcapturedr r r �test_captured_stdin6 s
$z TestSupport.test_captured_stdinc C s t �� d S r )r Z
gc_collectr r r r �test_gc_collect> s zTestSupport.test_gc_collectc C s | � t�� t� d S r )ZassertIsInstancer Zpython_is_optimized�boolr r r r �test_python_is_optimizedA s z$TestSupport.test_python_is_optimizedc C s G dd� d�}|� }d|_ t�|dd��*}| �|j d� | �|d� W d � n1 sX0 Y | �|j d� t�|dd��(}| �|jd� | �|� W d � n1 s�0 Y | �t|d�� t�|dd�� |`W d � n1 s�0 Y | �t|d�� d S )Nc @ s e Zd ZdS )z'TestSupport.test_swap_attr.<locals>.ObjN)�__name__�
__module__� __qualname__r r r r �ObjE s r� r` �xr� �y)r� r r, r r� � assertIsNoner! �hasattr)r
r� �objr� r� r r r �test_swap_attrD s *("zTestSupport.test_swap_attrc C s� ddi}t �|dd��,}| �|d d� | �|d� W d � n1 sH0 Y | �|d d� t �|dd��*}| �|d d� | �|� W d � n1 s�0 Y | �d|� t �|dd�� |d= W d � n1 s�0 Y | �d|� d S )Nr� r` r� r� )r Z swap_itemr r� r )r
�Dr� r� r r r �test_swap_itemU s *( $zTestSupport.test_swap_itemc @ s e Zd ZdZdZdZdZdS )zTestSupport.RefClassN)r� r� r� �
attribute1�
attribute2Z_hidden_attribute1� __magic_1__r r r r �RefClassc s r� c @ s e Zd ZdZdZdZdZdS )zTestSupport.OtherClassN)r� r� r� r� �
attribute3r� � __magic_2__r r r r �
OtherClassi s 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_mismatcho 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__ignorex 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 )NZtempdir�template)�extra� blacklistZTextTestResultZinstallHandler> Z TestProgramZ
BaseTestSuiteZ
load_tests)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�� }|dkr"t�d� t�� }t�� t j }t j}z�dt _t �
� }t�� |krd| � d� t j
}z|t _
t �� W |t _
n|t _
0 t jr�q�t�d� qNd| }| �||�� � | �t j� W |t _n|t _0 t �� d S )Nr FZtimeoutg�������?z2Warning -- reap_children() reaped child process %s)r Z
reap_childrenr" ri �_exit�time� monotonicZ
SHORT_TIMEOUTZenvironment_altered�io�StringIOZfailr �
__stderr__�sleepr r� rL )r
�pidZt0ZdeadlineZ was_alteredr� �
old_stderrrx r r r �test_reap_children� s2
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()))rj c S s i | ]\}}|� d �s||�qS )ZPYTHON)re )r] �key�valuer r r �
<dictcomp>� s
�z-TestSupport.check_options.<locals>.<dictcomp>T)r r� Zuniversal_newlines�envr )r �
executabler" �environ�items�
subprocess�run�PIPEZDEVNULLr 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 � q\1 s�0 Y q\| �g d�ddg� d S )Nz-B�-sz-S�-Ez-vz-bz-q�-Iz-bbz-vvv�-Wignore�-X�dev)r� r� r� Z faulthandlerZ
importtimeZ showrefcount� tracemallocz
tracemalloc=3��optsZargs_from_interpreter_flags)r� r� r� �ZsubTestr� �r
r� r r r � test_args_from_interpreter_flags� s2 �, �z,TestSupport.test_args_from_interpreter_flagsc C sT g dgdgdgfD ]<}| j |d�� | �|d� W d � q1 sD0 Y qd 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�0 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 � n1 �s�0 Y d S )Nc @ s e Zd Zdd� Zdd� ZdS )z)TestSupport.test_match_test.<locals>.Testc S s
|| _ d S r ��test_id)r
r� r r r �__init__ s z2TestSupport.test_match_test.<locals>.Test.__init__c S s | j S r 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)Zignore_patterns)r r, Zset_match_testsrL Z
match_testr� r! )r
r� r� Z
test_chdirr r r �test_match_test sz
0 zTestSupport.test_match_testc C sN t �� }t�ttj�}zt �� }W t�|� n t�|� 0 | �|| d� d S )Nr` )r Zfd_countr" r �__file__�O_RDONLYrA r )r
�startr� Zmorer r r �
test_fd_countp s
zTestSupport.test_fd_countc C sF t �� }tj}z|t_t�|� W |t_n|t_0 | �|�� |� d S r )r� r� r r� r Z
print_warningr r� )r
rx r� r� r� r r r �check_print_warning| s z TestSupport.check_print_warningc C s | � dd� | � dd� d S )Nrx zWarning -- msg
za
bzWarning -- a
Warning -- b
)r� r r r r �test_print_warning� s ��z TestSupport.test_print_warning)N)7r� r� r� r r r r
�skipr r r& r2 r= rD rE rF rO rP rV rh Z
skipUnlessr� r" rl rq ru rw ry 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 sh
�
/
" i r c C s t g} tj| � d S r )r r Z run_unittest)Ztestsr r r � test_main� s r� �__main__)r� r9 r� r" rU r> r. r� r rG rk r� r
�testr Z test.supportr r r ZTestCaser r� r� r r r r �<module> s2