Your IP : 216.73.216.213
��Yf�- �
@ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z e j
j d d d g �Z y d d l
Z
Wn e k
r� d d � Z Yn
Xd d � Z Gd d � d e j � Z Gd
d � d � Z Gd d
� d
e e j � Z e j e j d d � � Z Gd d � d e e j � Z Gd d � d e e j � Z d e j Z Gd d � d e e j � Z Gd d � d e e j � Z e d k r�e j � d S)� N� ensurepipZblocked�sslc C s t j t j � } | | � S)N)�unittest�skipr Z_MISSING_SSL_MESSAGE)�fZdeco� r �8/opt/alt/python35/lib64/python3.5/test/test_ensurepip.py�requires_usable_pip s r c C s | S)Nr )r r r r r s c @ s e Z d Z d d � Z d S)�TestEnsurePipVersionc C s | j t j t j � � d S)N)� assertEqualr � _PIP_VERSION�version)�selfr r r �test_returns_version s z)TestEnsurePipVersion.test_returns_versionN)�__name__�
__module__� __qualname__r r r r r r
s r
c @ s e Z d Z d d � Z d S)�EnsurepipMixinc C s� t j j d � } | j � | _ | j | j � t j } t j j d � } | j � } | j | j � | | _ t j | _ t j
j � | _ | _
d S)Nzensurepip._run_pipz ensurepip.os)
r �mockZpatch�start�run_pip�
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 @ s� e Z d Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d
� � Z e d d � � Z e d
d � � Z
e d d � � Z e d d � � Z e d d � � Z
d d � Z e d d � � Z e d d � � Z d S) �
TestBootstrapc C si t j � | j j d d d t j j d d g t j j � | j j d d } | j t | � d � d S) N�installz
--no-indexz --find-links�
setuptools�pipr � � )
r � bootstrapr �assert_called_once_withr r �ANY� call_argsr �len)r �additional_pathsr r r �test_basic_bootstrapping2 s
z&TestBootstrap.test_basic_bootstrappingc C sK t j d d � | j j d d d t j j d d d d g t j j � d S) N�rootz /foo/bar/r! z
--no-indexz --find-linksz--rootr" r# )r r& r r' r r r( )r r r r � test_bootstrapping_with_rootA s z*TestBootstrap.test_bootstrapping_with_rootc C sH t j d d � | j j d d d t j j d d d g t j j � d S) N�userTr! z
--no-indexz --find-linksz--userr" r# )r r& r r' r r r( )r r r r � test_bootstrapping_with_userN s
z*TestBootstrap.test_bootstrapping_with_userc C sH t j d d � | j j d d d t j j d d d g t j j � d S) NZupgradeTr! z
--no-indexz --find-linksz --upgrader" r# )r r& r r' r r r( )r r r r � test_bootstrapping_with_upgradeZ s
z-TestBootstrap.test_bootstrapping_with_upgradec C sH t j d d � | j j d d d t j j d d d g t j j � d S) N� verbosityr$ r! z
--no-indexz --find-linksz-vr" r# )r r& r r' r r r( )r r r r �#test_bootstrapping_with_verbosity_1f s
z1TestBootstrap.test_bootstrapping_with_verbosity_1c C sH t j d d � | j j d d d t j j d d d g t j j � d S) Nr2 r% r! z
--no-indexz --find-linksz-vvr" r# )r r& r r' r r r( )r r r r �#test_bootstrapping_with_verbosity_2r s
z1TestBootstrap.test_bootstrapping_with_verbosity_2c C sH t j d d � | j j d d d t j j d d d g t j j � d S) Nr2 � r! z
--no-indexz --find-linksz-vvvr" r# )r r& r r' r r r( )r r r r �#test_bootstrapping_with_verbosity_3~ s
z1TestBootstrap.test_bootstrapping_with_verbosity_3c C s% t j � | j | j d d � d S)N�ENSUREPIP_OPTIONSr! )r r& r r )r r r r �'test_bootstrapping_with_regular_install� s
z5TestBootstrap.test_bootstrapping_with_regular_installc C s+ t j d d � | j | j d d � d S)N�
altinstallTr7 )r r& r r )r r r r �#test_bootstrapping_with_alt_install� s z1TestBootstrap.test_bootstrapping_with_alt_installc C s' t j d d � | j d | j � d S)N� default_pipTr7 )r r& � assertNotInr )r r r r �#test_bootstrapping_with_default_pip� s z1TestBootstrap.test_bootstrapping_with_default_pipc C sD | j t � � t j d d d d � Wd QRX| j | j j � d S)Nr9 Tr; )� 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 j � | j d | j � d S)Nz test fodder�PIP_THIS_SHOULD_GO_AWAY)r r r& r< )r r r r �&test_pip_environment_variables_removed� s
z4TestBootstrap.test_pip_environment_variables_removedc C s( t j � | j | j d t j � 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. r0 r1 r3 r4 r6 r8 r: r= rB rD rF r r r r r 0 s
r c # s� � d k r d } n" G� f d d � d � } | � } t � } t j j d | � } | t j d <z | VWd | | k r� t j d =n
| t j d <Xd S)Nc s e Z d Z � Z d S)zfake_pip.<locals>.FakePipN)r r r � __version__r )r
r r �FakePip� s rH r# )�object�sys�modules�get)r
r# rH �sentinelZorig_pipr )r
r �fake_pip� s
rN c @ s� e Z d Z d d � Z d d � Z e d d � � Z e d d � � Z e d d
� � Z e d d � � Z e d
d � � Z
e d d � � Z d S)�
TestUninstallc C s5 t d � � t j � Wd QRX| j | j j � d S)N)rN r �_uninstall_helperr@ r rA )r r r r �)test_uninstall_skipped_when_not_installed� s
z7TestUninstall.test_uninstall_skipped_when_not_installedc C sp t d � �( t j j � � } t j � Wd QRXWd QRX| j � j � } | j d | � | j | j
j � d S)Nznot a valid versionz!only uninstall a matching version) rN �test�support�captured_stderrr rP �getvalue�strip�assertInr@ r rA )r �stderr�warningr r r �5test_uninstall_skipped_with_warning_for_wrong_version� s
zCTestUninstall.test_uninstall_skipped_with_warning_for_wrong_versionc
C s> t � � t j � Wd QRX| j j d d d d d g � d S)N� uninstallz-yz--disable-pip-version-checkr# r" )rN r rP r r' )r r r r �test_uninstall� s
z TestUninstall.test_uninstallc C sG t � � t j d d � Wd QRX| j j d d d d d d g � d S) Nr2 r$ r[ z-yz--disable-pip-version-checkz-vr# r" )rN r rP r r' )r r r r � test_uninstall_with_verbosity_1� s
z-TestUninstall.test_uninstall_with_verbosity_1c C sG t � � t j d d � Wd QRX| j j d d d d d d g � d S) Nr2 r% r[ z-yz--disable-pip-version-checkz-vvr# r" )rN r rP r r' )r r r r � test_uninstall_with_verbosity_2� s
z-TestUninstall.test_uninstall_with_verbosity_2c C sG t � � t j d d � Wd QRX| j j d d d d d d g � d S) Nr2 r5 r[ z-yz--disable-pip-version-checkz-vvvr# r" )rN r rP r r' )r r r r � test_uninstall_with_verbosity_3� s
z-TestUninstall.test_uninstall_with_verbosity_3c
C s? d | j d <t � � t j � Wd QRX| j d | j � d S)Nz test fodderrC )r rN r rP r< )r r r r rD s
z4TestUninstall.test_pip_environment_variables_removedc
C s9 t � � t j � Wd QRX| j | j d t j � d S)NrE )rN r rP r r r r )r r r r rF s
z+TestUninstall.test_pip_config_file_disabledN) r r r rQ rZ r r\ r] r^ r_ rD rF r r r r rO � s rO c sF e Z d Z � f d d � Z d d � Z d d � Z d d � Z � S) �TestMissingSSLc s3 t t j d <| j d d � � } t � j � d S)Nr c S s t t j d <d S)Nr )r rJ rK r r r r �restore_module s z,TestMissingSSL.setUp.<locals>.restore_module)�ensurepip_no_sslrJ rK r �superr )r ra )� __class__r r r s
zTestMissingSSL.setUpc
C s[ d | j d <| j t d � � t j � Wd QRX| j | j j � | j d | j � d S)Nz test fodder�PIP_THIS_SHOULD_STAYzrequires SSL/TLS) r �assertRaisesRegex� RuntimeErrorrb r& r@ r rA rW )r r r r �test_bootstrap_requires_ssl s
z*TestMissingSSL.test_bootstrap_requires_sslc C sl d | j d <| j t d � � t � � t j � Wd QRXWd QRX| j | j j � | j d | j � d S)Nz test fodderre zrequires SSL/TLS)
r rf rg rN rb rP r@ r rA rW )r r r r �test_uninstall_requires_ssl" s
z*TestMissingSSL.test_uninstall_requires_sslc
C sk t j j � � } t j d g � Wd QRX| j � j � } | j | j d � | � | j | j
j � d S)Nz --versionzrequires SSL/TLS) rR rS rT rb �_mainrU rV Z
assertTrue�endswithr@ r rA )r rX rY r r r �"test_main_exits_early_with_warning* s
z1TestMissingSSL.test_main_exits_early_with_warning)r r r r rh ri rl r r )rd r r` s r` zpip c @ s4 e Z d Z e d d � � Z e d d � � Z d S)� TestBootstrappingMainFunctionc C sy t j j � �. } | j t � � t j d g � Wd QRXWd QRX| j � j � } | j | t
� | j | j j
� d S)Nz --version)rR rS �captured_stdoutr> �
SystemExitr rj rU rV r �EXPECTED_VERSION_OUTPUTr@ r rA )r �stdout�resultr r r �test_bootstrap_version7 s z4TestBootstrappingMainFunction.test_bootstrap_versionc C sl t j g � | j j d d d t j j d d g t j j � | j j d d } | j t | � d � d S) Nr! z
--no-indexz --find-linksr" r# r r$ r% )
r rj r r' r r r( r) r r* )r r+ r r r r, @ s
z6TestBootstrappingMainFunction.test_basic_bootstrappingN)r r r r rs r, r r r r rm 5 s rm c @ s. e Z d Z d d � Z e d d � � Z d S)� TestUninstallationMainFunctionc C s| t j j � �1 } | j t � � t j j d g � Wd QRXWd QRX| j � j � } | j
| t � | j | j
j � d S)Nz --version)rR rS rn r> ro r �
_uninstallrj rU rV r rp r@ r rA )r rq rr r r r �test_uninstall_versionQ s !z5TestUninstallationMainFunction.test_uninstall_versionc
C sD t � � t j j g � Wd QRX| j j d d d d d g � d S)Nr[ z-yz--disable-pip-version-checkr# r" )rN r ru rj r r' )r r r r �test_basic_uninstallY s
z3TestUninstallationMainFunction.test_basic_uninstallN)r r r rv r rw r r r r rt O s rt �__main__) r Z
unittest.mockZ test.supportrR r Zos.path�
contextlibrJ r Zensurepip._uninstallrS Zimport_fresh_modulerb r � ImportErrorr ZTestCaser
r r �contextmanagerr rN rO r` rp rm rt r �mainr r r r �<module> s6
} R !