Your IP : 216.73.216.213
a
XC?h�f � @ s2 d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z m
Z
mZmZmZmZmZmZ ddlZddlZddlmZ z ddlZW n ey� dZY n0 e�ejej kp�ej ej kd�Z ddd�Z G dd � d ej!�Z"G d
d � d e"�Z#e G d d
� d
e"��Z$e%dk�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_synchronize)�patchz: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/python39/lib64/python3.9/test/test_venv.py� check_output& s �
�r c @ sJ e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d �d
d�Z dd� Z
dS )�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 �setUp5 s"
�
�zBaseTest.setUpc C s t | j� d S �N)r r+ �r7 r r r �tearDownH s zBaseTest.tearDownc O sj t � �D}t� � }||i |�� W d � n1 s20 Y W d � n1 sP0 Y |�� |�� fS r: )r r �getvalue)r7 �func�args�kwargs�output�errorr r r �run_with_captureK s JzBaseTest.run_with_capturec G s t jj| jg|�R � S r: )r' r( �joinr+ )r7 r? r r r � get_env_fileQ s zBaseTest.get_env_filezutf-8�r
c G s@ t | j|� d|d��}|�� }W d � n1 s20 Y |S )N�rrF )�openrE �read)r7 r
r? �f�resultr r r �get_text_file_contentsT s &z BaseTest.get_text_file_contentsc C s$ |� |�s | �d|�d|��� d S )NzString z does not end with )�endswith�fail)r7 �string�tailr r r �assertEndsWithY s
zBaseTest.assertEndsWithN) �__name__�
__module__� __qualname__�__doc__ZmaxDiffr9 r< rC rE rL rQ r r r r r 1 s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d � �Z e
j d krHd
Z n4ddddde
j
dd� fdde
j
dd� dffZ dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Ze�e� d"�d#d$� �Zed%d&� �Ze�e� d"�d'd(� �Ze�ej d)kd*�d+d,� �Z e�ej d)kd*�d-d.� �Z e�ej d)kd/�d0d1� �Z e�ej d)kd/�d2d3� �Z ed4d5� �Z!e�ej d)kd6�d7d8� �Z"e�e
j d9kd:�d;d<� �Z#dS )=� BasicTestz Test venv module functionality.c G s | j |� }| �tj�|�� d S r: )rE �
assertTruer' r( �isdir)r7 r? �fnr r r rX ` 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.
Zlib64�P� �posix�darwin�
pyvenv.cfgz home = %s�Contents of %r:� %r�File %r should exist.N) r r+ rC �venv�createrX r- r$ r! rE �structZcalcsizer' r, r. r3 rW r( �islink� assertFalser5 rL r0 �dirname�assertInr2 �print�listdir)r7 r Z
conditions�datar8 r( rY �bdr r r �
test_defaultsd s,
�
zBasicTest.test_defaultsc C s0 t j�| j�d }t| j� t�� }| �|j| j� |� | j�}| �
d�}| � |j d| � | �
d|� t| j� tjdd�}| �|j| j� |� | j�}| �
d�}| � |j d� | �d|� t| j� tjd d�}t j�t �� �}| �|j| j� |� | j�}| �
d�}| � |j d| � | �d
| |� d S ) N� r^ z(%s) z prompt = z My prompt)�promptz (My prompt) zprompt = 'My prompt'
�.zprompt = '%s'
)r' r( r1 r+ r rb �
EnvBuilderrC rc �ensure_directoriesrL � assertEqualro Z assertNotInrh �basename�getcwd)r7 �env_name�builder�contextrk �cwdr r r � test_prompt� s.
zBasicTest.test_promptc s� t �� }tjdkrdnd}tj�tj�d }t� � ��}tj�
tj� |||��� tjdkrltj�
tj� � ��� � �fdd�}|�
|�}td|�� |�|� W d � n1 s�0 Y W d � n1 s�0 Y d S )Nr& r r rn c
s2 t j�| d �| d< ��| � ddddddg� d S )Nr �-m�pipZinstallz --upgradeZ
setuptools)r' r( �normcasers )r �Z
expect_exer7 r r �pip_cmd_checker� s ��z<BasicTest.test_upgrade_dependencies.<locals>.pip_cmd_checkerzvenv.subprocess.check_call)rb rq r. r3 r' r( r1 r8 r* �TemporaryDirectoryr} rD r) rr r
Zupgrade_dependencies)r7 rw Zbin_pathZ
python_exeZ fake_env_dirr Z fake_contextr r~ r �test_upgrade_dependencies� s
�
z#BasicTest.test_upgrade_dependenciesc 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+ rC rb rc r' r( rD r- r2 r. r� r� r rs �strip�encode)r7 �envpyr r� Zexpectedr r r r r �
test_prefixes� s
� zBasicTest.test_prefixesr& ))r )r r )r �
site-packages)r )r$ )r! r! r" Nr# r� c C sp |D ]f}t jj| jg|�R � }t �|� t j�||�}t|d��}|�d� W d � q1 s`0 Y qdS )zn
Create some files in the environment which are unrelated
to the virtual environment.
�wb� Still here?N)r' r( rD r+ �mkdirrH �write)r7 �paths�filename�subdirs�drY rJ r r r �create_contents� s
zBasicTest.create_contentsc C s� | � | jd� t�| j� | jD ]j}tjj| jg|d �R � }| �tj� |�� t
|d�� }| � |� � d� W d � q 1 s�0 Y q tj
dd�}|�| j� | jD ]0}tjj| jg|d �R � }| �tj� |�� q�dS )zE
Test creating environment in an existing directory.
�foo)r� �rbr� NT��clear)r� � ENV_SUBDIRSrb rc r+ r' r( rD rW r5 rH rs rI rq rf )r7 r� rY rJ rw r r r �test_overwrite_existing� s
0
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 r: ) r' rj r( rD re �isfile�removerX r )r7 r( rY r r r �clear_directory� s zBasicTest.clear_directoryc C s� | j d d� D ]p}tjj| jg|�R � }t|d��}|�d� W d � n1 sR0 Y | �tt ft
j | j� | � | j� qd S )N� r� � )
r� r' r( rD r+ rH r� Z assertRaises�
ValueError�OSErrorrb rc r� )r7 r� rY rJ 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)�upgrader_ r` ra N)rb rq rC rc r+ rX r- r$ r! rE r2 r' r( r5 ri rj rW )r7 r� rw rY rl 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_packagesr^ z"include-system-site-packages = %s
N)rb rq rc r+ rL rh )r7 Zssp�srw rk 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)
rb rq rc r+ rE r- r2 r6 rf r' r( re rW )r7 Zuslrw rY 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.
r� �!import sys; print(sys.executable)N)r r+ rC rb rc r' r( rD r) r- r2 r rs r� r� �r7 r� r r r r r �test_executableC s
��
zBasicTest.test_executablec C sl t | j� tjddd�}|�| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� |�� � dS )r� Tr� r� r� N�r r+ rb rq rc r' r( rD r) r- r2 r rs r� r� �r7 rw r� r r r r r �test_executable_symlinksP s
��
z"BasicTest.test_executable_symlinksr z&contains invalid characters on Windowsc C s t | j� t�d�}|du r&| �d� d}tj�tj�| j�|�}t j
dd�}|� |� tj�|| j d�}tj�| jd�}t
|d ��(}|�d
t�|�� d �� W d � n1 s�0 Y t||g�\}} |�� }
| �|�� |
d v � | �|
d
|�� � dS )zK
Test that the template strings are quoted properly (bash)
�bashN�bash required for this test�
"';&&$e|'"Tr� �activateztest_special_chars.sh�w�source �r
python -c 'import sys; print(sys.executable)'
python -c 'import os; print(os.environ["VIRTUAL_ENV"])'
deactivate
r rn �r r+ �shutil�which�skipTestr' r( rD r) rb rq rc r- rH r� �shlexZquoter �
splitlinesrW r� rQ ) r7 r� rv r+ rw r� � test_scriptrJ r r �linesr r r �test_special_chars_bash_ s
6z!BasicTest.test_special_chars_bashc C s t | j� t�d�p t�d�}|du r0| �d� d}tj�tj�| j�|�}t j
dd�}|� |� tj�|| j d�}tj�| jd �}t
|d
��(}|�d t�|�� d �� W d � n1 s�0 Y t||g�\}} |�� }
| �|�� |
d
v � | �|
d |�� � dS )zJ
Test that the template strings are quoted properly (csh)
Ztcsh�cshNzcsh required for this testr� Tr� z activate.cshztest_special_chars.cshr� r� r� r rn r� ) r7 r� rv r+ rw r� r� rJ r r r� r r r �test_special_chars_cshy s
6z BasicTest.test_special_chars_cshzonly relevant on Windowsc
C s� t | j� d}tj�tj�| j�|�}tjdd�}|�|� tj�|| j d�}tj�| jd�}t
|d��2}|� d|� d| j � d | j � d
�� W d � n1 s�0 Y t
|g�\}}|�� } | �|�� | d v � | �| d
|�� � d S )zO
Test that the template strings are quoted properly on Windows
z'&&^$eTr� � activate.batztest_special_chars.batr� z @echo off
"z" & z* -c "import sys; print(sys.executable)" & z> -c "import os; print(os.environ['VIRTUAL_ENV'])" & deactivateNr rn )r r+ r' r( rD r) rb rq rc r- rH r� r2 r r� rW r� rQ )
r7 rv r+ rw r� Z
test_batchrJ r r r� r r r �test_special_chars_windows� s&
���(z$BasicTest.test_special_chars_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� r� �&r� zprint(0)ZoemrF �0N)r r+ r' r( rD r) rb rq rc r- r2 r rs r� )r7 r+ rw r� r� r r r r r �test_unicode_in_batch_file� 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.
r� zrfrom multiprocessing import Pool; pool = Pool(1); print(pool.apply_async("Python".lower).get(3)); pool.terminate()ZpythonN)r r r+ rC rb rc r' r( rD r) r- r2 r rs r� r� r� r r r �test_multiprocessing� s
��
z BasicTest.test_multiprocessingznot relevant on Windowsc C s� t �d�}|d u r | �d� t| j� tjdd�}|�| j� tj �
| j| j d�}tj �
| jd�}t |d��"}|�
d|� d �� W d � n1 s�0 Y t||g�\}}| �|d
�� � | �|d
�� � d S ) Nr� r� Tr� r� ztest_strict.shr� zset -euo pipefail
source z
deactivate
� )r� r� r� r r+ rb rq rc r' r( rD r- rH r� r rs r� )r7 r� rw r� r� rJ r r r r r �%test_deactivate_with_strict_bash_opts� s
�(z/BasicTest.test_deactivate_with_strict_bash_optsr] zonly relevant on macOSc C sf t | j� t�� }|�| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� d�� � d S )Nr� z5import os; print("__PYVENV_LAUNCHER__" in os.environ)�Falser� r� r r r �test_macos_env� s
��
zBasicTest.test_macos_env)$rR rS rT rU rX rm rz r� �requireVenvCreater� r. r3 r� r/ r� r� r� r� r� r� �unittest�
skipUnlessr r� r� r� ZskipIfr' r, r� r� r� r� r� r� r� r r r r rV ] sR
�
rV c @ sR 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 )Nr� z1try:
import pip
except ImportError:
print("OK")�latin-1r� ZOK) r' r( rD r) r+ r- r2 r �decoders r� r� r r r �assert_pip_not_installed� s ��
z&EnsurePipTest.assert_pip_not_installedc C s& t | j� | �tj| j� | �� d S r: �r r+ rC rb rc 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 sT t tjd�� }| �|�� d� W d � n1 s20 Y | �tj�tj�� d S )Nr� r� )rH r' �devnullrs rI rW r( r5 )r7 rJ r r r � test_devnull s .zEnsurePipTest.test_devnullc C s� t | j� t� ��0}d|d< d|d< t�� ��}||d< d}d}d}|fD ]b\}}tj�||�} t�| � tj�| |�}
t |
d ��} | �
|� W d � qH1 s�0 Y qHz| j t j
| j|d
d � W nL tj�y } z0| jjd d
�}
d}| �|�| |
�� W Y d } ~ n
d } ~ 0 0 W d � n1 �s*0 Y W d � n1 �sJ0 Y tj�tj�| j�| j| j�}t|ddddddg�\}}|�d�}| �|d� |�d�}d�t�� �}| �|d t|�� |� t� | j��d�}| � ||� t� �(}t|dddddg�\}}W d � n1 �s&0 Y |�d�}t j dd|t j d�}| �|�!� d� |�d�}| � d|� | � d |� |�s�| �"� d S ) N�eZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1)r| zpip.ini)z.pipzpip.confr� T)r� r� �replace)�errorsz {}
**Subprocess Output**
{}z-Wzignore::DeprecationWarningz-Ir{ r| 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* r� r' r( rD r� rH r� rC rb rc r r rA r� rN �formatr) r- r2 r rs � ensurepip�version�len�fsencoderh �re�sub� MULTILINE�rstripr� )r7 r� ZenvvarsZhome_dirZ
bad_configZ win_locationZposix_locationrg Zfname�dirpathZfpathrJ �excZdetails�msgr� r r Zexpected_versionr+ r r r �do_test_with_pip sf
* �
h
�
�*
�
z EnsurePipTest.do_test_with_pipzpip requires ctypesc C s | � d� | � d� d S )NFT)r� r; r r r �
test_with_pipm s
zEnsurePipTest.test_with_pipN)rR rS rT rU r� r� r� r� r� r� r� �ctypesr r� r r r r r� � s
Z
r� �__main__)N)'rU r� r' Zos.pathr� r� rd r r. r* r� Z test.supportr r r r r r r r r� rb Z
unittest.mockr
r� � ImportErrorr� r� r� r0 r8 r� r ZTestCaser rV r� rR �mainr r r r �<module> sD (
�
,