Your IP : 216.73.216.213
3
� f>e � @ s* d Z ddlZddlZddlmZ ddlmZmZmZmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlmZ ddlmZ e jjr�ejd��ddlZdadd � Zd
d � Z G d d
� d
ej �Z G dd� de �Z G dd� dej �Z!G dd� dej �Z"e#dk�r&ej$� 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 s| t jd d � atjrxtjjtj� rxy tjtj� tj tj� W n6 t
k
rv } zt j dtj| f ��W Y d d } ~ X 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 �3/opt/alt/python36/lib64/python3.6/test/test_site.py� setUpModule% s r c C s t tjd d �< d S )N)r r r
r r r r �tearDownModule4 s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z de
fdd�Z dd� Z dd� Z
dd� Zdd� Zejejd�dd� �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 j
� t j 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 �tearDownE s
z HelperFunctionsTests.tearDownc C sh d}t jj|� }tj|� \}}| jt jj|�|� |t jj|�krP| j||� n| jt jj|�|� d S )N� Beginning�End)r+ r, )r r
�joinr �makepath� assertEqual�abspath�normcase)r& �
path_partsZ original_dirZabs_dirZnorm_dirr r r �
test_makepathO s z"HelperFunctionsTests.test_makepathc C s< t j� }x.dd� tjD �D ]}| j||d||f � qW d S )Nc S s* g | ]"}|rt jj|�rtj|�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 | j |jtjd|j � | j tj|j�d tj� | jt jj
|j �� dS )z>Contain common code for testing results of reading a .pth filez%s not in sys.modulesr N) r9 �importedr �modulesr r. �
good_dir_pathr
� assertFalser r5 � bad_dir_path)r& �pth_filer r r �pth_file_testsd s z#HelperFunctionsTests.pth_file_testsc
C sL t � }|jdd� z*|j� tj|j|jt� � | j|� W d |j� X d S )NT)�prep) �PthFile�cleanup�creater �
addpackage�base_dir�filename�setrB )r& rA r r r �test_addpackagek s z$HelperFunctionsTests.test_addpackage�.c s\ t jj|�}|d }t jj||�� t� ddd�}| j� fdd�� |j|� |j� ||fS )Nz.pth�wzutf-8)�encodingc s
t j� �S )N)r �remover )�pth_fnr r �<lambda>� s z/HelperFunctionsTests.make_pth.<locals>.<lambda>)r r
r0 r- �openZ
addCleanup�write�close)r& �contents�pth_dirZpth_nameZ pth_basenamerA r )rP r �make_pthz s
z HelperFunctionsTests.make_pthc
C s� | j d�\}}t� �}tj||t� � W d Q R X | j|j� d� | j|j� tjt j
j ||��� | j|j� d� | j|j� d� | j|j� d� d S )Nzimport bad)syntax
zline 1� Tracebackzimport bad\)syntax� SyntaxError) rW r r rG rJ � assertRegex�getvalue�re�escaper r
r- )r& rV rP �err_outr r r �!test_addpackage_import_bad_syntax� s
z6HelperFunctionsTests.test_addpackage_import_bad_syntaxc
C s� | j d�\}}t� �}tj||t� � W d Q R X | j|j� d� | j|j� tjt j
j ||��� | j|j� d� | j|j� d� d S )Nz randompath
import nosuchmodule
zline 2rX �ModuleNotFoundError) rW r r rG rJ rZ r[ r\ r] r r
r- )r& rV rP r^ r r r � test_addpackage_import_bad_exec� s
z4HelperFunctionsTests.test_addpackage_import_bad_execc
C s� | j d�\}}t� �}tj||t� � W d Q R X | j|j� d� | j|j� tjt j
j ||��� | j|j� d� | j|j� d� d S )Nzabc def
zline 1rX �
ValueError) rW r r rG rJ rZ r[ r\ r] r r
r- )r& rV rP r^ r r r �#test_addpackage_import_bad_pth_file� s
z8HelperFunctionsTests.test_addpackage_import_bad_pth_filec
C sH t � }|jdd� z&|j� tj|jt� � | j|� W d |j� X d S )NT)rC )rD rE rF r r rH rJ rB )r& rA r r r �test_addsitedir� s z$HelperFunctionsTests.test_addsitedirz<requires access to PEP 370 user-site (site.ENABLE_USER_SITE)c C s" t j}| j|tj� tjj� }tj tj
dd| g|d�}| j |d� tjj� }tj tj
ddd| g|d�}|t j � d kr�| j |d� n | j |d� tjj� }d|d< tj tj
dd| g|d�}|t j � d kr�| j |d� n | j |d� tjj� }d |d
< tj tj
dd g|d�}| j |d� d S ) Nz-cz$import sys; sys.exit(%r in sys.path))�envr4 z-sr �1ZPYTHONNOUSERSITEz/tmp�PYTHONUSERBASEz=import sys, site; sys.exit(site.USER_BASE.startswith("/tmp")))
r r r9 r r
r �environr �
subprocess�call�
executabler/ �getsitepackages)r& Zusersitere �rcr r r �
test_s_option� s8
z"HelperFunctionsTests.test_s_optionc
C sh d t _t j� }| jt j|� d t _dd l}d |_t� �(}d|d< | jt j� jd�t j� � W d Q R X d S )Nr �xoxorg ) r r � getuserbaser/ r" r# r �
assertTrue�
startswith)r& � user_baser" rh r r r �test_getuserbase� s z%HelperFunctionsTests.test_getuserbasec C sL d t _d t _t j� }| jt j|� | j|jt j�|� | jt jt j� � d S )N)r r r �getusersitepackagesr/ rq rr rp )r& � user_siter r r �test_getusersitepackages� s z-HelperFunctionsTests.test_getusersitepackagesc C s dgt _t j� }tjdkr|tjd�r|dgt _t j� }| jt|�d� t j
j dtjd�dtj d d� d�}| j|d |� n�t j
d
kr�| jt|�d� t j
j dd d tj d d� d�}| j|d
|� n@| jt|�d� | j|d
d� t j
j ddd�}| j|d |� d S )Nro �darwin�PYTHONFRAMEWORKzPython.framework� z/Libraryz%d.%dz
site-packagesr4 �/�lib64z python%d.%dr �lib)r r rl r �platformr" �get_config_varr/ �lenr r
r- � version_info�sep)r& �dirsZwantedr r r �test_getsitepackages� s,
z)HelperFunctionsTests.test_getsitepackagesc % C s d t _d t _d t_t� �h}tjddd� ��L |d= |d= t j� }| j |j
dt j �|� t j
� }| j |j
|�|� W d Q R X W d Q R X tjddd ��f}tjjt d
��L}tjt d d ��2 t� }t j|� |j|� |j� | j|� 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 rQ s z=HelperFunctionsTests.test_no_home_directory.<locals>.<lambda>rg �APPDATA�~z
os.path.isdirF)Z return_valuer r
T)r r r r" r# r r Zpatchrp rq rr r r� ru �objectr Z swap_attrrJ �addusersitepackagesZassert_called_once_withZassert_not_calledr? )r& rh rs rv 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* r3 r; rB rK r rW r_ ra rc rd r �
skipUnlessr r
rn rt rw r� r� r r r r r 8 s"
&
r c @ s4 e Zd ZdZedddfdd�Zdd� Zd
d
d �Zd S )rD z/Helper class for handling testing of .pth filesZtimeZ __testdir__Z__badc C sj |d | _ tjjd�| _tjj| j| j �| _|| _|| _|| _ tjj| j| j�| _
tjj| j| j �| _ dS )z Initialize instance variablesz.pth� N) rI r r
r0 rH r- � file_pathr< � good_dirname� bad_dirnamer> r@ )r& �
filename_baser< r� r� r r r �__init__= s
zPthFile.__init__c
C sr t | jd�}zJtd|d� td|d� td| j |d� t| j|d� t| j|d� W d|j� X tj| 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.
rM z#import @bad module name)�file�
z import %sN)
rR r� �printr< r� r� rT r �mkdirr> )r& ZFILEr r r rF I s
zPthFile.createFc C s� t jj| j�rt j| j� |r@tjj| j�| _ | j rTtj| j= n| j rT| j tj| j< t jj| j
�rnt j | j
� t jj| j �r�t j | 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
r5 r� rO r r= �getr< Zimported_moduler> �rmdirr@ )r& rC r r r rE _ s zPthFile.cleanupN)F)r� r� r� r� r r� rF rE r r r r rD : s
rD c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zej d
�d d � �Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zejjd�ejjejejd dkd�ejeejd�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' t s zImportSideEffectTests.setUpc C s | j tjdd�< dS )zRestore sys.pathN)r r r
)r& r r r r* x s z ImportSideEffectTests.tearDownc C s� y"t jjt jjt j��}t j� }W n. tk
rP t jjt jjt j��\}}Y nX t|��� t j j
� }||d< d}dj |�}t j
tjdd |g|t jd
�}|j� \}}| j|jd � |j� d d � \} }
| jt jj| �� | jt jj|
�� t j
tjd |g|t jd
�}|j� \}}| j|jd � |j� d d � \} }
| jt jj| �d
j| jd��� | jt jj|
�d
j|
jd��� W d Q R X d S )N�
PYTHONPATH�import os, sys�9os_file = os.__file__.encode("ascii", "backslashreplace")�(sys.stdout.buffer.write(os_file + b"\n")�=os_cached = os.__cached__.encode("ascii", "backslashreplace")�*sys.stdout.buffer.write(os_cached + b"\n")r� z-Sz-c)re �stdoutr rz z expected absolute path, got {}�ascii)r� r� r� r� r� )r r
�relpath�dirname�__file__�getcwdrb �splitr rh r r- ri �Popenr rk �PIPE� communicater/ �
returncode�
splitlinesr? �isabsrq �format�decode) r& �parent�cwdre �codeZcommand�procr� �stderrZ
os__file__Z os__cached__r r r �test_abs_paths| sD
z$ImportSideEffectTests.test_abs_pathsc C s8 t j� t� }x$tjD ]}| j||� |j|� qW d S )N)r �removeduppathsrJ 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$ | j ttd�� | j ttd�� d S )N�quit�exit)rq �hasattr�builtins)r& r r r �test_setting_quit� s z'ImportSideEffectTests.test_setting_quitc C s4 | j ttd�� | j ttd�� | j ttd�� d S )N� copyright�credits�license)rq r� r� )r& r r r �test_setting_copyright� s z,ImportSideEffectTests.test_setting_copyrightc C s | j ttd�� d S )N�help)rq r� r� )r& r r r �test_setting_help� s z'ImportSideEffectTests.test_setting_helpc C sR t jdkrNdd l}|j� d jd�rNx(tjjj� D ]}|dkr2P q2W | jd� d S )N�win32r r4 Zcp�mbcszdid not alias mbcs) r r~ �localeZgetdefaultlocalerr � 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 | jd� 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 jj� d }tjj|dd�}y<tjj|��& tjj |��}|j
� }W d Q R X W d Q R X W n, tj j k
r� } z |j
}W Y d d }~X nX | j|dd| d� d S )Nr4 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� q s" 1
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zed d
� �Zed d � �Z ed
d� �Z
e j e
jdkd�dd� �Ze j e
jdkd�dd� �ZdS )�StartupImportTestsc
C s� t jtjddddgt jt jdd�}|j� \}}t|�}| jd|� dd d
d d h}tjd
krl| j |j
|�|� | j d||� tjd
kr�| j d||� ddddddddddh
j tj
�}tjd
kr�| j |j
|�|� d S )Nz-Iz-vz-cz#import sys; print(set(sys.modules))zutf-8)r� r� rN r r\ �_sre� sre_compile�
sre_constants� sre_parserx r� �copyreg� _collections� collections� functools�heapq� itertools�keyword�operator�reprlib�types�weakref)ri r� r rk r� r� �evalr9 r~ r? � intersectionr� �
difference�builtin_module_names)r& �popenr� r� r= Zre_modsZcollection_modsr r r �test_startup_imports� s(
z'StartupImportTests.test_startup_importsc C s& t jtjddg�j� }| j|d� d S )Nz-cz9import sys; sys.exit(hasattr(sys, "__interactivehook__"))z''__interactivehook__' not added by site)ri r� r rk �waitrq )r& �rr r r � test_startup_interactivehook s
z/StartupImportTests.test_startup_interactivehookc C s( t jtjdddg�j� }| j|d� d S )Nz-Iz-cz9import sys; sys.exit(hasattr(sys, "__interactivehook__"))z,'__interactivehook__' added in isolated mode)ri r� r rk r� r? )r& r� r r r �%test_startup_interactivehook_isolated s z8StartupImportTests.test_startup_interactivehook_isolatedc C s( t jtjdddg�j� }| j|d� d S )Nz-Iz-czYimport site, sys; site.enablerlcompleter(); sys.exit(hasattr(sys, "__interactivehook__"))z6'__interactivehook__' not added by enablerlcompleter())ri r� r rk r� rq )r& r� r r r �.test_startup_interactivehook_isolated_explicit! s zAStartupImportTests.test_startup_interactivehook_isolated_explicitc C s� t jjt jd�t jjtj�d �}tjtj|� t jj |�d d }y4t
|d�� }x|D ]}t ||d� qXW W d Q R X |S t j
j|� t j
j|� � Y nX d S )NZTEMPr4 r z._pthrM )r� )r r
r- �getenvr� r rk �shutilr �splitextrR r� r� r �unlink)r& �lines�exe_file� _pth_file�f�liner r r �_create_underpth_exe' s "
z'StartupImportTests._create_underpth_exec C s0 t jj|�d d }tjj|� tjj|� d S )Nr z._pth)r r
r� r� r r )r& r r r r r �_cleanup_underpth_exe7 s z(StartupImportTests._cleanup_underpth_exec C sJ g }x@|D ]8}| s
|d dkr"q
t jjt jj||��}|j|� q
W |S )Nr �#)r r
r0 r- �append)r& �
sys_prefixr r r Zabs_pathr r r �"_calc_sys_path_for_underpth_nosite= s
z5StartupImportTests._calc_sys_path_for_underpth_nositer� zonly supported on Windowsc s� t jjt jjtj��� t jjtj�}d� fdd�td�D �d�}| j|�}| j t jj|�|�}zDt j
j � }d|d< d j |t j
d
��|d
< tj|d d | g|d
�}W d | j|� X | j|d� d S )N�fake-path-namec s g | ]}� �qS r r )r6 �_)�libpathr r r7 M s z@StartupImportTests.test_underpth_nosite_file.<locals>.<listcomp>r� r� � # commentzfrom-envr� z{};{}�PATHz-czRimport sys; sys.exit(sys.flags.no_site and len(sys.path) > 200 and sys.path == %r))re zsys.path is incorrect)r )r� r )r r
r� r� r� r rk �ranger r rh r r� r� ri rj r rq )r& �
exe_prefixZ pth_linesr r re rm r )r r �test_underpth_nosite_fileG s(
z,StartupImportTests.test_underpth_nosite_filec s� t jjt jjtj��� t jjtj�}| jd� fdd�td�D �d��}t jj|�}z^t j j
� }d|d < d
j |t j d ��|d < t
j|d d
t jj|d�� t jj|d�f g|d�}W d | j|� X | j|d� d S )N�fake-path-namec s g | ]}� �qS r r )r6 r
)r r r r7 i s z9StartupImportTests.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)))re zsys.path is incorrect)r )r� r r )r r
r� r� r� r rk r r rh r r� r� ri rj r- r rq )r& r r r
re rm r )r r �test_underpth_filec s*
z%StartupImportTests.test_underpth_fileN)r� r� r� r� r� r� r� � classmethodr 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� Zurllib.requestr� Z urllib.errorr� ri r" r r �flags�no_siter r r r r ZTestCaser r� rD r� r� r� �mainr r r r �<module> s@
7