File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_ensurepip.cpython-37.pyc
B
��gP' � @ s� d dl Z d dlZ d dlZd dlZd dlZd dlZd dlZd dlZd dl ZG dd� de j
�Z G dd� d�Z G dd� de e j
�Z
ejejfdd ��ZG d
d � d e e j
�Zd ej ZG d
d� de e j
�ZG dd� de e j
�Zedkr�e �� dS )� Nc @ s e Zd Zdd� ZdS )�TestEnsurePipVersionc C s | � tjt�� � d S )N)� assertEqual� ensurepip� _PIP_VERSION�version)�self� r �//usr/local/lib/python3.7/test/test_ensurepip.py�test_returns_version s z)TestEnsurePipVersion.test_returns_versionN)�__name__�
__module__� __qualname__r
r r r r r
s r c @ s e Zd Zdd� ZdS )�EnsurepipMixinc C st t j�d�}|�� | _d| j_| �|j� tj }t j�d�}|�� }| �|j� ||_ tj
|_
tj � � | _
|_ d S )Nzensurepip._run_pipr z ensurepip.os)�unittest�mockZpatch�start�run_pip� return_valueZ
addCleanup�stop�os�devnull�path�environ�copy�
os_environ)r Z
run_pip_patchZ real_devnullZos_patchZ
patched_osr r r �setUp s
zEnsurepipMixin.setUpN)r r r
r r r r r r 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�installz--no-cache-dirz
--no-indexz --find-links�
setuptools�pipr � � )
r � bootstrapr �assert_called_once_withr r �ANY� call_argsr �len)r �additional_pathsr r r �test_basic_bootstrapping& s
z&TestBootstrap.test_basic_bootstrappingc C s8 t jdd� | j�ddddtjjdddd g tjj� d S )
Nz /foo/bar/)�rootr z--no-cache-dirz
--no-indexz --find-linksz--rootr r )r r" r r# r r r$ )r r r r � test_bootstrapping_with_root4 s
z*TestBootstrap.test_bootstrapping_with_rootc
C s6 t jdd� | j�ddddtjjddd gtjj� d S )
NT)�userr z--no-cache-dirz
--no-indexz --find-linksz--userr r )r r" r r# r r r$ )r r r r � test_bootstrapping_with_user@ s
z*TestBootstrap.test_bootstrapping_with_userc
C s6 t jdd� | j�ddddtjjddd gtjj� d S )
NT)Zupgrader z--no-cache-dirz
--no-indexz --find-linksz --upgrader r )r r" r r# r r r$ )r r r r � test_bootstrapping_with_upgradeK s
z-TestBootstrap.test_bootstrapping_with_upgradec
C s6 t jdd� | j�ddddtjjddd gtjj� d S )
Nr )� verbosityr z--no-cache-dirz
--no-indexz --find-linksz-vr r )r r" r r# r r r$ )r r r r �#test_bootstrapping_with_verbosity_1V s
z1TestBootstrap.test_bootstrapping_with_verbosity_1c
C s6 t jdd� | j�ddddtjjddd gtjj� d S )
Nr! )r. r z--no-cache-dirz
--no-indexz --find-linksz-vvr r )r r" r r# r r r$ )r r r r �#test_bootstrapping_with_verbosity_2a s
z1TestBootstrap.test_bootstrapping_with_verbosity_2c
C s6 t jdd� | j�ddddtjjddd gtjj� d S )
N� )r. r z--no-cache-dirz
--no-indexz --find-linksz-vvvr r )r r" r r# r r r$ )r r r r �#test_bootstrapping_with_verbosity_3l s
z1TestBootstrap.test_bootstrapping_with_verbosity_3c C s t �� | �| jd d� d S )N�ENSUREPIP_OPTIONSr )r r" r r )r r r r �'test_bootstrapping_with_regular_installw s z5TestBootstrap.test_bootstrapping_with_regular_installc C s" t jdd� | �| jd d� d S )NT)�
altinstallr3 r5 )r r" r r )r 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_pipr3 )r r" � assertNotInr )r r r r �#test_bootstrapping_with_default_pip s z1TestBootstrap.test_bootstrapping_with_default_pipc C s6 | � t�� tjddd� W d Q R X | �| jj� d S )NT)r5 r7 )� assertRaises�
ValueErrorr r" � assertFalser �called)r 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 fodder�PIP_THIS_SHOULD_GO_AWAY)r r r" r8 )r r r r �&test_pip_environment_variables_removed� s
z4TestBootstrap.test_pip_environment_variables_removedc C s t �� | �| jd tj� d S )N�PIP_CONFIG_FILE)r r" r r r r )r r r r � test_pip_config_file_disabled� s z+TestBootstrap.test_pip_config_file_disabledN)r r r
r( r* r, r- r/ r0 r2 r4 r6 r9 r> r@ rB r r r r r $ s r c # sr � d krd }nG � fdd�d�}|� }t � }tj�d|�}|tjd< z
|V W d ||krbtjd= n
|tjd< X d S )Nc s e Zd Z� ZdS )zfake_pip.<locals>.FakePipN)r r r
� __version__r )r r r �FakePip� s rD r )�object�sys�modules�get)r r rD �sentinelZorig_pipr )r r �fake_pip� s
rJ c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�
TestUninstallc C s. t d �� t�� W d Q R X | �| jj� d S )N)rJ r �_uninstall_helperr< r r= )r 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 Q R X W d Q R X |�� �� }| �d|� | � | j
j � d S )Nznot a valid versionz!only uninstall a matching version) rJ �test�supportZcaptured_stderrr rL �getvalue�stripZassertInr< r r= )r �stderrZwarningr r r �5test_uninstall_skipped_with_warning_for_wrong_version� s
zCTestUninstall.test_uninstall_skipped_with_warning_for_wrong_versionc C s4 t � � t�� W d Q R X | j�dddddg� d S )N� uninstallz-yz--disable-pip-version-checkr r )rJ r rL r r# )r r r r �test_uninstall� s
z TestUninstall.test_uninstallc C s: t � � tjdd� W d Q R X | j�ddddddg� d S ) Nr )r. rT z-yz--disable-pip-version-checkz-vr r )rJ r rL r r# )r r r r � test_uninstall_with_verbosity_1� s
z-TestUninstall.test_uninstall_with_verbosity_1c C s: t � � tjdd� W d Q R X | j�ddddddg� d S ) Nr! )r. rT z-yz--disable-pip-version-checkz-vvr r )rJ r rL r r# )r r r r � test_uninstall_with_verbosity_2� s
z-TestUninstall.test_uninstall_with_verbosity_2c C s: t � � tjdd� W d Q R X | j�ddddddg� d S ) Nr1 )r. rT z-yz--disable-pip-version-checkz-vvvr r )rJ r rL r r# )r r r r � test_uninstall_with_verbosity_3� s
z-TestUninstall.test_uninstall_with_verbosity_3c C s6 d| j d< t� � t�� W d Q R X | �d| j � d S )Nz test fodderr? )r rJ r rL r8 )r r r r r@ � s
z4TestUninstall.test_pip_environment_variables_removedc C s2 t � � t�� W d Q R X | �| jd tj� d S )NrA )rJ r rL r r r r )r r r r rB � s z+TestUninstall.test_pip_config_file_disabledN) r r r
rM rS rU rV rW rX r@ rB r r r r rK � s rK zpip c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestBootstrappingMainFunctionc
C sb t j�� �(}| �t�� t�dg� W d Q R X W d Q R X |�� �� }| � |t
� | � | j j
� d S )Nz --version)rN rO �captured_stdoutr: �
SystemExitr �_mainrP rQ 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 )
Nr z--no-cache-dirz
--no-indexz --find-linksr r r r r! )
r r\ r r# r r r$ r% r r& )r � exit_coder' r r r r( s
z6TestBootstrappingMainFunction.test_basic_bootstrappingc C s" d| j _t�g �}| �|d� d S )Nr! )r r r r\ r )r ra r r r � test_bootstrapping_error_code s
z;TestBootstrappingMainFunction.test_bootstrapping_error_codeN)r r r
r` r( rb r r r r rY � s rY c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestUninstallationMainFunctionc
C sd t j�� �*}| �t�� tj�dg� W d Q R X W d Q R X |�� � � }| �
|t � | � | j
j� d S )Nz --version)rN rO rZ r: r[ r �
_uninstallr\ rP rQ r r] r< r r= )r r^ r_ r r r �test_uninstall_version s " z5TestUninstallationMainFunction.test_uninstall_versionc C sD t � � tj�g �}W d Q R X | j�dddddg� | �|d� d S )NrT z-yz--disable-pip-version-checkr r r )rJ r rd r\ r r# r )r ra r r r �test_basic_uninstall! s z3TestUninstallationMainFunction.test_basic_uninstallc C s6 t � � d| j_tj�g �}W d Q R X | �|d� d S )Nr! )rJ r r r rd r\ r )r ra r r r �test_uninstall_error_code. s z8TestUninstallationMainFunction.test_uninstall_error_codeN)r r r
re rf rg r r r r rc s
rc �__main__)r Z
unittest.mockZ test.supportrN r Zos.path�
contextlibrF r Zensurepip._uninstallZTestCaser r r �contextmanagerr rJ rK r] rY rc r �mainr r r r �<module> s& qN