File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_venv.cpython-37.pyc
B
��g�O � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z m
Z
mZmZmZ ddlZddlZddlZy ddlZW n ek
r� dZY nX e�eed�p�ejejkd�Zddd�ZG dd � d ej �Z G d
d � d e �Z eG d d
� d
e ��Z e dk�re�!� 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�rmtree�_base_executablez:cannot run venv.create from within a venv on this platformc C sB t j| t jt j|d�}|�� \}}|jr:t �|j| ||��||fS )N)�stdout�stderr�encoding)�
subprocess�Popen�PIPEZ communicate�
returncode�CalledProcessError)�cmdr �p�out�err� r �*/usr/local/lib/python3.7/test/test_venv.py� check_output# s r c @ s< e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�BaseTestzBase class for venv tests.i� c C s� t j�t�� �| _t jdkr0d| _d| _d| _ n$d| _ddt
j d d� f| _d | _ t t
d
t
j
�}t j�|�d | _t
jd kr�t j�|�r�t j�|�s�d
| _nd| _d S )N�nt�Scripts)�Lib�Include�bin�libz python%d.%d� �includer ����win32TF)�os�path�realpath�tempfileZmkdtemp�env_dir�name�bindirr r �sys� version_info�getattr�
executable�split�exe�platform�lexists�exists�cannot_link_exe)�selfr- r r r �setUp2 s
zBaseTest.setUpc C s t | j� d S )N)r r' )r4 r r r �tearDownE s zBaseTest.tearDownc
O s> t � �"}t� �}|||� W d Q R X W d Q R X |�� |�� fS )N)r r �getvalue)r4 �func�args�kwargs�output�errorr r r �run_with_captureH s zBaseTest.run_with_capturec G s t jj| jf|�� S )N)r# r$ �joinr' )r4 r9 r r r � get_env_fileN s zBaseTest.get_env_filec G s( t | j|� d��}|�� }W d Q R X |S )N�r)�openr? �read)r4 r9 �f�resultr r r �get_text_file_contentsQ s z BaseTest.get_text_file_contentsN)
�__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@d Z n4d d
ddde j dd� fdde j dd� dffZ dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Ze�e� d �d!d"� �Zed#d$� �Ze�e� d �d%d&� �Ze�ejd'kd(�d)d*� �Zed+d,� �Z e� ejd'kd-�d.d/� �Z e�e j
d0kd1�d2d3� �Z dS )4� BasicTestz Test venv module functionality.c G s | j |� }| �tj�|�� d S )N)r? �
assertTruer# r$ �isdir)r4 r9 �fnr r r rL Y s
zBasicTest.isdirc C s, t | j� | �tj| j� | �| j� | �| j� | j| j� | � d�}t
� d�dkoht j
dkohtjdk}|r�| �t j�|�� n| �t j�|�� | �d�}ttdtj�}t j�|�}| �d| |� | � | j| j�}t j�|��s| � | j�}td | � td
t � |� � | �t j�|�d | � d S )
zB
Test the create function with default arguments.
Zlib64�P� �posix�darwinz
pyvenv.cfgr z home = %szContents of %r:z %rzFile %r should exist.N) r r' r= �venv�createrL r) r r r? �struct�calcsizer# r( r* r0 rK r$ �islink� assertFalser2 rE r, r- �dirname�assertInr/ �print�listdir)r4 r Z
conditions�datar- r$ rM �bdr r r �
test_defaults] s*
zBasicTest.test_defaultsc C sb t j�| j�d }t�� }|�| j�}| �|jd| � tjdd�}|�| j�}| �|jd� d S )N� z(%s) z My prompt)�promptz (My prompt) ) r# r$ r. r' rR �
EnvBuilderZensure_directories� assertEqualr` )r4 Zenv_name�builder�contextr r r � test_prompty s zBasicTest.test_promptc C s� t | j� | �tj| j� tj�| j| j| j �}|ddg}x\d| jfd| jfdt
j fdt
j ffD ]4\}}d| |d< t
|�\}}| �|�� |�� � 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 )r r' r= rR rS r# r$ r> r) r/ r* rg rh r rb �strip�encode)r4 �envpyr rf Zexpectedr r r r r �
test_prefixes� s
zBasicTest.test_prefixesr" ))r )r )r )r z
site-packages)r )r )r r z python%d.%dNr z
site-packagesc
C s^ xX|D ]P}t jj| jf|�� }t �|� t j�||�}t|d��}|�d� 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� | � | jd� t�| j� x^| jD ]T}tjj| jf|d �� }| �tj� |�� t
|d��}| � |� � d� W dQ R X q"W tj
dd�}|�| j� x8| jD ].}tjj| jf|d �� }| �tj� |�� q�W dS )zE
Test creating environment in an existing directory.
�foo)ru �rbs Still here?NT)�clear)rt � ENV_SUBDIRSrR rS r' r# r$ r> rK r2 rA rb rB ra rW )r4 rr rM rC rc r r r �test_overwrite_existing� s z!BasicTest.test_overwrite_existingc C s^ xXt �|�D ]J}t j�||�}t j�|�s6t j�|�rBt �|� q t j�|�r t|� q W d S )N) r# r[ r$ r> rV �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��}|�d� W d Q R X | �tt ft
j | j� | � | j� qW d S )N� rm � )
rx r# r$ r> r' rA ro Z assertRaises�
ValueError�OSErrorrR rS r| )r4 rp 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
�}t j �
|�s�| � | j�}td| � tdt �|� � | �t j �
|�d| � qW dS )zC
Test upgrading an existing environment directory.
)FT)�upgradezContents of %r:z %rzFile %r should exist.N)rR ra r= rS r' rL r) r r r? r/ r# r$ r2 rZ r[ rK )r4 r� rc rM r] r r r � test_upgrade� s
zBasicTest.test_upgradec C sJ xDdD ]<\}}t jd|d�}|�| j� | �d�}| �d| |� qW dS )z:
Test isolation from system site-packages
))T�true)FZfalseT)rw �system_site_packagesz
pyvenv.cfgz"include-system-site-packages = %s
N)rR ra rS r' rE rY )r4 Zssp�src r\ r r r �test_isolation� s
zBasicTest.test_isolationzNeeds symlinksc C sl xfdD ]^}t jd|d�}|�| j� | �| j| j�}|r| jrR| �t j
� |�� q| � t j
� |�� qW dS )z3
Test symlinking works as expected
)FTT)rw �symlinksN)
rR ra rS r' r? r) r/ r3 rW r# r$ rV rK )r4 Zuslrc rM r r r �test_symlinking� s
zBasicTest.test_symlinkingc C sb t | j� | �tj| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� |�� � 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) r/ r rb ri rj )r4 rk r r r r r �test_executable
s
zBasicTest.test_executablec C sl t | j� tjddd�}|�| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� |�� � dS )zD
Test that the sys.executable value is as expected.
T)rw r� z-cz!import sys; print(sys.executable)N)r r' rR ra rS r# r$ r> r% r) r/ r rb ri rj )r4 rc rk r r r r r �test_executable_symlinks s
z"BasicTest.test_executable_symlinksr zonly relevant on Windowsc C s� t | j� tj�tj�| j�d�}tjdd�}|�|� tj�|| j d�}tj�|| j | j
�}t |d| j
ddgdd �\}}| � |�
� d
� d S ) z0
Test handling of Unicode paths
u ϼўТλФЙT)rw z activate.bat�&z-czprint(0)Zoem)r �0N)r r' r# r$ r> r% rR ra rS r) r/ r rb ri )r4 r' rc �activaterk r r r r r �test_unicode_in_batch_file( s
z$BasicTest.test_unicode_in_batch_filec C sb t | j� | �tj| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� d�� � dS )zA
Test that the multiprocessing is able to spawn.
z-czrfrom multiprocessing import Pool; pool = Pool(1); print(pool.apply_async("Python".lower).get(3)); pool.terminate()ZpythonN)r r' r= rR rS r# r$ r> r% r) r/ r rb ri rj )r4 rk r r r r r �test_multiprocessing9 s
z BasicTest.test_multiprocessingznot relevant on Windowsc C s� t �d�}|d kr | �d� t| j� tjdd�}|�| j� tj �
| j| j d�}tj �
| jd�}t |d��}|�
d|� d �� W d Q R X t||g�\}}| �|d
�� � | �|d
�� � d S ) N�bashzbash required for this testT)rw r� ztest_strict.sh�wzset -euo pipefail
source z
deactivate
� )�shutilZwhichZskipTestr r' rR ra rS r# r$ r> r) rA ro r rb rj )r4 r� rc r� Z test_scriptrC r r r r r �%test_deactivate_with_strict_bash_optsI s
z/BasicTest.test_deactivate_with_strict_bash_optsrQ zonly relevant on macOSc C sf t | j� t�� }|�| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� d�� � d S )Nz-cz5import os; print("__PYVENV_LAUNCHER__" in os.environ)�False)r r' rR ra rS r# r$ r> r% r) r/ r rb ri rj )r4 rc rk r r r r r �test_macos_env\ s
zBasicTest.test_macos_env) rF rG rH rI rL r^ re �requireVenvCreaterl r* r0 rx r+ rt ry r| r� r� r� �unittest�
skipUnlessr r� r� r� r# r( r� r� ZskipIfr� r� r r r r rJ V s2
rJ c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze �
e d �e d
d� ��Z
dS )�
EnsurePipTestz%Test venv module installation of pip.c C sd t j�t j�| j�| j| j�}t|ddg�\}}|�d�}| � |d� |�d�}| � |�
� d� d S )Nz-cz1try:
import pip
except ImportError:
print("OK")zlatin-1r� ZOK) r# r$ r> r% r' r) r/ r �decoderb ri )r4 rk r r r r r �assert_pip_not_installedk s
z&EnsurePipTest.assert_pip_not_installedc C s& t | j� | �tj| j� | �� d S )N)r r' r= rR rS r� )r4 r r r �test_no_pip_by_defaultx s
z$EnsurePipTest.test_no_pip_by_defaultc C s* t | j� | jtj| jdd� | �� d S )NF)�with_pip)r r' r= rR rS r� )r4 r r r �test_explicit_no_pip} s
z"EnsurePipTest.test_explicit_no_pipc C sd t tjd��}| �|�� d� W d Q R X tj�� dkrL| �tj�tj�� n| � tj�tj�� d S )Nrv r~ Znul)
rA r# �devnullrb rB �lowerrW r$ r2 rK )r4 rC r r r � test_devnull� s
zEnsurePipTest.test_devnullc C s: t | j� t� ��}d|d< d|d< t�� ��}||d< d}d}d}xX|fD ]N\}}tj�||�} t�| � tj�| |�}
t |
d ��} | �
|� W d Q R X qHW y| j t j
| j|d
d � W nH tjk
r� } z(| jjd d
�}
d}| �|�| |
�� W d d } ~ X Y nX W d Q R X W d Q R X tj�tj�| j�| j| j�}t|ddddddg�\}}|�d�}| �|d� |�d�}d�t�� �}| �|d t|�� |� t� | j��d�}| � ||� t� � }t|dddddg�\}}W d Q R X |�d�}t j dd|t j d�}| �|�!� d� |�d�}| � d|� | � d |� |�s6| �"� d S ) N�eZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1)�pipzpip.ini)z.pipzpip.confr� T)r� r� �replace)�errorsz {}
**Subprocess Output**
{}z-Wzignore::DeprecationWarningz-Iz-mr� z --versionzlatin-1r� zpip {}zensurepip._uninstallzl^(WARNING: )?The directory .* or its parent directory is not owned or is not writable by the current user.*$)�flagsz Successfully uninstalled pipz#Successfully uninstalled setuptools)#r r' r r& ZTemporaryDirectoryr# r$ r> rn rA ro r= rR rS r r r; r� Zfail�formatr% r) r/ r rb � 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_pip� s^
8
z EnsurePipTest.do_test_with_pipzpip requires ctypesc C s | � d� | � 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� �ctypesr r� r r r r r� h s
Z
r� �__main__)N)"rI r� r# Zos.pathr� r� rT r r* r& Z test.supportr r r r r r Z threadingr� rR r� � ImportErrorr� �hasattrrf rg r� r ZTestCaser rJ r� rF �mainr r r r �<module> s>
(