File: //usr/local/lib/python3.10/test/__pycache__/test_site.cpython-310.pyc
o
�i2o � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z m
Z
m Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZ dd l m Z ej j rye� d
��ddl Z e j!duZ"da#d d � Z$d
d� Z%G dd� dej&�Z'G dd� de(�Z)G dd� dej&�Z*G dd� dej&�Z+e�,ej-dkd�G dd� dej&��Z.e/dkr�e�0� dS dS ) zTests for 'site'.
Tests assume the initial paths in sys.path once the interpreter has begun
executing have not been removed.
� N)�support)� os_helper)�
socket_helper)�captured_stderr)�TESTFN�EnvironmentVarGuard�
change_cwd)�mock)�copyzPython was invoked with -Sc
C sv t jd d � atjr7tj�tj�s9zt�tj� t� tj� W d S t
y6 } z
t � dtj| f ��d } ~ ww d S d S )Nz-unable to create user site directory (%r): %s)
�sys�path� OLD_SYS_PATH�site�ENABLE_USER_SITE�os�isdir� USER_SITE�makedirs�
addsitedir�PermissionError�unittest�SkipTest)�exc� r �+/usr/local/lib/python3.10/test/test_site.py� setUpModule, s ����r c C s t tjd d �< d S �N)r
r r r r r r �tearDownModule; s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z de
fdd�Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Ze�eed �d �d d � �Ze�ejd!kd"�d#d$� �Zd%d&� Ze�ed'�d(d)� �Z e�e j d*�d+d,� �Z e�ed'�d-d.� �Z e�ed'�d/d0� �Z!d1d2� Z"e�ed'�d3d4� �Z#d5d6� Z$d7S )8�HelperFunctionsTestsz Tests for helper functions.
c C s@ t jdd� | _tj| _tj| _tj| _ t
j | _ t
t
j �| _dS )zSave a copy of sys.pathN)r r �sys_pathr � USER_BASE�old_baser �old_site�PREFIXES� old_prefixes� sysconfig� _CONFIG_VARS�
original_varsr
�old_vars��selfr r r �setUpC s zHelperFunctionsTests.setUpc C sZ | j tjdd�< | jt_| jt_| jt_ | j
t _ t j dur+t j �
� t j �| j� dS dS �zRestore sys.pathN)r r r r! r r r" r r$ r# r' r% r& �clear�updater( r) r r r �tearDownL s
�z HelperFunctionsTests.tearDownc C sj d}t jj|� }tj|� \}}| �t j�|�|� |t j�|�kr)| �||� d S | �t j�|�|� d S )N)Z BeginningZEnd)r r �joinr �makepath� assertEqual�abspath�normcase)r* �
path_partsZ original_dirZabs_dirZnorm_dirr r r �
test_makepathX s z"HelperFunctionsTests.test_makepathc C s8 t �� }dd� tjD �D ]
}| �||d||f � q d S )Nc S s* g | ]}|rt j�|�rt�|�d �qS )� )r r �existsr r1 )�.0r r r r �
<listcomp>g s
�
�z;HelperFunctionsTests.test_init_pathinfo.<locals>.<listcomp>zB%s from sys.path not found in set returned by _init_pathinfo(): %s)r �_init_pathinfor r �assertIn)r* Zdir_set�entryr r r �test_init_pathinfoe s ���z'HelperFunctionsTests.test_init_pathinfoc C sJ | � |jtjd|j � | � t�|j�d tj� | �t j�
|j �� dS )z>Contain common code for testing results of reading a .pth filez%s not in sys.modulesr N) r< �importedr �modulesr r1 �
good_dir_pathr � assertFalser r8 � bad_dir_path�r* �pth_filer r r �pth_file_testsm s
�z#HelperFunctionsTests.pth_file_testsc C sR t � }|jdd� z|�� t�|j|jt� � | �|� W |�� d S |�� w �NT��prep) �PthFile�cleanup�creater �
addpackage�base_dir�filename�setrF rD r r r �test_addpackaget s z$HelperFunctionsTests.test_addpackage�.c sz t j�|�}|d }t j�||�� t� ddd��}| �� fdd�� |�|� W d � ||fS 1 s4w Y ||fS )N�.pth�w�utf-8)�encodingc s
t �� �S r )r �remover ��pth_fnr r �<lambda>� s
z/HelperFunctionsTests.make_pth.<locals>.<lambda>)r r r3 r0 �open�
addCleanup�write)r* �contents�pth_dirZpth_nameZ pth_basenamerE r rX r �make_pth� s
��z HelperFunctionsTests.make_pthc C s� | � d�\}}t� �}t�||t� � W d � n1 s w Y | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� | �|�� d� d S )Nzimport bad-syntax
zline 1� Tracebackzimport bad-syntax� SyntaxError� r` r r rM rP Z assertRegex�getvalue�re�escaper r r0 �r* r_ rY �err_outr r r �!test_addpackage_import_bad_syntax� s �
�z6HelperFunctionsTests.test_addpackage_import_bad_syntaxc C s� | � d�\}}t� �}t�||t� � W d � n1 s w Y | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� d S )Nz randompath
import nosuchmodule
zline 2ra �ModuleNotFoundErrorrc rg r r r � test_addpackage_import_bad_exec� s �
�z4HelperFunctionsTests.test_addpackage_import_bad_execc C s0 | � d�\}}t�||t� �}| �|t� � d S )Nz
)r` r rM rP r2 )r* r_ rY � known_pathsr r r �test_addpackage_empty_lines� s z0HelperFunctionsTests.test_addpackage_empty_linesc C s� | � d�\}}t� �}| �t�||t� �� W d � n1 s w Y d | _| �|�� d� t j
D ]
}t |t �r@| �
d|� q3d S )Nzabc def
� zabc def)r` r rB r rM rP ZmaxDiffr2 rd r r �
isinstance�str� assertNotIn)r* r_ rY rh r r r r �#test_addpackage_import_bad_pth_file� s �
��z8HelperFunctionsTests.test_addpackage_import_bad_pth_filec C sN t � }|jdd� z|�� t�|jt� � | �|� W |�� d S |�� w rG )rJ rK rL r r rN rP rF rD r r r �test_addsitedir� s z$HelperFunctionsTests.test_addsitedirc C sp t d�}|jdd� z(|�� t�|jt� � | �t�|j �d t
j � | � |jt
j � W |�� d S |�� w )Nz.dotfileTrH r )
rJ rK rL r r rN rP rq r1 rA r r r< rD r r r �test_addsitedir_dotfile� s z,HelperFunctionsTests.test_addsitedir_dotfile�chflagsztest needs os.chflags()c C s� t � }|jdd� z9|�� t�|j�}t�|j|jtjB � t �
|j t � � | �
t �|j�d tj� | �|j tj� W |�� d S |�� w )NTrH r )rJ rK rL r �stat� file_pathru �st_flags� UF_HIDDENr r rN rP rq r1 rA r r r< )r* rE �str r r � test_addsitedir_hidden_flags� s z1HelperFunctionsTests.test_addsitedir_hidden_flags�win32ztest needs Windowsc C s� t � }|jdd� z1|�� t�dd|jg� t�|jt � � | �
t� |j �d t
j� | �|jt
j� W |�� d S |�� w )NTrH Zattribz+Hr )rJ rK rL �
subprocess�
check_callrw r r rN rP rq r1 rA r r r< rD r r r �%test_addsitedir_hidden_file_attribute� s z:HelperFunctionsTests.test_addsitedir_hidden_file_attributec C s | � t�� t�� � d S r )r2 r � _getuserbaser% r) r r r �test__getuserbase� s z&HelperFunctionsTests.test__getuserbasezneed user sitec C sJ t jdkr t jr d}ntjd }| �tj�t� t�
� ��t � d|�� d S )N�darwinZosx_framework_user�_userZpurelib)
r �platform�
_frameworkr �namer2 r �normpathr � _get_pathr� r% Zget_path)r* �schemer r r �
test_get_path� s
�z"HelperFunctionsTests.test_get_pathz<requires access to PEP 370 user-site (site.ENABLE_USER_SITE)c C s0 t j�tj�}| �|tj� t j�� }t j
tj dd| g|d�}| � |d� t j�� }t j
tj ddd| g|d�}|t�
� d krH| � |d� n| � |dd� t j�� }d|d < t j
tj dd| g|d�}|t�
� d krt| � |d� n| � |dd
� t j�� }d |d < t j
tj dd
g|d�}| � |dd� d S )N�-cz$import sys; sys.exit(%r in sys.path)��envr7 z-sr z%User site still added to path with -s�1ZPYTHONNOUSERSITEz3User site still added to path with PYTHONNOUSERSITEz/tmp�PYTHONUSERBASEz=import sys, site; sys.exit(site.USER_BASE.startswith("/tmp"))z#User base not set by PYTHONUSERBASE)r r r� r r r< r �environr
r} �call�
executabler2 �getsitepackages)r* Zusersiter� �rcr r r �
test_s_option� sP
��
��
���
���z"HelperFunctionsTests.test_s_optionc C s~ d t _t �� }| �t j|� d t _dd l}d |_t� �}d|d< | �t �� �d�t �� � W d � d S 1 s8w Y d S )Nr �xoxor� ) r r � getuserbaser2 r% r& r �
assertTrue�
startswith)r* � user_baser% r� r r r �test_getuserbase# s �"�z%HelperFunctionsTests.test_getuserbasec C sL d t _d t _t �� }| �t j|� | �|�t j�|� | �t jt �� � d S r )r r r �getusersitepackagesr2 r� r� r� )r* � user_siter r r �test_getusersitepackages5 s z-HelperFunctionsTests.test_getusersitepackagesc C s� dgt _t �� }tjdkrVtjdkr4| �t|�d� tj �
dtjdtj d d� d�}| �|d |� n| �t|�d� tj �
dddtj d d� d�}| �|d |� d S | �t|�d� | �|d d� tj �
ddd�}| �|d |� d S )
Nr� �/�lib� z python%d.%dz
site-packagesr r7 ���) r r# r� r �sepr �
platlibdirr2 �lenr r0 � version_info)r* �dirsZwantedr r r �test_getsitepackages@ s(
�
�z)HelperFunctionsTests.test_getsitepackagesc
C st d t _d t _t� �C}t�ddd� ��+ |d= |d= t �� }| �|�dt j
�|� t � � }| �|�|�|� W d � n1 s@w Y W d � n1 sOw Y tjddd ��U}tj� t d
��5}t
�t d d �� t� }t �|� |�|� |�� | �|� W d � n1 s�w Y W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )
Nzos.path.expanduserc S s | S r r )r r r r rZ a s z=HelperFunctionsTests.test_no_home_directory.<locals>.<lambda>r� �APPDATA�~z
os.path.isdirF)Z return_valuer r T)r r r r r �patchr� r� r� r r� r� �objectr Z swap_attrrP �addusersitepackagesZassert_called_once_withZassert_not_calledrB )r* r� r� r� Z
mock_isdirZmock_addsitedirrl r r r �test_no_home_directoryX s< �� ��
� �
��P z+HelperFunctionsTests.test_no_home_directoryc
C s� d}d|d fdfD ]D\}}t �dt j|d���. t �dt�� �� t�|� | �tj �
� |� W d � n1 s:w Y W d � n1 sIw Y q
d S )Nz bla-bla-blaT�
)Frn z sys.flags)�verbosez
sys.stderr) r r� ZMock�io�StringIOr �_tracer2 r �stderrrd )r* �messager� �outr r r �
test_tracez s �
�� ��z HelperFunctionsTests.test_traceN)%�__name__�
__module__� __qualname__�__doc__r+ r/ r6 r> rF rQ r r` ri rk rm rr rs rt r �
skipUnless�hasattrr r{ r r� r r� �
HAS_USER_SITEr� r r r� r� r� r� r� r� r r r r r ? s@
'
!r c @ s4 e Zd ZdZedddfdd�Zdd� Zd
d
d �Zd S )rJ z/Helper class for handling testing of .pth files�timeZ __testdir__Z__badc C sj |d | _ tj�d�| _tj�| j| j �| _|| _|| _|| _ tj�| j| j�| _
tj�| j| j �| _ dS )z Initialize instance variablesrS rn N) rO r r r3 rN r0 rw r? � good_dirname� bad_dirnamerA rC )r* �
filename_baser? r� r� r r r �__init__� s
zPthFile.__init__c C sz t | jd�}z)td|d� td|d� td| j |d� t| j|d� t| j|d� W |�� n|�� w t�| j � dS )ao Create a .pth file with a comment, blank lines, an ``import
<self.imported>``, a line with self.good_dirname, and a line with
self.bad_dirname.
Creation of the directory for self.good_dir_path (based off of
self.good_dirname) is also performed.
Make sure to call self.cleanup() to undo anything done by this method.
rT z#import @bad module name��filer� z import %sN)
r[ rw �printr? r� r� �closer �mkdirrA )r* �FILEr r r rL � s zPthFile.createFc C s� t j�| j�r
t �| j� |r tj�| j�| _ | j r tj| j= n
| j r*| j tj| j< t j�| j
�r7t � | j
� t j�| j �rFt � | j � dS dS )z�Make sure that the .pth file is deleted, self.imported is not in
sys.modules, and that both self.good_dirname and self.bad_dirname are
not existing directories.N)
r r r8 rw rW r r@ �getr? Zimported_modulerA �rmdirrC )r* rI r r r rK � s
� �zPthFile.cleanupN)F)r� r� r� r� r r� rL rK r r r r rJ � s
� rJ c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Ze� d
�d d � �Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zej�d�ejje�eejd�d�dd� ���Zd S ) �ImportSideEffectTestsz(Test side-effects from importing 'site'.c C s t jdd� | _dS )zMake a copy of sys.pathN)r r r r) r r r r+ � � zImportSideEffectTests.setUpc C s | j tjdd�< dS r, )r r r r) r r r r/ � r� z ImportSideEffectTests.tearDownc C s* dt jd _t�� | �t jd j� dS )z�Test for __cached__ is None.
Regarding to PEP 3147, __cached__ can be None.
See also: https://bugs.python.org/issue30167
N�test)r r@ �
__cached__r � abs_pathsZ assertIsNoner) r r r �test_abs_paths_cached_None� s z0ImportSideEffectTests.test_abs_paths_cached_Nonec C s4 t �� t� }tjD ]
}| �||� |�|� q
d S r )r �removeduppathsrP r r rq �add)r* Z
seen_pathsr r r r �test_no_duplicate_paths� s
�z-ImportSideEffectTests.test_no_duplicate_pathsztest not implementedc C s d S r r r) r r r �test_add_build_dir� s z(ImportSideEffectTests.test_add_build_dirc C s$ | � ttd�� | � ttd�� d S )N�quit�exit�r� r� �builtinsr) r r r �test_setting_quit� s z'ImportSideEffectTests.test_setting_quitc C s4 | � ttd�� | � ttd�� | � ttd�� d S )N� copyright�credits�licenser� r) r r r �test_setting_copyright� s z,ImportSideEffectTests.test_setting_copyrightc C s | � ttd�� d S )N�helpr� r) r r r �test_setting_help� r� z'ImportSideEffectTests.test_setting_helpc C sZ t jdkr)dd l}|�� d �d�r+tjj�� D ] }|dkr! d S q| �d� d S d S d S )Nr| r r7 Zcp�mbcszdid not alias mbcs) r r� �locale�getdefaultlocaler� � encodings�aliases�values�fail)r* r� �valuer r r �test_aliasing_mbcs� s
��z(ImportSideEffectTests.test_aliasing_mbcsc C s> dt jvr zdd l}W n
ty Y d S w | �d� d S d S )N�
sitecustomizer z(sitecustomize not imported automatically)r r@ r� � ImportErrorr� )r* r� r r r �test_sitecustomize_executed� s
��z1ImportSideEffectTests.test_sitecustomize_executedZnetworkZ HTTPSHandlerz$need SSL support to download licensec
C s� t j�� d }tjj|dd�}| �tjj� z1t� |��" tj�
|�� }|� � }W d � n1 s2w Y W d � n1 sAw Y W n tj j
y^ } z |j}W Y d }~nd }~ww | j|dd| d� d S )Nr7 ZHEAD)�method�� z Can't find )�msg)r� �_Printer__data�split�urllib�requestZRequestr\ �
urlcleanupr Ztransient_internet�urlopenZgetcode�error� HTTPError�coder2 )r* �urlZreq�datar� �er r r �test_license_exists_at_url s
�� ����z0ImportSideEffectTests.test_license_exists_at_urlN)r� r� r� r� r+ r/ r� r� r �skipr� r� r� r� r� r� r� r Zrequires_resourceZsystem_must_validate_certr� r� r� r� r� r r r r r� � s&
�r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�StartupImportTestsc
C s8 t jtjdddgt jdd�}|�� d }| �|jdt|�� t |�}|D ] }t
�
t j �
t
�|�d��}|rB| �dt|�� d |� �� q$t jtjdd
dd gt jt jdd �}|�� \}}| �|jd||f� t |�}| �d
|� h d�}| �|�|�|� | �d||� | �d||� h d��tj�} | �|�| �|� d S )N�-Ir� z!import sys; print(repr(sys.path))rU )�stdoutrV r z*.pthzfound z .pth files in: z-vz#import sys; print(set(sys.modules)))r r� rV r > re �_sre�
sre_constants� sre_parse� sre_compiler� �copyreg>
�reprlib� functools� _collections�operator�heapq� itertools�weakref�types�keyword� collections)r} �Popenr r� �PIPE� communicater2 �
returncode�repr�eval�globr r r0 rf ZskipTestr� r< rB � intersectionrq �
difference�builtin_module_names)
r* �popenr Zisolated_pathsr Z pth_filesr� r@ Zre_modsZcollection_modsr r r �test_startup_imports sB �� ��� �z'StartupImportTests.test_startup_importsc C s& t �tjddg��� }| �|d� d S )Nr� �9import sys; sys.exit(hasattr(sys, "__interactivehook__"))z''__interactivehook__' not added by site�r} r r r� �waitr� �r* �rr r r � test_startup_interactivehookI s
��z/StartupImportTests.test_startup_interactivehookc C �( t �tjdddg��� }| �|d� d S )Nr� r� r z,'__interactivehook__' added in isolated mode)r} r r r� r rB r r r r �%test_startup_interactivehook_isolatedN � ��z8StartupImportTests.test_startup_interactivehook_isolatedc C r" )Nr� r� zYimport site, sys; site.enablerlcompleter(); sys.exit(hasattr(sys, "__interactivehook__"))z6'__interactivehook__' not added by enablerlcompleter()r r r r r �.test_startup_interactivehook_isolated_explicitT r$ zAStartupImportTests.test_startup_interactivehook_isolated_explicitN)r� r� r� r r! r# r% r r r r r� s
0 r� r| zonly supported on Windowsc @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�
_pthFileTestsTc
C s6 dd l }t�� }| �tj|� tj�|tj� t
j �d �}|� t
j
�}tj�|tj� |�d �}t�t
j |� t�||� t�tj�tj� |�d d��D ]}t�|tj�|tj� |�d �� qM|rntj�|�d d } n
tj�|�d d } t| d��}
|D ]} t| |
d� q�W d � |S 1 s�w Y |S )Nr r7 zvcruntime*.dllz._pthrT r� )�_winapi�tempfile�mkdtempr\ r �rmtreer r r0 r� r r� ZGetModuleFileNameZ dllhandle�shutilr
r �splitextr[ r� ) r* �lines�exe_pthr' Ztemp_dir�exe_fileZ dll_src_fileZdll_file�fnZ _pth_file�f�liner r r �_create_underpth_exe] s* $$ �
��z"_pthFileTests._create_underpth_exec C sD g }|D ]}|r|d dkrqt j�t j�||��}|�|� q|S )Nr �#)r r r3 r0 �append)r* �
sys_prefixr- r r2 Zabs_pathr r r �"_calc_sys_path_for_underpth_nositeq s z0_pthFileTests._calc_sys_path_for_underpth_nositec s� t j�t j�tj��� t j�tj�}dg� fdd�td�D ��d�d�}| �|�}| � t j�|�|�}t j
� � }d|d< d � |t �
d
��|d
< tj|d d g|d
d�}|�� �d�}| �|d� | �||d� d S )N�fake-path-namec � g | ]}� �qS r r �r9 �_�Zlibpathr r r: � z;_pthFileTests.test_underpth_nosite_file.<locals>.<listcomp>r� rn � # comment�from-env�
PYTHONPATH�{};{}�PATHr� zCimport sys; print("\n".join(sys.path) if sys.flags.no_site else "")�ansi)r� rV r� zsys.flags.no_site was False�sys.path is incorrect)r r �dirnamer� �__file__r r� �ranger3 r7 r� r
�format�getenvr} � check_output�rstripr� r� r2 )r* �
exe_prefixZ pth_linesr/ r r� �outputZactual_sys_pathr r< r �test_underpth_nosite_filez s< ����
�
�� �z'_pthFileTests.test_underpth_nosite_filec
s� t j�t j�tj��� t j�tj�}| �dg� fdd�td�D ��d�d�d��}t j�|�}t j �
� }d|d < d
� |t � d ��|d < t
j|d d
t j�|d�� t j�|d�f g|d�}| �|d� d S )Nr8 c r9 r r r: r< r r r: � r= z4_pthFileTests.test_underpth_file.<locals>.<listcomp>r� rn r> � import siter? r@ rA rB r� ��import sys; sys.exit(not sys.flags.no_site and %r in sys.path and %r in sys.path and %r not in sys.path and all("\r" not in p and "\n" not in p for p in sys.path))r� rD �r r rE r� rF r r� r3 rG r� r
rH rI r} r� r0 r� �r* rL r/ r6 r� r� r r< r �test_underpth_file� s8 �����
����z _pthFileTests.test_underpth_filec
s� t j�t j�tj��� t j�tj�}| jdg� fdd�td�D ��d�d�d�dd �}t j�|�}t j �
� }d
|d < d � |t � d
��|d
< t
j|ddt j�|d�� t j�|d
�f g|d�}| �|d� d S )Nr8 c r9 r r r: r< r r r: � r= z8_pthFileTests.test_underpth_dll_file.<locals>.<listcomp>r� rn r> rO F)r. r? r@ rA rB r� rP r� rD rQ rR r r< r �test_underpth_dll_file� s< ������
����z$_pthFileTests.test_underpth_dll_fileN)T)r� r� r� r3 r7 rN rS rT r r r r r& Z s
r& �__main__)1r� r Z test.supportr� r r r r Ztest.support.os_helperr r r r� r� r r� r re r+ rv r} r r% r( Z urllib.errorr� Zurllib.requestr r
�flags�no_siter r r r� r
r r ZTestCaser r� rJ r� r� r� r� r& r� �mainr r r r �<module> sT
F7]Cm �