Your IP : 216.73.216.85
3
� f G � @ s* d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z m
Z
mZmZ ddlZddlZy ddlZW n ek
r� dZY nX y ddlZW n ek
r� dZY nX ejejejkd�Zddd�ZG dd� dej�ZG d d
� d
e�Z eG d d � d e��Z e d
k�r&ej � dS )z|
Test harness for the venv module.
Copyright (C) 2011-2012 Vinay Sajip.
Licensed to the PSF under a contributor agreement.
� 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 �3/opt/alt/python36/lib64/python3.6/test/test_venv.py� check_output# s r c @ s< e Zd ZdZdZdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�BaseTestzBase class for venv tests.�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__�__doc__ZmaxDiffr5 r6 r= r? rE r r r r r . s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zedd � �Ze j
d
kr@d0Z n4d1d2d3dde j dd� fdde j dd� dffZ dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zeje� d!�d"d#� �Zed$d%� �Zeje� d!�d&d'� �Zejejd(kd)�d*d+� �ZdS )4� BasicTestz Test venv module functionality.c G s | j |� }| jtjj|�� d S )N)r? �
assertTruer& r' �isdir)r4 r9 �fnr r r rL 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 )
zB
Test the create function with default arguments.
�lib64�P� �posixr" z
pyvenv.cfgr# z home = %szContents of %r:z %rzFile %r should exist.N) r r* r= �venv�createrL r, r! r r? �structZcalcsizer& r+ r- r/ rK r' �islink� assertFalse�existsrE r0 r1 �dirname�assertInr3 �print�listdir)r4 r Z
conditions�datar1 r' rM �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* rR �
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�}|ddg}x\d| jfd| jfdtjfdtjffD ]4\}}d| |d< t|�\}}| j |j� |j� � q�W dS )z>
Test that the prefix values are as expected.
z-cN�prefix� base_prefix�base_exec_prefixzimport sys; print(sys.%s)r )ra r- rf re rg � exec_prefixr r* r= rR rS r& r' r> r, r3 r �strip�encode)r4 �envpyr re Zexpectedr r r r r �
test_prefixes� s
zBasicTest.test_prefixesZwin32r 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 dQ R X qW dS )zn
Create some files in the environment which are unrelated
to the virtual environment.
�wbs Still here?N)r& r' r> r* �mkdirrA �write)r4 �paths�filename�subdirs�drM 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 dQ R X q"W tj
dd�}|j| j� x8| jD ].}tjj| jf|d �� }| jtjj |�� q�W dS ) zE
Test creating environment in an existing directory.
�foo�rbs Still here?NT)�clear)rv )rv )ru � ENV_SUBDIRSrR rS r* r& r' r> rK rW rA ra rB r` rV )r4 rs rM 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�removerL r )r4 r' rM 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� rn � )
ry r& r' r> r* rA rp Z assertRaises�
ValueError�OSErrorrR rS r} )r4 rq rM 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 dS ) zC
Test upgrading an existing environment directory.
FT)�upgradezContents of %r:z %rzFile %r should exist.N)FT)rR r` r= rS r* rL r, r! r r? r3 r& r' rW rZ r[ rK )r4 r� rb rM r] r r r � test_upgrade� s
zBasicTest.test_upgradec C sJ xDd D ]<\}}t jd|d�}|j| j� | jd�}| jd| |� qW dS ) z:
Test isolation from system site-packages
T�trueF�false)rx �system_site_packagesz
pyvenv.cfgz"include-system-site-packages = %s
N�Tr� �Fr� )r� r� )rR r` rS 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 dS )z3
Test symlinking works as expected
FT)rx �symlinksN)FT) rR r` rS r* r? r, r3 rK r& r' rU )r4 Zuslrb rM r r r �test_symlinking� s
zBasicTest.test_symlinkingc C sb t | j� | jtj| j� tjjtjj| j�| j | j
�}t |ddg�\}}| j |j
� |j� � dS )zD
Test that the sys.executable value is as expected.
z-cz!import sys; print(sys.executable)N)r r* r= rR rS 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� � dS )zD
Test that the sys.executable value is as expected.
T)rx r� z-cz!import sys; print(sys.executable)N)r r* rR r` rS 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 ) z0
Test handling of Unicode paths
u ϼўТλФЙT)rx z activate.bat�&z-czprint(0)Zoem)r
�0N)r r* r& r' r> r( rR r` rS 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 rm )r� r� r� r� )r )r! )r ) rF rG rH rI rL r^ rd �
skipInVenvrl r- r/ ry r. ru rz r} r� r� r� �unittest�
skipUnlessr r� r� r� r& r+ r� r r r r rJ S s2
rJ c @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze j
e d �e j
e d
�e
dd� ���ZdS )�
EnsurePipTestz%Test venv module installation of pip.c 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= rR rS 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= rR rS 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 )Nrw r Znul)
rA r& �devnullra rB �lowerrV r' rW rK )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> ro rA rp r= rR rS r r r; r� Zfail�formatr( r, r3 r ra � ensurepip�version�len�fsencoderY �re�sub� MULTILINE�rstripr� )r4 r� ZenvvarsZhome_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 rI r� r� r� r� r� r� r� � threading�ctypesr r� r r r r r� 9 s
T
r� �__main__)N) rI r� r& Zos.pathr� rT r r- r) Z test.supportr r r r r r r� rR r� � ImportErrorr� ZskipIfre rf r� r ZTestCaser rJ r� rF �mainr r r r �<module> s<
% g