Your IP : 216.73.216.213
�
g f�@ �
@ 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 m Z 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 Xe j e j e j k d � Z e j j � d k re j Z n d d � Z Gd d � d e j � Z Gd
d � d e � Z e Gd d
� d
e � � Z d d � Z e! d k r�e � n 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� run_unittest� can_symlink�EnvironmentVarGuard�rmtreez Test not appropriate in a venvZnulc C s | S)N� )�fr r �3/opt/alt/python34/lib64/python3.4/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
�setUp) s zBaseTest.setUpc C s t | j � d S)N)r r )r' r r r
�tearDown9 s zBaseTest.tearDownc O sG t � �& } t � � } | | | � Wd QXWd QX| j � | j � f S)N)r r �getvalue)r' �func�args�kwargs�output�errorr r r
�run_with_capture< s zBaseTest.run_with_capturec G s t j j | j | � S)N)r r �joinr )r' r, r r r
� get_env_fileB s zBaseTest.get_env_filec G s1 t | j | � d � � } | j � } Wd QX| S)N�r)�openr2 �read)r' r, r �resultr r r
�get_text_file_contentsE s z BaseTest.get_text_file_contentsN) �__name__�
__module__� __qualname__�__doc__r( r) r0 r2 r7 r r r r
r & 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)r2 �
assertTruer r �isdir)r' r, �fnr r r
r> M 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 | � � n | 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 r0 �venv�creater> r r r r2 �structZcalcsizer r r r" r= r �islink� assertFalse�existsr7 r# r$ �dirname�assertInr& �print�listdir)r' �pZ
conditions�datar$ r r? �bdr r r
�
test_defaultsQ 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 rS rR rT � exec_prefixr r r0 rD rE r r r1 r r& �
subprocess�Popen�PIPE� communicate�strip�encode)r' �envpy�cmdrR ZexpectedrN �out�errr r r
�
test_prefixesq s
! zBasicTest.test_prefixes�win32r r r �
site-packagesr r r z python%d.%dNrU c
C st xm | D]e } t j j | j | � } t j | � t j j | | � } t | d � � } | j d � Wd QXq Wd S)zn
Create some files in the environment which are unrelated
to the virtual environment.
�wbs Still here?N)r r r1 r �mkdirr4 �write)r' �paths�filename�subdirs�dr? r r r r
�create_contents� s
zBasicTest.create_contentsc C s | j | j d � t j | j � xt | j D]i } t j j | j | d � } | j t j j | � � t
| d � � } | j | j � d � Wd QXq- 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)ro )ro )rn � ENV_SUBDIRSrD rE r r r r1 r= rI r4 rX r5 �
EnvBuilderrH )r' rl r? r �builderr r r
�test_overwrite_existing� s z!BasicTest.test_overwrite_existingc C s� x t j | � D]n } t j j | | � } t j j | � sO t j j | � r_ t j | � q t j j | � r t | � q q Wd S)N) r rM r r1 rG �isfile�remover> r )r' r r? r r r
�clear_directory� s $zBasicTest.clear_directoryc C s� x� | j d d � D]r } t j j | j | � } t | d � � } | j d � Wd QX| j t t f t
j | j � | j | j � q Wd S)Nr rg � )
rr r r r1 r r4 ri Z assertRaises�
ValueError�OSErrorrD rE rx )r' rj 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 | � � n | 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)rD rs r0 rE r r> r r r r2 r& r r rI rL rM r= )r' r} rt r? rP 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�falserq Zsystem_site_packagesz
pyvenv.cfgz"include-system-site-packages = %s
N�Tztrue�Fr� )r� r� )rD rs rE r r7 rK )r' Zssp�srt rO r r r
�test_isolation� s
zBasicTest.test_isolationzNeeds symlinksc C sw xp d D]h } t j d d d | � } | j | j � | j | j | j � } | r | j t j j
| � � q q Wd S)z3
Test symlinking works as expected
FTrq �symlinksN)FT) rD rs rE r r2 r r& r= r r rG )r' Zuslrt 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)rV rW N)r r r0 rD rE r r r1 r r r& rZ r[ r\ r] rX r^ r_ )r' r` ra rN rb rc 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.
rq Tr� z-cz!import sys; print(sys.executable)rV rW N)r r rD rs rE r r r1 r r r& rZ r[ r\ r] rX r^ r_ )r' rt r` ra rN rb rc r r r
�test_executable_symlinks s
- z"BasicTest.test_executable_symlinks�zScripts�zInclude��Lib�r� rf )r� r� r� r� )zbin)zinclude)zlib)r8 r9 r: r; r> rQ �
skipInVenvrd r r" rr � version_inforn ru rx r| r~ r� �unittestZ
skipUnlessr r� r� r� r r r r
r< J s. #
!r< c @ ss e Z d Z d Z d d � Z d d � Z d d � Z e d d � � Z e j
e d
k e j
� 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-crV rW zlatin-1� ZOK)r r r1 r r r r& rZ r[ r\ r] �decoderX r^ )r' r` Z
try_importra rN rb rc 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 r0 rD rE r� )r' r r r
�test_no_pip_by_default, 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 r0 rD rE r� )r' r r r
�test_explicit_no_pip1 s
z"EnsurePipTest.test_explicit_no_pipc
C sQ | j t j j t j � � t t j d � � } | j | j � d � Wd QXd S)Nrp ry )r= r r rI �devnullr4 rX r5 )r' r r r r
� test_devnull_exists_and_is_empty6 s z.EnsurePipTest.test_devnull_exists_and_is_emptyNc ' C s t | j � t � �A} d | d <d | d <t j � �} | | d <d } d } d } xs | f D]h \ } } t j j | | � } t j | � t j j | | � } t | d � � }
|
j
| � Wd QXqb Wy | j t j
| j d d
�Wn\ t j k
rL} z9 | j j d d � } d }
| j |
j | | � � WYd d } ~ Xn XWd QXWd QXt 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 QX| j d � } | j | d � | j d � } | j d | � | j d | � | j � d S) N�eZPYTHONWARNINGS�1ZPIP_NO_INSTALL�HOMEz[global]
no-install=1�pip�pip.ini�.pip�pip.conf�wr� T�errors�replacez {}
**Subprocess Output**
{}z-Imz --versionrV rW zlatin-1r� zpip {}zensurepip._uninstallz Successfully uninstalled pipz#Successfully uninstalled setuptools)r� r� )r� r� )!r r r r ZTemporaryDirectoryr r r1 rh r4 ri r0 rD rE rZ ZCalledProcessErrorr. r� Zfail�formatr r r& r[ r\ r] rX � ensurepipr! �len�fsencoderK r� )r' ZenvvarsZhome_dirZ
bad_configZ win_locationZposix_locationrJ Zfname�dirpathZfpathr �excZdetails�msgr` ra rN rb rc Zexpected_versionr r r r
�
test_with_pipA sZ
8- zEnsurePipTest.test_with_pip)r8 r9 r: r; r� r� r� r r� r� �skipIf�sslr� Z_MISSING_SSL_MESSAGEr� r r r r
r� s r� c C s t t t � d S)N)r r< r� r r r r
� test_main� s r� �__main__)"r; r� r Zos.pathrF rZ r r Z test.supportr r r r r r �textwrapr� rD r� � ImportErrorr� rR rS r� r� �lowerZexpectedFailurer ZTestCaser r< r� r� r8 r r r r
�<module> s6 .
$� w