Your IP : 216.73.216.213
3
��2a G � @ s& d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
m Z m Z m
Z
mZ d dlZd dlZy d dlZW n ek
r� dZY nX y d dlZW n ek
r� dZY nX ejejejkd�Zd
dd�ZG dd� dej�ZG dd � d e�ZeG d
d � d e��Z e d k�r"ej � dS )� N)�captured_stdout�captured_stderr�
requires_zlib� can_symlink�EnvironmentVarGuard�rmtreez Test not appropriate in a venvc C sB t j| t jt j|d�}|j� \}}|jr:t j|j| ||��||fS )N)�stdout�stderr�encoding)�
subprocess�Popen�PIPEZ communicate�
returncode�CalledProcessError)�cmdr
�p�out�err� r �./opt/alt/python36/lib64/python3.6/test_venv.py� check_output# s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�BaseTest�P �2 c C s� t jjtj� �| _t jdkr0d| _d
| _d| _ n$d| _ddt
j d d� f| _d | _ t
j d
krtd t j
krtt j
d }nt
j}t jj|�d | _d S )N�nt�Scripts�Lib�Include�bin�libz python%d.%d� �include�darwin�__PYVENV_LAUNCHER__� )r ���)�os�path�realpath�tempfileZmkdtemp�env_dir�name�bindirr r! �sys� version_info�platform�environ�
executable�split�exe)�selfr1 r r r �setUp2 s
zBaseTest.setUpc C s t | j� d S )N)r r* )r4 r r r �tearDownB s zBaseTest.tearDownc O s> t � �"}t� �}|||� W d Q R X W d Q R X |j� |j� fS )N)r r �getvalue)r4 �func�args�kwargs�output�errorr r r �run_with_captureE s zBaseTest.run_with_capturec G s t jj| jf|�� S )N)r& r' �joinr* )r4 r9 r r r � get_env_fileK s zBaseTest.get_env_filec G s( t | j|� d��}|j� }W d Q R X |S )N�r)�openr? �read)r4 r9 �f�resultr r r �get_text_file_contentsN s z BaseTest.get_text_file_contentsNi� ) �__name__�
__module__� __qualname__ZmaxDiffr5 r6 r= r? rE r r r r r . s r c @ s� e Zd Zdd� Zdd� Zdd� Zedd� �Zej d kr<d/Z
n4d0d1d2ddej dd� fddej dd� d
ffZ
dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zeje� d �d!d"� �Zed#d$� �Zeje� d �d%d&� �Zejejd'kd(�d)d*� �ZdS )3� BasicTestc G s | j |� }| jtjj|�� d S )N)r? �
assertTruer& r' �isdir)r4 r9 �fnr r r rK V s
zBasicTest.isdirc C sD t | j� | jtj| j� | j| j� | j| j� | j| j� | j d�}t
j d�dkoht j
dkohtjdk}|r�| jt jj|�� n| jt jj|�� | jd�}tjdkr�dt jkr�t jd }ntj}t jj|�}| jd| |� | j | j| j�}t jj|��s(| j | j�}td | � td
t j |� � | jt jj|�d | � d S ) N�lib64�P� �posixr" z
pyvenv.cfgr# z home = %szContents of %r:z %rzFile %r should exist.) r r* r= �venv�createrK r, r! r r? �struct�calcsizer& r+ r- r/ rJ r' �islink� assertFalse�existsrE r0 r1 �dirname�assertInr3 �print�listdir)r4 r Z
conditions�datar1 r' rL �bdr r r �
test_defaultsZ s0
zBasicTest.test_defaultsc C sb t jj| j�d }tj� }|j| j�}| j|jd| � tjdd�}|j| j�}| j|jd� d S )Nr$ z(%s) z My prompt)�promptz (My prompt) ) r& r' r2 r* rQ �
EnvBuilderZensure_directories� assertEqualr_ )r4 Zenv_name�builder�contextr r r � test_promptz s zBasicTest.test_promptc C s� | j tjtj� | j tjtj� t| j� | jt j
| j� t j j
| j| j| j�}|dd g}x\d| jfd| jfdtjfdtjffD ]4\}}d| |d< t|�\}}| j |j� |j� � q�W d S )Nz-c�prefix� base_prefix�base_exec_prefixzimport sys; print(sys.%s)r )ra r- rf re rg � exec_prefixr r* r= rQ rR r& r' r> r, r3 r �strip�encode)r4 �envpyr re Zexpectedr r r r r �
test_prefixes� s
zBasicTest.test_prefixes�win32r r r �
site-packagesr r! r z python%d.%dNr c
C s^ xX|D ]P}t jj| jf|�� }t j|� t jj||�}t|d��}|jd� W d Q R X qW d S )N�wbs Still here?)r& r' r> r* �mkdirrA �write)r4 �paths�filename�subdirs�drL rC r r r �create_contents� s
zBasicTest.create_contentsc C s� | j | jd� tj| j� x^| jD ]T}tjj| jf|d �� }| jtjj |�� t
|d��}| j |j � d� W d Q R X q"W tj
dd�}|j| j� x8| jD ].}tjj| jf|d �� }| jtjj |�� q�W d S )N�foo�rbs Still here?T)�clear)rw )rw )rv � ENV_SUBDIRSrQ rR r* r& r' r> rJ rW rA ra rB r` rV )r4 rt rL rC rb r r r �test_overwrite_existing� s z!BasicTest.test_overwrite_existingc C s^ xXt j|�D ]J}t jj||�}t jj|�s6t jj|�rBt j|� q t jj|�r t|� q W d S )N) r& r[ r' r> rU �isfile�removerK r )r4 r' rL r r r �clear_directory� s zBasicTest.clear_directoryc C sr xl| j d d� D ]Z}tjj| jf|�� }t|d��}|jd� W d Q R X | jtt ft
j | j� | j | j� qW d S )N� ro � )
rz r& r' r> r* rA rq Z assertRaises�
ValueError�OSErrorrQ rR r~ )r4 rr rL rC r r r �test_unoverwritable_fails� s z#BasicTest.test_unoverwritable_failsc C s� x�dD ]�}t j|d�}| j|j| j� | j| j� | j| j� | j| j� | j | j| j
�}t j j
|�s�| j | j�}td| � tdt j|� � | jt j j
|�d| � qW d S )NFT)�upgradezContents of %r:z %rzFile %r should exist.)FT)rQ r` r= rR r* rK r, r! r r? r3 r& r' rW rZ r[ rJ )r4 r� rb rL r] r r r � test_upgrade� s
zBasicTest.test_upgradec C sJ xDd
D ]<\}}t jd|d�}|j| j� | jd�}| jd| |� qW d S ) NT�trueF�false)ry �system_site_packagesz
pyvenv.cfgz"include-system-site-packages = %s
�Tr� �Fr� )r� r� )rQ r` rR r* rE rY )r4 Zssp�srb r\ r r r �test_isolation� s
zBasicTest.test_isolationzNeeds symlinksc C sR xLdD ]D}t jd|d�}|j| j� | j| j| j�}|r| jtj j
|�� qW d S )NFT)ry �symlinks)FT) rQ r` rR r* r? r, r3 rJ r& r' rU )r4 Zuslrb rL r r r �test_symlinking� s
zBasicTest.test_symlinkingc C sb t | j� | jtj| j� tjjtjj| j�| j | j
�}t |ddg�\}}| j |j
� |j� � d S )Nz-cz!import sys; print(sys.executable))r r* r= rQ rR r& r' r> r( r, r3 r ra ri rj )r4 rk r r r r r �test_executable
s
zBasicTest.test_executablec C sl t | j� tjddd�}|j| j� tjjtjj| j�| j | j
�}t |ddg�\}}| j |j
� |j� � d S )NT)ry r� z-cz!import sys; print(sys.executable))r r* rQ r` rR r& r' r> r( r, r3 r ra ri rj )r4 rb rk r r r r r �test_executable_symlinks s
z"BasicTest.test_executable_symlinksr zonly relevant on Windowsc C s� t | j� tjjtjj| j�d�}tjdd�}|j|� tjj|| j d�}tjj|| j | j
�}t |d| j
ddgdd �\}}| j |j
� d
� d S ) Nu ϼўТλФЙT)ry z activate.bat�&z-czprint(0)Zoem)r
�0)r r* r& r' r> r( rQ r` rR r, r3 r ra ri )r4 r* rb Zactivaterk r r r r r �test_unicode_in_batch_file( s
z$BasicTest.test_unicode_in_batch_file�r �r �r �r rn )r� r� r� r� )r )r! )r )rF rG rH rK r^ rd �
skipInVenvrl r- r/ rz r. rv r{ r~ r� r� r� �unittest�
skipUnlessr r� r� r� r& r+ r� r r r r rI S s0
rI c @ sX e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej e
d �ej e d �e d
d� ���Z
dS )�
EnsurePipTestc C sd t jjt jj| j�| j| j�}t|ddg�\}}|jd�}| j |d� |jd�}| j |j
� d� d S )Nz-cz1try:
import pip
except ImportError:
print("OK")zlatin-1� ZOK) r& r' r> r( r* r, r3 r �decodera ri )r4 rk r r r r r �assert_pip_not_installed< s
z&EnsurePipTest.assert_pip_not_installedc C s& t | j� | jtj| j� | j� d S )N)r r* r= rQ rR r� )r4 r r r �test_no_pip_by_defaultI s
z$EnsurePipTest.test_no_pip_by_defaultc C s* t | j� | jtj| jdd� | j� d S )NF)�with_pip)r r* r= rQ rR r� )r4 r r r �test_explicit_no_pipN s
z"EnsurePipTest.test_explicit_no_pipc
C sd t tjd��}| j|j� d� W d Q R X tjj� dkrL| jtjjtj�� n| j tjjtj�� d S )Nrx r� Znul)
rA r& �devnullra rB �lowerrV r' rW rJ )r4 rC r r r � test_devnullS s
zEnsurePipTest.test_devnullc ) C s2 t | j� t� � ��}d|d< d|d< tj� ��}||d< d}d}d }xX|fD ]N\}}tjj||�} tj| � tjj| |�}
t |
d ��} | j
|� W d Q R X qJW y| j t j
| j|d d
� W nJ tjk
�r } z*| jjdd�}
d}| j|j| |
�� W Y d d } ~ X nX W d Q R X W d Q R X tjjtjj| j�| j| j�}t|dddg�\}}|jd�}| j|d� |jd�}djtj� �}| j|d t|�� |� tj | j�jd�}| j ||� t� �}t|ddg�\}}W d Q R X |jd�}t j dd|t j d�}| j|j!� d� |jd�}| j d|� | j d|� |�s.| j"� d S ) N�eZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1�pip�pip.ini�.pip�pip.conf�wT)r� r� �replace)�errorsz {}
**Subprocess Output**
{}z-Imz --versionzlatin-1r� zpip {}zensurepip._uninstallzN^The directory .* or its parent directory is not owned by the current user .*$)�flagsz Successfully uninstalled pipz#Successfully uninstalled setuptools)r� r� )r� r� )#r r* r r) ZTemporaryDirectoryr& r' r> rp rA rq r= rQ rR r r r; r� Zfail�formatr( r, r3 r ra � ensurepip�version�len�fsencoderY �re�sub� MULTILINE�rstripr� )r4 r� �envvarsZhome_dirZ
bad_configZ win_locationZposix_locationrX Zfname�dirpathZfpathrC �excZdetails�msgrk r r Zexpected_versionr* r r r �do_test_with_pipa sV
8
z EnsurePipTest.do_test_with_pipz@some dependencies of pip import threading module unconditionallyzpip requires ctypesc C s | j d� | j d� d S )NFT)r� )r4 r r r �
test_with_pip� s
zEnsurePipTest.test_with_pipN)rF rG rH r� r� r� r� r� r� r� � threading�ctypesr r� r r r r r� 9 s
T
r� �__main__)N) r� r& Zos.pathr� rS r r- r) Z test.supportr r r r r r r� rQ r� � ImportErrorr� ZskipIfre rf r� r ZTestCaser rI r� rF �mainr r r r �<module> s:
% g