Your IP : 216.73.216.213
�
g f� � @ s� d d l Z d d l Z d d l 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
j 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 m Z m Z m Z m Z m Z m Z m! Z! m" Z" m# Z# m$ Z$ d d l m% Z% e j& e j' d � Z( d d � Z) e j* d d d d
� � Z+ Gd d � d e j, � Z- e( Gd
d � d e j, � � Z. Gd d � d e j, � Z/ Gd d � d e j, � Z0 Gd d � d e j, � Z1 Gd d � d e j, � Z2 Gd d � d e j, � Z3 Gd d � d e j, � Z4 e# Gd d � d e j, � � Z5 e# Gd d � d e j, � � Z6 Gd d � d e j, � Z7 e8 d! k r�e j9 � n d S)"� N)�_get_sourcefile)�EnvironmentVarGuard�TESTFN�check_warnings�forget� is_jython�make_legacy_pyc�rmtree� run_unittest� swap_attr� swap_item�
temp_umask�unlink�unload�create_empty_file� cpython_only�TESTFN_UNENCODABLE)�
script_helperz*test meaningful only when writing bytecodec C sL x; | d | d | d | d | d f D] } t | � q* Wt d � d S)Nz.pyz.pycz.pyoz.pywz $py.class� __pycache__)r r )�name�f� r �5/opt/alt/python34/lib64/python3.4/test/test_import.py� remove_files s
r � c c s� | p d } t j � �� } t j | | | � } t j j | d � } z2 t j j d | � | | f Vt j j | � Wd | d k r� | t j | <n | t j k r� t j | =n XWd QXd S)NZspamr ) r Ztemp_dir� make_script�sys�modules�pop�path�insert�remove)r �sourceZtempdirr Z
old_moduler r r �_ready_to_import* s r# c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d d
� Z d d � Z d d � Z
d d � Z e d d � � Z
d d � Z d d � Z d d � Z d d � Z e d d � � Z d d! � Z d" d# � Z d$ S)%� ImportTestsc C s t t � t j � d S)N)r r � importlib�invalidate_caches)�selfr r r �setUpA s
zImportTests.setUpc C s t t � d S)N)r r )r' r r r �tearDownE s zImportTests.tearDownc C s& | j t � � d d l } Wd QXd S)Nr )� assertRaises� ImportError�RAnDoM)r' r, r r r �test_case_sensitivityH s z!ImportTests.test_case_sensitivityc C s d d l m } d S)Nr )� double_const)�testr. )r' r. r r r �test_double_constN s z ImportTests.test_double_constc s� � f d d � } t j j d t j � zP | d � t j j d � rw x- d d d d d
d g D] } | | � q` Wn Wd t j d =Xd S) Nc s� t | } t d } t r' t d } n
t d } t | d � �a } t d | d d | �t j d � } t j d � } t d | d | �t d
| d | �Wd QXt t j k r� t j t =n t j � z� y t
t � } Wn< t k
r} z � j d | | f � WYd d } ~ Xn X� j
| j | d | � � j
| j | d | � Wd t t � t | � t | � t | � Xd S)
Nz.pyoz $py.classz.pyc�wz)# This tests Python's ability to import azfile.�filei� za =zb =zimport from %s failed: %sz'module loaded (%s) but contents invalid)r r �open�print�randomZ randranger r r% r& �
__import__r+ �fail� assertEqual�a�br r ) �extr" ZpyoZpycr r9 r: �mod�err)r' r r �test_with_extensionT s8
*
z4ImportTests.test_import.<locals>.test_with_extensionr z.py�winz.PYz.Pyz.pYz.pywz.PYWz.pYw)r r r �os�curdir�platform�
startswith)r' r> r; r )r' r � test_importS s $
zImportTests.test_importZlonglistc " C s | d } t | d � �E } | j d � x! t d � D] } | j d � q6 W| j d � Wd QXz t j | � Wd t | � Xt j j d � t j
� i } z" t | � t d | d | � Wd t j d =t | d
� t | d � | j
� y t j | =Wn t k
rYn XXd S)
Nz.pyr1 zd = [
i� z"",
�]r zimport � �c�o���)r3 �write�range�
py_compile�compiler r r �appendr% r& r �exec�clearr �KeyError)r' �module�filenamer �i� namespacer r r � test_module_with_large_stack� s.
z(ImportTests.test_module_with_large_stackc C s� t d } t | d � � } t d d | �Wd QXt j j d t j � t j � t t j
k rn t j
t =n zK xD d d d g D]3 } | j t t
t � | j t t j
d | � q� WWd t j d =t t � Xd S)
Nz.pyr1 za = 1/0r2 r rF � � z'damaged module in sys.modules on %i try)r r3 r4 r r r r@ rA r% r& r r* �ZeroDivisionErrorr6 Z assertNotInr )r' r" r rT r r r �test_failing_import_sticks� s
z&ImportTests.test_failing_import_sticksc C ss d d l } d d l } | j | | | j � | j t | j d � � d d l j } | j | | j | j � d S)Nr �__file__)r/ � test.support�assertIs�__name__�
assertTrue�hasattr�support)r' �xr/ �yr r r �test_import_name_binding� s z$ImportTests.test_import_name_bindingc C sT t t j d } t | d � � } | j d � Wd QXt j j d t j � z� t t � } | j
t t j � | j | j
d d � | j | j d d � t t � t | d � � } | j d � Wd QX| j t t j | � t j j t � } | j | d � | j | j
d
d � | j | j d d � Wd t j d =t t � t t � Xd S) N�pyr1 z
a = 1
b=2
r rF z!module has wrong attribute valuesrW za = 10
b=20//0
z$expected module to be in sys.modules�
)r r@ �extsepr3 rJ r r r rA r6 �assertInr r8 r9 r: r r* rY r% �reload�getZassertIsNotNoner )r' r" r r< r r r �test_failing_reload� s(
z ImportTests.test_failing_reloadc C s t d } t | d � � } | j d � Wd QXt j j d t j � z� t t � } | j | j
j d � � t j | � t j
t =t | � t j � t t � } t j j | j
� \ } } | j | d � Wd t j d =t t � t t j
k rt j
t =n Xd S)Nz.pyr1 z test = None
r �.pyc�.pyo)rl rm )r r3 rJ r r r r@ rA r6 r_ r[ �endswithr! r r r% r& �splitextrh r )r' r" r r< �baser; r r r �test_file_to_source� s$
z ImportTests.test_file_to_sourcec C s� t j j t � } t j � } y | j | � Wn( t k
rY | j d j | � � Yn X| j
t � � } t | � Wd QXd S)Nzpath is not encodable to {})
r@ r �abspathr r �getfilesystemencoding�encode�UnicodeEncodeError�skipTest�formatr* r+ r6 )r' r �encodingrG r r r �test_import_by_filename s
z#ImportTests.test_import_by_filenamec C s/ t j d t t j d � � } t j | � d S)Nr z� import sys
class C:
def __del__(self):
import importlib
sys.argv.insert(0, C())
)r r r �textwrap�dedent�assert_python_ok)r' Ztestfnr r r �!test_import_in_del_does_not_crash s z-ImportTests.test_import_in_del_does_not_crashc C s t j j d t j � z� t d } t j j | � } t | d � � } Wd QXy t j
| d � Wnz t k
r� | j d � Yn\ t
k
r� } z< | j t t d d � k r� � n | j d j | � � WYd d } ~ Xn Xt t � t j | � Wd t j d =t t � Xd S)Nr z.pyr1 rW �! � z-cannot set modification time to large integerZ EOVERFLOWz2cannot set modification time to large integer ({})l � �� l � �� )r� r� )r r r r@ rA r r% �util�cache_from_sourcer3 �utime�
OverflowErrorrv �OSError�errno�getattrrw r6 �statr )r' r" Zcompiledr �er r r �test_timestamp_overflow s$
)
z#ImportTests.test_timestamp_overflowc C s= y t d d d g �Wn t k
r8 | j d � Yn Xd S)NZhttp�fromlistZblahz fromlist must allow bogus names)r6 r+ r7 )r' r r r �test_bogus_fromlist3 s
z ImportTests.test_bogus_fromlistc C sA d d g } t j | � } | j � \ } } | j d | � d S)Nz-cz&del __builtins__.__import__; import oss ImportError)r Z spawn_pythonZ communicaterh )r' �args�popen�stdout�stderrr r r �test_delete_builtins_import9 s z'ImportTests.test_delete_builtins_importc C s- | j t d � � d d l m } Wd QXd S)Nz^No module named 'bogus'r )�foo)�assertRaisesRegexr+ �bogusr� )r' r� r r r �/test_from_import_message_for_nonexistent_module@ s z;ImportTests.test_from_import_message_for_nonexistent_modulec C s- | j t d � � d d l m } Wd QXd S)Nz^cannot import name 'bogus'r )r� )r� r+ �rer� )r' r� r r r �,test_from_import_message_for_existing_moduleD s z8ImportTests.test_from_import_message_for_existing_moduleN)r^ �
__module__� __qualname__r( r) r- r0 rD rV rZ rd rk �skip_if_dont_write_bytecoderq ry r} r� r� r r� r� r� r r r r r$ ? s" .( &
r$ c @ s� e Z d Z e j e j d k d � d d � � Z e j e j d k d � d d � � Z e j e j d k d � d d � � Z d d
� Z
d S) �FilePermissionTests�posixz%test meaningful only on posix systemsc C s� d } t | � �l t � �\ \ } } t j j | � } t | � } t j j | � se | j d � n t j
| � } Wd QXWd QX| j t t
j
| j � � t d | @� � d S)N� zC__import__ did not result in creation of either a .pyc or .pyo filei� )r
r# r% r� r� r6 r@ r �existsr7 r� r8 �oct�S_IMODE�st_mode)r' �maskr r � cached_pathrR � stat_infor r r �test_creation_modeM s z&FilePermissionTests.test_creation_modec C s� d } t d � �z t � �j \ } } t j j | � } t j | | � t | � t j j | � ss | j
d � n t j | � } Wd QXWd QX| j t
t j | j � � t
| � � d S)Ni� r� zC__import__ did not result in creation of either a .pyc or .pyo file)r
r# r% r� r� r@ �chmodr6 r r� r7 r� r8 r� r� r� )r' �moder r r� r� r r r �test_cached_mode_issue_2051^ s
z/FilePermissionTests.test_cached_mode_issue_2051c C s� d } t d � �z t � �j \ } } t j j | � } t j | | � t | � t j j | � ss | j
d � n t j | � } Wd QXWd QX| d B} | j t
t j | j � � t
| � � d S)N� r� zC__import__ did not result in creation of either a .pyc or .pyo file� )r
r# r% r� r� r@ r� r6 r r� r7 r� r8 r� r� r� )r' r� r r r� r� Zexpectedr r r �test_cached_readonlyn s
z(FilePermissionTests.test_cached_readonlyc C sj t � �[\ } } t | d � � } | j d � Wd QXt j | � } t j | | j | j d f � t j | d � t | � } | j
| j d � t j | d � t | d � � } | j d � Wd QXt | � t
j � t | � } | j
| j d � t | � t | � t
j � | d } t j t
j j | � | � t | � } | j
| j d � Wd QXd S)
Nr1 zx = 'original'
i �r� Zoriginali� zx = 'rewritten'
Z rewrittenrH )r# r3 rJ r@ r� r� �st_atime�st_mtimer� r6 r8 rb r r% r& r �renamer� r� )r' r r r �s�mZ
bytecode_onlyr r r �test_pyc_always_writable~ s,
z,FilePermissionTests.test_pyc_always_writableN) r^ r� r� �unittest�
skipUnlessr@ r r� r� r� r� r r r r r� I s r� c @ s� e Z d Z d Z d Z e j j e � Z e j j
e e � e j d Z e
j j e � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z d S)�PycRewritingTestsZunlikely_module_namez�
import sys
code_filename = sys._getframe().f_code.co_filename
module_filename = __file__
constant = 1
def func():
pass
func_filename = func.__code__.co_filename
re c C s� t j d d � | _ t j j | j d � | _ t j | j � t
| j d � � } | j | j
� Wd QXt j j d | j � t j � d S)Nr1 r )r r �sys_pathr r � module_name� orig_moduler@ �mkdir�dir_namer3 � file_namerJ �
module_sourcer r% r& )r' r r r r r( � s zPycRewritingTests.setUpc C ss | j t j d d � <| j d k r; | j t j | j <n
t | j � t | j � t | j � t
| j � d S)N) r� r r r� r r� r r r� �
compiled_namer r� )r' r r r r) � s
zPycRewritingTests.tearDownc C s* t � } t | j | | � t j | j S)N)�globalsr6 r� r r )r' �nsr r r �
import_module� s z PycRewritingTests.import_modulec C s� | j � } | j | j | j � | j | j | j � | j | j | j � t j | j =| j � } | j | j | j � | j | j | j � | j | j | j � d S)N) r� r8 �module_filenamer� �
code_filename�
func_filenamer r r� )r' r< r r r � test_basics� s
z PycRewritingTests.test_basicsc C sh t j | j d d �| j � } | j | j | j � | j | j | j � | j | j | j � d S)N�dfilezanother_module.py)rL rM r� r� r8 r� r� r� )r' r< r r r �test_incorrect_code_name� s
z*PycRewritingTests.test_incorrect_code_namec C s� d } t j | j d | �t j | j � t | j � } t j � | j � } | j | j
| � | j | j | � | j | j | � d S)Nzanother_module.pyr� )
rL rM r� r@ r! r r% r& r� r8 r� r� r� )r' �target�pyc_filer< r r r �test_module_without_source� s
z,PycRewritingTests.test_module_without_sourcec C sJ t j | j � t | j d � �% } | j d � } t j | � } Wd QXt | j � } t
j j } | j
d � } | | | <t | � | j | j | j | j | j | j t | � | j | j | j | j | j | j | j | j � } t | j d � �$ } | j | � t j | | � Wd QX| j � } | j | j! j | j � d S)N�rb� rF �wb)"rL rM r� r3 r� �read�marshal�load�list� co_constsr% r� �__code__�index�type� co_argcount�co_kwonlyargcount�
co_nlocals� co_stacksize�co_flags�co_code�tuple�co_names� co_varnames� co_filename�co_name�co_firstlineno� co_lnotab� co_freevars� co_cellvarsrJ �dumpr8 Zconstant)r' r �header�codeZ constantsZ foreign_code�posr< r r r �test_foreign_code� s&
z#PycRewritingTests.test_foreign_codeN)r^ r� r� r� r� r@ r rr r r� �joinrg r� r% r� r� r� r( r) r� r� r� r� r� r r r r r� � s
r� c @ sg e Z d Z d Z e Z d d � Z d d � Z d d
� Z e j
e j d k d � d
d � � Z
d S)�
PathsTestsr/ � testäöüß� testéè�
test°³²c C s* t j | j � t j d d � | _ d S)N)r@ r� r r �syspath)r' r r r r( s zPathsTests.setUpc C s' t | j � | j t j d d � <d S)N)r r r� r )r' r r r r) s
zPathsTests.tearDownc C s{ t t j j | j d � d � � } | j d � Wd QXt j j | j d � t d � } | j | j d � t
d � d S)Nztest_trailing_slash.pyr1 z testdata = 'test_trailing_slash'�/�test_trailing_slash) r3 r@ r r� rJ r rN r6 r8 �testdatar )r' r r< r r r r� s $ z PathsTests.test_trailing_slash�win32zWindows-specificc % C s� t t j j | j d � d � � } | j d � Wd QXt j � t j j | j � } d d l } | j � } | d } d | | f } | | d d � 7} y t j
| � WnZ t k
r } z: | j t j
t j f k r� | j d | f � n � WYd d } ~ Xn Xt j j d | � y t d � } Wn< t k
rk} z | j d | | f � WYd d } ~ Xn X| j | j d � | j | j j | � | j � t d � d S)
Nztest_unc_path.pyr1 ztestdata = 'test_unc_path'r z\\%s\%s$rW z%cannot access administrative share %rZ
test_unc_pathz,could not import 'test_unc_path' from %r: %r)r3 r@ r r� rJ r% r& rr �socketZ gethostname�listdirr� r� ZEPERMZEACCESrv r r r6 r+ r7 r8 r� r_ r[ rC r ) r' r r r� ZhnZdriveZuncr� r< r r r �
test_UNC_path s0 $
! zPathsTests.test_UNC_pathN)ztestr� r� r� )r^ r� r� ZSAMPLESr r r( r) r� r� r� r rB r� r r r r r� � s r� c @ sF e Z d Z d d � Z e Z d d � Z d d � Z d d � Z d S)
�RelativeImportTestsc C s t d � d S)Nztest.relimport)r )r' r r r r) 5 s z RelativeImportTests.tearDownc C s* d d l m } | j t | d � � d S)NrF )� relimportr� )r r� r_ r` )r' r� r r r �test_relimport_star9 s z'RelativeImportTests.test_relimport_starc s� � f d d � } t d d d d � � | � t d d d d � � | � t d d d d � � | j t | � t d d d d � � | j t | � t d t � � � | j t | � d S) Nc s t d � � d S)Nzfrom . import relimport)rO r )r� r r �check_relativeD s z:RelativeImportTests.test_issue3221.<locals>.check_relative� __package__r/ r^ ztest.notarealmoduleznotarealpkg.notarealmoduler� )�dictr* � SystemError�object� TypeError)r' r� r )r� r �test_issue3221> s z"RelativeImportTests.test_issue3221c C s7 | j t � �" d d l m } | j d � Wd QXd S)NrF )�sepz>explicit relative import triggered an implicit absolute import)r* r+ r@ r r7 )r' r r r r �#test_absolute_import_without_future[ s z7RelativeImportTests.test_absolute_import_without_futureN)r^ r� r� r) r( r� r� r r r r r r� 3 s
r� c @ s e Z d Z d d � Z d S)� OverridingImportBuiltinTestsc C s� d d l } d d � } | j | � | � t t d d d � � � | j | � d � Wd QXt t � d d d � � � | j | � | � Wd QXd S) Nr c S s d d l } | S)Nr )r@ )r@ r r r r� j s z?OverridingImportBuiltinTests.test_override_builtin.<locals>.foor6 c W s d S)Nr r )rb r r r �<lambda>o s zDOverridingImportBuiltinTests.test_override_builtin.<locals>.<lambda>r c W s d S)Nr r )rb r r r r u s )r@ r8 r �builtinsr r� )r' r@ r� r r r �test_override_builtinf s z2OverridingImportBuiltinTests.test_override_builtinN)r^ r� r� r r r r r r e s r c @ s e Z d Z e j j Z d d � Z d d � Z d d � Z e
d d � � Z e j
e j d k d
� e j e e d � o� e j � d k d
� e
d d � � � � Z e
d d � � Z e
d d � � Z d d � Z e
d d � � Z e
d d � � Z d d � Z d d � Z d S) � PycacheTestsc C s% t t � t d � t | j � d S)Nr )r r r r r" )r' r r r �_clean~ s
zPycacheTests._cleanc C sf t d | _ | j � t | j d � � } t d d | �Wd QXt j j d t j � t
j � d S)Nz.pyr1 z/# This is a test file written by test_import.pyr2 r ) r r" r r3 r4 r r r r@ rA r% r& )r' �fpr r r r( � s
zPycacheTests.setUpc C s t j d =| j � d S)Nr )r r r )r' r r r r) � s
zPycacheTests.tearDownc C s� | j t j j d � � t t � | j t j j d � � | j t j j t j j d d j t | j t
rr d n d � � � � d S)Nr z
{}.{}.py{}rG rH ) � assertFalser@ r r� r6 r r_ r� rw �tag� __debug__)r' r r r �test_import_pyc_path� s
z!PycacheTests.test_import_pyc_pathr� z%test meaningful only on posix systems�geteuidr z=due to varying filesystem permission semantics (issue #11956)c
C sq t d � � t t � Wd QX| j t j j d � � | j t j j t j j d d j t | j
� � � � d S)N� r z {}.{}.pyc) r
r6 r r_ r@ r r� r r� rw r
)r' r r r �test_unwritable_directory� s
z&PycacheTests.test_unwritable_directoryc C ss t t � t j j | j � } | j t j j | � � t j
| j � t t � t j � | j
t t t � d S)N)r6 r r% r� r� r" r_ r@ r r� r! r r& r* r+ )r' r� r r r �test_missing_source� s
z PycacheTests.test_missing_sourcec C s~ t t � t | j � } t j | j � t t � t j � t t � } | j | j
t j j t j
t j j | � � � d S)N)r6 r r r" r@ r! r r% r& r8 r[ r r� rA �relpath)r' r� r� r r r �test_missing_source_legacy� s
z'PycacheTests.test_missing_source_legacyc C sK t t � } t j j t d � } | j | j t j j t j
| � � d S)Nz.py) r6 r r% r� r� r8 �
__cached__r@ r r� rA )r' r� r� r r r �test___cached__� s z PycacheTests.test___cached__c C s~ t t � t | j � } t j | j � t t � t j � t t � } | j | j
t j j t j
t j j | � � � d S)N)r6 r r r" r@ r! r r% r& r8 r r r� rA r )r' r� r� r r r �test___cached___legacy_pyc� s
z'PycacheTests.test___cached___legacy_pycc C s d d � } t j d � | j | � t t j j d d � d � � Wd QXt t j j d d � d � � Wd QXt j � t d � } t j j
t j j d d � � } | j | j t j j t j
| � � t j j
t j j d d � � } | j t j d j t j j t j
| � � d S)Nc S s" t d � t d � t d � d S)N�pep3147z pep3147.foo)r r r r r r �cleanup� s
z5PycacheTests.test_package___cached__.<locals>.cleanupr z __init__.pyr1 zfoo.pyz pep3147.foo)r@ r� �
addCleanupr3 r r� r% r& r6 r� r� r8 r rA r r )r' r r� �init_pyc�foo_pycr r r �test_package___cached__� s
%!z$PycacheTests.test_package___cached__c C sG d d � } t j d � | j | � t t j j d d � d � � Wd QXt t j j d d � d � � Wd QXt j � t d � } t d � t d � t j � t d � } t j
j t j j d d � � } | j | j
t j j t j | � � t j
j t j j d d � � } | j t j d j
t j j t j | � � d S)Nc S s" t d � t d � t d � d S)Nr z pep3147.foo)r r r r r r r � s
z>PycacheTests.test_package___cached___from_pyc.<locals>.cleanupr z __init__.pyr1 zfoo.pyz pep3147.foo)r@ r� r r3 r r� r% r& r6 r r� r� r8 r rA r r )r' r r� r r r r r � test_package___cached___from_pyc� s&
%!z-PycacheTests.test_package___cached___from_pycc C sb t t � t t � t | j d � � } t d d | �Wd QXt t � } | j | j d � d S)Nr9 zx = 5r2 r )r6 r r r3 r" r4 r8 rb )r' r r� r r r � test_recompute_pyc_same_second s
z+PycacheTests.test_recompute_pyc_same_secondN)r^ r� r� r �implementation� cache_tagr
r r( r) r� r r� r� r@ r �skipIfr` r
r r r r r r r r r r r r r y s" $
r c @ s~ e Z d Z d Z e d Z d d � Z d d � Z e j e e
d � pY e
j � d k d
� e j
j d d � � � Z d
S)� TestSymbolicallyLinkedPackageZsamplez-taggedc C s� t j j | j � t j j | j � t j d d � | _ t j | j � | j
t j j | j � t j j | j d � } t j j | � t j
| j | j d d �| j
t j j | j � t j � | j t j j | j � d � d S)Nz __init__.pyZtarget_is_directoryT)r/ ra r �tagged� package_namer r �
orig_sys_pathr@ r� r r� r �symlinkr r% r& r8 �isdir)r' Z init_filer r r r( s
z#TestSymbolicallyLinkedPackage.setUpc C s | j t j d d � <d S)N)r$ r r )r' r r r r) 1 s z&TestSymbolicallyLinkedPackage.tearDown�getwindowsversion� r z Windows Vista or later requiredc C s* d g t j d d � <t j | j � d S)N�.)r r r% r� r# )r' r r r � test_symlinked_dir_importable5 s z;TestSymbolicallyLinkedPackage.test_symlinked_dir_importableN)r( r )r^ r� r� r# r" r( r) r� r� r` r r' r/ ra Zskip_unless_symlinkr* r r r r r! s
r! c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�ImportlibBootstrapTestsc C s t j d } | j | � d S)N�_frozen_importlib)r r r_ )r' r< r r r �test_frozen_importlibH s
z-ImportlibBootstrapTests.test_frozen_importlibc C sv d d l m } t j d } | j | | � | j | j d � | j | j d � | j | j j
d � | j � d S)Nr )�
_bootstrapr, zimportlib._bootstrapr% z
_bootstrap.py) r% r. r r r] r8 r^ r� r_ r[ rn )r' r. r< r r r �"test_frozen_importlib_is_bootstrapL s
z:ImportlibBootstrapTests.test_frozen_importlib_is_bootstrapc C s7 d d l m } t j d } | j | j | j � d S)Nr )� machineryr, )r% r0 r r r] �
FileFinder)r' r0 r< r r r �test_there_can_be_only_oneT s
z2ImportlibBootstrapTests.test_there_can_be_only_oneN)r^ r� r� r- r/ r2 r r r r r+ D s r+ c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �GetSourcefileTestsz�Test importlib._bootstrap._get_sourcefile() as used by the C API.
Because of the peculiarities of the need of this function, the tests are
knowingly whitebox tests.
c
C sO t j d � �: } d | _ t d } t d } | j t | � | � Wd QXd S)Nz!importlib._bootstrap._path_isfileTz.pycz.py)�mock�patch� return_valuer r8 r )r' � _path_isfiler Zexpectr r r �test_get_sourcefileg s
z&GetSourcefileTests.test_get_sourcefilec
C sE t j d � �0 } d | _ t d } | j t | � | � Wd QXd S)Nz!importlib._bootstrap._path_isfileFz.pyc)r4 r5 r6 r r8 r )r' r7 r r r r � test_get_sourcefile_no_sourcep s
z0GetSourcefileTests.test_get_sourcefile_no_sourcec C s$ t d } | j t | � | � d S)Nz.bad_ext)r r8 r )r' r r r r �test_get_sourcefile_bad_extx s
z.GetSourcefileTests.test_get_sourcefile_bad_extN)r^ r� r� �__doc__r8 r9 r: r r r r r3 ] s r3 c @ s� e Z d Z d d � Z d d � Z d d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z e d d � � Z e j e d � d! d" � � Z d# S)$�ImportTracebackTestsc C s: t j t � t j d d � | _ t j j d t � d S)Nr )r@ r� r r r �old_pathr )r' r r r r( � s
zImportTracebackTests.setUpc C s$ | j t j d d � <t t � d S)N)r= r r r r )r' r r r r) � s z ImportTracebackTests.tearDownz.pyc C s\ t j j t | | � } t | d � � } | j | � Wd QX| j t | � t j � | S)Nr1 )
r@ r r� r r3 rJ r r r% r& )r' r< �contentsr; Zfnamer r r r �
create_module� s
z"ImportTracebackTests.create_modulec C s� g } xO | rW | j j } | j } | s; | | d k rK | j | � n | j } q W| j t | � t | � | � x- t | | � D] \ } } | j | | � q� Wd S)NrF rI ) �tb_frame�f_coder� rN �tb_nextr8 �len�ziprh )r' �tb�filesZ
deduped_filesr� �fnZpatr r r �assert_traceback� s
z%ImportTracebackTests.assert_tracebackc C se y d d l } Wn. t k
r@ } z | j } WYd d } ~ Xn X| j d � | j | t g � d S)Nr z#ImportError should have been raised)�nonexistent_xyzzyr+ �
__traceback__r7 rH r[ )r' rI r� rE r r r �test_nonexistent_module� s
z,ImportTracebackTests.test_nonexistent_modulec C sx | j d d � y d d l } Wn. t k
rP } z | j } WYd d } ~ Xn X| j d � | j | t d g � d S)Nr� zimport nonexistent_xyzzyr z#ImportError should have been raisedzfoo.py)r? r� r+ rJ r7 rH r[ )r' r� r� rE r r r � test_nonexistent_module_nested� s
z3ImportTracebackTests.test_nonexistent_module_nestedc C sx | j d d � y d d l } Wn. t k
rP } z | j } WYd d } ~ Xn X| j d � | j | t d g � d S)Nr� z1/0r z)ZeroDivisionError should have been raisedzfoo.py)r? r� rY rJ r7 rH r[ )r' r� r� rE r r r �test_exec_failure� s
z&ImportTracebackTests.test_exec_failurec C s� | j d d � | j d d � y d d l } Wn. t k
r` } z | j } WYd d } ~ Xn X| j d � | j | t d d g � d S) Nr� z
import bar�barz1/0r z)ZeroDivisionError should have been raisedzfoo.pyzbar.py)r? r� rY rJ r7 rH r[ )r' r� r� rE r r r �test_exec_failure_nested� s
z-ImportTracebackTests.test_exec_failure_nestedc C su | j d d � y d d l } Wn. t k
rP } z | j } WYd d } ~ Xn X| j d � | j | t g � d S)Nr� zinvalid syntax is invalidr z#SyntaxError should have been raised)r? r� � SyntaxErrorrJ r7 rH r[ )r' r� r� rE r r r �test_syntax_error� s
z&ImportTracebackTests.test_syntax_errorc C s� d } | j t | � t j j t | � } t j | � t j j | d � } t | d � � } | j | � Wd QXt j j | d � } t | d � � } | j | � Wd QXt j
� | | f S)N� _parent_fooz __init__.pyr1 zbar.py) r r r@ r r� r r� r3 rJ r% r& )r' �parentZchildZpkg_nameZpkg_path� init_pathr �bar_pathr r r �_setup_broken_package� s
z*ImportTracebackTests._setup_broken_packagec C s� | j d d � \ } } y d d l } Wn. t k
rX } z | j } WYd d } ~ Xn X| j d � | j | t | g � d S)Nr z1/0r z)ZeroDivisionError should have been raised)rV �_parent_foo.barrY rJ r7 rH r[ )r' rT rU rR r� rE r r r �test_broken_submodule� s
z*ImportTracebackTests.test_broken_submodulec C s� | j d d � \ } } y d d l m } Wn. t k
r\ } z | j } WYd d } ~ Xn X| j d � | j | t | g � d S)Nr z1/0r )rN z#ImportError should have been raised)rV rR rN rY rJ r7 rH r[ )r' rT rU rN r� rE r r r �test_broken_from� s
z%ImportTracebackTests.test_broken_fromc C s� | j d d � \ } } y d d l } Wn. t k
rX } z | j } WYd d } ~ Xn X| j d � | j | t | g � d S)Nz1/0r r z)ZeroDivisionError should have been raised)rV rW rY rJ r7 rH r[ )r' rT rU rR r� rE r r r �test_broken_parent� s
z'ImportTracebackTests.test_broken_parentc C s� | j d d � \ } } y d d l m } Wn. t k
r\ } z | j } WYd d } ~ Xn X| j d � | j | t | g � d S)Nz1/0r r )rN z)ZeroDivisionError should have been raised)rV rR rN rY rJ r7 rH r[ )r' rT rU rN r� rE r r r �test_broken_parent_from� s
z,ImportTracebackTests.test_broken_parent_fromc C s� | j d d � t j d } d t | j � k rA | j j } n d } z� d d � } | | j _ y d d l } Wn. t k
r� } z | j } WYd d } ~ Xn X| j d � | j
| t d t g � Wd | d k r� | j ` n | | j _ Xd S)
Nr� r r, � load_modulec W s d d d S)NrF r r )r� r r r � exec_module s z9ImportTracebackTests.test_import_bug.<locals>.exec_moduler z)ZeroDivisionError should have been raisedz<frozen importlib) r? r r �vars� SourceLoaderr] r� rY rJ r7 rH r[ )r' r% Zold_exec_moduler] r� r� rE r r r �test_import_bug s"
z$ImportTracebackTests.test_import_bugzneed TESTFN_UNENCODABLEc C sS t j d t d � } | j t | � | d d � } t j d d | d d �d S) Nr �passrX z-czmod = __import__(%a)Z
__isolatedF���)r r r r r r| )r' Zpynamer r r r �test_unencodable_filename$ s
z.ImportTracebackTests.test_unencodable_filenameN)r^ r� r� r( r) r? rH rK rL rM rO rQ rV rX rY rZ r[ r r` r� r� r rc r r r r r< s
r< �__main__):r% �importlib.utilZimportlib._bootstrapr r r� r@ rB rL r5 r� r r� Z
unittest.mockr4 rz r� Zshutil�
contextlibr\ r/ r r r r r r r r
r r r
r r r r r r r �dont_write_bytecoder� r �contextmanagerr# ZTestCaser$ r� r� r� r� r r r! r+ r3 r< r^ �mainr r r r �<module> sT j
� V^52�.!�