Your IP : 216.73.216.213
3
� f̱ � @ s. d dl Z d dlZ d dlmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dljZd dlZd dlZd dlZd dlZd dlZd dlmZmZmZmZmZm 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 dlm)Z) d dl*m+Z+ ej,e j-d�Z.dd� Z/ej0d%d
d ��Z1G d d
� d
ej2�Z3e.G dd� dej2��Z4G dd� dej2�Z5G dd� dej2�Z6G dd� dej2�Z7G dd� dej2�Z8G dd� dej2�Z9G dd� dej2�Z:e%G d d � d ej2��Z;e%G d d � d ej2��Z<G d d!� d!ej2�Z=G d"d#� d#ej2�Z>e?d$k�r*ej@� dS )&� 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�temp_dir�
DirsOnSysPath)�
script_helper)�uncachez*test meaningful only when writing bytecodec C s: x,| d | d | d | d fD ] }t |� q W td� d S )Nz.pyz.pycz.pywz $py.class� __pycache__)r r )�name�f� r �>/opt/alt/python36/lib64/python3.6/test/test_import/__init__.py� remove_files$ s r � c c s� | pd} t � �t}tj|| |�}tjj| d �}z(tjjd|� | |fV tjj|� W d |d k rj|tj| < n| tjkr|tj| = X W d Q R X d S )NZspamr ) r r � make_script�sys�modules�pop�path�insert�remove)r �sourceZtempdirr" Z
old_moduler r r �_ready_to_import- s
r& c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
d1dd�Z dd� Z dd� Z
dd� Zedd� �Zd d � Zd d � Zed d!� �Zd"d#� Zed$d%� �Zd&d'� Zd(d)� Zd*d+� Zed,d-� �Zd.d/� Zd0S )2� ImportTestsc C s t t� tj� d S )N)r r � importlib�invalidate_caches)�selfr r r �setUpD s zImportTests.setUpc C s t t� d S )N)r r )r* r r r �tearDownH s zImportTests.tearDownc C s" | j t�� dd l}W d Q R X d S )Nr )� assertRaises�ModuleNotFoundError�(something_that_should_not_exist_anywhere)r* r/ r r r �&test_import_raises_ModuleNotFoundErrorK s z2ImportTests.test_import_raises_ModuleNotFoundErrorc C s& | j t�� ddlm} W d Q R X d S )Nr )�blah)r- r. r/ r1 )r* r1 r r r �:test_from_import_missing_module_raises_ModuleNotFoundErrorO s zFImportTests.test_from_import_missing_module_raises_ModuleNotFoundErrorc C s& | j t�� ddlm} W d Q R X d S )Nr )r/ )r- � ImportErrorr( r/ )r* r/ r r r �0test_from_import_missing_attr_raises_ImportErrorS s z<ImportTests.test_from_import_missing_attr_raises_ImportErrorc C s" | j t�� dd l}W d Q R X d S )Nr )r- r3 �RAnDoM)r* r5 r r r �test_case_sensitivityW s z!ImportTests.test_case_sensitivityc C s ddl m} d S )Nr )� double_const)�testr7 )r* r7 r r r �test_double_const] s z ImportTests.test_double_constc sZ � fdd�}t jjdtj� z.|d� t jjd�rHxd D ] }||� q8W W d t jd= X d S )
Nc s t | }trt d }nt d }t|d��F}td| d|d� tjd�}tjd�}td||d� td ||d� W d Q R X t tjkr�tjt = tj � zny t
t �}W n4 t k
r� } z� j d
| |f � W Y d d }~X nX � j
|j|d | � � j
|j|d | � W d tt � t|� t|� X d S ) Nz $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__r3 �fail� assertEqual�a�br r )�extr% Zpycr rB rC �mod�err)r* r r �test_with_extensionc s4
$
z4ImportTests.test_import.<locals>.test_with_extensionr z.py�win�.PY�.Py�.pY�.pyw�.PYW�.pYw)rI rJ rK rL rM rN )r r" r# �os�curdir�platform�
startswith)r* rG rD r )r* r � test_importb s "
zImportTests.test_import�longlistc # C s� |d }t |d��6}|jd� xtd�D ]}|jd� q(W |jd� W d Q R X ztj|� W d t|� X tjjd� t j
� i }z t |� t d| d |� W d tjd = t|d
� t|d � |j
� y tj|= W n tk
r� Y nX X d S )
Nz.pyr: zd = [
i� z"",
�]r zimport � �c�o���)r< �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��}td|d� W d Q R X tjjdtj� tj � t tj
krTtj
t = z4x.d
D ]&}| j t t
t � | jt tj
d | � q\W W d tjd= tt � X d S ) Nz.pyr: za = 1/0)r; r rV � � z'damaged module in sys.modules on %i try)rV rg rh )r r< r= r r" r# rO rP r( r) r r- �ZeroDivisionErrorr? � assertNotInr )r* r% r rd r r r �test_failing_import_sticks� s
z&ImportTests.test_failing_import_sticksc C sR dd l }dd l}| j|||j� | jt|jd�� dd lj}| j||j|j� d S )Nr �__file__)r8 � test.support�assertIs�__name__�
assertTrue�hasattr�support)r* �xr8 �yr r r �test_import_name_binding� s
z$ImportTests.test_import_name_bindingc C s t tj d }t|d��}|jd� W d Q R X tjjdtj� z�t t �}| j
t tj � | j |j
dd� | j |jdd� tt � t|d��}|jd� W d Q R X | jttj|� tj jt �}| j|d � | j |j
d
d� | j |jdd� W d tjd= tt � tt � X d S ) N�pyr: z
a = 1
b=2
r rV z!module has wrong attribute valuesrg za = 10
b=20//0
z$expected module to be in sys.modules�
)r rO �extsepr< rZ r r" r# rP r? �assertInr rA rB rC r r- ri r( �reload�getZassertIsNotNoner )r* r% r rE r r r �test_failing_reload� s( z ImportTests.test_failing_reloadc C s� t d }t|d��}|jd� W d Q R X tjjdtj� zftt �}| j |j
j d�� tj |� tj
t = t|� tj� tt �}tjj|j
�\}}| j|d� W d tjd= tt � t tj
kr�tj
t = X d S )Nz.pyr: z test = None
r z.pyc)r r< rZ r r" r# rO rP r? rp rl �endswithr$ r r r( r) �splitextrA r )r* r% r rE �baserD r r r �test_file_to_source� s$
z ImportTests.test_file_to_sourcec C sj t jjt�}tj� }y|j|� W n$ tk
rF | jdj |�� Y nX | j
t ��}t |� W d Q R X d S )Nzpath is not encodable to {})
rO r" �abspathr r �getfilesystemencoding�encode�UnicodeEncodeError�skipTest�formatr- r3 r? )r* r" �encodingrW r r r �test_import_by_filename s z#ImportTests.test_import_by_filenamec C s" t jdttjd��}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 jjdtj� z�td }tjj|�}t |d��}W d Q R X ytj
|d� W nn t k
rj | j d� Y nR t
k
r� } z6|jttdd �ttd d �fkr�� | j d
j|�� W Y d d }~X nX tt� tj|� W d t jd= tt� X d S )Nr z.pyr: rg �! � z-cannot set modification time to large integerZ EOVERFLOWZEINVALz2cannot set modification time to large integer ({})l � �� l � �� )r� r� )r r" r# rO rP r r( �util�cache_from_sourcer< �utime�
OverflowErrorr� �OSError�errno�getattrr� r? �statr )r* r% Zcompiledr �er r r �test_timestamp_overflow( s&
"z#ImportTests.test_timestamp_overflowc C s6 yt ddgd� W n tk
r0 | jd� Y nX d S )NZhttpr1 )�fromlistz fromlist must allow bogus names)r? r3 r@ )r* r r r �test_bogus_fromlistB s z ImportTests.test_bogus_fromlistc C s. ddg}t j|� }|j� \}}| jd|� d S )Nz-cz&del __builtins__.__import__; import oss ImportError)r Z spawn_pythonZ communicatery )r* �args�popen�stdout�stderrr r r �test_delete_builtins_importH s
z'ImportTests.test_delete_builtins_importc C s( | j td�� ddlm} W d Q R X d S )Nz^No module named 'bogus'r )�foo)�assertRaisesRegexr3 �bogusr� )r* r� r r r �/test_from_import_message_for_nonexistent_moduleO s z;ImportTests.test_from_import_message_for_nonexistent_modulec C s( | j td�� ddlm} W d Q R X d S )Nz^cannot import name 'bogus'r )r� )r� r3 �rer� )r* r� r r r �,test_from_import_message_for_existing_moduleS s z8ImportTests.test_from_import_message_for_existing_modulec C sP G dd� d�}d}| j t|� |� tj|< | jt�� ddlm} W d Q R X d S )Nc @ s e Zd Zdd� ZdS )zIImportTests.test_from_import_AttributeError.<locals>.AlwaysAttributeErrorc S s t �d S )N)�AttributeError)r* �_r r r � __getattr__[ s zUImportTests.test_from_import_AttributeError.<locals>.AlwaysAttributeError.__getattr__N)ro �
__module__� __qualname__r� r r r r �AlwaysAttributeErrorZ s r� � test_from_import_AttributeErrorr )�does_not_exist)�
addCleanupr r r r- r3 r� r� )r* r� � module_namer� r r r r� W s z+ImportTests.test_from_import_AttributeErrorc C sZ t tdd ��D | jt�� ddlm} W d Q R X | jt�� tj W d Q R X W d Q R X d S )Nro r )r� )r rO r- r3 r� r� )r* r� r r r �test_issue31492d s
zImportTests.test_issue31492c s� t jjdtjjtjjt�d�� z�d �� �fdd��x�td�D ]x}tj � � �fdd�td�D �}z*t
j j |� j
�� tjd� W d Q R X W d t jjd d � t jjd
d � X �d k r>��q>W W d t jd= X d S ) Nr �datac s@ � j � y dd l} W n& tk
r: } z
|�W Y d d }~X nX d S )Nr )�wait�package�
BaseException)r� r� )�event�excr r �runt s
z)ImportTests.test_concurrency.<locals>.runrw c s g | ]}t j� d ��qS ))�target)� threadingZThread)�.0rs )r� r r �
<listcomp>~ s z0ImportTests.test_concurrency.<locals>.<listcomp>rg r� zpackage.submodule)r r" r# rO �join�dirnamerl r[ r� ZEventr8 rr Z
start_threads�set�timeZsleepr r! )r* rd Zthreadsr )r� r� r� r �test_concurrencyp s z ImportTests.test_concurrencyN)rT ) ro r� r� r+ r, r0 r2 r4 r6 r9 rS rf rk ru r| �skip_if_dont_write_bytecoder� r� r� r� r� r r� r� r� r� r� r� r r r r r' B s. ,
( &
r' c @ sb e Zd Zejejdkd�dd� �Zejejdkd�dd� �Zejejdkd�dd� �Z d d
� Z
d S ) �FilePermissionTests�posixz%test meaningful only on posix systemsc C s� d}t |��P t� �>\}}tjj|�}t|�}tjj|�sD| j d� tj
|�}W d Q R X W d Q R X | j t t
j
|j��t d| @ �� d S )N� z4__import__ did not result in creation of a .pyc filei� )r
r& r( r� r� r? rO r" �existsr@ r� rA �oct�S_IMODE�st_mode)r* �maskr r" � cached_pathrb � stat_infor r r �test_creation_mode� s
z&FilePermissionTests.test_creation_modec C s� d}t d��\ t� �J\}}tjj|�}tj||� t|� tjj |�sP| j
d� tj |�}W d Q R X W d Q R X | j t
t j|j��t
|�� d S )Ni� r� z4__import__ did not result in creation of a .pyc file)r
r& r( r� r� rO �chmodr? r" r� r@ r� rA 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��\ t� �J\}}tjj|�}tj||� t|� tjj |�sP| j
d� tj |�}W d Q R X W d Q R X |dB }| j t
t j|j��t
|�� d S )N� r� z4__import__ did not result in creation of a .pyc file� )r
r& r( r� r� rO r� r? r" r� r@ r� rA r� r� r� )r* r� r r" r� r� Zexpectedr r r �test_cached_readonly� s
z(FilePermissionTests.test_cached_readonlyc C s t � ��
\}}t|d��}|jd� W d Q R X tj|�}tj||j|jd f� tj|d� t |�}| j
|j d� tj|d� t|d��}|jd� W d Q R X t |� t
j� t |�}| j
|j d� t|� t |� t
j� |d }tjt
jj|�|� t |�}| j
|j d� W d Q R X d S )
Nr: zx = 'original'
i �r� Zoriginali� zx = 'rewritten'
Z rewrittenrW )r& r< rZ rO r� r� �st_atime�st_mtimer� r? rA rs 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) ro r� r� �unittest�
skipUnlessrO r r� r� r� r� r r r r r� � s r� c @ s| e Zd ZdZdZejje�Z ejj
e e�ej d Z e
jje �Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� ZdS )�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
rv c
C sn t jd d � | _t jj| jd �| _tj| j � t
| j d��}|j | j
� W d Q R X t jjd| j � tj� d S )Nr: r )r r" �sys_pathr r! r� � orig_modulerO �mkdir�dir_namer< � file_namerZ �
module_sourcer# r( r) )r* r r r r r+ � s zPycRewritingTests.setUpc C sV | j tjd d �< | jd k r*| jtj| 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||� tj| j S )N)�globalsr? 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� tj| j= | j � }| j|j| j� | j|j| j� | j|j| j� d S )N) r� rA �module_filenamer� �
code_filename�
func_filenamer r r� )r* rE r r r � test_basics
s
z PycRewritingTests.test_basicsc C sL t j| jdd� | j� }| j|j| j� | j|j| j� | j|j| j� d S )Nzanother_module.py)�dfile)r\ r] r� r� rA r� r� r� )r* rE r r r �test_incorrect_code_name s
z*PycRewritingTests.test_incorrect_code_namec C sh d}t j| j|d� tj| j� t| j�}tj� | j� }| j |j
|� | j |j |� | j |j |� d S )Nzanother_module.py)r� )
r\ r] r� rO r$ r r( r) r� rA r� r� r� )r* r� �pyc_filerE r r r �test_module_without_source s
z,PycRewritingTests.test_module_without_sourcec C s� t j| j� t| jd��}|jd�}tj|�}W d Q R X t|j �}t
j j }|j
d�}|||< t|�|j|j|j|j|j|jt|�|j|j|j|j|j|j|j |j �}t| jd�� }|j |� tj ||� W d Q R X | j � }| j |j!j|j� d S )N�rb� rV �wb)"r\ r] r� r< 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_cellvarsrZ �dumprA Zconstant)r* r �header�codeZ constantsZ foreign_code�posrE r r r �test_foreign_code' s&
z#PycRewritingTests.test_foreign_codeN)ro r� r� r� r� rO r" r� r r� r� rx r� r( r� r� r� r+ r, r� r� r� r� r r r r r r� � s
r� c @ sF e Zd ZdZeZdd� Zdd� Zd d
� Ze j
e j d kd �d
d� �Z
dS )�
PathsTestsr8 � testäöüß� testéè�
test°³²c C s t j| j� tjd d � | _d S )N)rO r� r" r �syspath)r* r r r r+ B s zPathsTests.setUpc C s t | j� | jtjd d �< d S )N)r r" r r )r* r r r r, F s
zPathsTests.tearDownc
C s` t tjj| jd�d��}|jd� W d Q R X tjj| jd � td�}| j|j d� t
d� d S )Nztest_trailing_slash.pyr: z testdata = 'test_trailing_slash'�/�test_trailing_slash) r< rO r" r� rZ r r^ r? rA �testdatar )r* r rE r r r r K s z PathsTests.test_trailing_slashZwin32zWindows-specificc & C sR t tjj| jd�d��}|jd� W d Q R X tj� tjj| j�}dd l}|j � }|d }d||f }||dd � 7 }ytj
|� W nJ t k
r� } z.|j t j
t jt jfkr�| jd|f � � W Y d d }~X nX tjjd|� y td�}W n6 tk
�r } z| jd ||f � W Y d d }~X nX | j|jd� | j|jj|�|j� td� d S )
Nztest_unc_path.pyr: ztestdata = 'test_unc_path'r z\\%s\%s$rg z%cannot access administrative share %rZ
test_unc_pathz,could not import 'test_unc_path' from %r: %r) r< rO r" r� rZ r( r) r� �socketZ gethostname�listdirr� r� ZEPERMZEACCES�ENOENTr� r r# r? r3 r@ rA r rp rl rR r ) r* r r" r ZhnZdriveZuncr� rE r r r �
test_UNC_pathT s0 zPathsTests.test_UNC_pathN)r8 r r r )ro r� r� ZSAMPLESr r" r+ r, r r� r� r rQ r r r r r r = s r c @ s@ e Zd Zdd� ZeZdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�RelativeImportTestsc C s t d� d S )Nztest.relimport)r )r* r r r r, t s z RelativeImportTests.tearDownc C s ddl m} | jt|d�� d S )Nrg )� relimportr )r r rp rq )r* r r r r �test_relimport_starx s z'RelativeImportTests.test_relimport_starc s| � fdd�}t ddd�� |� t ddd�� |� t ddd�� | jt|� t ddd�� | jt|� t t� d�� | jt|� d S ) Nc s t d� � d S )Nzfrom . import relimport)r_ r )r� r r �check_relative� s z:RelativeImportTests.test_issue3221.<locals>.check_relativer8 ztest.notarealmodule)� __package__ro znotarealpkg.notarealmoduler� )r )�dictr- r. �object� TypeError)r* r r )r� r �test_issue3221} s z"RelativeImportTests.test_issue3221c C s0 | j t�� ddlm} | jd� W d Q R X d S )NrV )�sepz>explicit relative import triggered an implicit absolute import)r- r3 rO r$ r@ )r* r$ r r r �#test_absolute_import_without_future� s z7RelativeImportTests.test_absolute_import_without_futurec C s� t jjt jjt�dd�}tdd��T t|��@ | jt�� dd l }W d Q R X | j
dt j � | j
dt j � W d Q R X W d Q R X d S )Nr� �package2�
submodule1�
submodule2r )
rO r" r� r� rl r r r- r3 r' rj r r )r* r" r' r r r � test_import_from_non_package� s z0RelativeImportTests.test_import_from_non_packagec C sR t ddd��< ttjjtjjt�d��� dd l}|jj W d Q R X W d Q R X d S )Nr& zpackage2.submodule1zpackage2.submodule2r� r )
r r rO r" r� r� rl Zpackage2.submodule1r' r( )r* r& r r r �!test_import_from_unloaded_package� s z5RelativeImportTests.test_import_from_unloaded_packageN)
ro r� r� r, r+ r r# r% r) r* r r r r r r s r c @ s e Zd Zdd� ZdS )� OverridingImportBuiltinTestsc C sx dd l }dd� }| j|� |� ttddd� �� | j|� d� W d Q R X tt� ddd� �� | j|� |� W d Q R X d S ) Nr c S s dd l } | S )Nr )rO )rO r r r r� � s z?OverridingImportBuiltinTests.test_override_builtin.<locals>.foor? c W s dS )Nr� r )rs r r r �<lambda>� s zDOverridingImportBuiltinTests.test_override_builtin.<locals>.<lambda>r� c W s dS )Nr� r )rs r r r r, � s )rO rA r �builtinsr r� )r* rO r� r r r �test_override_builtin� s z2OverridingImportBuiltinTests.test_override_builtinN)ro r� r� r. r r r r r+ � s r+ c @ s� e Zd Zdd� Zdd� Zdd� Zedd� �Zej e
j d kd
�ej e
e
d �oTe
j� d kd
�edd� ���Zedd� �Zedd� �Zdd� Zedd� �Zedd� �Zdd� Zd d � Zd S ) � PycacheTestsc C s t t� td� t| j� d S )Nr )r r r r r% )r* r r r �_clean� s zPycacheTests._cleanc C sR t d | _| j� t| jd��}td|d� W d Q R X tjjdtj � t
j � d S )Nz.pyr: z/# This is a test file written by test_import.py)r; r ) r r% r0 r< r= r r" r# rO rP r( r) )r* �fpr r r r+ � s
zPycacheTests.setUpc C s, t jd tjkstd��t jd= | j� d S )Nr zUnexpected sys.path[0])r r" rO rP �AssertionErrorr0 )r* r r r r, � s zPycacheTests.tearDownc C sZ | j tjjd�� tt� | jtjjd�� tjj | j
�}| jtjj|�dj |t�� d S )Nr z*bytecode file {!r} for {!r} does not exist) � assertFalserO r" r� r? r rp r( r� r� r% r� )r* �pyc_pathr 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 s\ t d�� tt� W d Q R X | jtjjd�� tjj | j
�}| j tjj|�dj |t�� d S )N� r z"bytecode file {!r} for {!r} exists)
r
r? r rp rO r" r� r( r� r� r% r3 r� )r* r4 r r r �test_unwritable_directory� s
z&PycacheTests.test_unwritable_directoryc C sV t t� tjj| j�}| jtjj |�� tj
| j� t t� tj � | j
tt t� d S )N)r? r r( r� r� r% rp rO r" r� r$ r r) r- r3 )r* r� r r r �test_missing_source� s z PycacheTests.test_missing_sourcec C sn t t� t| j�}tj| j� tt� tj� t t�}z&| j |j
tj j tj
tj j|��� W d tj|� X d S )N)r? r r r% rO r$ r r( r) rA rl r" r� rP �relpath)r* r� r� r r r �test_missing_source_legacy s
z'PycacheTests.test_missing_source_legacyc C s6 t t�}tjjtd �}| j|jtjj tj
|�� d S )Nz.py) r? r r( r� r� rA �
__cached__rO r" r� rP )r* r� r� r r r �test___cached__ s z PycacheTests.test___cached__c C s\ t t� t| j�}tj| j� tt� tj� t t�}| j |j
tj j tj
tj j|��� d S )N)r? r r r% rO r$ r r( r) rA r< r" r� rP r: )r* r� r� r r r �test___cached___legacy_pyc s
z'PycacheTests.test___cached___legacy_pycc C s� dd� }t jd� | j|� tt jjdd�d�� W d Q R X tt jjdd�d�� W d Q R X tj� td�}tj j
t jjdd��}| j |j t jjt j
|�� tj j
t jjdd��}| j tjd j t jjt j
|�� d S )Nc S s t d� td� td� d S )N�pep3147z pep3147.foo)r r r r r r �cleanup- s z5PycacheTests.test_package___cached__.<locals>.cleanupr? z __init__.pyr: zfoo.pyz pep3147.foo)rO r� r� r< r" r� r( r) r? r� r� rA r< rP r r )r* r@ r� �init_pyc�foo_pycr r r �test_package___cached__* s
z$PycacheTests.test_package___cached__c C s� dd� }t jd� | j|� tt jjdd�d�� W d Q R X tt jjdd�d�� W d Q R X tj� td�}t d� t d� tj� td�}tj
j t jjdd��}| j |j
t jjt j|�� tj
j t jjdd��}| j tjd j
t jjt j|�� d S )Nc S s t d� td� td� d S )Nr? z pep3147.foo)r r r r r r r@ D s z>PycacheTests.test_package___cached___from_pyc.<locals>.cleanupr? z __init__.pyr: zfoo.pyz pep3147.foo)rO r� r� r< r" r� r( r) r? r r� r� rA r< rP r r )r* r@ r� rA rB r r r � test_package___cached___from_pycA s&
z-PycacheTests.test_package___cached___from_pycc C sN t t� tt� t| jd��}td|d� W d Q R X t t�}| j|jd� d S )NrB zx = 5)r; r� )r? r r r< r% r= rA rs )r* r1 r� r r r � test_recompute_pyc_same_second\ s z+PycacheTests.test_recompute_pyc_same_secondN)ro r� r� r0 r+ r, r� r5 r� r� rO r �skipIfrq r6 r8 r9 r; r= r> rC rD rE r r r r r/ � s
r/ c @ sX e Zd ZdZed Zdd� Zdd� Zeje e
d� p>e
j � dkd
�e j
jd d � ��Zd
S )� TestSymbolicallyLinkedPackageZsamplez-taggedc C s� t jj| j� t jj| j� tjd d � | _tj | j� | j
t jj| j� tjj | jd�}t jj |� tjj
|�svt�tj| j| jdd� | j
t jj| j� tj� | jtjj| j�d� tjjtjj | jd��s�t�d S )Nz __init__.pyT)Ztarget_is_directory)r8 rr r �tagged� package_namer r" �
orig_sys_pathrO r� r� r� r r� r2 �symlinkr r( r) rA �isdir�isfile)r* Z init_filer r r r+ k s z#TestSymbolicallyLinkedPackage.setUpc C s | j tjd d �< d S )N)rJ r r" )r* r r r r, � s z&TestSymbolicallyLinkedPackage.tearDown�getwindowsversion� r z Windows Vista or later requiredc C sN dgt jd d �< tjj| j�s"t�tjjtjj| jd��s>t�tj| j� d S )N�.z __init__.py) r r" rO r� rI r2 r� r( r� )r* r r r � test_symlinked_dir_importable� s z;TestSymbolicallyLinkedPackage.test_symlinked_dir_importableN)rO r )ro r� r� rI rH r+ r, r� r� rq r rN r8 rr Zskip_unless_symlinkrQ r r r r rG g s
rG c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ImportlibBootstrapTestsc C s t jd }| j|� d S )N�_frozen_importlib)r r rp )r* rE r r r �test_frozen_importlib� s
z-ImportlibBootstrapTests.test_frozen_importlibc C sX ddl m} tjd }| j||� | j|jd� | j|jd� | j|j j
d�|j � d S )Nr )�
_bootstraprS zimportlib._bootstrapr( z
_bootstrap.py) r( rU r r rn rA ro r rp rl r} )r* rU rE r r r �"test_frozen_importlib_is_bootstrap� s
z:ImportlibBootstrapTests.test_frozen_importlib_is_bootstrapc C sX ddl m} tjd }| j||� | j|jd� | j|jd� | j|j j
d�|j � d S )Nr )�_bootstrap_external�_frozen_importlib_externalz importlib._bootstrap_externalr( z_bootstrap_external.py) r( rW r r rn rA ro r rp rl r} )r* rW rE r r r �4test_frozen_importlib_external_is_bootstrap_external� s
zLImportlibBootstrapTests.test_frozen_importlib_external_is_bootstrap_externalc C s* ddl m} tjd }| j|j|j� d S )Nr )� machineryrS )r( rZ r r rn �
ModuleSpec)r* rZ rE r r r �test_there_can_be_only_one� s
z2ImportlibBootstrapTests.test_there_can_be_only_oneN)ro r� r� rT rV rY r\ r r r r rR � s rR c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �GetSourcefileTestsz�Test importlib._bootstrap_external._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 s@ t jd��,}d|_td }td }| jt|�|� W d Q R X d S )Nz*importlib._bootstrap_external._path_isfileTz.pycz.py)�mock�patch� return_valuer rA r )r* � _path_isfiler" Zexpectr r r �test_get_sourcefile� s
z&GetSourcefileTests.test_get_sourcefilec
C s8 t jd��$}d|_td }| jt|�|� W d Q R X d S )Nz*importlib._bootstrap_external._path_isfileFz.pyc)r^ r_ r` r rA r )r* ra r" r r r � test_get_sourcefile_no_source� s z0GetSourcefileTests.test_get_sourcefile_no_sourcec C s t d }| jt|�|� d S )Nz.bad_ext)r rA r )r* r" r r r �test_get_sourcefile_bad_ext� s z.GetSourcefileTests.test_get_sourcefile_bad_extN)ro r� r� �__doc__rb rc rd r r r r r] � s r] c @ s� e Zd Zdd� Zdd� Zd$dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zed d � �Zejed �d!d"� �Zd#S )%�ImportTracebackTestsc C s, t jt� tjd d � | _tjjdt� d S )Nr )rO r� r r r" �old_pathr# )r* r r r r+ � s
zImportTracebackTests.setUpc C s | j tjd d �< tt� d S )N)rg r r" r r )r* r r r r, � s z ImportTracebackTests.tearDown�.pyc
C sJ t jjt|| �}t|d��}|j|� W d Q R X | jt|� tj � |S )Nr: )
rO r" r� r r< rZ r� r r( r) )r* rE �contentsrD Zfnamer r r r �
create_module� s z"ImportTracebackTests.create_modulec C s| g }x8|r<|j j}|j}| s*||d kr4|j|� |j}qW | jt|�t|�|� x"t||�D ]\}}| j||� q`W d S )NrV rY ) �tb_frame�f_coder r^ �tb_nextrA �len�zipry )r* �tb�filesZ
deduped_filesr �fnZpatr r r �assert_traceback� s
z%ImportTracebackTests.assert_tracebackc C sR y dd l }W n( tk
r4 } z |j}W Y d d }~X n X | jd� | j|tg� d S )Nr z#ImportError should have been raised)�nonexistent_xyzzyr3 �
__traceback__r@ rs rl )r* rt r� rp r r r �test_nonexistent_module� s
z,ImportTracebackTests.test_nonexistent_modulec C s` | j dd� y dd l}W n( tk
r@ } z |j}W Y d d }~X n X | jd� | j|tdg� d S )Nr� zimport nonexistent_xyzzyr z#ImportError should have been raisedzfoo.py)rj r� r3 ru r@ rs rl )r* r� r� rp r r r � test_nonexistent_module_nested s
z3ImportTracebackTests.test_nonexistent_module_nestedc C s` | j dd� y dd l}W n( tk
r@ } z |j}W Y d d }~X n X | jd� | j|tdg� d S )Nr� z1/0r z)ZeroDivisionError should have been raisedzfoo.py)rj r� ri ru r@ rs rl )r* r� r� rp r r r �test_exec_failure s
z&ImportTracebackTests.test_exec_failurec C sn | j dd� | j dd� y dd l}W n( tk
rL } z |j}W Y d d }~X n X | jd� | j|tddg� d S ) Nr� z
import bar�barz1/0r z)ZeroDivisionError should have been raisedzfoo.pyzbar.py)rj r� ri ru r@ rs rl )r* r� r� rp r r r �test_exec_failure_nested s
z-ImportTracebackTests.test_exec_failure_nestedc C s^ | j dd� y dd l}W n( tk
r@ } z |j}W Y d d }~X n X | jd� | j|tg� d S )Nr� zinvalid syntax is invalidr z#SyntaxError should have been raised)rj r� � SyntaxErrorru r@ rs rl )r* r� r� rp r r r �test_syntax_error! s
z&ImportTracebackTests.test_syntax_errorc C s� d}| j t|� tjjt|�}tj|� tjj|d�}t|d��}|j|� W d Q R X tjj|d�}t|d��}|j|� W d Q R X t j
� ||fS )N� _parent_fooz __init__.pyr: zbar.py) r� r rO r" r� r r� r< rZ r( r) )r* �parentZchildZpkg_nameZpkg_path� init_pathr �bar_pathr r r �_setup_broken_package+ s
z*ImportTracebackTests._setup_broken_packagec C sd | j dd�\}}y dd l}W n( tk
rD } z |j}W Y d d }~X n X | jd� | j|t|g� d S )Nr z1/0r z)ZeroDivisionError should have been raised)r� �_parent_foo.barri ru r@ rs rl )r* r r� r} r� rp r r r �test_broken_submodule: s
z*ImportTracebackTests.test_broken_submodulec C sh | j dd�\}}yddlm} W n( tk
rH } z |j}W Y d d }~X n X | jd� | j|t|g� d S )Nr z1/0r )ry z#ImportError should have been raised)r� r} ry ri ru r@ rs rl )r* r r� ry r� rp r r r �test_broken_fromD s
z%ImportTracebackTests.test_broken_fromc C sd | j dd�\}}y dd l}W n( tk
rD } z |j}W Y d d }~X n X | jd� | j|t|g� d S )Nz1/0r r z)ZeroDivisionError should have been raised)r� r� ri ru r@ rs rl )r* r r� r} r� rp r r r �test_broken_parentN s
z'ImportTracebackTests.test_broken_parentc C sh | j dd�\}}yddlm} W n( tk
rH } z |j}W Y d d }~X n X | jd� | j|t|g� d S )Nz1/0r r )ry z)ZeroDivisionError should have been raised)r� r} ry ri ru r@ rs rl )r* r r� ry r� rp r r r �test_broken_parent_fromX s
z,ImportTracebackTests.test_broken_parent_fromc C s� | j dd� tjd }dt|j�kr.|jj}nd }zfdd� }||j_y dd l}W n( tk
rx } z |j}W Y d d }~X n X | j d� | j
|t d t g� W d |d kr�|j`n||j_X d S )
Nr� r rX � load_modulec W s dd d S )NrV r r )r� r r r � exec_modulem s z9ImportTracebackTests.test_import_bug.<locals>.exec_moduler z)ZeroDivisionError should have been raisedz<frozen importlib) rj r r �vars� SourceLoaderr� r� ri ru r@ rs rl )r* r( Zold_exec_moduler� r� r� rp r r r �test_import_bugb s"
z$ImportTracebackTests.test_import_bugzneed TESTFN_UNENCODABLEc C s> t jdtd�}| jt|� |d d� }t jdd| dd� d S ) Nr �passrh z-czmod = __import__(%a)F)Z
__isolated���)r r r r� r r� )r* Zpynamer r r r �test_unencodable_filename} s
z.ImportTracebackTests.test_unencodable_filenameN)rh )ro r� r� r+ r, rj rs rv rw rx rz r| r� r� r� r� r� r r� r� r� r r� r r r r rf � s
rf c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�CircularImportTestszQSee the docstrings of the modules being imported for the purpose of the
test.c C s. x(t tjj� �D ]}|jd�rtj|= qW dS )zSMake sure no modules pre-exist in sys.modules which are being used to
test.z&test.test_import.data.circular_importsN)r� r r �keysrR )r* �keyr r r r, � s
z CircularImportTests.tearDownc C s0 y dd l }W n tk
r* | jd� Y nX d S )Nr z/circular import through relative imports failed)Z,test.test_import.data.circular_imports.basicr3 r@ )r* r8 r r r � test_direct� s z CircularImportTests.test_directc C s0 y dd l }W n tk
r* | jd� Y nX d S )Nr z@relative import in module contributing to circular import failed)Z/test.test_import.data.circular_imports.indirectr3 r@ )r* r8 r r r �
test_indirect� s z!CircularImportTests.test_indirectc C s0 y dd l }W n tk
r* | jd� Y nX d S )Nr z-circular import involving a subpackage failed)Z1test.test_import.data.circular_imports.subpackager3 r@ )r* r8 r r r �test_subpackage� s z#CircularImportTests.test_subpackagec C sT ydd l jjjj}W n tk
r2 | jd� Y nX ddlm} | j |j|j� d S )Nr z9circular import with rebinding of module attribute failed)r� )
Z0test.test_import.data.circular_imports.rebindingrS r� Zcircular_imports� rebindingr3 r@ Z-test.test_import.data.circular_imports.subpkgr� rn )r* r� r� r r r �test_rebinding� s z"CircularImportTests.test_rebindingN) ro r� r� re r, r� r� r� r� r r r r r� � s r� �__main__)Nr )Ar( �importlib.utilZ importlib._bootstrap_externalr r- r� rO rQ r\ r> r� r r� r� r� Z
unittest.mockr^ r� r� Zshutil�
contextlibrm r8 r r r r r r r r
r r r
r r r r r r r r Ztest.test_importlib.utilr rF �dont_write_bytecoder� r �contextmanagerr& ZTestCaser' r� r� r r r+ r/ rG rR r] rf r� ro �mainr r r r �<module> sb
P K
S^5@ ".
! 1(