shell bypass 403
o
a bY � @ s0 d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z m
Z
mZ ddlmZmZmZ ddlZddlZddlmZ zddlZW n ey[ dZY nw e�ejejkpiej ej kd�Z ddd�Z G d d
� d
ej �Z!G d d � d e!�Z"e G d
d� de!��Z#e$dkr�e�%� dS 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�*skip_if_broken_multiprocessing_synchronize)� can_symlink�EnvironmentVarGuard�rmtree)�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�PIPE� communicate�
returncode�CalledProcessError)�cmdr �p�out�err� r �7/usr/local/python-3.10/lib/python3.10/test/test_venv.py� check_output$ s �
�r c @ sB e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d �d
d�Z dS )�BaseTestzBase class for venv tests.i� c C s� t j�t�� �| _t jdkrd| _d| _d| _ nd| _ddt
j d d� f| _d | _ t
j }t j�
|�d
| _t
jd krLt j�|�rLt j�|�sLd | _d S d
| _d S )N�nt�Scripts��Lib�Include�bin�lib� python%d.%d� �include����win32TF)�os�path�realpath�tempfile�mkdtemp�env_dir�name�bindirr! r$ �sys� version_info�_base_executable�split�exe�platform�lexists�exists�cannot_link_exe)�self�
executabler r r �setUp3 s"
�
�
zBaseTest.setUpc C s t | j� d S �N)r r, �r8 r r r �tearDownF s zBaseTest.tearDownc O sj t � �"}t� �}||i |�� W d � n1 sw Y W d � n1 s(w Y |�� |�� fS r; )r r �getvalue)r8 �func�args�kwargs�output�errorr r r �run_with_captureI s �� �zBaseTest.run_with_capturec G s t jj| jg|�R � S r; )r' r( �joinr, )r8 r@ r r r � get_env_fileO s zBaseTest.get_env_filezutf-8�r c G sB t | j|� d|d��
}|�� }W d � |S 1 sw Y |S )N�rrG )�openrF �read)r8 r r@ �f�resultr r r �get_text_file_contentsR s
��z BaseTest.get_text_file_contentsN)
�__name__�
__module__� __qualname__�__doc__ZmaxDiffr: r= rD rF rM r r r r r / s r c @ sB e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d � �Z e
j d kr$d
Z nddddde
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�ejd)kd*�d+d,� �Z ed-d.� �Z e� ejd)kd/�d0d1� �Z e�e
j d2kd3�d4d5� �Z dS )6� BasicTestz Test venv module functionality.c G s | j |� }| �tj�|�� d S r; )rF �
assertTruer' r( �isdir)r8 r@ �fnr r r rT Z s
zBasicTest.isdirc C s" t | j� | �tj| j� | �| j� | �| j� | j| j� | � d�}t
� d�dko4t j
dko4tjdk}|rA| �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, rD �venv�createrT r. r$ r! rF �struct�calcsizer' r- r/ r4 rS r( �islink� assertFalser6 rM r1 �dirname�assertInr3 �print�listdir)r8 r Z
conditions�datar9 r( rU �bdr r r �
test_defaults^ 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� rZ z(%s) z prompt = z My prompt)�promptz (My prompt) zprompt = 'My prompt'
�.zprompt = '%s'
)r' r( r2 r, r r^ �
EnvBuilderrD r_ �ensure_directoriesrM � assertEqualrl Z assertNotInre �basename�getcwd)r8 Zenv_name�builder�contextrh �cwdr r r � test_promptz s.
zBasicTest.test_promptc s� t �� }tjdkr dnd}tj�tj�d }t� � �R}tj�
tj� |||��� tjdkr6tj�
tj� � ��� � �fdd�}|�
|�}td|��
|�|� W d � n1 sWw Y W d � d S W d � d S 1 sow Y d S )Nr& r r rk c
s2 t j�| d �| d< ��| � ddddddg� d S )Nr �-m�pip�installz --upgradeZ
setuptools)r' r( �normcaserp )r �Z
expect_exer8 r r �pip_cmd_checker� s ��z<BasicTest.test_upgrade_dependencies.<locals>.pip_cmd_checkerzvenv.subprocess.check_call)r^ rn r/ r4 r' r( r2 r9 r* �TemporaryDirectoryrz rE r) ro r Zupgrade_dependencies)r8 rs 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 ]\}}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, rD r^ r_ r' r( rE r. r3 r/ r� r� r rp �strip�encode)r8 �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 ]3}t jj| jg|�R � }t �|� t j�||�}t|d��
}|�d� W d � n1 s0w Y qdS )zn
Create some files in the environment which are unrelated
to the virtual environment.
�wb� Still here?N)r' r( rE r, �mkdirrI �write)r8 �paths�filename�subdirs�drU rK r r r �create_contents� s
���zBasicTest.create_contentsc C s� | � | jd� t�| j� | jD ]5}tjj| jg|d �R � }| �tj� |�� t
|d��}| � |� � d� W d � n1 s@w Y qtj
dd�}|�| j� | jD ]}tjj| jg|d �R � }| �tj� |�� qUdS )zE
Test creating environment in an existing directory.
�foo)r� �rbr� NT��clear)r� � ENV_SUBDIRSr^ r_ r, r' r( rE rS r6 rI rp rJ rn rc )r8 r� rU rK rs r r r �test_overwrite_existing� s
��
�z!BasicTest.test_overwrite_existingc C sZ t �|�D ]%}t j�||�}t j�|�st j�|�r t �|� qt j�|�r*t|� qd S r; ) r' rg r( rE rb �isfile�removerT r )r8 r( rU r r r �clear_directory� s ��zBasicTest.clear_directoryc C s� | j d d� D ]8}tjj| jg|�R � }t|d��
}|�d� W d � n1 s)w Y | �tt ft
j | j� | � | j� qd S )N� r� � )
r� r' r( rE r, rI r� Z assertRaises�
ValueError�OSErrorr^ r_ r� )r8 r� rU rK r r r �test_unoverwritable_fails� s ��z#BasicTest.test_unoverwritable_failsc C s� dD ]Q}t j|d�}| �|j| j� | �| j� | �| j� | j| j� | � | j| j
�}t j �
|�sG| � | j�}td| � tdt �|� � | �t j �
|�d| � qdS )zC
Test upgrading an existing environment directory.
�FT)�upgrader[ r\ r] N)r^ rn rD r_ r, rT r. r$ r! rF r3 r' r( r6 rf rg rS )r8 r� rs rU ri 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_packagesrZ z"include-system-site-packages = %s
N)r^ rn r_ r, rM re )r8 Zssp�srs rh r r r �test_isolation s
�zBasicTest.test_isolationzNeeds symlinksc C sh dD ]/}t jd|d�}|�| j� | �| j| j�}|r1| jr(| �t j
� |�� q| � t j
� |�� qdS )z3
Test symlinking works as expected
r� T�r� �symlinksN)
r^ rn r_ r, rF r. r3 r7 rc r' r( rb rS )r8 Zuslrs rU 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, rD r^ r_ r' r( rE r) r. r3 r rp r� r� �r8 r� 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� r r� N�r r, r^ rn r_ r' r( rE r) r. r3 r rp r� r� �r8 rs r� r r r r r �test_executable_symlinksJ 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�&r zprint(0)ZoemrG �0N)r r, r' r( rE r) r^ rn r_ r. r3 r rp r� )r8 r, rs �activater� r r r r r �test_unicode_in_batch_fileX 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, rD r^ r_ r' r( rE r) r. r3 r rp r� r� r� r r r �test_multiprocessingi 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 sJw Y t||g�\}}| �|d
�� � | �|d
�� � d S ) N�bashzbash required for this testTr� r� ztest_strict.sh�wzset -euo pipefail
source z
deactivate
� )�shutil�whichZskipTestr r, r^ rn r_ r' r( rE r. rI r� r rp r� )r8 r� rs r� Z test_scriptrK r r r r r �%test_deactivate_with_strict_bash_opts~ s
� �z/BasicTest.test_deactivate_with_strict_bash_optsrY 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)!rN rO rP rQ rT rj rv r~ �requireVenvCreater� r/ r4 r� r0 r� r� r� r� r� r� �unittest�
skipUnlessr r� r� r� r' r- r� r� ZskipIfr� r� r r r r rR W sH
�
rR 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( rE r) r, r. r3 r �decoderp 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, rD 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 sT t tjd��}| �|�� d� W d � n1 sw Y | �tj�tj�� d S )Nr� r� )rI r' �devnullrp rJ rS r( r6 )r8 rK r r r � test_devnull� s �zEnsurePipTest.test_devnullc C s� t | j� t� ��}d|d< d|d< t�� �z}||d< d}d}d}|fD ]1\}}tj�||�} t�| � tj�| |�}
t |
d ��
} | �
|� W d � n1 sOw Y q#z
| j t j
| j|d
d � W n% tjy� } z| jjd d
�}
d}| �|�| |
�� W Y d } ~ nd } ~ ww W d � n1 s�w Y W d � n1 s�w Y tj�tj�| j�| j| j�}t|ddddddddg �\}}|�d�}| �|d� |�d�}d�t�� �}| �|d t|�� |� t� | j��d�}| � ||� t� �}d|d< t|dddddddg�\}}W d � n 1 �sw Y |�d�}t j dd|t j d�}| �|�!� d� |�d�}| � d |� | � d |� |�sM| �"� d S d S ) N�ignoreZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1)rx zpip.ini)z.pipzpip.confr� T)r� r� �replace)�errorsz {}
**Subprocess Output**
{}z-Wzignore::DeprecationWarningzignore::ImportWarningz-Irw rx 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( rE r� rI r� rD r^ r_ r
r rB r� Zfail�formatr) r. r3 r rp � ensurepip�version�len�fsencodere �re�sub� MULTILINE�rstripr� )r8 r� �envvarsZhome_dirZ
bad_configZ win_locationZposix_locationrd Zfname�dirpathZfpathrK �excZdetails�msgr� r r Zexpected_versionr, r r r �do_test_with_pip� s�
��
� ��� �� � )
�
� �
�
�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)rN rO rP rQ r� r� r� r� r� r� r� �ctypesr r� r r r r r� � s
_r� �__main__r; )&rQ r� r' Zos.pathr� r� r` r
r/ r* Z test.supportr r r r Ztest.support.os_helperr r r r� r^ Z
unittest.mockr r� � ImportErrorr� r� r� r1 r9 r� r ZTestCaser rR r� rN �mainr r r r �<module> sH �
�
( H �