File: //usr/local/lib/python3.10/test/__pycache__/test_ensurepip.cpython-310.opt-2.pyc
o
�i�0 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z G dd� dej �Z G dd� d�Z
G dd� de
ej �Ze je �� fdd ��ZG d
d � d e
ej �Zd e �� ZG d
d� de
ej �ZG dd� de
ej �Zedkr{e�� dS dS )� Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestPackagesc C s t j�||�}t|d��� d S )N�wb)�os�path�join�open�close)�selfZ directory�filename�fullname� r �0/usr/local/lib/python3.10/test/test_ensurepip.py�touch s zTestPackages.touchc
C s� t �� �W}| �|d� | �|d� tjj�tdd ��) tjj�td|�� | �t� � d� W d � n1 s7w Y W d � n1 sFw Y W d � d S W d � d S 1 s^w Y d S )Nz pip-1.2.3b1-py2.py3-none-any.whl�"setuptools-49.1.3-py3-none-any.whl� _PACKAGES�_WHEEL_PKG_DIRz1.2.3b1)
�tempfile�TemporaryDirectoryr �unittest�mock�patch�object� ensurepip� assertEqual�version)r �tmpdirr r r
� test_version s
� �� �"�zTestPackages.test_versionc C s� t jj�tdd ��. t jj�tdd �� t�� }| �tjt�� � W d � n1 s+w Y W d � n1 s:w Y | � |d j
� | � |d j
� d S )Nr r �pip�
setuptools) r r r r r �
_get_packagesr Z _PIP_VERSIONr ZassertIsNotNoneZ
wheel_name)r �packagesr r r
�test_get_packages_no_dir s � �� z%TestPackages.test_get_packages_no_dirc
C s6 d}d}t �� ��}| �||� | �||� | �|d� tjj�tdd ��% tjj�td|�� t�� }W d � n1 s=w Y W d � n1 sLw Y | � |d j
d� | � |d j t j
�||�� | � |d j
d � | � |d j t j
�||�� | � t|�ddg� W d � d S 1 s�w Y d S )
Nr z pip-20.2.2-py2.py3-none-any.whlz!wheel-0.34.2-py2.py3-none-any.whlr r r z49.1.3r z20.2.2)r r r r r r r r r r r Z
wheel_pathr r r �sorted)r Zsetuptools_filenameZ pip_filenamer r r r r
�test_get_packages_with_dir) s.
�
�� � �"�z'TestPackages.test_get_packages_with_dirN)�__name__�
__module__� __qualname__r r r! r# r r r r
r s
r c @ s e Zd Zdd� ZdS )�EnsurepipMixinc C s| t j�d�}|�� | _d| j_| �|j� tj }t j�d�}|�� }tj
|_
| �|j� ||_ tj |_ tj �
� | _|_ d S )Nzensurepip._run_pipr z ensurepip.os)r r r �start�run_pip� return_valueZ
addCleanup�stopr �devnull�listdirr �environ�copy�
os_environ)r Z
run_pip_patchZ real_devnullZos_patchZ
patched_osr r r
�setUpE s
zEnsurepipMixin.setUpN)r$ r% r& r1 r r r r
r' C s r' c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �
TestBootstrapc C sP t �� | j�ddddtjjddgtjj� | jjd d }| �t |�d � d S �
N�install�--no-cache-dir�
--no-index� --find-linksr r r � � )
r � bootstrapr) �assert_called_once_withr r �ANY� call_argsr �len)r �additional_pathsr r r
�test_basic_bootstrappingY s
��z&TestBootstrap.test_basic_bootstrappingc C s8 t jdd� | j�ddddtjjdddd g tjj� d S )
Nz /foo/bar/)�rootr4 r5 r6 r7 z--rootr r �r r: r) r; r r r< �r r r r
� test_bootstrapping_with_rootg s
��z*TestBootstrap.test_bootstrapping_with_rootc
C �6 t jdd� | j�ddddtjjddd gtjj� d S )
NT)�userr4 r5 r6 r7 z--userr r rB rC r r r
� test_bootstrapping_with_users � ��z*TestBootstrap.test_bootstrapping_with_userc
C rE )
NT)Zupgrader4 r5 r6 r7 z --upgrader r rB rC r r r
� test_bootstrapping_with_upgrade~ rH z-TestBootstrap.test_bootstrapping_with_upgradec
C rE )
Nr8 �� verbosityr4 r5 r6 r7 �-vr r rB rC r r r
�#test_bootstrapping_with_verbosity_1� rH z1TestBootstrap.test_bootstrapping_with_verbosity_1c
C rE )
Nr9 rJ r4 r5 r6 r7 �-vvr r rB rC r r r
�#test_bootstrapping_with_verbosity_2� rH z1TestBootstrap.test_bootstrapping_with_verbosity_2c
C rE )
N� rJ r4 r5 r6 r7 �-vvvr r rB rC r r r
�#test_bootstrapping_with_verbosity_3� rH z1TestBootstrap.test_bootstrapping_with_verbosity_3c C s t �� | �| jd d� d S )N�ENSUREPIP_OPTIONSr4 �r r: r r0 rC r r r
�'test_bootstrapping_with_regular_install� s z5TestBootstrap.test_bootstrapping_with_regular_installc C s" t jdd� | �| jd d� d S )NT)�
altinstallrS rV rT rC r r r
�#test_bootstrapping_with_alt_install� s z1TestBootstrap.test_bootstrapping_with_alt_installc C s t jdd� | �d| j� d S )NT)� default_piprS )r r: � assertNotInr0 rC r r r
�#test_bootstrapping_with_default_pip� s z1TestBootstrap.test_bootstrapping_with_default_pipc C sJ | � t�� tjddd� W d � n1 sw Y | �| jj� d S )NT)rV rX )� assertRaises�
ValueErrorr r: � assertFalser) �calledrC r r r
�$test_altinstall_default_pip_conflict� s �z2TestBootstrap.test_altinstall_default_pip_conflictc C s$ d| j d< t�� | �d| j � d S �Nz test fodderZPIP_THIS_SHOULD_GO_AWAY)r0 r r: rY rC r r r
�&test_pip_environment_variables_removed� s
z4TestBootstrap.test_pip_environment_variables_removedc C s t �� | �| jd tj� d S �NZPIP_CONFIG_FILE)r r: r r0 r r, rC r r r
� test_pip_config_file_disabled� s z+TestBootstrap.test_pip_config_file_disabledN)r$ r% r& r@ rD rG rI rM rO rR rU rW rZ r_ ra rc r r r r
r2 W s r2 c # s� � � d u rd }n G � fdd�d�}|� }t � }tj�d|�}|tjd< z|V W ||u r2tjd= d S |tjd< d S ||u rBtjd= w |tjd< w )Nc s e Zd Z� ZdS )zfake_pip.<locals>.FakePipN)r$ r% r& � __version__r �r r r
�FakePip� s rf r )r �sys�modules�get)r r rf �sentinelZorig_pipr re r
�fake_pip� s �
�rk c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�
TestUninstallc C sB t d �� t�� W d � n1 sw Y | �| jj� d S )N)rk r �_uninstall_helperr] r) r^ rC r r r
�)test_uninstall_skipped_when_not_installed� s
�z7TestUninstall.test_uninstall_skipped_when_not_installedc C s� t d��! tj�� � }t�� W d � n1 sw Y W d � n1 s(w Y |�� �� }| �d|� | � | j
j � d S )Nznot a valid versionz!only uninstall a matching version) rk �test�supportZcaptured_stderrr rm �getvalue�stripZassertInr] r) r^ )r �stderr�warningr r r
�5test_uninstall_skipped_with_warning_for_wrong_version� s
�� � zCTestUninstall.test_uninstall_skipped_with_warning_for_wrong_versionc C sB t � � t�� W d � n1 sw Y | j�g d�� d S )N�� uninstall�-y�--disable-pip-version-checkr r �rk r rm r) r; rC r r r
�test_uninstall� s
��z TestUninstall.test_uninstallc C �F t � � tjdd� W d � n1 sw Y | j�g d�� d S )Nr8 rJ )rw rx ry rL r r rz rC r r r
� test_uninstall_with_verbosity_1� � ��z-TestUninstall.test_uninstall_with_verbosity_1c C r| )Nr9 rJ )rw rx ry rN r r rz rC r r r
� test_uninstall_with_verbosity_2 r~ z-TestUninstall.test_uninstall_with_verbosity_2c C r| )NrP rJ )rw rx ry rQ r r rz rC r r r
� test_uninstall_with_verbosity_3 r~ z-TestUninstall.test_uninstall_with_verbosity_3c C sJ d| j d< t� � t�� W d � n1 sw Y | �d| j � d S r` )r0 rk r rm rY rC r r r
ra s
�z4TestUninstall.test_pip_environment_variables_removedc C sF t � � t�� W d � n1 sw Y | �| jd tj� d S rb )rk r rm r r0 r r, rC r r r
rc s
�z+TestUninstall.test_pip_config_file_disabledN) r$ r% r& rn ru r{ r} r r� ra rc r r r r
rl � s rl zpip c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )� TestBootstrappingMainFunctionc C s� t j�� �#}| �t�� t�dg� W d � n1 s w Y W d � n1 s+w Y |�� �� }| � |t
� | � | j j
� d S �Nz --version)ro rp �captured_stdoutr[ �
SystemExitr �_mainrq rr r �EXPECTED_VERSION_OUTPUTr] r) r^ �r �stdout�resultr r r
�test_bootstrap_version- s �� � z4TestBootstrappingMainFunction.test_bootstrap_versionc C s^ t �g �}| j�ddddtjjddgtjj� | jjd d }| �t |�d � | �|d� d S r3 )
r r� r) r; r r r< r= r r> )r � exit_coder? r r r
r@ 5 s
��z6TestBootstrappingMainFunction.test_basic_bootstrappingc C s" d| j _t�g �}| �|d� d S �Nr9 )r) r* r r� r �r r� r r r
� test_bootstrapping_error_codeD s
z;TestBootstrappingMainFunction.test_bootstrapping_error_codeN)r$ r% r& r� r@ r� r r r r
r� + s r� c @ r� )� TestUninstallationMainFunctionc C s� t j�� �$}| �t�� tj�dg� W d � n1 s w Y W d � n1 s,w Y |�� � � }| �
|t � | � | j
j� d S r� )ro rp r� r[ r� r �
_uninstallr� rq rr r r� r] r) r^ r� r r r
�test_uninstall_versionL s �� � z5TestUninstallationMainFunction.test_uninstall_versionc C sR t � � tj�g �}W d � n1 sw Y | j�g d�� | �|d� d S )Nrv r )rk r r� r� r) r; r r� r r r
�test_basic_uninstallT s ��z3TestUninstallationMainFunction.test_basic_uninstallc C sJ t � � d| j_tj�g �}W d � n1 sw Y | �|d� d S r� )rk r) r* r r� r� r r� r r r
�test_uninstall_error_codea s
�z8TestUninstallationMainFunction.test_uninstall_error_codeN)r$ r% r& r� r� r� r r r r
r� J s
r� �__main__)�
contextlibr Zos.pathrg r Z test.supportro r Z
unittest.mockr Zensurepip._uninstallZTestCaser r' r2 �contextmanagerr rk rl r� r� r� r$ �mainr r r r
�<module> s, 5q N �