File: //usr/local/lib/python3.10/test/__pycache__/test_venv.cpython-310.opt-2.pyc
o
�i�Z � @ s6 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z m Z m
Z
mZ d dlmZmZmZ d dlZd dlZd dlmZ zd dlZW n ey^ dZY nw e�ejejkplej ej kd�Z ddd�Z G dd � d ej �Z!G d
d � d e!�Z"e G d d
� d
e!��Z#e$dkr�e�%� dS dS )� 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 �+/usr/local/lib/python3.10/test/test_venv.py� check_output% s �
�r c @ s@ e Zd Z dZdd� Zdd� Zdd� Zdd � Zd
d �d d
�ZdS )�BaseTesti� 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 �setUp4 s"
�
�
zBaseTest.setUpc C s t | j� d S �N)r r, �r8 r r r �tearDownG 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_captureJ s �� �zBaseTest.run_with_capturec G s t jj| jg|�R � S r; )r' r( �joinr, )r8 r@ r r r � get_env_fileP 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_contentsS s
��z BaseTest.get_text_file_contentsN) �__name__�
__module__� __qualname__ZmaxDiffr: r= rD rF rM r r r r r 0 s r c @ s@ e Zd Z dd� Zdd� Zdd� Zdd� Zed d
� �Ze j
d kr#d Z nd
dddde 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/d0� �Z e�e j
d1kd2�d3d4� �Z dS )5� BasicTestc G s | j |� }| �tj�|�� d S r; )rF �
assertTruer' r( �isdir)r8 r@ �fnr r r rS [ s
zBasicTest.isdirc C s$ t | j� | �tj| j� | �| j� | �| j� | j| j� | � d�}t
� d�dko5t j
dko5tjdk}|rB| �t j�|�� n | �t j�|�� | �d�}tj}t j�|�}| �d| |� | � | j| j�}t j�|�s�| � | j�}td| � td t �|� � | �t j�|�d
| � d S ) NZlib64�P� �posix�darwin�
pyvenv.cfgz home = %s�Contents of %r:� %r�File %r should exist.) r r, rD �venv�createrS r. r$ r! rF �struct�calcsizer' r- r/ r4 rR r( �islink� assertFalser6 rM r1 �dirname�assertInr3 �print�listdir)r8 r Z
conditions�datar9 r( rT �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� rY 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 � assertEqualrk Z assertNotInrd �basename�getcwd)r8 Zenv_name�builder�contextrg �cwdr r r � test_prompt{ 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 rj c
s2 t j�| d �| d< ��| � ddddddg� d S )Nr �-m�pip�installz --upgradeZ
setuptools)r' r( �normcasero )r rA �Z
expect_exer8 r r �pip_cmd_checker� s ��z<BasicTest.test_upgrade_dependencies.<locals>.pip_cmd_checkerz venv.subprocess.check_output)r] rm r/ r4 r' r( r2 r9 r* �TemporaryDirectoryry rE r) rn r Zupgrade_dependencies)r8 rr Zbin_pathZ
python_exeZ fake_env_dirr{ Z fake_contextr rz r �test_upgrade_dependencies� s"
�
��"�z#BasicTest.test_upgrade_dependenciesc C s� t | j� | �tj| j� tj�| j| j| j �}|dd g}d| jfd| jfdt
j fdt
j ffD ]\}}d| |d< t
|�\}}| �|�� |�� � q0d S )N�-c�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 ro �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 sr |D ]3}t jj| jg|�R � }t �|� t j�||�}t|d��
}|�d� W d � n1 s1w Y qd S )N�wb� Still here?)r' r( rE r, �mkdirrI �write)r8 �paths�filename�subdirs�drT 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 sAw Y qtj
dd�}|�| j� | jD ]}tjj| jg|d �R � }| �tj� |�� qVd S )N�foo)r� �rbr� T��clear)r� � ENV_SUBDIRSr] r^ r, r' r( rE rR r6 rI ro rJ rm rb )r8 r� rT rK rr 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' rf r( rE ra �isfile�removerS r )r8 r( rT 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� rT 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 �
|�sH| � | j�}td| � tdt �|� � | �t j �
|�d| � qd S )N�FT)�upgraderZ r[ r\ )r] rm rD r^ r, rS r. r$ r! rF r3 r' r( r6 re rf rR )r8 r� rr rT rh r r r � test_upgrade s �zBasicTest.test_upgradec C sH dD ] \}}t jd|d�}|�| j� | �d�}| �d| |� qd S )N))T�true)FZfalseT)r� �system_site_packagesrY z"include-system-site-packages = %s
)r] rm r^ r, rM rd )r8 Zssp�srr rg r r r �test_isolation s
�zBasicTest.test_isolationzNeeds symlinksc C sj dD ]/}t jd|d�}|�| j� | �| j| j�}|r2| jr)| �t j
� |�� q| � t j
� |�� qd S )Nr� T�r� �symlinks)
r] rm r^ r, rF r. r3 r7 rb r' r( ra rR )r8 Zuslrr rT r r r �test_symlinking& s ��zBasicTest.test_symlinkingc C sd t | j� | �tj| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� |�� � d S )Nr~ �!import sys; print(sys.executable))r r, rD r] r^ r' r( rE r) r. r3 r ro r� r� �r8 r� r r r r r �test_executable> s
�
�zBasicTest.test_executablec C sn t | j� tjddd�}|�| j� tj�tj�| j�| j | j
�}t |ddg�\}}| � |�
� |�� � d S )NTr� r~ r� �r r, r] rm r^ r' r( rE r) r. r3 r ro r� r� �r8 rr r� r r r r r �test_executable_symlinksK 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 ) Nu ϼўТλФЙTr� z activate.bat�&r~ zprint(0)ZoemrG �0)r r, r' r( rE r) r] rm r^ r. r3 r ro r� )r8 r, rr �activater� r r r r r �test_unicode_in_batch_fileY s
�z$BasicTest.test_unicode_in_batch_filec C sj t � t| j� | �tj| j� tj�tj� | j�| j
| j �}t |ddg�\}}| �
|�� d�� � d S )Nr~ zrfrom multiprocessing import Pool; pool = Pool(1); print(pool.apply_async("Python".lower).get(3)); pool.terminate()Zpython)r r r, rD r] r^ r' r( rE r) r. r3 r ro r� r� r� r r r �test_multiprocessingj 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] rm r^ r' r( rE r. rI r� r ro r� )r8 r� rr r� Z test_scriptrK r r r r r �%test_deactivate_with_strict_bash_opts s
� �z/BasicTest.test_deactivate_with_strict_bash_optsrX 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 rS ri ru 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 rQ X sH
�
rQ c @ s^ e Zd Z dd� Zdd� Zdd� Zdd� Zd d
� Zej d d � �Z
e � e
d
�e� dd� ��ZdS )�
EnsurePipTestc 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 �decodero 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' �devnullro rJ rR r( r6 )r8 rK r r r � test_devnull� s �zEnsurePipTest.test_devnullc C s� t | j� t� ��}d|d< d|d< t�� �f}||d< d}d}d}|fD ]1\}}tj�||�} t�| � tj�| |�}
t |
d ��
} | �
|� W d � n1 sOw Y q#| � � � | j t
j| j|d
d � W d � n1 sow Y W d � n1 s~w Y W d � n1 s�w Y tj�tj�| j�| j| j�} t| d d
d dddddg �\}
}|�d�}| �|d� |
�d�}
d�t�� �}| �|
d t|�� |� t�| j��d�}| �||
� t� �/}| � � � d|d< t| d d
d ddddg�\}
}W d � n 1 �sw Y W d � n 1 �sw Y |�d�}tj dd|tj d�}| �|� � d� |
�d�}
| �d|
� | �d|
� |�sN| � � d S d S )N�ignoreZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1)rw zpip.ini)z.pipzpip.confr� T)r� r� z-Wzignore::DeprecationWarningzignore::ImportWarningz-Irv rw 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� � nicer_errorrD r] r^ r) r. r3 r r� ro �format� ensurepip�version�len�fsencoderd �re�sub� MULTILINE�rstripr� )r8 r� �envvarsZhome_dirZ
bad_configZ win_locationZposix_locationrc Zfname�dirpathZfpathrK r� r r Zexpected_versionr, r r r �do_test_with_pip� s|
��
� �� �� � #
�
� �� �
�
�z EnsurePipTest.do_test_with_pipc
c sx � zd V W d S t jy; } z&|jpdjdd�}|jp djdd�}| �|� d|� d|� �� W Y d }~d S d }~ww )Nr� �replace)�errorsz
**Subprocess Output**
z
**Subprocess Error**
)r
r rB r� r Zfail)r8 �excr r r r r r� s � �����zEnsurePipTest.nicer_errorzpip 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 r� r� r� r� r� �
contextlib�contextmanagerr� r� r� �ctypesr r� r r r r r� � s
Y
r� �__main__r; )&r� 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 rQ r� rN �mainr r r r �<module> sJ �
�
( H �