File: //usr/local/lib/python3.7/test/test_import/__pycache__/__init__.cpython-37.pyc
B
��g�� � @ 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 dlmZ d dlZd dlZd dlZd dlZd dlmZmZmZmZmZmZm 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* e�+e j,d�Z-dd� Z.ej/d%d
d ��Z0G d d
� d
ej1�Z2e-G dd� dej1��Z3G dd� dej1�Z4G dd� dej1�Z5G dd� dej1�Z6G dd� dej1�Z7G dd� dej1�Z8G dd� dej1�Z9e$G d d � d ej1��Z:e$G d d � d ej1��Z;G d d!� d!ej1�Z<G d"d#� d#ej1�Z=e>d$k�r$e�?� 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 �5/usr/local/lib/python3.7/test/test_import/__init__.py� remove_files# s r � c
c s� | pd} t � �t}t�|| |�}tj�| d �}z(tj�d|� | |fV tj�|� 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 � Ze d
d� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd;dd�Zd d � Zd d � Zd d!� Zd"d#� Zed$d%� �Zd&d'� Zd(d)� Zed*d+� �Zd,d-� Ze d.d/� �Zd0d1� Z d2d3� Z d4d5� Z e d6d7� �Z d8d9� Z d:S )<� ImportTestsc C s t t� t�� d S )N)r r � importlib�invalidate_caches)�selfr r r �setUpC s zImportTests.setUpc C s t t� d S )N)r r )r* r r r �tearDownG s zImportTests.tearDownc C s" | � 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_ModuleNotFoundErrorJ s z2ImportTests.test_import_raises_ModuleNotFoundErrorc C s& | � 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_ModuleNotFoundErrorN s zFImportTests.test_from_import_missing_module_raises_ModuleNotFoundErrorc C s& | � 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_ImportErrorR s z<ImportTests.test_from_import_missing_attr_raises_ImportErrorc C sZ | � t��}ddlm} W d Q R X | �|jjd� | �|jjtj� | � t
|j�d� d S )Nr )� i_dont_exist�osz7cannot import name 'i_dont_exist' from 'os' \(.*os.py\)) r- r3 r6 r5 � assertEqual� exceptionr r" �__file__� assertRegex�str)r* �cmr5 r r r �/test_from_import_missing_attr_has_name_and_pathV s
z;ImportTests.test_from_import_missing_attr_has_name_and_pathc C sb dd l }| �t��}ddl m} W d Q R X | �|jjd� | �|jj|j� | � t
|j�d� d S )Nr )r5 � _testcapizCcannot import name 'i_dont_exist' from '_testcapi' \(.*\.(so|pyd)\)) r> r- r3 r5 r7 r8 r r" r9 r: r; )r* r> r< r5 r r r �2test_from_import_missing_attr_has_name_and_so_path] s z>ImportTests.test_from_import_missing_attr_has_name_and_so_pathc C sD | � t��}ddlm} W d Q R X | �|jjd� | �|jj� d S )Nr )r5 �_warning) r- r3 r@ r5 r7 r8 r Z assertIsNoner" )r* r< r5 r r r �&test_from_import_missing_attr_has_namef s z2ImportTests.test_from_import_missing_attr_has_namec C sF | � t��}ddlm} W d Q R X | �|jjddh� | �|j� d S )Nr )r5 � posixpathZntpath)r- r3 Zos.pathr5 �assertInr8 r �assertIsNotNone)r* r< r5 r r r �/test_from_import_missing_attr_path_is_canonicalm s z;ImportTests.test_from_import_missing_attr_path_is_canonicalc C s" | � t�� dd l}W d Q R X d S )Nr )r- r3 �RAnDoM)r* rF r r r �test_case_sensitivitys s z!ImportTests.test_case_sensitivityc C s ddl m} d S )Nr )� double_const)�testrH )r* rH r r r �test_double_consty s z ImportTests.test_double_constc sZ � fdd�}t j�dtj� z.|d� t j�d�rHxdD ] }||� q8W W d t jd= X d S )Nc
s t | }trt d }nt d }t|d��F}td| d|d� t�d�}t�d�}td||d� td ||d� W d Q R X t tjkr�tjt = t� � zny t
t �}W n4 t k
r� } z� � d
| |f � W d d }~X Y nX � �
|j|d | � � �
|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 �failr7 �a�br r )�extr% Zpycr rR rS �mod�err)r* r r �test_with_extension s4
$
z4ImportTests.test_import.<locals>.test_with_extensionr z.py�win)z.PYz.Pyz.pYz.pywz.PYWz.pYw)r r" r# r6 �curdir�platform�
startswith)r* rW rT r )r* r � test_import~ s "
zImportTests.test_import�longlistc C s� |d }t |d��6}|�d� xtd�D ]}|�d� q(W |�d� W d Q R X zt�|� W d t|� X tj�d� t �
� i }z t |� t d| d |� W d tjd = t|d
� t|d � |�
� y tj|= W n tk
r� Y nX X d S ) Nz.pyrK zd = [
i� z"",
�]r zimport ����c�o)rM �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 tj�dtj� t� � t tj
krTtj
t = z4x.dD ]&}| � t t
t � | �t tj
d| � q\W W d tjd= tt � X d S )Nz.pyrK za = 1/0)rL r )� � � z'damaged module in sys.modules on %i try)r rM rN r r" r# r6 rY r( r) r r- �ZeroDivisionErrorrP � assertNotInr )r* r% r rl r r r �test_failing_import_sticks� s
z&ImportTests.test_failing_import_sticksc C sT dd l }dd l}| �|||j� | �t|jd�� dd lm} | �||j|j� d S )Nr r9 )rI � test.support�assertIs�__name__�
assertTrue�hasattr�support)r* �xrI �yr r r �test_import_name_binding� s z$ImportTests.test_import_name_bindingc C sr zd}W d dd l m m} X d}x&|dk rHdd l m m} |d7 }q$W x"td�D ]}dd l m m} qTW d S )N.r �
ro rp )Ztest.support.script_helperrz r rc )r* r{ rl r r r �test_issue31286� s
zImportTests.test_issue31286c C s t tj d }t|d��}|�d� W d Q R X tj�dtj� z�t t �}| �
t tj � | � |j
dd� | � |jdd� tt � t|d��}|�d� W d Q R X | �ttj|� tj �t �}| �|d � | � |j
d
d� | � |jdd� W d tjd= tt � tt � X d S ) N�pyrK z
a = 1
b=2
r ro z!module has wrong attribute valuesrp za = 10
b=20//0
z$expected module to be in sys.modulesr~ )r r6 �extseprM rb r r" r# rY rP rC r r7 rR rS r r- rr r( �reload�getrD r )r* r% r rU r r r �test_failing_reload s( z ImportTests.test_failing_reloadc C s� t d }t|d��}|�d� W d Q R X tj�dtj� zftt �}| � |j
� d�� t� |� tj
t = t|� t�� tt �}tj�|j
�\}}| �|d� W d tjd= tt � t tj
kr�tj
t = X d S )Nz.pyrK z test = None
r z.pyc)r rM rb r r" r# r6 rY rP rx r9 �endswithr$ r r r( r) �splitextr7 r )r* r% r rU �baserT r r r �test_file_to_source( s$
z ImportTests.test_file_to_sourcec C sj t j�t�}t�� }y|�|� W n$ tk
rF | �d� |�� Y nX | �
t ��}t |� W d Q R X d S )Nzpath is not encodable to {})
r6 r" �abspathr r �getfilesystemencoding�encode�UnicodeEncodeError�skipTest�formatr- r3 rP )r* r" �encodingr` r r r �test_import_by_filename@ s z#ImportTests.test_import_by_filenamec C s" t �dtt�d��}t �|� 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_crashJ s z-ImportTests.test_import_in_del_does_not_crashc
C s� t j�dtj� z�td }tj�|�}t |d��}W d Q R X yt�
|d� W nn t k
rj | � d� Y nR t
k
r� } z4|jttdd �ttdd �fkr�� | � d�|�� W d d }~X Y nX tt� t�|� W d t jd= tt� X d S ) Nr z.pyrK )l �� l �� z-cannot set modification time to large integerZ EOVERFLOWZEINVALz2cannot set modification time to large integer ({}))r r" r# r6 rY r r( �util�cache_from_sourcerM �utime�
OverflowErrorr� �OSError�errno�getattrr� rP �statr )r* r% Zcompiledr �er r r �test_timestamp_overflowU s&
"z#ImportTests.test_timestamp_overflowc C s6 yt ddgd� W n tk
r0 | �d� Y nX d S )NZhttpr1 )�fromlistz fromlist must allow bogus names)rP r3 rQ )r* r r r �test_bogus_fromlisto s z ImportTests.test_bogus_fromlistc C s. ddg}t j|� }|�� \}}| �d|� d S )Nz-cz&del __builtins__.__import__; import oss ImportError)r Z spawn_pythonZ communicaterC )r* �args�popen�stdout�stderrr r r �test_delete_builtins_importu s
z'ImportTests.test_delete_builtins_importc C s( | � 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_module| s z;ImportTests.test_from_import_message_for_nonexistent_modulec C s( | � 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_module� s z8ImportTests.test_from_import_message_for_existing_modulec C sb G dd� d�}d}| � t|� |� tj|< | �t��}ddlm} W d Q R X | �t |j
�d� 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)rw �
__module__� __qualname__r� r r r r �AlwaysAttributeError� s r� � test_from_import_AttributeErrorr )�does_not_existzScannot import name 'does_not_exist' from '<unknown module name>' (unknown location)) �
addCleanupr r r r- r3 r� r� r7 r; r8 )r* r� � module_namer< r� r r r r� � s z+ImportTests.test_from_import_AttributeErrorc
C sZ t tdd ��D | �t�� ddlm} W d Q R X | �t�� tj W d Q R X W d Q R X d S )Nrw r )r� )r r6 r- r3 r� r� )r* r� r r r �test_issue31492� s
zImportTests.test_issue31492c s� t j�dtj�tj�t�d�� z�d �� �fdd��x�td�D ]x}t� � � �fdd�td�D �}z*t
j � |� j
�� t�d� W d Q R X W d t j�d d � t j�d
d � X �d k r>��q>W W d t jd= X d S ) Nr �datac
s@ � � � y dd l} W n& tk
r: } z|�W d d }~X Y nX d S )Nr )�wait�package�
BaseException)r� r� )�event�excr r �run� s
z)ImportTests.test_concurrency.<locals>.runr~ c s g | ]}t j� d ��qS ))�target)� threadingZThread)�.0r{ )r� r r �
<listcomp>� s z0ImportTests.test_concurrency.<locals>.<listcomp>rp r� zpackage.submodule)r r" r# r6 �join�dirnamer9 rc r� ZEventrI rz Z
start_threads�set�timeZsleepr r! )r* rl Zthreadsr )r� r� r� r �test_concurrency� s z ImportTests.test_concurrencyN)r] )!rw r� r� r+ r, r0 r2 r4 r= r r? rA rE rG rJ r\ rn rt r} r r� �skip_if_dont_write_bytecoder� r� r� r� r� r� r� r� r� r� r� r r r r r' A s8 ,
( &
r' c @ sb e Zd Ze�ejdkd�dd� �Ze�ejdkd�dd� �Ze�ejdkd�dd� �Z d d
� Z
d S ) �FilePermissionTests�posixz%test meaningful only on posix systemsc
C s� d}t |��P t� �>\}}tj�|�}t|�}tj�|�sD| � d� t�
|�}W d Q R X W d Q R X | � t t
�
|j��t d| @ �� d S )N� z4__import__ did not result in creation of a .pyc filei� )r
r& r( r� r� rP r6 r" �existsrQ r� r7 �oct�S_IMODE�st_mode)r* �maskr r" � cached_pathrj � stat_infor r r �test_creation_mode� s
z&FilePermissionTests.test_creation_modec
C s� d}t d��\ t� �J\}}tj�|�}t�||� t|� tj� |�sP| �
d� t� |�}W d Q R X W d Q R X | � t
t �|j��t
|�� d S )Ni� r� z4__import__ did not result in creation of a .pyc file)r
r& r( r� r� r6 �chmodrP r" r� rQ r� r7 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\}}tj�|�}t�||� t|� tj� |�sP| �
d� t� |�}W d Q R X W d Q R X |dB }| � t
t �|j��t
|�� d S )N� r� z4__import__ did not result in creation of a .pyc file� )r
r& r( r� r� r6 r� rP r" r� rQ r� r7 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��}|�d� W d Q R X t�|�}t�||j|jd f� t�|d� t |�}| �
|j d� t�|d� t|d��}|�d� W d Q R X t |� t
�� t |�}| �
|j d� t|� t |� t
�� |d }t�t
j�|�|� t |�}| �
|j d� W d Q R X d S )
NrK zx = 'original'
i �r� Zoriginali� zx = 'rewritten'
Z rewrittenr` )r& rM rb r6 r� r� �st_atime�st_mtimer� rP r7 r{ 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) rw r� r� �unittest�
skipUnlessr6 r r� r� r� r� r r r r r� � s r� c @ s| e Zd ZdZdZej�e�Z ej�
e e�ej d Z e
j�e �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
r� c C sn t jd d � | _t j�| jd �| _t�| j � t
| j d��}|� | j
� W d Q R X t j�d| j � t�� d S )NrK r )r r" �sys_pathr r! r� � orig_moduler6 �mkdir�dir_namerM � file_namerb �
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)�globalsrP r� r r )r* �nsr r r �
import_module5 s z PycRewritingTests.import_modulec C s~ | � � }| �|j| j� | �|j| j� | �|j| j� tj| j= | � � }| �|j| j� | �|j| j� | �|j| j� d S )N) r� r7 �module_filenamer� �
code_filename�
func_filenamer r r� )r* rU r r r � test_basics: s
z PycRewritingTests.test_basicsc C sL t j| jdd� | �� }| �|j| j� | �|j| j� | �|j| j� d S )Nzanother_module.py)�dfile)rd re r� r� r7 r� r� r� )r* rU r r r �test_incorrect_code_nameE s
z*PycRewritingTests.test_incorrect_code_namec C sh d}t j| j|d� t�| j� t| j�}t�� | �� }| � |j
|� | � |j |� | � |j |� d S )Nzanother_module.py)r� )
rd re r� r6 r$ r r( r) r� r7 r� r� r� )r* r� �pyc_filerU r r r �test_module_without_sourceL s
z,PycRewritingTests.test_module_without_sourcec C s� t �| j� t| jd��}|�d�}t�|�}W d Q R X t|j �}t
j j }|�
d�}|||< t|�|j|j|j|j|j|jt|�|j|j|j|j|j|j|j |j �}t| jd�� }|� |� t� ||� W d Q R X | � � }| � |j!j|j� d S )N�rb� ro �wb)"rd re r� rM 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_cellvarsrb �dumpr7 Zconstant)r* r �header�codeZ constantsZ foreign_code�posrU r r r �test_foreign_codeW s&
z#PycRewritingTests.test_foreign_codeN)rw r� r� r� r� r6 r" r� r r� r� r� 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� Zdd� Ze �
e j dkd �d
d � �Z
d S )
�
PathsTests)rI u testäöüßu testéèu
test°³²c C s t �| j� tjd d � | _d S )N)r6 r� r" r �syspath)r* r r r r+ r s zPathsTests.setUpc C s t | j� | jtjd d �< d S )N)r r" r r )r* r r r r, v s
zPathsTests.tearDownc C s` t tj�| jd�d��}|�d� W d Q R X tj�| jd � td�}| �|j d� t
d� d S )Nztest_trailing_slash.pyrK z testdata = 'test_trailing_slash'�/�test_trailing_slash) rM r6 r" r� rb r rf rP r7 �testdatar )r* r rU r r r r { s z PathsTests.test_trailing_slash�win32zWindows-specificc
C sR t tj�| jd�d��}|�d� W d Q R X t�� tj�| j�}dd l}|� � }|d }d||f }||dd � 7 }yt�
|� W nJ t k
r� } z,|j t j
t jt jfkr�| �d|f � � W d d }~X Y nX tj�d|� y td�}W n6 tk
�r } z| �d ||f � W d d }~X Y nX | �|jd� | �|j�|�|j� td� d S )
Nztest_unc_path.pyrK ztestdata = 'test_unc_path'r z\\%s\%s$rp z%cannot access administrative share %rZ
test_unc_pathz,could not import 'test_unc_path' from %r: %r) rM r6 r" r� rb r( r) r� �socketZ gethostname�listdirr� r� ZEPERMZEACCES�ENOENTr� r r# rP r3 rQ r7 r rx r9 r[ r ) r* r r" r ZhnZdriveZuncr� rU r r r �
test_UNC_path� s0 zPathsTests.test_UNC_pathN)rw r� r� ZSAMPLESr r" r+ r, r r� r� r rZ r r r r r r m s r c @ sH e Zd Zdd� ZeZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )�RelativeImportTestsc C s t d� d S )Nztest.relimport)r )r* r r r r, � s z RelativeImportTests.tearDownc C s ddl m} | �t|d�� d S )Nrp )� relimportr )r r rx ry )r* r r r r �test_relimport_star� s z'RelativeImportTests.test_relimport_starc s| � fdd�}t ddd�� |� t ddd�� |� t ddd�� | �t|� t ddd�� | �t|� t t� d�� | �t|� d S ) Nc s t d� � d S )Nzfrom . import relimport)rg r )r� r r �check_relative� s z:RelativeImportTests.test_issue3221.<locals>.check_relativerI ztest.notarealmodule)� __package__rw znotarealpkg.notarealmoduler� )r! )�dictr- r. �object� TypeError)r* r r )r� r �test_issue3221� s z"RelativeImportTests.test_issue3221c C s t �dddd� d S )Nz-W�ignorez-czfoo = 1; from . import foo)r Zassert_python_failure)r* r r r �)test_parentless_import_shadowed_by_global� s
z=RelativeImportTests.test_parentless_import_shadowed_by_globalc C s0 | � t�� ddlm} | �d� W d Q R X d S )Nro )�sepz>explicit relative import triggered an implicit absolute import)r- r3 r6 r( rQ )r* r( r r r �#test_absolute_import_without_future� s z7RelativeImportTests.test_absolute_import_without_futurec C s� t j�t j�t�dd�}tdd��T t|��@ | �t�� dd l }W d Q R X | �
dt j � | �
dt j � W d Q R X W d Q R X d S )Nr� �package2�
submodule1�
submodule2r )
r6 r" r� r� r9 r r r- r3 r+ rs r r )r* r" r+ r r r � test_import_from_non_package� s z0RelativeImportTests.test_import_from_non_packagec
C sR t ddd��< ttj�tj�t�d��� dd l}|jj W d Q R X W d Q R X d S )Nr* zpackage2.submodule1zpackage2.submodule2r� r )
r r r6 r" r� r� r9 Zpackage2.submodule1r+ r, )r* r* r r r �!test_import_from_unloaded_package� s z5RelativeImportTests.test_import_from_unloaded_packageN) rw 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� }| �|� |� ttddd� �� | �|� d� W d Q R X tt� ddd� �� | �|� |� W d Q R X d S ) Nr c S s dd l } | S )Nr )r6 )r6 r r r r� � s z?OverridingImportBuiltinTests.test_override_builtin.<locals>.foorP c W s dS )N� r )r{ r r r �<lambda>� � zDOverridingImportBuiltinTests.test_override_builtin.<locals>.<lambda>r0 c W s dS )Nr0 r )r{ r r r r1 � r2 )r6 r7 r �builtinsr r� )r* r6 r� r r r �test_override_builtin� s z2OverridingImportBuiltinTests.test_override_builtinN)rw r� r� r4 r r r r r/ � s r/ c @ s� e Zd Zdd� Zdd� Zdd� Zedd� �Ze� e
j d kd
�e� e
e
d �oTe
�� 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 | _| �� t| jd��}td|d� W d Q R X tj�dtj � t
� � d S )Nz.pyrK z/# This is a test file written by test_import.py)rL r ) r r% r6 rM rN r r" r# r6 rY r( r) )r* �fpr r r r+ s
zPycacheTests.setUpc C s, t jd tjkstd��t jd= | �� d S )Nr zUnexpected sys.path[0])r r" r6 rY �AssertionErrorr6 )r* r r r r, s zPycacheTests.tearDownc C sZ | � tj�d�� tt� | �tj�d�� tj� | j
�}| �tj�|�d� |t�� d S )Nr z*bytecode file {!r} for {!r} does not exist) � assertFalser6 r" r� rP r rx 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 | �tj�d�� tj� | j
�}| � tj�|�d� |t�� d S )N� r z"bytecode file {!r} for {!r} exists)
r
rP r rx r6 r" r� r( r� r� r% r9 r� )r* r: r r r �test_unwritable_directory s
z&PycacheTests.test_unwritable_directoryc C sV t t� tj�| j�}| �tj� |�� t�
| j� t t� t� � | �
tt t� d S )N)rP r r( r� r� r% rx r6 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�}t�| j� tt� t�� t t�}z&| � |j
tj � tj
tj �|��� W d t�|� X d S )N)rP r r r% r6 r$ r r( r) r7 r9 r" r� rY �relpath)r* r� r� r r r �test_missing_source_legacy6 s
z'PycacheTests.test_missing_source_legacyc C s6 t t�}tj�td �}| �|jtj� tj
|�� d S )Nz.py) rP r r( r� r� r7 �
__cached__r6 r" r� rY )r* r� r� r r r �test___cached__I s z PycacheTests.test___cached__c C s\ t t� t| j�}t�| j� tt� t�� t t�}| � |j
tj � tj
tj �|��� d S )N)rP r r r% r6 r$ r r( r) r7 rB r" r� rY r@ )r* r� r� r r r �test___cached___legacy_pycO s
z'PycacheTests.test___cached___legacy_pycc C s� dd� }t �d� | �|� tt j�dd�d�� W d Q R X tt j�dd�d�� W d Q R X t�� td�}tj �
t j�dd��}| � |j t j�t j
|�� tj �
t j�dd��}| � tjd j t j�t j
|�� d S )Nc S s t d� td� td� d S )N�pep3147z pep3147.foo)r r r r r r �cleanupb s z5PycacheTests.test_package___cached__.<locals>.cleanuprE z __init__.pyrK zfoo.pyz pep3147.foo)r6 r� r� rM r" r� r( r) rP r� r� r7 rB rY r r )r* rF r� �init_pyc�foo_pycr r r �test_package___cached___ s
z$PycacheTests.test_package___cached__c C s� dd� }t �d� | �|� tt j�dd�d�� W d Q R X tt j�dd�d�� W d Q R X t�� td�}t d� t d� t�� td�}tj
� t j�dd��}| � |j
t j�t j|�� tj
� t j�dd��}| � tjd j
t j�t j|�� d S )Nc S s t d� td� td� d S )NrE z pep3147.foo)r r r r r r rF y s z>PycacheTests.test_package___cached___from_pyc.<locals>.cleanuprE z __init__.pyrK zfoo.pyz pep3147.foo)r6 r� r� rM r" r� r( r) rP r r� r� r7 rB rY r r )r* rF r� rG rH r r r � test_package___cached___from_pycv s&
z-PycacheTests.test_package___cached___from_pycc C sN t t� tt� t| jd��}td|d� W d Q R X t t�}| �|jd� d S )NrR zx = 5)rL r0 )rP r r rM r% rN r7 r{ )r* r7 r� r r r � test_recompute_pyc_same_second� s z+PycacheTests.test_recompute_pyc_same_secondN)rw r� r� r6 r+ r, r� r; r� r� r6 r �skipIfry r< r> r? rA rC rD rI rJ rK r r r r r5 � s
r5 c @ sX e Zd ZdZed Zdd� Zdd� Ze�e e
d� p>e
� � dkd �e j
jd
d � ��Zd S )
� TestSymbolicallyLinkedPackageZsamplez-taggedc C s� t j�| j� t j�| j� tjd d � | _t� | j� | �
t jj| j� tj� | jd�}t j� |� tj�
|�svt�tj| j| jdd� | �
t jj| j� t�� | �tj�| j�d� tj�tj� | jd��s�t�d S )Nz __init__.pyT)Ztarget_is_directory)rI rz r �tagged� package_namer r" �
orig_sys_pathr6 r� r� r� r r� r8 �symlinkr r( r) r7 �isdir�isfile)r* Z init_filer r r r+ � s z#TestSymbolicallyLinkedPackage.setUpc C s | j tjd d �< d S )N)rP r r" )r* r r r r, � s z&TestSymbolicallyLinkedPackage.tearDown�getwindowsversion)� r z Windows Vista or later requiredc C sN dgt jd d �< tj�| j�s"t�tj�tj�| jd��s>t�t�| j� d S )N�.z __init__.py) r r" r6 r� rO r8 r� r( r� )r* r r r � test_symlinked_dir_importable� s z;TestSymbolicallyLinkedPackage.test_symlinked_dir_importableN)rw r� r� rO rN r+ r, r� r� ry r rT rI rz Zskip_unless_symlinkrW r r r r rM � s
rM c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ImportlibBootstrapTestsc C s t jd }| �|� d S )N�_frozen_importlib)r r rx )r* rU r r r �test_frozen_importlib� s
z-ImportlibBootstrapTests.test_frozen_importlibc C sX ddl m} tjd }| �||� | �|jd� | �|jd� | �|j �
d�|j � d S )Nr )�
_bootstraprY zimportlib._bootstrapr( z
_bootstrap.py) r( r[ r r rv r7 rw r! rx r9 r� )r* r[ rU r r r �"test_frozen_importlib_is_bootstrap� s
z:ImportlibBootstrapTests.test_frozen_importlib_is_bootstrapc C sX ddl m} tjd }| �||� | �|jd� | �|jd� | �|j �
d�|j � d S )Nr )�_bootstrap_external�_frozen_importlib_externalz importlib._bootstrap_externalr( z_bootstrap_external.py) r( r] r r rv r7 rw r! rx r9 r� )r* r] rU 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� d S )Nr )� machineryrY )r( r` r r rv �
ModuleSpec)r* r` rU r r r �test_there_can_be_only_one� s
z2ImportlibBootstrapTests.test_there_can_be_only_oneN)rw r� r� rZ r\ r_ rb r r r r rX � s rX 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 �d��,}d|_td }td }| �t|�|� W d Q R X d S )Nz*importlib._bootstrap_external._path_isfileTz.pycz.py)�mock�patch� return_valuer r7 r )r* � _path_isfiler" �expectr r r �test_get_sourcefile� s
z&GetSourcefileTests.test_get_sourcefilec C s8 t �d��$}d|_td }| �t|�|� W d Q R X d S )Nz*importlib._bootstrap_external._path_isfileFz.pyc)rd re rf r r7 r )r* rg r" r r r � test_get_sourcefile_no_source� s z0GetSourcefileTests.test_get_sourcefile_no_sourcec C s t d }| �t|�|� d S )Nz.bad_ext)r r7 r )r* r" r r r �test_get_sourcefile_bad_ext s z.GetSourcefileTests.test_get_sourcefile_bad_extN)rw r� r� �__doc__ri rj rk r r r r rc � s rc 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 � �Ze�ed �d!d"� �Zd#S )%�ImportTracebackTestsc C s, t �t� tjd d � | _tj�dt� d S )Nr )r6 r� r r r" �old_pathr# )r* r r r r+ s
zImportTracebackTests.setUpc C s | j tjd d �< tt� d S )N)rn r r" r r )r* r r r r, s z ImportTracebackTests.tearDown�.pyc C sJ t j�t|| �}t|d��}|�|� W d Q R X | �t|� t� � |S )NrK )
r6 r" r� r rM rb r� r r( r) )r* rU �contentsrT Zfnamer r r r �
create_module s z"ImportTracebackTests.create_modulec C sz g }x6|r:|j j}|j}|r(||d kr2|�|� |j}qW | �t|�t|�|� x"t||�D ]\}}| �||� q^W d S )Nr_ ) �tb_frame�f_coder rf �tb_nextr7 �len�ziprC )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 d d }~X Y n X | �d� | �|tg� d S )Nr z#ImportError should have been raised)�nonexistent_xyzzyr3 �
__traceback__rQ rz r9 )r* r{ r� rw r r r �test_nonexistent_module, s
z,ImportTracebackTests.test_nonexistent_modulec
C s` | � dd� y dd l}W n( tk
r@ } z
|j}W d d }~X Y n X | �d� | �|tdg� d S )Nr� zimport nonexistent_xyzzyr z#ImportError should have been raisedzfoo.py)rq r� r3 r| rQ rz r9 )r* r� r� rw r r r � test_nonexistent_module_nested6 s
z3ImportTracebackTests.test_nonexistent_module_nestedc
C s` | � dd� y dd l}W n( tk
r@ } z
|j}W d d }~X Y n X | �d� | �|tdg� d S )Nr� z1/0r z)ZeroDivisionError should have been raisedzfoo.py)rq r� rr r| rQ rz r9 )r* r� r� rw r r r �test_exec_failure@ s
z&ImportTracebackTests.test_exec_failurec
C sn | � dd� | � dd� y dd l}W n( tk
rL } z
|j}W d d }~X Y n X | �d� | �|tddg� d S ) Nr� z
import bar�barz1/0r z)ZeroDivisionError should have been raisedzfoo.pyzbar.py)rq r� rr r| rQ rz r9 )r* r� r� rw r r r �test_exec_failure_nestedJ s
z-ImportTracebackTests.test_exec_failure_nestedc
C s^ | � dd� y dd l}W n( tk
r@ } z
|j}W d d }~X Y n X | �d� | �|tg� d S )Nr� zinvalid syntax is invalidr z#SyntaxError should have been raised)rq r� � SyntaxErrorr| rQ rz r9 )r* r� r� rw r r r �test_syntax_errorV s
z&ImportTracebackTests.test_syntax_errorc C s� d}| � t|� tj�t|�}t�|� tj�|d�}t|d��}|�|� W d Q R X tj�|d�}t|d��}|�|� W d Q R X t �
� ||fS )N� _parent_fooz __init__.pyrK zbar.py) r� r r6 r" r� r r� rM rb r( r) )r* �parentZchildZpkg_nameZpkg_path� init_pathr �bar_pathr r r �_setup_broken_package` s
z*ImportTracebackTests._setup_broken_packagec
C sd | � dd�\}}y dd l}W n( tk
rD } z
|j}W d d }~X Y n X | �d� | �|t|g� d S )Nr z1/0r z)ZeroDivisionError should have been raised)r� �_parent_foo.barrr r| rQ rz r9 )r* r� r� r� r� rw r r r �test_broken_submoduleo s
z*ImportTracebackTests.test_broken_submodulec
C sh | � dd�\}}yddlm} W n( tk
rH } z
|j}W d d }~X Y n X | �d� | �|t|g� d S )Nr z1/0r )r� z#ImportError should have been raised)r� r� r� rr r| rQ rz r9 )r* r� r� r� r� rw r r r �test_broken_fromy s
z%ImportTracebackTests.test_broken_fromc
C sd | � dd�\}}y dd l}W n( tk
rD } z
|j}W d d }~X Y n X | �d� | �|t|g� d S )Nz1/0r r z)ZeroDivisionError should have been raised)r� r� rr r| rQ rz r9 )r* r� r� r� r� rw r r r �test_broken_parent� s
z'ImportTracebackTests.test_broken_parentc
C sh | � dd�\}}yddlm} W n( tk
rH } z
|j}W d d }~X Y n X | �d� | �|t|g� d S )Nz1/0r r )r� z)ZeroDivisionError should have been raised)r� r� r� rr r| rQ rz r9 )r* r� r� r� r� rw r r r �test_broken_parent_from� s
z,ImportTracebackTests.test_broken_parent_fromc
C s� | � dd� tjd }dt|j�kr.|jj}nd }zfdd� }||j_y dd l}W n( tk
rx } z
|j}W d d }~X Y n X | � d� | �
|t d t g� W d |d kr�|j`n||j_X d S )
Nr� r r^ � load_modulec W s dd d S )Nro r r )r� r r r � exec_module� s z9ImportTracebackTests.test_import_bug.<locals>.exec_moduler z)ZeroDivisionError should have been raisedz<frozen importlib) rq r r �vars� SourceLoaderr� r� rr r| rQ rz r9 )r* r( Zold_exec_moduler� r� r� rw r r r �test_import_bug� s"
z$ImportTracebackTests.test_import_bugzneed TESTFN_UNENCODABLEc C s> t �dtd�}| �t|� |d d� }t jdd| dd� d S )Nr �pass���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)ro )rw r� r� r+ r, rq rz r} r~ r r� r� r� r� r� r� r� r r� r� r� r r� r r r r rm
s
rm c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�CircularImportTestszQSee the docstrings of the modules being imported for the purpose of the
test.c C s. x(t tj�� �D ]}|�d�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 �keysr[ )r* �keyr r r r, � s
z CircularImportTests.tearDownc C s0 y dd l }W n tk
r* | �d� Y nX d S )Nr z/circular import through relative imports failed)Z,test.test_import.data.circular_imports.basicr3 rQ )r* rI r r r � test_direct� s z CircularImportTests.test_directc C s0 y dd l }W n tk
r* | �d� Y nX d S )Nr z@relative import in module contributing to circular import failed)Z/test.test_import.data.circular_imports.indirectr3 rQ )r* rI r r r �
test_indirect� s z!CircularImportTests.test_indirectc C s0 y dd l }W n tk
r* | �d� Y nX d S )Nr z-circular import involving a subpackage failed)Z1test.test_import.data.circular_imports.subpackager3 rQ )r* rI r r r �test_subpackage� s z#CircularImportTests.test_subpackagec C sb y"dd l m m m m} W n tk
r@ | �d� Y nX ddlm} | � |j|j� d S )Nr z9circular import with rebinding of module attribute failed)r� )
Z0test.test_import.data.circular_imports.rebindingr\ r� Zcircular_imports� rebindingr3 rQ Z-test.test_import.data.circular_imports.subpkgr� rv )r* r� r� r r r �test_rebinding� s " z"CircularImportTests.test_rebindingc C s0 y dd l }W n tk
r* | �d� Y nX d S )Nr z9circular import with binding a submodule to a name failed)Z.test.test_import.data.circular_imports.bindingr3 rQ )r* rI r r r � test_binding� s z CircularImportTests.test_bindingN)
rw r� r� rl r, r� r� r� r� r� r r r r r� � s r� �__main__)Nr )@r( �importlib.utilZ importlib._bootstrap_externalr r3 r� r6 rZ rd rO r� r r� r� r� Z
unittest.mockrd r� r� �
contextlibru rI r r r r r r r r
r r r
r r r r r r r r Ztest.test_importlib.utilr rL �dont_write_bytecoder� r �contextmanagerr& ZTestCaser' r� r� r r r/ r5 rM rX rc rm r� rw �mainr r r r �<module> s` P |
S^5E ".
! 1.