Your IP : 216.73.216.147
U
i�f�Q � @ s( d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z m
Z
mZmZmZmZmZ ddlZddlZddlZz ddlZW n ek
r� dZY nX e�ejejkp�ejej kd�Z ddd�Z G dd� dej �Z G d d
� d
e �Z!e G d d � d e ��Z"e#d
k�r$e�$� 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�
import_module�*skip_if_broken_multiprocessing_synchronizez: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 �3/opt/alt/python38/lib64/python3.8/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
j }t j�
|�d
| _t
jd kr�t j�|�r�t j�|�s�d | _nd
| _d S )N�nt�Scripts��Lib�Include�bin�lib� python%d.%d� �include����win32TF)�os�path�realpath�tempfileZmkdtemp�env_dir�name�bindirr r# �sys� version_info�_base_executable�split�exe�platform�lexists�exists�cannot_link_exe)�self�
executabler r r �setUp4 s"
�
�zBaseTest.setUpc C s t | j� d S �N)r r* �r6 r r r �tearDownG s zBaseTest.tearDownc
O s> t � �"}t� �}|||� W 5 Q R X W 5 Q R X |�� |�� fS r9 )r r �getvalue)r6 �func�args�kwargs�output�errorr r r �run_with_captureJ s zBaseTest.run_with_capturec G s t jj| jf|�� S r9 )r& r' �joinr* )r6 r> r r r � get_env_fileP s zBaseTest.get_env_filec G s( t | j|� d��}|�� }W 5 Q R X |S )N�r)�openrD �read)r6 r> �f�resultr r r �get_text_file_contentsS s z BaseTest.get_text_file_contentsN)
�__name__�
__module__� __qualname__�__doc__ZmaxDiffr8 r; rB rD rJ r r r r r 0 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 r9 )rD �
assertTruer& r' �isdir)r6 r> �fnr r r rQ [ 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�}tj}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.
�lib64�P� �posix�darwin�
pyvenv.cfgz home = %s�Contents of %r:� %r�File %r should exist.N) r r* rB �venv�createrQ r, r# r rD �structZcalcsizer& r+ r- r2 rP r' �islink� assertFalser4 rJ r/ �dirname�assertInr1 �print�listdir)r6 r Z
conditions�datar7 r' rR �bdr r r �
test_defaults_ s,
�
zBasicTest.test_defaultsc C s� t j�| j�d }t| j� t�� }| �|j| j� |� | j�}| �
d�}| � |j d| � | �
d|� t| j� tjdd�}| �|j| j� |� | j�}| �
d�}| � |j d� | �d|� d S ) N� rX z(%s) z prompt = z My prompt)�promptz (My prompt) zprompt = 'My prompt'
)r& r' r0 r* r r\ �
EnvBuilderrB r] Zensure_directoriesrJ � assertEqualri Z assertNotInrb )r6 Zenv_name�builder�contextre r r r � test_prompt{ s
zBasicTest.test_promptc C s� t | j� | �tj| j� tj�| j| j| j �}|ddg}d| jfd| jfdt
j fdt
j ffD ]4\}}d| |d< t
|�\}}| �|�� |�� � q^dS ) z>
Test that the prefix values are as expected.
�-cN�prefix� exec_prefix� base_prefix�base_exec_prefixzimport sys; print(sys.%s)r" )r r* rB r\ r] r& r' rC r, r1 r- rr rs r rk �strip�encode)r6 �envpyr rp Zexpectedr r r r r �
test_prefixes� s
� zBasicTest.test_prefixesr% ))r )r r )r �
site-packages)r )r# )r r r! Nr" rx c
C sZ |D ]P}t jj| jf|�� }t �|� t j�||�}t|d��}|�d� W 5 Q R X qdS )zn
Create some files in the environment which are unrelated
to the virtual environment.
�wb� Still here?N)r& r' rC r* �mkdirrF �write)r6 �paths�filename�subdirs�drR rH r r r �create_contents� s
zBasicTest.create_contentsc
C s� | � | jd� t�| j� | jD ]T}tjj| jf|d �� }| �tj� |�� t
|d��}| � |� � d� W 5 Q R X q tj
dd�}|�| j� | jD ].}tjj| jf|d �� }| �tj� |�� q�dS )zE
Test creating environment in an existing directory.
�foo)r� �rbrz T��clearN)r� � ENV_SUBDIRSr\ r] r* r& r' rC rP r4 rF rk rG rj r` )r6 r rR rH rl r r r �test_overwrite_existing� s
z!BasicTest.test_overwrite_existingc C sZ t �|�D ]J}t j�||�}t j�|�s4t j�|�r@t �|� q
t j�|�r
t|� q
d S r9 ) r& rd r' rC r_ �isfile�removerQ r )r6 r' rR r r r �clear_directory� s zBasicTest.clear_directoryc
C sn | j d d� D ]Z}tjj| jf|�� }t|d��}|�d� W 5 Q R X | �tt ft
j | j� | � | j� qd S )N� ry � )
r� r& r' rC r* rF r| Z assertRaises�
ValueError�OSErrorr\ r] r� )r6 r} rR rH r r r �test_unoverwritable_fails� s z#BasicTest.test_unoverwritable_failsc C s� dD ]�}t j|d�}| �|j| j� | �| j� | �| j� | j| j� | � | j| j
�}t j �
|�s�| � | j�}td| � tdt �|� � | �t j �
|�d| � qdS )zC
Test upgrading an existing environment directory.
�FT)�upgraderY rZ r[ N)r\ rj rB r] r* rQ r, r# r rD r1 r& r' r4 rc rd rP )r6 r� rl rR rf r r r � test_upgrade� s zBasicTest.test_upgradec C sF dD ]<\}}t jd|d�}|�| j� | �d�}| �d| |� qdS )z:
Test isolation from system site-packages
))T�true)FZfalseT)r� �system_site_packagesrX z"include-system-site-packages = %s
N)r\ rj r] r* rJ rb )r6 Zssp�srl re r r r �test_isolation� s
zBasicTest.test_isolationzNeeds symlinksc C sh dD ]^}t jd|d�}|�| j� | �| j| j�}|r| jrP| �t j
� |�� q| � t j
� |�� qdS )z3
Test symlinking works as expected
r� T�r� ZsymlinksN)
r\ rj r] r* rD r, r1 r5 r` r& r' r_ rP )r6 Zuslrl rR r r r �test_symlinking� s zBasicTest.test_symlinkingc C sb t | j� | �tj| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� |�� � dS )�D
Test that the sys.executable value is as expected.
ro �!import sys; print(sys.executable)N)r r* rB r\ r] r& r' rC r( r, r1 r rk rt ru �r6 rv 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 )r� Tr� ro r� N�r r* r\ rj r] r& r' rC r( r, r1 r rk rt ru �r6 rl rv 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 ϼўТλФЙTr� z activate.bat�&ro zprint(0)Zoem)r �0N)r r* r& r' rC r( r\ rj r] r, r1 r rk rt )r6 r* rl �activaterv r r r r r �test_unicode_in_batch_file2 s
�
z$BasicTest.test_unicode_in_batch_filec C sh t � t| j� | �tj| j� tj�tj� | j�| j
| j �}t |ddg�\}}| �
|�� d�� � dS )zA
Test that the multiprocessing is able to spawn.
ro zrfrom multiprocessing import Pool; pool = Pool(1); print(pool.apply_async("Python".lower).get(3)); pool.terminate()�pythonN)r r r* rB r\ r] r& r' rC r( r, r1 r rk rt ru r� r r r �test_multiprocessingC 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 5 Q R X t||g�\}}| �|d
�� � | �|d
�� � d S ) N�bashzbash required for this testTr� r� ztest_strict.sh�wzset -euo pipefail
source z
deactivate
� )�shutilZwhichZskipTestr r* r\ rj r] r& r' rC r, rF r| r rk ru )r6 r� rl r� Z test_scriptrH r r r r r �%test_deactivate_with_strict_bash_optsX s
z/BasicTest.test_deactivate_with_strict_bash_optsrW zonly relevant on macOSc C sf t | j� t�� }|�| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� d�� � d S )Nro z5import os; print("__PYVENV_LAUNCHER__" in os.environ)�Falser� r� r r r �test_macos_envk s
��
zBasicTest.test_macos_env) rK rL rM rN rQ rg rn �requireVenvCreaterw r- r2 r� r. r� r� r� r� r� r� �unittest�
skipUnlessr r� r� r� r& r+ r� r� ZskipIfr� r� r r r r rO X sD
�
rO 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 )Nro z1try:
import pip
except ImportError:
print("OK")�latin-1r� ZOK) r& r' rC r( r* r, r1 r �decoderk rt r� r r r �assert_pip_not_installedz s ��
z&EnsurePipTest.assert_pip_not_installedc C s& t | j� | �tj| j� | �� d S r9 �r r* rB r\ r] r� r: r r r �test_no_pip_by_default� s
z$EnsurePipTest.test_no_pip_by_defaultc C s* t | j� | jtj| jdd� | �� d S )NF)�with_pipr� r: r r r �test_explicit_no_pip� s
z"EnsurePipTest.test_explicit_no_pipc C s@ t tjd��}| �|�� d� W 5 Q R X | �tj�tj�� d S )Nr� r� )rF r& �devnullrk rG rP r' r4 )r6 rH r r r � test_devnull� s zEnsurePipTest.test_devnullc C s6 t | j� t� ��}d|d< d|d< t�� ��}||d< d}d}d}|fD ]N\}}tj�||�} t�| � tj�| |�}
t |
d ��} | �
|� W 5 Q R X qFz| j t j
| j|d
d � W nH tjk
r� } z(| jjd d
�}
d}| �|�| |
�� W 5 d } ~ X Y nX W 5 Q R X W 5 Q R X tj�tj�| j�| j| j�}t|ddddddg�\}}|�d�}| �|d� |�d�}d�t�� �}| �|d t|�� |� t� | j��d�}| � ||� t� � }t|dddddg�\}}W 5 Q R X |�d�}t j dd|t j d�}| �|�!� d� |�d�}| � d|� | � d |� |�s2| �"� 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 --versionr� r� 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' rC r{ rF r| rB r\ r] r
r r@ r� Zfail�formatr( r, r1 r rk � ensurepip�version�len�fsencoderb �re�sub� MULTILINE�rstripr� )r6 r� ZenvvarsZhome_dirZ
bad_configZ win_locationZposix_locationra Zfname�dirpathZfpathrH �excZdetails�msgrv r r Zexpected_versionr* r r r �do_test_with_pip� sr
�
8
�
�
�
z EnsurePipTest.do_test_with_pipzpip requires ctypesc C s | � d� | � d� d S )NFT)r� r: r r r �
test_with_pip� s
zEnsurePipTest.test_with_pipN)rK rL rM rN r� r� r� r� r� r� r� �ctypesr r� r r r r r� w s
Z
r� �__main__)N)%rN r� r& Zos.pathr� r� r^ r
r- r) Z test.supportr r r r r r r r Z threadingr� r\ r� � ImportErrorr� rp rr r/ r7 r� r ZTestCaser rO r� rK �mainr r r r �<module> s@ (
�
( !