Your IP : 216.73.216.213
��Yf�E � @ s� d 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 d d l m
Z
m Z m Z m
Z
m Z d d l Z d d l Z d d l Z y d d l Z Wn e k
r� d Z Yn Xy d d l Z Wn e k
rd Z Yn Xy d d l Z Wn e k
r2d Z Yn Xe j e j e j k d � Z e j j � d k rre j Z n d d � Z Gd d � d e j � Z Gd
d � d e � Z e Gd d
� d
e � � Z! e" d k r�e j# � d S)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� can_symlink�EnvironmentVarGuard�rmtreez Test not appropriate in a venvZnulc C s | S)N� )�fr r �3/opt/alt/python35/lib64/python3.5/test/test_venv.py�failsOnWindows. s r
c @ sR e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
�BaseTestzBase class for venv tests.c C s� t j j t j � � | _ t j d k rH d | _ d
| _ d | _ n2 d | _ d d t
j d d � f | _ d | _ t
j d
k r� d t j
k r� t j
d } n t
j } t j j | � d | _ d S)N�nt�Scripts�Lib�Include�bin�libzpython%s� �include�darwin�__PYVENV_LAUNCHER__� )r ���)�os�path�realpath�tempfileZmkdtemp�env_dir�name�bindirr r �sys�version�platform�environ�
executable�split�exe)�selfr# r r r �setUp4 s zBaseTest.setUpc C s t | j � d S)N)r r )r& r r r �tearDownD s zBaseTest.tearDownc O sI t � �' } t � � } | | | � Wd QRXWd QRX| j � | j � f S)N)r r �getvalue)r& �func�args�kwargs�output�errorr r r �run_with_captureG s zBaseTest.run_with_capturec G s t j j | j | � S)N)r r �joinr )r& r+ r r r � get_env_fileM s zBaseTest.get_env_filec G s2 t | j | � d � � } | j � } Wd QRX| S)N�r)�openr1 �read)r& r+ r �resultr r r �get_text_file_contentsP s z BaseTest.get_text_file_contentsN) �__name__�
__module__� __qualname__�__doc__r' r( r/ r1 r6 r r r r r 1 s r c @ s: e Z d Z d Z d d � Z d d � Z e d d � � Z e j d k rT d* Z
nF d+ d, d- d d e j d d � f d d e j d d � d f f Z
d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z e j e � d � d d! � � Z e d" d# � � Z e j e � d � d$ d% � � Z d S).� BasicTestz Test venv module functionality.c G s, | j | � } | j t j j | � � d S)N)r1 �
assertTruer r �isdir)r& r+ �fnr r r r= X s zBasicTest.isdirc C s� t | j � | j t j | j � | j | j � | j | j � | j | j � | j d � } t
j d � d k o� t j
d k o� t j d k } | r� | j t j j | � � n | j t j j | � � | j d � } t j d k r
d t j k r
t j d } n t j } t j j | � } | j d | | � | j | j | j � } t j j | � s�| j | j � } t d | � t d
t j | � � | j t j j | � d | � d S)
zB
Test the create function with default arguments.
�lib64�P� �posixr z
pyvenv.cfgr z home = %szContents of %r:z %rzFile %r should exist.N) r r r/ �venv�creater= r r r r1 �structZcalcsizer r r r! r< r �islink� assertFalse�existsr6 r" r# �dirname�assertInr% �print�listdir)r& �pZ
conditions�datar# r r> �bdr r r �
test_defaults\ s0
$ zBasicTest.test_defaultsc C s' | j t j t j � | j t j t j � t | j � | j t j
| j � t j j
| j | j | j � } | d d g } x� d | j f d | j f d t j f d t j f f D]i \ } } d | | d <t j | d t j d t j �} | j � \ } } | j | j � | j � � q� Wd S)
z>
Test that the prefix values are as expected.
z-cN�prefix� base_prefix�base_exec_prefixzimport sys; print(sys.%s)� �stdout�stderr)� assertEqualr rR rQ rS � exec_prefixr r r/ rC rD r r r0 r r% �
subprocess�Popen�PIPE� communicate�strip�encode)r& �envpy�cmdrQ ZexpectedrM �out�errr r r �
test_prefixes| s
! zBasicTest.test_prefixes�win32r
r r �
site-packagesr r r z python%d.%dNrT c
C su xn | D]f } t j j | j | � } t j | � t j j | | � } t | d � � } | j d � Wd QRXq Wd S)zn
Create some files in the environment which are unrelated
to the virtual environment.
�wbs Still here?N)r r r0 r �mkdirr3 �write)r& �paths�filename�subdirs�dr> r r r r �create_contents� s
zBasicTest.create_contentsc C s
| j | j d � t j | j � xu | j D]j } t j j | j | d � } | j t j j | � � t
| d � � } | j | j � d � Wd QRXq- Wt j
d d � } | j | j � xF | j D]; } t j j | j | d � } | j t j j | � � q� Wd S) zE
Test creating environment in an existing directory.
�foo�rbs Still here?N�clearT)rn )rn )rm � ENV_SUBDIRSrC rD r r r r0 r< rH r3 rW r4 �
EnvBuilderrG )r& rk r> r �builderr r r �test_overwrite_existing� s ! z!BasicTest.test_overwrite_existingc C s� x| t j | � D]k } t j j | | � } t j j | � sO t j j | � r_ t j | � q t j j | � r t | � q Wd S)N) r rL r r0 rF �isfile�remover= r )r& r r> r r r �clear_directory� s $zBasicTest.clear_directoryc C s� x� | j d d � D]s } t j j | j | � } t | d � � } | j d � Wd QRX| j t t f t
j | j � | j | j � q Wd S)Nr rf � )
rq r r r0 r r3 rh Z assertRaises�
ValueError�OSErrorrC rD rw )r& ri r> r r r r �test_unoverwritable_fails� s z#BasicTest.test_unoverwritable_failsc C s� x� d D]� } t j d | � } | j | j | j � | j | j � | j | j � | j | j � | j | j | j
� } t j j
| � s� | j | j � } t d | � t d t j | � � | j t j j
| � d | � q Wd S) zC
Test upgrading an existing environment directory.
FT�upgradezContents of %r:z %rzFile %r should exist.N)FT)rC rr r/ rD r r= r r r r1 r% r r rH rK rL r< )r& r| rs r> rO r r r � test_upgrade� s
zBasicTest.test_upgradec C sf x_ d D]W \ } } t j d d d | � } | j | j � | j d � } | j d | | � q Wd S)
z:
Test isolation from system site-packages
T�trueF�falserp �system_site_packagesz
pyvenv.cfgz"include-system-site-packages = %s
N�Tr~ �Fr )r� r� )rC rr rD r r6 rJ )r& Zssp�srs rN r r r �test_isolation� s
zBasicTest.test_isolationzNeeds symlinksc C st xm d D]e } t j d d d | � } | j | j � | j | j | j � } | r | j t j j
| � � q Wd S)z3
Test symlinking works as expected
FTrp �symlinksN)FT) rC rr rD r r1 r r% r< r r rF )r& Zuslrs r> r r r �test_symlinking� s
zBasicTest.test_symlinkingc C s� t | j � | j t j | j � t j j t j j | j � | j | j
� } | d d g } t j | d t j
d t j
�} | j � \ } } | j | j � | j � � d S)zD
Test that the sys.executable value is as expected.
z-cz!import sys; print(sys.executable)rU rV N)r r r/ rC rD r r r0 r r r% rY rZ r[ r\ rW r] r^ )r& r_ r` rM ra rb r r r �test_executable s
- zBasicTest.test_executablec C s� t | j � t j d d d d � } | j | j � t j j t j j | j � | j | j
� } | d d g } t j | d t j
d t j
�} | j � \ } } | j | j � | j � � d S) zD
Test that the sys.executable value is as expected.
rp Tr� z-cz!import sys; print(sys.executable)rU rV N)r r rC rr rD r r r0 r r r% rY rZ r[ r\ rW r] r^ )r& rs r_ r` rM ra rb r r r �test_executable_symlinks s
- z"BasicTest.test_executable_symlinks�r
�r �r �r re )r� r� r� r� )r )r )r )r7 r8 r9 r: r= rP �
skipInVenvrc r r! rq � version_inform rt rw r{ r} r� �unittest�
skipUnlessr r� r� r� r r r r r; U s. #
!r; c @ s� e Z d Z d Z d d � Z d d � Z d d � Z e d d � � Z d
d � Z e
j e d k e
j � e
j e d
� e
j e d � d d � � � � Z d S)�
EnsurePipTestz%Test venv module installation of pip.c C s� t j j t j j | j � | j | j � } d } | d | g } t j | d t j d t j �} | j
� \ } } | j d � } | j | d � | j d � } | j | j
� d � d S)Nz1try:
import pip
except ImportError:
print("OK")z-crU rV zlatin-1� ZOK)r r r0 r r r r% rY rZ r[ r\ �decoderW r] )r& r_ Z
try_importr` rM ra rb r r r �assert_pip_not_installed' s z&EnsurePipTest.assert_pip_not_installedc C s1 t | j � | j t j | j � | j � d S)N)r r r/ rC rD r� )r& r r r �test_no_pip_by_default7 s
z$EnsurePipTest.test_no_pip_by_defaultc C s7 t | j � | j t j | j d d �| j � d S)N�with_pipF)r r r/ rC rD r� )r& r r r �test_explicit_no_pip< s
z"EnsurePipTest.test_explicit_no_pipc
C sR | j t j j t j � � t t j d � � } | j | j � d � Wd QRXd S)Nro rx )r< r r rH �devnullr3 rW r4 )r& r r r r � test_devnull_exists_and_is_emptyA s z.EnsurePipTest.test_devnull_exists_and_is_emptyc ) C s= t | j � t � �I} d | d <d | d <t j � � } | | d <d } d } d } xt | f D]i \ } } t j j | | � } t j | � t j j | | � }
t |
d � � } | j
| � Wd QRXqb Wy& | j t j
| j d | d
d �Wn\ t j k
rS} z9 | j j d d � }
d } | j | j | |
� � WYd d } ~ Xn XWd QRXWd QRXt j j t j j | j � | j | j � } | d d d g } t j | d t j d t j �} | j � \ } } | j d � } | j | d � | j d � } d j t j � � } | j | d t | � � | � t j | j � j d � } | j | | � | d d g } t � �: } t j | d t j d t j �} | j � \ } } Wd QRX| j d � } t j! d d | d t j" �} | j | j# � d � | j d � } | j d | � | j d | � | s9| j$ � d S) N�eZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1�pip�pip.ini�.pip�pip.conf�wr� r� T�errors�replacez {}
**Subprocess Output**
{}z-Imz --versionrU rV zlatin-1r� zpip {}zensurepip._uninstallzN^The directory .* or its parent directory is not owned by the current user .*$�flagsz Successfully uninstalled pipz#Successfully uninstalled setuptools)r� r� )r� r� )%r r r r ZTemporaryDirectoryr r r0 rg r3 rh r/ rC rD rY ZCalledProcessErrorr- r� Zfail�formatr r r% rZ r[ r\ rW � ensurepipr �len�fsencoderJ �re�sub� MULTILINE�rstripr� )r& r� ZenvvarsZhome_dirZ
bad_configZ win_locationZposix_locationrI Zfname�dirpathZfpathr �excZdetails�msgr_ r` rM ra rb Zexpected_versionr r r r �do_test_with_pipK sb
:- z EnsurePipTest.do_test_with_pipNz@some dependencies of pip import threading module unconditionallyzpip requires ctypesc C s | j d � | j d � d S)NFT)r� )r& r r r �
test_with_pip� s
zEnsurePipTest.test_with_pip)r7 r8 r9 r: r� r� r� r
r� r� r� �skipIf�sslr� Z_MISSING_SSL_MESSAGEr� � threading�ctypesr� r r r r r� $ s
[r� �__main__)$r: r� r Zos.pathr� rE rY r r Z test.supportr r r r r �textwrapr� rC r� � ImportErrorr� r� r� rQ rR r� r� �lowerZexpectedFailurer
ZTestCaser r; r� r7 �mainr r r r �<module> sF (
$� �