Your IP : 216.73.216.213
��Yf�L � @ sv d Z d d l Z d d l Z d d l m Z m Z m 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 Z d d l m Z e j j r� e j d � � d d l Z d a d d � Z d d � Z Gd
d � d e j � Z Gd d
� d
e � Z Gd d � d e j � Z Gd d � d e j � Z e d k rre j � d S)zTests for 'site'.
Tests assume the initial paths in sys.path once the interpreter has begun
executing have not been removed.
� N)�captured_stderr�TESTFN�EnvironmentVarGuard)�copyzPython was invoked with -Sc C s� t j d d � a t j r� t j j t j � r� y$ t j t j � t j t j � WnA t
k
r� } z! t j d t j | f � � WYd d } ~ Xn Xd S)Nz-unable to create user site directory (%r): %s)
�sys�path� OLD_SYS_PATH�site�ENABLE_USER_SITE�os�isdir� USER_SITE�makedirs�
addsitedir�PermissionError�unittest�SkipTest)�exc� r �3/opt/alt/python35/lib64/python3.5/test/test_site.py� setUpModule! s r c C s t t j d d � <d S)N)r r r r r r r �tearDownModule0 s r c @ s e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d e
d d � Z d d � Z d d � Z
e j e j d k d � d d � � Z d d � Z e j e j d � d d � � Z d d � Z d d! � Z d" d# � Z d$ S)%�HelperFunctionsTestsz Tests for helper functions.
c C s\ t j d d � | _ t j | _ t j | _ t j | _ t
j | _ t
t
j � | _ d S)zSave a copy of sys.pathN)r r �sys_pathr � USER_BASE�old_baser
�old_site�PREFIXES� old_prefixes� sysconfig� _CONFIG_VARS�
original_varsr �old_vars)�selfr r r �setUp8 s zHelperFunctionsTests.setUpc C sj | j t j d d � <| j t _ | j t _ | j t _ | j
t _ t j j
� t j j | j � d S)zRestore sys.pathN)r r r r r r r r
r r r! r r �clear�updater" )r# r r r �tearDownA s
z HelperFunctionsTests.tearDownc C s� d } t j j | � } t j | � \ } } | j t j j | � | � | t j j | � k rt | j | | � n | j t j j | � | � d S)N� Beginning�End)r( r) )r r �joinr �makepath� assertEqual�abspath�normcase)r# �
path_partsZ original_dirZabs_dirZnorm_dirr r r �
test_makepathK s z"HelperFunctionsTests.test_makepathc C sN t j � } x; d d � t j D� D]# } | j | | d | | f � q# Wd S)Nc S s; g | ]1 } | r t j j | � r t j | � d � q S)� )r r r r r+ )�.0r r r r �
<listcomp>Z s z;HelperFunctionsTests.test_init_pathinfo.<locals>.<listcomp>zB%s from sys.path not found in set returned by _init_pathinfo(): %s)r �_init_pathinfor r �assertIn)r# Zdir_set�entryr r r �test_init_pathinfoX s
z'HelperFunctionsTests.test_init_pathinfoc C sc | j | j t j d | j � | j t j | j � d t j � | j t j j
| j � � d S)z>Contain common code for testing results of reading a .pth filez%s not in sys.modulesr N) r5 �importedr �modulesr r+ �
good_dir_pathr � assertFalser �exists� bad_dir_path)r# �pth_filer r r �pth_file_tests` s #z#HelperFunctionsTests.pth_file_testsc
C sb t � } | j d d � z7 | j � t j | j | j t � � | j | � Wd | j � Xd S)N�prepT) �PthFile�cleanup�creater �
addpackage�base_dir�filename�setr? )r# r> r r r �test_addpackageg s
z$HelperFunctionsTests.test_addpackage�.c s� t j j | � } | d } t j j | | � � t � d d d �} | j � f d d � � | j | � | j � | | f S)Nz.pth�w�encodingzutf-8c s
t j � � S)N)r �remover )�pth_fnr r �<lambda>| s z/HelperFunctionsTests.make_pth.<locals>.<lambda>)r r r- r* �openZ
addCleanup�write�close)r# �contents�pth_dirZpth_nameZ pth_basenamer> r )rM r �make_pthv s
z HelperFunctionsTests.make_pthc
C s� | j d � \ } } t � � } t j | | t � � Wd QRX| j | j � d � | j | j � t j t j
j | | � � � | j | j � d � | j | j � d � | j | j � d � d S)Nzimport bad)syntax
zline 1� Tracebackzimport bad\)syntax� SyntaxError) rT r r rD rG � assertRegex�getvalue�re�escaper r r* )r# rS rM �err_outr r r �!test_addpackage_import_bad_syntax� s z6HelperFunctionsTests.test_addpackage_import_bad_syntaxc
C s� | j d � \ } } t � � } t j | | t � � Wd QRX| j | j � d � | j | j � t j t j
j | | � � � | j | j � d � | j | j � d � d S)Nz randompath
import nosuchmodule
zline 2rU � ImportError) rT r r rD rG rW rX rY rZ r r r* )r# rS rM r[ r r r � test_addpackage_import_bad_exec� s z4HelperFunctionsTests.test_addpackage_import_bad_exec�win32zIWindows does not raise an error for file paths containing null charactersc
C s� | j d � \ } } t � � } t j | | t � � Wd QRX| j | j � d � | j | j � t j t j
j | | � � � | j | j � d � | j | j � d � d S)Nzabc def
zline 1rU �
ValueError) rT r r rD rG rW rX rY rZ r r r* )r# rS rM r[ r r r �#test_addpackage_import_bad_pth_file� s z8HelperFunctionsTests.test_addpackage_import_bad_pth_filec
C s\ t � } | j d d � z1 | j � t j | j t � � | j | � Wd | j � Xd S)Nr@ T)rA rB rC r r rE rG r? )r# r> r r r �test_addsitedir� s
z$HelperFunctionsTests.test_addsitedirz<requires access to PEP 370 user-site (site.ENABLE_USER_SITE)c C s� t j } | j | t j � t j j � } t j t j
d d | g d | �} | j | d � t j j � } t j t j
d d d | g d | �} | t j � d k r� | j | d � n | j | d � t j j � } d | d <t j t j
d d | g d | �} | t j � d k r7| j | d � n | j | d � t j j � } d | d
<t j t j
d d g d | �} | j | d � d S) Nz-cz$import sys; sys.exit(%r in sys.path)�envr1 z-sr �1ZPYTHONNOUSERSITEz/tmp�PYTHONUSERBASEz=import sys, site; sys.exit(site.USER_BASE.startswith("/tmp")))
r r
r5 r r r �environr �
subprocessZcall�
executabler, �getsitepackages)r# Zusersiterc �rcr r r �
test_s_option� s8
z"HelperFunctionsTests.test_s_optionc
C s� d t _ t j � } | j t j | � d t _ d d l } d | _ t � �6 } d | d <| j t j � j d � t j � � Wd QRXd S)Nr �xoxore ) r r � getuserbaser, r r r �
assertTrue�
startswith)r# � user_baser rf r r r �test_getuserbase� s
z%HelperFunctionsTests.test_getuserbasec C sQ d t _ d t _ t j � } | j t j | � | j | j t j � | � d S)N)r r
r �getusersitepackagesr, rn ro )r# � user_siter r r �test_getusersitepackages� s
z-HelperFunctionsTests.test_getusersitepackagesc C sn d g t _ t j � } t j d k r� t j d � r� d g t _ t j � } | j t | � d � t j
j d t j d � t j d d � d � } | j | d | � n� t j
d
k r| j t | � d � t j
j d d d t j d d � d � } | j | d
| � nV | j t | � d � | j | d
d � t j
j d d d � } | j | d | � d S)Nrl �darwin�PYTHONFRAMEWORKzPython.framework� z/Library� z
site-packagesr1 �/�lib64�pythonr �lib)r r ri r �platformr �get_config_varr, �lenr r r* �version�sep)r# �dirsZwantedr r r �test_getsitepackages� s* # z)HelperFunctionsTests.test_getsitepackagesN)�__name__�
__module__� __qualname__�__doc__r$ r' r0 r7 r? rH r rT r\ r^ r ZskipIfr r} ra rb �
skipUnlessr r
rk rq rt r� r r r r r 4 s
'
!& r c @ sI e Z d Z d Z e d d d d d � Z d d � Z d d
d � Z d S)
rA z/Helper class for handling testing of .pth filesZtimeZ __testdir__Z__badc C s� | d | _ t j j d � | _ t j j | j | j � | _ | | _ | | _ | | _ t j j | j | j � | _
t j j | j | j � | _ d S)z Initialize instance variablesz.pth� N) rF r r r- rE r* � file_pathr8 � good_dirname� bad_dirnamer: r= )r# �
filename_baser8 r� r� r r r �__init__ s
zPthFile.__init__c
C s� t | j d � } za t d d | �t d d | �t d | j d | �t | j d | �t | j d | �Wd | j � Xt j | j � d S)ao Create a .pth file with a comment, blank lines, an ``import
<self.imported>``, a line with self.good_dirname, and a line with
self.bad_dirname.
Creation of the directory for self.good_dir_path (based off of
self.good_dirname) is also performed.
Make sure to call self.cleanup() to undo anything done by this method.
rJ z#import @bad module name�file�
z import %sN)
rO r� �printr8 r� r� rQ r �mkdirr: )r# ZFILEr r r rC " s zPthFile.createFc C s� t j j | j � r% t j | j � | r\ t j j | j � | _ | j rx t j | j =n | j rx | j t j | j <t j j | j
� r� t j | j
� t j j | j � r� t j | j � d S)z�Make sure that the .pth file is deleted, self.imported is not in
sys.modules, and that both self.good_dirname and self.bad_dirname are
not existing directories.N)
r r r< r� rL r r9 �getr8 Zimported_moduler: �rmdirr= )r# r@ r r r rB 8 s zPthFile.cleanupN)r� r� r� r� r r� rC rB r r r r rA s
rA c @ s e Z d Z d Z d d � Z d d � Z d d � Z d d � Z e j d
� d d � � Z
d
d � Z d d � Z d d � Z
d d � Z d d � Z e j j d � e j j e j e j d d k d � e j e e j d � d � d d � � � � � Z d S) �ImportSideEffectTestsz(Test side-effects from importing 'site'.c C s t j d d � | _ d S)zMake a copy of sys.pathN)r r r )r# r r r r$ M s zImportSideEffectTests.setUpc C s | j t j d d � <d S)zRestore sys.pathN)r r r )r# r r r r' Q s z ImportSideEffectTests.tearDownc
C s� t j j t j j t j � � } t j j � } | | d <d } d j | � } t j t
j d d | g d
| d t j �} | j
� \ } } | j | j d � | j � d d
� \ } } | j t j j | � � | j t j j | � � t j t
j d | g d
| d t j �} | j
� \ } } | j | j d � | j � d d
� \ } } | j t j j | � d j | j d � � � | j t j j | � d j | j d � � � d S)N�
PYTHONPATH�import os, sys�9os_file = os.__file__.encode("ascii", "backslashreplace")�(sys.stdout.buffer.write(os_file + b"\n")�=os_cached = os.__cached__.encode("ascii", "backslashreplace")�*sys.stdout.buffer.write(os_cached + b"\n")r� z-Sz-crc �stdoutr rw z expected absolute path, got {}�ascii)r� r� r� r� r� )r r �relpath�dirname�__file__rf r r* rg �Popenr rh �PIPE� communicater, �
returncode�
splitlinesr; �isabsrn �format�decode)
r# �parentrc �codeZcommand�procr� �stderrZ
os__file__Z os__cached__r r r �test_abs_pathsU s: !
z$ImportSideEffectTests.test_abs_pathsc C sH t j � t � } x. t j D]# } | j | | � | j | � q Wd S)N)r �removeduppathsrG r r � assertNotIn�add)r# Z
seen_pathsr r r r �test_no_duplicate_paths s
z-ImportSideEffectTests.test_no_duplicate_pathsztest not implementedc C s d S)Nr )r# r r r �test_add_build_dir� s z(ImportSideEffectTests.test_add_build_dirc C s0 | j t t d � � | j t t d � � d S)N�quit�exit)rn �hasattr�builtins)r# r r r �test_setting_quit� s z'ImportSideEffectTests.test_setting_quitc C sF | j t t d � � | j t t d � � | j t t d � � d S)N� copyright�credits�license)rn r� r� )r# r r r �test_setting_copyright� s z,ImportSideEffectTests.test_setting_copyrightc C s | j t t d � � d S)N�help)rn r� r� )r# r r r �test_setting_help� s z'ImportSideEffectTests.test_setting_helpc C so t j d k rk d d l } | j � d j d � rk x4 t j j j � D] } | d k rG PqG W| j d � d S)Nr_ r r1 �cp�mbcszdid not alias mbcs) r r} �localeZgetdefaultlocalero � encodings�aliases�values�fail)r# r� �valuer r r �test_aliasing_mbcs� s z(ImportSideEffectTests.test_aliasing_mbcsc C sE d t j k rA y d d l } Wn t k
r3 Yn X| j d � d S)N�
sitecustomizer z(sitecustomize not imported automatically)r r9 r� r] r� )r# r� r r r �test_sitecustomize_executed� s
z1ImportSideEffectTests.test_sitecustomize_executedZnetworkrx �finalzonly for released versionsZ HTTPSHandlerz$need SSL support to download licensec C s� t j j � d } t j j | d d �} yF t j j | � �- t j j | � � } | j
� } Wd QRXWd QRXWn4 t j j k
r� } z | j
} WYd d } ~ Xn X| j | d d d | �d S)Nr1 �methodZHEAD�� �msgz Can't find )r� �_Printer__data�split�urllib�requestZRequest�test�supportZtransient_internetZurlopenZgetcode�errorZ HTTPErrorr� r, )r# ZurlZreq�datar� �er r r �test_license_exists_at_url� s z0ImportSideEffectTests.test_license_exists_at_urlN)r� r� r� r� r$ r' r� r� r �skipr� r� r� r� r� r� r� r� Zrequires_resourceZsystem_must_validate_certr� r � version_infor� r� r� r� r r r r r� J s" *
r� c @ s e Z d Z d d � Z d S)�StartupImportTestsc
C s> t j t j d d d d g d t j d t j �} | j � \ } } | j d � } | j d � } t | � } | j d | � d d
d d d
h } t j d k r� | j
| j | � | � | j d | | � t j d k r� | j d | | � d d d d d d d d d d h
j
t j � } | j
| j | � | � d S)Nz-Iz-vz-cz#import sys; print(set(sys.modules))r� r� zutf-8r rY �_sre� sre_compile�
sre_constants� sre_parseru r� �copyreg� _collections� collections� functools�heapq� itertools�keyword�operator�reprlib�types�weakref)rg r� r rh r� r� r� �evalr5 r} r; � intersectionr� �
difference�builtin_module_names)r# �popenr� r� r9 Zre_modsZcollection_modsr r r �test_startup_imports� s( z'StartupImportTests.test_startup_importsN)r� r� r� r� r r r r r� � s r� �__main__)!r� r Z test.supportr� r r r r� r r rY r� Zurllib.requestr� Z urllib.errorrg r r �flags�no_siter r r r r ZTestCaser �objectrA r� r� r� �mainr r r r �<module> s2 �7|"