Your IP : 216.73.217.32
���]�E � @ 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 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
r,d Z Yn Xe j e j e j k d � Z e j j � d k rle 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)� N)�captured_stdout�captured_stderr� can_symlink�EnvironmentVarGuard�rmtreez Test not appropriate in a venvZnulc C s | S)N� )�fr r �./opt/alt/python35/lib64/python3.5/test_venv.py�failsOnWindows. s r
c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �BaseTestc 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�tempfile�mkdtemp�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)�openr2 �read)r' r, r �resultr r r �get_text_file_contentsP s z BaseTest.get_text_file_contentsN)�__name__�
__module__� __qualname__r( r) r0 r2 r7 r r r r r 1 s
r c @ s4 e Z d Z d d � Z d d � Z e d d � � Z e j d k rN 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)-� BasicTestc G s, | j | � } | j t j j | � � d S)N)r2 �
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) N�lib64�P� �posixr z
pyvenv.cfgr z home = %szContents of %r:z %rzFile %r should exist.) r r r0 �venv�creater= r r r r2 �struct�calcsizer r r r" r< r �islink� assertFalse�existsr7 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) Nz-c�prefix� base_prefix�base_exec_prefixzimport sys; print(sys.%s)� �stdout�stderr)� assertEqualr rS rR rT � exec_prefixr r r0 rC rD r r r1 r r& �
subprocess�Popen�PIPE� communicate�strip�encode)r' �envpy�cmdrR ZexpectedrN �out�errr r r �
test_prefixes| s
! zBasicTest.test_prefixes�win32r
r r �
site-packagesr r r z python%d.%dNrU c
C su xn | D]f } t j j | j | � } t j | � t j j | | � } t | d � � } | j d � Wd QRXq Wd S)N�wbs Still here?)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 � 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)N�foo�rbs Still here?�clearT)ro )ro )rn � ENV_SUBDIRSrC rD 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]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 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]s } t j j | j | � } t | d � � } | j d � Wd QRX| j t t f t
j | j � | j | j � q Wd S)Nr rg � )
rr r r r1 r r4 ri Z assertRaises�
ValueError�OSErrorrC rD 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 | � � | j t j j
| � d | � q Wd S)NFT�upgradezContents of %r:z %rzFile %r should exist.)FT)rC rs r0 rD 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) NT�trueF�falserq �system_site_packagesz
pyvenv.cfgz"include-system-site-packages = %s
�Tr �Fr� )r� r� )rC rs rD r r7 rK )r' Zssp�srt rO 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)NFTrq �symlinks)FT) rC rs rD 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)Nz-cz!import sys; print(sys.executable)rV rW )r r r0 rC rD 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)Nrq Tr� z-cz!import sys; print(sys.executable)rV rW )r r rC rs rD 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�r
�r �r �r rf )r� r� r� r� )r )r )r )r8 r9 r: r= rQ �
skipInVenvrd r r" rr � version_inforn ru rx r| r~ r� �unittest�
skipUnlessr r� r� r� r r r r r; U s, #
!r; c @ s� e 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)�
EnsurePipTestc 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 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 r0 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)Nrp ry )r< r r rI �devnullr4 rX r5 )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 --versionrV rW 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 �TemporaryDirectoryr r r1 rh r4 ri r0 rC rD rZ �CalledProcessErrorr. r� Zfail�formatr r r& r[ r\ r] rX � ensurepipr! �len�fsencoderK �re�sub� MULTILINE�rstripr� )r' r� �envvarsZhome_dirZ
bad_configZ win_locationZposix_locationrJ Zfname�dirpathZfpathr �excZdetails�msgr` ra rN rb rc 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)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 Zos.pathr� rE rZ r r Z test.supportr r r r r �textwrapr� rC r� � ImportErrorr� r� r� rR rS r� r� �lowerZexpectedFailurer
ZTestCaser r; r� r8 �mainr r r r �<module> sD (
$� �