File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_site.cpython-37.pyc
B
��g6l � @ s^ d Z ddlZddlZddlmZ ddlmZmZmZmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlZddlZddlmZ ddlmZ e jjr�e�d��ddlZda dd � Z d
d � Z G d d
� d
ej �Z G dd� de!�Z"G dd� dej �Z#G dd� dej �Z$e�%e j&dkd�G dd� dej ��Z'e(dk�rZe�)� dS )zTests for 'site'.
Tests assume the initial paths in sys.path once the interpreter has begun
executing have not been removed.
� N)�support)�captured_stderr�TESTFN�EnvironmentVarGuard�
change_cwd)�mock)�copyzPython was invoked with -Sc
C sz t jd d � atjrvtj�tj�svy t�tj� t� tj� W n6 t
k
rt } zt � dtj| f ��W d 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 �*/usr/local/lib/python3.7/test/test_site.py� setUpModule' s r c C s t tjd d �< d S )N)r r r
r r r r �tearDownModule6 s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z de
fdd�Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Ze�ejd �d d � �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))�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 �setUp> s zHelperFunctionsTests.setUpc C sL | j tjdd�< | jt_| jt_| jt_ | j
t _ t j �
� t j �| j� dS )zRestore sys.pathN)r r r
r r r r r r! r r$ r" r# �clear�updater% )r& r r r �tearDownG 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_makepathQ s z"HelperFunctionsTests.test_makepathc C s< t �� }x.dd� tjD �D ]}| �||d||f � qW d S )Nc S s* g | ]"}|rt j�|�rt�|�d �qS )� )r r
�existsr r, )�.0r
r r r �
<listcomp>` 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 �� dS )z>Contain common code for testing results of reading a .pth filez%s not in sys.modulesr N) r7 �importedr �modulesr r, �
good_dir_pathr
� assertFalser r3 � bad_dir_path)r& �pth_filer r r �pth_file_testsf s z#HelperFunctionsTests.pth_file_testsc C sL t � }|jdd� z*|�� t�|j|jt� � | �|� W d |�� X d S )NT)�prep) �PthFile�cleanup�creater �
addpackage�base_dir�filename�setr@ )r& r? r r r �test_addpackagem s z$HelperFunctionsTests.test_addpackage�.c s\ t j�|�}|d }t j�||�� t� ddd�}| �� fdd�� |�|� |�� ||fS )Nz.pth�wzutf-8)�encodingc s
t �� �S )N)r �remover )�pth_fnr r �<lambda>� � z/HelperFunctionsTests.make_pth.<locals>.<lambda>)r r
r. r+ �open�
addCleanup�write�close)r& �contents�pth_dirZpth_nameZ pth_basenamer? r )rN r �make_pth| s
z HelperFunctionsTests.make_pthc C s� | � d�\}}t� �}t�||t� � W d Q R X | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� | �|�� d� d S )Nzimport bad)syntax
zline 1� Tracebackzimport bad\)syntax� SyntaxError) rW r r rE rH � assertRegex�getvalue�re�escaper r
r+ )r& rV rN �err_outr r r �!test_addpackage_import_bad_syntax� s
z6HelperFunctionsTests.test_addpackage_import_bad_syntaxc C s� | � d�\}}t� �}t�||t� � W d Q R X | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� d S )Nz randompath
import nosuchmodule
zline 2rX �ModuleNotFoundError) rW r r rE rH rZ r[ r\ r] r r
r+ )r& rV rN r^ r r r � test_addpackage_import_bad_exec� s
z4HelperFunctionsTests.test_addpackage_import_bad_execc C s� | � d�\}}t� �}t�||t� � W d Q R X | �|�� d� | �|�� t�t j
� ||��� | �|�� d� | �|�� d� d S )Nzabc def
zline 1rX �
ValueError) rW r r rE rH rZ r[ r\ r] r r
r+ )r& rV rN 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 d |�� X d S )NT)rA )rB rC rD r r rF rH r@ )r& r? r r r �test_addsitedir� s z$HelperFunctionsTests.test_addsitedirc C s | � t�� t�� � d S )N)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_userZ_userZpurelib) r �platform�
_frameworkr �namer- r � _get_pathre 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 )Nz-cz$import sys; sys.exit(%r in sys.path))�envr2 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 r7 r r
r �environr �
subprocess�call�
executabler- �getsitepackages)r& Zusersiterm �rcr r r �
test_s_option� s<
z"HelperFunctionsTests.test_s_optionc C sh d t _t �� }| �t j|� d t _dd l}d |_t� �(}d|d< | �t �� �d�t �� � W d Q R X d S )Nr �xoxoro ) r r � getuserbaser- r" r# r �
assertTrue�
startswith)r& � user_baser" rp 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 )N)r r r �getusersitepackagesr- ry rz rx )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 ) Nrw �/r2 �libz python%d.%d� z
site-packagesr ) r r rt r �sepr- �lenr
r+ r � version_info)r& �dirsZwantedr r r �test_getsitepackages s
z)HelperFunctionsTests.test_getsitepackagesc C s� d t _d t _t� �h}t�ddd� ��L |d= |d= t �� }| �|�dt j
�|� t � � }| �|�|�|� W d Q R X W d Q R X tjddd ��f}tj� t d
��L}t
�t d d ��2 t� }t �|� |�|� |�� | �|� W d Q R X W d Q R X W d Q R X d S )
Nzos.path.expanduserc S s | S )Nr )r
r r r rO % rP z=HelperFunctionsTests.test_no_home_directory.<locals>.<lambda>ro �APPDATA�~z
os.path.isdirF)Z return_valuer r
T)r r r r r Zpatchrx ry rz r r� r} �objectr Z swap_attrrH �addusersitepackagesZassert_called_once_withZassert_not_calledr= )r& rp r{ r~ Z
mock_isdirZmock_addsitedir� known_pathsr r r �test_no_home_directory s& &
z+HelperFunctionsTests.test_no_home_directoryN)�__name__�
__module__� __qualname__�__doc__r' r* r1 r9 r@ rI r rW r_ ra rc rd rf rl r �
skipUnlessr r
rv r| r r� r� r r r r r : s&
)
r c @ s4 e Zd ZdZedddfdd�Zdd� Zd
d
d �Zd S )rB z/Helper class for handling testing of .pth filesZtimeZ __testdir__Z__badc C sj |d | _ tj�d�| _tj�| j| j �| _|| _|| _|| _ tj�| j| j�| _
tj�| j| j �| _ dS )z Initialize instance variablesz.pth� N) rG r r
r. rF r+ � file_pathr: � good_dirname� bad_dirnamer< r> )r& �
filename_baser: r� r� r r r �__init__B s
zPthFile.__init__c C sr t | jd�}zJtd|d� td|d� td| j |d� t| j|d� t| j|d� W d|�� X 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.
rK z#import @bad module name)�file�
z import %sN)
rQ r� �printr: r� r� rT r �mkdirr< )r& �FILEr r r rD N 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 � 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
r3 r� rM r r; �getr: Zimported_moduler< �rmdirr> )r& rA r r r rC d s zPthFile.cleanupN)F)r� r� r� r� r r� rD rC r r r r rB ? s
rB 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�ejd dkd�e�eejd�d �d d � ����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' y s zImportSideEffectTests.setUpc C s | j tjdd�< dS )zRestore sys.pathN)r r r
)r& r r r r* } s z ImportSideEffectTests.tearDownc C s� y"t j�t j�t j��}t �� }W n. tk
rP t j�t j�t j��\}}Y nX t|��� t j �
� }||d< d}d� |�}t j
tjdd|g|t jd�}|�� \}}| �|jd� |�� d d� \} }
| �t j�| �� | �t j�|
�� t j
tjd|g|t jd�}|�� \}}| �|jd� |�� d d� \} }
| �t j�| �d �| �d
��� | �t j�|
�d �|
�d
��� W d Q R X d S ) N�
PYTHONPATH)zimport os, sysz9os_file = os.__file__.encode("ascii", "backslashreplace")z(sys.stdout.buffer.write(os_file + b"\n")z=os_cached = os.__cached__.encode("ascii", "backslashreplace")z*sys.stdout.buffer.write(os_cached + b"\n")r� z-Sz-c)rm �stdoutr r� z expected absolute path, got {}�ascii)r r
�relpath�dirname�__file__�getcwdrb �splitr rp r r+ rq �Popenr rs �PIPE� communicater- �
returncode�
splitlinesr= �isabsry �format�decode) r& �parent�cwdrm �codeZcommand�procr� �stderrZ
os__file__Z os__cached__r r r �test_abs_paths� s<
z$ImportSideEffectTests.test_abs_pathsc C s8 t �� t� }x$tjD ]}| �||� |�|� qW d S )N)r �removeduppathsrH r r
� assertNotIn�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 )Nr )r& r r r �test_add_build_dir� s z(ImportSideEffectTests.test_add_build_dirc C s$ | � ttd�� | � ttd�� d S )N�quit�exit)ry �hasattr�builtins)r& r r r �test_setting_quit� s z'ImportSideEffectTests.test_setting_quitc C s4 | � ttd�� | � ttd�� | � ttd�� d S )N� copyright�credits�license)ry r� r� )r& r r r �test_setting_copyright� s z,ImportSideEffectTests.test_setting_copyrightc C s | � ttd�� d S )N�help)ry r� r� )r& r r r �test_setting_help� s z'ImportSideEffectTests.test_setting_helpc C sR t jdkrNdd l}|�� d �d�rNx(tjj�� D ]}|dkr2P q2W | �d� d S )N�win32r r2 Zcp�mbcszdid not alias mbcs) r rh �locale�getdefaultlocalerz � encodings�aliases�values�fail)r& r� �valuer r r �test_aliasing_mbcs� s
z(ImportSideEffectTests.test_aliasing_mbcsc C s: dt jkr6y 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_executedZnetwork� �finalzonly for released versionsZ HTTPSHandlerz$need SSL support to download licensec
C s� t j�� d }tjj|dd�}y<tj�|��& tj� |��}|�
� }W d Q R X W d Q R X W n, tj j k
r� } z
|j
}W d d }~X Y nX | j|dd| d� d S )Nr2 ZHEAD)�method�� z Can't find )�msg)r� �_Printer__datar� �urllib�requestZRequest�testr Ztransient_internetZurlopenZgetcode�errorZ HTTPErrorr� r- )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� r �skipr� r� r� r� r� r� r� r Zrequires_resourceZsystem_must_validate_certr� r r� r� r� r� r� r r r r r� v s" 1
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�StartupImportTestsc
C sL t jtjdddgt jdd�}|�� d }| �|jdt|�� t |�}x>|D ]6}t
�
t j �
|d��}|rJ| �dt|�� d |� �� qJW t jtjdd
dd gt jt jdd �}|�� \}}| �|jd||f� t |�}| �d
|� dddddh}| �|�|�|� | �d||� | �d||� dddddddd d d h
�tj�} | �|�| �|� d S ) Nz-Iz-cz!import sys; print(repr(sys.path))zutf-8)r� rL r z*.pthzfound z .pth files in: z-vz#import sys; print(set(sys.modules)))r� r� rL r r\ �_sre� sre_compile�
sre_constants� sre_parser� �copyreg� _collections� collections� functools�heapq� itertools�keyword�operator�reprlib�types�weakref)rq r� r rs r� r� r- r� �repr�eval�globr r
r+ ZskipTestr� r7 r= � intersectionr� �
difference�builtin_module_names)
r& �popenr� Zisolated_pathsr
Z pth_filesr� r; Zre_modsZcollection_modsr r r �test_startup_imports� s:
z'StartupImportTests.test_startup_importsc C s& t �tjddg��� }| �|d� d S )Nz-cz9import sys; sys.exit(hasattr(sys, "__interactivehook__"))z''__interactivehook__' not added by site)rq r� r rs �waitry )r& �rr r r � test_startup_interactivehook+ s
z/StartupImportTests.test_startup_interactivehookc C s( t �tjdddg��� }| �|d� d S )Nz-Iz-cz9import sys; sys.exit(hasattr(sys, "__interactivehook__"))z,'__interactivehook__' added in isolated mode)rq r� r rs r� r= )r& r� r r r �%test_startup_interactivehook_isolated0 s z8StartupImportTests.test_startup_interactivehook_isolatedc C s( t �tjdddg��� }| �|d� d S )Nz-Iz-czYimport site, sys; site.enablerlcompleter(); sys.exit(hasattr(sys, "__interactivehook__"))z6'__interactivehook__' not added by enablerlcompleter())rq r� r rs r� ry )r& r� r r r �.test_startup_interactivehook_isolated_explicit6 s zAStartupImportTests.test_startup_interactivehook_isolated_explicitN)r� r� r� r� r� r� r� r r r r r� � s 0r� r� 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�� } x|D ]}
t|
| d� q�W W d Q R X |S )Nr r2 z._pthrK )r� )�_winapi�tempfileZmkdtemprR r� r Zrmtreer r
r+ r� r rs ZGetModuleFileNameZ dllhandle�shutilr �splitextrQ 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 sH g }x>|D ]6}|r
|d dkr q
t j�t j�||��}|�|� q
W |S )Nr �#)r r
r. r+ �append)r& �
sys_prefixr r r Zabs_pathr r r �"_calc_sys_path_for_underpth_nositeR 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 )r4 �_)�libpathr r r5 ` s z;_pthFileTests.test_underpth_nosite_file.<locals>.<listcomp>r� r� � # commentzfrom-envr� z{};{}�PATHz-czCimport sys; print("\n".join(sys.path) if sys.flags.no_site else "")�ansi)rm rL r� zsys.flags.no_site was Falsezsys.path is incorrect)r )r� r )r r
r� r� r� r rs �ranger r
rp r r� �getenvrq Z check_output�rstripr� ry r- )r& �
exe_prefixZ pth_linesr r rm �outputZactual_sys_pathr )r r �test_underpth_nosite_file[ 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 )N�fake-path-namec s g | ]}� �qS r r )r4 r )r r r r5 | s z4_pthFileTests.test_underpth_file.<locals>.<listcomp>r� r� � # comment� import sitezfrom-envr� z{};{}r z-cz�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)))rm zsys.path is incorrect)r )r� r r )r r
r� r� r� r rs r r rp r r� r rq rr r+ ry )r& r r r rm ru r )r r �test_underpth_filew s&
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 )N�fake-path-namec s g | ]}� �qS r r )r4 r )r r r r5 � s z8_pthFileTests.test_underpth_dll_file.<locals>.<listcomp>r� r� � # comment� import siteF)r zfrom-envr� z{};{}r z-cz�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)))rm zsys.path is incorrect)r )r� r r )r r
r� r� r� r rs r r rp r r� r rq rr r+ ry )r& r r r rm ru r )r r �test_underpth_dll_file� s(
z$_pthFileTests.test_underpth_dll_fileN)T)r� r� r� r r
r r r! r r r r r� = s
r� �__main__)*r� r Z test.supportr� r r r r r r� r� r r r\ r� Zurllib.requestr� Z urllib.errorr rq r" r r r �flags�no_siter r r r r ZTestCaser r� rB r� r� r� rh r� r� �mainr r r r �<module> sF
7 D
k