File: //opt/cppython/lib/python3.8/test/__pycache__/test_site.cpython-38.opt-2.pyc
U
.D�g�h � @ sb d dl Z d dlZd dlmZ d dlmZmZmZmZ d dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl mZ d dlmZ e jjr�e �d��d dlZda dd� Z d d
� Z G d d � d e j �Z G d
d� de!�Z"G dd� de j �Z#G dd� de j �Z$e �%e j&dkd�G dd� de j ��Z'e(dk�r^e �)� dS )� N)�support)�captured_stderr�TESTFN�EnvironmentVarGuard�
change_cwd)�mock)�copyzPython was invoked with -Sc
C sz t jd d � atjrvtj�tj�svz t�tj� t� tj� W n6 t
k
rt } zt � dtj| f ��W 5 d } ~ X Y nX 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 �-/opt/cppython/lib/python3.8/test/test_site.py� setUpModule( s �r c C s t tjd d �< d S �N)r r r
r r r r �tearDownModule7 s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
e fdd�Z
dd� Z dd� Z dd� Z
dd� Zdd� Ze�eed�d�d d � �Ze�ejd kd �d d!� �Zd"d#� Zd$d%� Ze�ejd&�d'd(� �Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z d1S )2�HelperFunctionsTestsc C s@ t jd d � | _tj| _tj| _tj| _ t
j | _ t
t
j �| _d S r )r r
�sys_pathr � USER_BASE�old_baser �old_site�PREFIXES� old_prefixes� sysconfig� _CONFIG_VARS�
original_varsr �old_vars��selfr r r �setUp? s zHelperFunctionsTests.setUpc C sL | j tjd d �< | jt_| jt_| jt_ | j
t _ t j �
� t j �| j� d S r )r r r
r r r r r r" r! r% r# r$ �clear�updater& r' r r r �tearDownH s
z HelperFunctionsTests.tearDownc C sh d}t jj|� }tj|� \}}| �t j�|�|� |t j�|�krP| �||� n| �t j�|�|� d S )N)Z BeginningZEnd)r r
�joinr �makepath� assertEqual�abspath�normcase)r( �
path_partsZ original_dirZabs_dirZnorm_dirr r r �
test_makepathR s z"HelperFunctionsTests.test_makepathc C s8 t �� }dd� tjD �D ]}| �||d||f � qd S )Nc S s* g | ]"}|rt j�|�rt�|�d �qS )� )r r
�existsr r. )�.0r
r r r �
<listcomp>a 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_pathinfo_ s ��z'HelperFunctionsTests.test_init_pathinfoc C sJ | � |jtjd|j � | � t�|j�d tj� | �t j�
|j �� d S )Nz%s not in sys.modulesr ) r9 �importedr �modulesr r. �
good_dir_pathr
� assertFalser r5 � bad_dir_path�r( �pth_filer r r �pth_file_testsg s
�z#HelperFunctionsTests.pth_file_testsc C sL t � }|jdd� z*|�� t�|j|jt� � | �|� W 5 |�� X d S �NT��prep) �PthFile�cleanup�creater �
addpackage�base_dir�filename�setrC rA r r r �test_addpackagen s z$HelperFunctionsTests.test_addpackage�.c s` t j�|�}|d }t j�||�� t� ddd��"}| �� fdd�� |�|� W 5 Q R X ||fS )N�.pth�w�utf-8)�encodingc s
t �� �S r )r �remover ��pth_fnr r �<lambda>� � z/HelperFunctionsTests.make_pth.<locals>.<lambda>)r r
r0 r- �open�
addCleanup�write)r( �contents�pth_dirZpth_nameZ pth_basenamerB r rU r �make_pth} s z HelperFunctionsTests.make_pthc C s� | � d�\}}t� �}t�||t� � W 5 Q R X | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� | �|�� d� d S )Nzimport bad-syntax
zline 1� Tracebackzimport bad-syntax� SyntaxError� r^ r r rJ rM Z assertRegex�getvalue�re�escaper r
r- �r( r] rV �err_outr r r �!test_addpackage_import_bad_syntax� s
�z6HelperFunctionsTests.test_addpackage_import_bad_syntaxc C s� | � d�\}}t� �}t�||t� � W 5 Q R X | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� d S )Nz randompath
import nosuchmodule
zline 2r_ �ModuleNotFoundErrorra re r r r � test_addpackage_import_bad_exec� s
�z4HelperFunctionsTests.test_addpackage_import_bad_execc C sr | � d�\}}t� � }| �t�||t� �� W 5 Q R X d | _| �|�� d� t j
D ]}t |t �rR| �
d|� qRd S )Nzabc def
� zabc def)r^ r r? r rJ rM ZmaxDiffr/ rb r r
�
isinstance�str� assertNotIn)r( r] rV rf r
r r r �#test_addpackage_import_bad_pth_file� s
z8HelperFunctionsTests.test_addpackage_import_bad_pth_filec C sH t � }|jdd� z&|�� t�|jt� � | �|� W 5 |�� X d S rD )rG rH rI r r rK rM rC rA r r r �test_addsitedir� s z$HelperFunctionsTests.test_addsitedirc C sj t d�}|jdd� zF|�� t�|jt� � | �t�|j �d t
j � | � |jt
j � W 5 |�� X d S )Nz.dotfileTrE r )
rG rH rI r r rK rM rm r. r> r r
r9 rA r r r �test_addsitedir_dotfile� s z,HelperFunctionsTests.test_addsitedir_dotfile�chflagsztest needs os.chflags()c C s� t � }|jdd� zh|�� t�|j�}t�|j|jtjB � t �
|j t � � | �
t �|j�d tj� | �|j tj� W 5 |�� X d S )NTrE r )rG rH rI r �stat� file_pathrq �st_flags� UF_HIDDENr r rK rM rm r. r> r r
r9 )r( rB �str r r � test_addsitedir_hidden_flags� s z1HelperFunctionsTests.test_addsitedir_hidden_flags�win32ztest needs Windowsc C sz t � }|jdd� zX|�� t�dd|jg� t�|jt � � | �
t� |j �d t
j� | �|jt
j� W 5 |�� X d S )NTrE Zattribz+Hr )rG rH rI �
subprocess�
check_callrs r r rK rM rm r. r> r r
r9 rA r r r �%test_addsitedir_hidden_file_attribute� s z:HelperFunctionsTests.test_addsitedir_hidden_file_attributec C s | � t�� t�� � d S r )r/ r � _getuserbaser# r' r r r �test__getuserbase� s z&HelperFunctionsTests.test__getuserbasec C sB t jdkrt jrd}n
tjd }| �t�t�� �t �
d|�� d S )N�darwinZosx_framework_user�_userZpurelib) r �platform�
_frameworkr �namer/ r � _get_pathr| r# Zget_path)r( Zschemer r r �
test_get_path� s
�z"HelperFunctionsTests.test_get_pathz<requires access to PEP 370 user-site (site.ENABLE_USER_SITE)c C s( t j}| �|tj� tj�� }tj tj
dd| g|d�}| � |d� tj�� }tj tj
ddd| g|d�}|t � � d kr�| � |d� n| � |dd� tj�� }d|d < tj tj
dd| g|d�}|t � � d kr�| � |d� n| � |dd
� tj�� }d |d < tj tj
dd
g|d�}| � |dd� d S )N�-cz$import sys; sys.exit(%r in sys.path)��envr4 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 r9 r r
r �environr ry �call�
executabler/ �getsitepackages)r( Zusersiter� �rcr r r �
test_s_option� sP
��
��
���
���z"HelperFunctionsTests.test_s_optionc C sh d t _t �� }| �t j|� d t _dd l}d |_t� �(}d|d< | �t �� �d�t �� � W 5 Q R X d S )Nr �xoxor� ) r r � getuserbaser/ 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 �getusersitepackagesr/ r� r� r� )r( � user_siter r r �test_getusersitepackages' s z-HelperFunctionsTests.test_getusersitepackagesc C s� dgt _t �� }tjdkr\| �t|�d� tj�dddt j
d d� d�}| �|d |� n@| �t|�d� | �|d d� tj�ddd�}| �|d |� d S ) Nr� �/r4 �libz python%d.%d� z
site-packagesr ) r r! r� r �sepr/ �lenr
r- r � version_info)r( �dirsZwantedr r r �test_getsitepackages1 s
�z)HelperFunctionsTests.test_getsitepackagesc C s� d t _d t _t� �h}t�ddd� ��L |d= |d= t �� }| �|�dt j
�|� t � � }| �|�|�|� W 5 Q R X W 5 Q R X tjddd ��f}tj� t d
��L}t
�t d d ��2 t� }t �|� |�|� |�� | �|� W 5 Q R X W 5 Q R X W 5 Q R X d S )
Nzos.path.expanduserc S s | S r r )r
r r r rW J rX z=HelperFunctionsTests.test_no_home_directory.<locals>.<lambda>r� �APPDATA�~z
os.path.isdirF)Z return_valuer r
T)r r r r r Zpatchr� r� r� r r� r� �objectr Z swap_attrrM �addusersitepackagesZassert_called_once_withZassert_not_calledr? )r( r� r� r� Z
mock_isdirZmock_addsitedir� known_pathsr r r �test_no_home_directoryB s0 ��& � �
z+HelperFunctionsTests.test_no_home_directoryN)!�__name__�
__module__� __qualname__r) r, r3 r; rC rN r r^ rg ri rn ro rp r �
skipUnless�hasattrr rw r r� r{ r} r� r r
r� r� r� r� r� r r r r r ; s0
'
r c @ s0 e Zd Zedddfdd�Zdd� Zd d d
�Zd S )
rG �timeZ __testdir__Z__badc C sj |d | _ tj�d�| _tj�| j| j �| _|| _|| _|| _ tj�| j| j�| _
tj�| j| j �| _ d S )NrP rj ) rL r r
r0 rK r- rs r< � good_dirname� bad_dirnamer> r@ )r( �
filename_baser< r� r� r r r �__init__g s
zPthFile.__init__c C sr t | jd�}zJtd|d� td|d� td| j |d� t| j|d� t| j|d� W 5 |�� X t�| j � d S )NrQ z#import @bad module name��file�
z import %s)
rY rs �close�printr< r� r� r �mkdirr> )r( �FILEr r r rI s s
zPthFile.createFc C s� t j�| j�rt �| j� |r@tj�| j�| _ | j rTtj| j= n| j rT| j tj| j< t j�| j
�rnt � | j
� t j�| j �r�t � | j � d S r )
r r
r5 rs rT r r= �getr< Zimported_moduler> �rmdirr@ )r( rF r r r rH � s zPthFile.cleanupN)F)r� r� r� r r� rI rH r r r r rG d s �
rG c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Ze�d �d
d � �Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zej�d�ejje�eejd�d�dd� ���ZdS ) �ImportSideEffectTestsc C s t jd d � | _d S r )r r
r r' r r r r) � s zImportSideEffectTests.setUpc C s | j tjd d �< d S r )r r r
r' r r r r, � s z ImportSideEffectTests.tearDownc C s* d t jd _t�� | �t jd j� d S )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 ]}| �||� |�|� qd S r )r �removeduppathsrM r r
rm �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� s z'ImportSideEffectTests.test_setting_helpc C sP t jdkrLdd l}|�� d �d�rLtjj�� D ]}|dkr0 qLq0| �d� d S )Nrx r r4 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 jkr6z dd l}W n tk
r* Y n X | �d� 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� z<tj �
|��& tj� |��}|� � }W 5 Q R X W 5 Q R X W n, tj
jk
r� } z
|j}W 5 d }~X Y nX | j|dd| d� d S )Nr4 ZHEAD)�method�� z Can't find )�msg)r� �_Printer__data�split�urllib�requestZRequestrZ �
urlcleanupr� r Ztransient_internet�urlopenZgetcode�error� HTTPError�coder/ )r( ZurlZreq�datar� �er r r �test_license_exists_at_url� s z0ImportSideEffectTests.test_license_exists_at_urlN)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 sN t jtjdddgt jdd�}|�� d }| �|jdt|�� t |�}|D ]<}t
�
t j �
t
�|�d��}|rH| �dt|�� d |� �� qHt jtjdd
dd gt jt jdd �}|�� \}}| �|jd||f� t |�}| �d
|� dddddh}| �|�|�|� | �d||� | �d||� dddddddd d d h
�tj�} | �|�| �|� d S ) N�-Ir� z!import sys; print(repr(sys.path))rR )�stdoutrS r z*.pthzfound z .pth files in: z-vz#import sys; print(set(sys.modules)))r� �stderrrS r rc �_sre� sre_compile�
sre_constants� sre_parser� �copyreg� _collections� collections� functools�heapq� itertools�keyword�operator�reprlib�types�weakref)ry �Popenr r� �PIPE� communicater/ �
returncode�repr�eval�globr r
r- rd ZskipTestr� r9 r? � intersectionrm �
difference�builtin_module_names)
r( �popenr� Zisolated_pathsr
Z pth_filesr� r= Zre_modsZcollection_modsr r r �test_startup_imports� sP �� �� ��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�ry r� r r� �waitr� �r( �rr r r � test_startup_interactivehook* s
�
z/StartupImportTests.test_startup_interactivehookc C s( t �tjdddg��� }| �|d� d S )Nr� r� r z,'__interactivehook__' added in isolated mode)ry r� r r� r r? r r r r �%test_startup_interactivehook_isolated/ s �
z8StartupImportTests.test_startup_interactivehook_isolatedc C s( t �tjdddg��� }| �|d� d S )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_explicit5 s �
zAStartupImportTests.test_startup_interactivehook_isolated_explicitN)r� r� r� r r r r r r r r r� � s 0r� rx zonly supported on Windowsc @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�
_pthFileTestsTc C s� dd l }t�� }| �tjj|� tj� |tj�
t j �d �}|�
t j�}tj� |tj�
|�d �}t�t j |� t�||� |r�tj�|�d d }ntj�|�d d }t|d�� } |D ]}
t|
| d� q�W 5 Q R X |S )Nr r4 z._pthrQ r� )�_winapi�tempfile�mkdtemprZ r� r �rmtreer r
r- r� r r� ZGetModuleFileNameZ dllhandle�shutilr �splitextrY r� ) r( �lines�exe_pthr Ztemp_dir�exe_fileZ dll_src_fileZdll_fileZ _pth_file�f�liner r r �_create_underpth_exe? s z"_pthFileTests._create_underpth_exec C sD g }|D ]6}|r|d dkr qt j�t j�||��}|�|� q|S )Nr �#)r r
r0 r- �append)r( �
sys_prefixr r r 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�}d� fdd�td�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 s g | ]}� �qS r r �r6 �_�Zlibpathr r r7 _ s z;_pthFileTests.test_underpth_nosite_file.<locals>.<listcomp>r� rj � # comment�from-env�
PYTHONPATH�{};{}�PATHr� zCimport sys; print("\n".join(sys.path) if sys.flags.no_site else "")�ansi)r� rS r� zsys.flags.no_site was False�sys.path is incorrect)r" )rj r& )r r
�dirnamer� �__file__r r� �ranger r! r� r �format�getenvry � check_output�rstripr� r� r/ )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�}| �d� fdd�td�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 )Nr" c s g | ]}� �qS r r r# r% r r r7 { s z4_pthFileTests.test_underpth_file.<locals>.<listcomp>r� rj r& � import siter' r( r) r* 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� r, )r" )rj r& r7 �r r
r- r� r. r r� r r/ r� r r0 r1 ry r� r- r� �r( r4 r r r� r� r r% r �test_underpth_filev s2 � �
����z _pthFileTests.test_underpth_filec
s� t j�t j�tj��� t j�tj�}| jd� fdd�td�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 )Nr" c s g | ]}� �qS r r r# r% r r r7 � s z8_pthFileTests.test_underpth_dll_file.<locals>.<listcomp>r� rj r& r7 F)r r' r( r) r* r� r8 r� r, )r" )rj r& r7 r9 r: r r% r �test_underpth_dll_file� s6 � ��
����z$_pthFileTests.test_underpth_dll_fileN)T)r� r� r� r r! r6 r; r< r r r r r < s
r �__main__)*r Z test.supportr� r r r r r r� r r r rc r� Zurllib.requestr� Z urllib.errorr rr ry r# r r r �flags�no_siter r r r r ZTestCaser r� rG r� r� r� r� r r� �mainr r r r �<module> sD
+7]Dk