Your IP : 216.73.216.213
�
g f;K � @ s� 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 � � n d d l Z e j r"e j j e j � r"e j e j � e j e j � n 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 r�e j# � n 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 @ 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)�sys�path�sys_path�site� USER_BASE�old_base� USER_SITE�old_site�PREFIXES� old_prefixes� sysconfig� _CONFIG_VARS�
original_varsr �old_vars)�self� r �3/opt/alt/python34/lib64/python3.4/test/test_site.py�setUp& 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 �tearDown/ 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 )�osr �joinr
�makepath� assertEqual�abspath�normcase)r �
path_partsZ original_dirZabs_dirZnorm_dirr r r �
test_makepath9 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 �isdirr
r )�.0r r r r �
<listcomp>H 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_pathinfoF 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) r+ �importedr �modulesr
r �
good_dir_pathr � assertFalser �exists� bad_dir_path)r �pth_filer r r �pth_file_testsN 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�setr5 )r r4 r r r �test_addpackageU 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>j s z/HelperFunctionsTests.make_pth.<locals>.<lambda>)r r r" r �openZ
addCleanup�write�close)r �contents�pth_dirZpth_nameZ pth_basenamer4 r )rC r �make_pthd s
z HelperFunctionsTests.make_pthc C s� | j d � \ } } t � � } t j | | t � � Wd QX| 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) rJ r r
r: r= � assertRegex�getvalue�re�escaper r r )r rI rC �err_outr r r �!test_addpackage_import_bad_syntaxo s z6HelperFunctionsTests.test_addpackage_import_bad_syntaxc C s� | j d � \ } } t � � } t j | | t � � Wd QX| j | j � d � | j | j � t j t j
j | | � � � | j | j � d � | j | j � d � d S)Nz randompath
import nosuchmodule
zline 2rK � ImportError) rJ r r
r: r= rM rN rO rP r r r )r rI rC rQ 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 QX| j | j � d � | j | j � t j t j
j | | � � � | j | j � d � | j | j � d � d S)NZabc def
zline 1rK � TypeError) rJ r r
r: r= rM rN rO rP r r r )r rI rC rQ 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)Nr6 T)r7 r8 r9 r
�
addsitedirr; r= r5 )r r4 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)�envr& z-sr �1ZPYTHONNOUSERSITEz/tmp�PYTHONUSERBASEz=import sys, site; sys.exit(site.USER_BASE.startswith("/tmp")))
r
r
r+ r r r �environr �
subprocess�call�
executabler! �getsitepackages)r ZusersiterZ �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 QXd S)Nr �xoxor\ ) r
r � getuserbaser! r r r �
assertTrue�
startswith)r � user_baser r] 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! rf rg )r � user_siter r r �test_getusersitepackages� s
z-HelperFunctionsTests.test_getusersitepackagesc C s� 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 | � t j
j d d
d t j d d � d � } | j | d | � t j
j d d
d � } | j | d | � nV | j t | � d � | j | d d � t j
j d d
d � } | j | d | � d S)Nrd �darwin�PYTHONFRAMEWORKzPython.framework� z/Libraryz
site-packages� �/�lib64�pythonr �libr& z site-python)r
r ra r �platformr �get_config_varr! �lenr r r �version�sep)r �dirsZwantedr r r �test_getsitepackages� s4 # # z)HelperFunctionsTests.test_getsitepackagesN)�__name__�
__module__� __qualname__�__doc__r r r% r- r5 r> r rJ rR rT �unittestZskipIfr ru rW rY �
skipUnlessr
�ENABLE_USER_SITErc ri rl r{ r r r r r " 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)
r7 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) r< r r r" r; r � file_pathr. � good_dirname� bad_dirnamer0 r3 )r �
filename_baser. 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.
r@ z#import @bad module name�file�
z import %sN)
rE r� �printr. r� r� rG r �mkdirr0 )r ZFILEr r r r9 s zPthFile.createFc C s� t j j | j � r( t j | j � n | rb t j j | j � | _ | j r� t j | j =q� n | j r� | j t j | j <n t j j | j
� r� t j | j
� n t j j | j � r� t j | j � n 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 r2 r� rB r r/ �getr. Zimported_moduler0 �rmdirr3 )r r6 r r r r8 + s zPthFile.cleanupN)r| r} r~ r r r� r9 r8 r r r r r7 s
r7 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 @ s zImportSideEffectTests.setUpc C s | j t j d d � <d S)zRestore sys.pathN)r r r )r r r r r D 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 | � � | j t j j | � � 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-crZ �stdoutr rp )r� r� r� r� r� )r r �relpath�dirname�__file__r] r r r^ �Popenr r` �PIPE� communicater! �
returncode�
splitlinesr1 �isabsrf )
r �parentrZ �codeZcommand�procr� �stderrZ
os__file__Z os__cached__r r r �test_abs_pathsH s2 !
z$ImportSideEffectTests.test_abs_pathsc C sH t j � t � } x. t j D]# } | j | | � | j | � q Wd S)N)r
�removeduppathsr= r r � assertNotIn�add)r Z
seen_pathsr r r r �test_no_duplicate_pathsn s
z-ImportSideEffectTests.test_no_duplicate_pathsztest not implementedc C s d S)Nr )r r r r �test_add_build_dirw s z(ImportSideEffectTests.test_add_build_dirc C s0 | j t t d � � | j t t d � � d S)N�quit�exit)rf �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)rf 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)rf r� r� )r r r r �test_setting_help� s z'ImportSideEffectTests.test_setting_helpc C sx t j d k rt d d l } | j � d j d � rt x: t j j j � D] } | d k rG PqG qG W| j d � qt n d S)NrU r r& �cp�mbcszdid not alias mbcs) r ru �localeZgetdefaultlocalerg � encodings�aliases�values�fail)r r� �valuer r r �test_aliasing_mbcs� s z(ImportSideEffectTests.test_aliasing_mbcsc C sH d t j k rD y d d l } Wn t k
r3 YqD X| j d � n d S)N�
sitecustomizer z(sitecustomize not imported automatically)r r/ r� rS r� )r r� r r r �test_sitecustomize_executed� s
z1ImportSideEffectTests.test_sitecustomize_executedZnetworkro �finalzonly for released versionsZ HTTPSHandlerz$need SSL support to download licensec C s� t j j � d } t j j | d d �} yD t j j | � �, t j j | � � } | j
� } Wd QXWd QXWn4 t j j k
r� } z | j
} WYd d } ~ Xn X| j | d d d | �d S)Nr& �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� = s" &
r� c @ s e Z d Z d d � Z d S)�StartupImportTestsc
C sD 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 | � | � n | j d | | � t j d k r� | j d | | � n 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
rO �_sre� sre_compile�
sre_constants� sre_parserm r� �copyreg� _collections� collections� functools�heapq� itertools�keyword�operator�reprlib�types�weakref)r^ r� r r` r� r� �decode�evalr+ ru r1 � intersectionr� �
difference�builtin_module_names)r �popenr� r� r/ 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 rO r� Zurllib.requestr� Z urllib.errorr^ r r �flags�no_siteZSkipTestr
r� r r' r
�makedirsrX ZTestCaser �objectr7 r� r� r| �mainr r r r �<module> s2 �7x"