Your IP : 216.73.216.213
3
��2a6G � @ sZ d dl mZ ejd�Zejd�Zejd�ZddlZddlZddl m
Z
ddl Z ddl Z ddl
Z
G dd � d �Zejeed
�\ZZG d d � d �ZG d
d� de�Zejeed
�\ZZG dd� de�Zejeed
�\ZZG dd� d�Zejeeed�\ZZG dd� d�Z eje ed
�\Z Z G dd� de j �Z G dd� d�Z!eje!ed�\Z"Z#e$dk�rVe j%� dS ) � )�util� importlibzimportlib.utilzimportlib.machinery� N)�supportc @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�ImportModuleTestsc C sN t jd��:}t j|gd�� | jjd�}| j|jd� W d Q R X W d Q R X d S )N� top_level)� meta_path)� test_util� mock_modules� import_state�init�
import_module� assertEqual�__name__)�self�mock�module� r �-/opt/alt/python36/lib64/python3.6/test_api.py�test_module_import s z$ImportModuleTests.test_module_importc C sh d}dj |�}dj |�}tj||��:}tj|gd�� | jj|�}| j|j|� W d Q R X W d Q R X d S )N�pkgz {0}.__init__z{0}.mod)r )�formatr r
r r r
r r )r �pkg_name�
pkg_long_name�namer r r r r � test_absolute_package_import s
z.ImportModuleTests.test_absolute_package_importc C s� d}dj |�}d}dj ||�}dj |�}tj||��H}tj|gd��. | jj|� | jj||�}| j|j|� W d Q R X W d Q R X d S )Nr z {0}.__init__�modz{0}.{1}z.{0})r )r r r
r r r
r r )r r r Z module_nameZ
absolute_nameZ
relative_namer r r r r �$test_shallow_relative_package_import$ s
z6ImportModuleTests.test_shallow_relative_package_importc C sr dddg}t j|� �T}t j|gd��: | jjd� | jjd� | jjdd�}| j|jd� W d Q R X W d Q R X d S )Nz
a.__init__z a.b.__init__za.c)r �aza.bz..c)r r
r r r
r r )r �modulesr r r r r �!test_deep_relative_package_import1 s
z3ImportModuleTests.test_deep_relative_package_importc C sv d}dj |�}dj |�}tj||��H}tj|gd��. | jj|� | jj||�}| j|j|� W d Q R X W d Q R X d S )Nr z {0}.__init__z{0}.mod)r )r r r
r r r
r r )r r r r r r r r r �!test_absolute_import_with_package: s
z3ImportModuleTests.test_absolute_import_with_packagec C s& | j t�� | jjd� W d Q R X d S )Nz.support)� assertRaises� TypeErrorr r
)r r r r � test_relative_import_wo_packageF s z1ImportModuleTests.test_relative_import_wo_packagec s� d� �fdd�}� fdd�}||d�}ddg}t j|d |i��,}t j|gd
�� �jjd� W d Q R X W d Q R X �j� d � d S ) Nr c s � j jd� d S )Nza.b)r r
r )r r r �load_aR s z2ImportModuleTests.test_loaded_once.<locals>.load_ac s � d7 � d S )Nr r r )� b_load_countr r �load_bT s z2ImportModuleTests.test_loaded_once.<locals>.load_b)r za.bz
a.__init__za.b� module_code)r r )r r
r r r
r )r r% r' �coder r r )r& r r �test_loaded_onceM s
z"ImportModuleTests.test_loaded_onceN)
r �
__module__� __qualname__r r r r r! r$ r* r r r r r s
r )r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�FindLoaderTestsNc C st d}t j|��\ tj|�}d}||_|tj|< tj� � tj dt
� | j j |�}W d Q R X | j
||� W d Q R X d S )N�some_modz a loader!�ignore)r �uncache�types�
ModuleType�
__loader__�sysr �warnings�catch_warnings� simplefilter�DeprecationWarningr � find_loaderr )r r r �loader�foundr r r �test_sys_modulesh s
z FindLoaderTests.test_sys_modulesc C sz d}t j|��b tj|�}d |_|tj|< | jt��2 t j
� � t j dt � | j
j|� W d Q R X W d Q R X W d Q R X d S )Nr. r/ )r r0 r1 r2 r3 r4 r r" �
ValueErrorr5 r6 r7 r8 r r9 )r r r r r r � test_sys_modules_loader_is_Noneu s
z/FindLoaderTests.test_sys_modules_loader_is_Nonec C s� d}t j|��z tj|�}y|`W n tk
r6 Y nX |tj|< | jt ��2 t
j � � t
j dt
� | jj|� W d Q R X W d Q R X W d Q R X d S )Nr. r/ )r r0 r1 r2 r3 �AttributeErrorr4 r r" r= r5 r6 r7 r8 r r9 )r r r r r r �"test_sys_modules_loader_is_not_set� s
z2FindLoaderTests.test_sys_modules_loader_is_not_setc C sr d}t j|��Z t j| jgd��> tj� �* tjdt� | j|d f| j j
|�� W d Q R X W d Q R X W d Q R X d S )Nr. )r r/ ) r r0 r �FakeMetaFinderr5 r6 r7 r8 r r r9 )r r r r r � test_success� s
z FindLoaderTests.test_successc C sx d}d}t j|��\ t j| jgd��@ tj� �, tjdt� | j||f| j j
||�� W d Q R X W d Q R X W d Q R X d S )Nr. zpath to some place)r r/ ) r r0 r rA r5 r6 r7 r8 r r r9 )r r �pathr r r �test_success_path� s
z!FindLoaderTests.test_success_pathc
C s6 t j� �$ t jdt� | j| jjd�� W d Q R X d S )Nr/ Znevergoingtofindthismodule)r5 r6 r7 r8 Z assertIsNoner r9 )r r r r � test_nothing� s
z FindLoaderTests.test_nothing)
r r+ r, rA r< r> r@ rB rD rE r r r r r- d s
r- c @ s e Zd ZG dd� d�ZdS )�FindLoaderPEP451Testsc @ s e Zd Zeddd��ZdS )z$FindLoaderPEP451Tests.FakeMetaFinderNc C s t d j| | |f�S )N�Source)� machinery�
ModuleSpec)r rC �targetr r r � find_spec� s z.FindLoaderPEP451Tests.FakeMetaFinder.find_spec)NN)r r+ r, � staticmethodrK r r r r rA � s rA N)r r+ r, rA r r r r rF � s rF c @ s e Zd ZG dd� d�ZdS )�FindLoaderPEP302Testsc @ s e Zd Zeddd��ZdS )z$FindLoaderPEP302Tests.FakeMetaFinderNc C s | |fS )Nr )r rC r r r � find_module� s z0FindLoaderPEP302Tests.FakeMetaFinder.find_module)N)r r+ r, rL rN r r r r rA � s rA N)r r+ r, rA r r r r rM � s rM c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� ReloadTestsc C sX xRdD ]J}| j |d��4 tj|�� | jj|�}| jj|� W d Q R X W d Q R X qW d S )N�tokenize�time�marshal)r )rP rQ rR )�subTestr � CleanImportr r
�reload)r r r r r r �test_reload_modules� s
z ReloadTests.test_reload_modulesc C s� dd� }t jdd|id�}|�^ t j|gd��D | jjd�}| jj|�}tjd }| j|j d� | j|j d� W d Q R X W d Q R X d S )Nc S s( dd l } t| �d�}d|_|| jd< d S )Nr r � )r4 �type�spamr )r4 r r r r r) � s z.ReloadTests.test_module_replaced.<locals>.coder )r( )r rW )
r r
r r r
rU r4 r r rY )r r) r r �reloaded�actualr r r �test_module_replaced� s
z ReloadTests.test_module_replacedc
C sh t jd��T dd l}|j}|`| jj|�}| j||� | jtjd |� | j |jj
|j
� W d Q R X d S )Nr1 r ) r rT r1 r3 r rU �assertIsr4 r r rC )r r1 r: rZ r r r �test_reload_missing_loader� s z&ReloadTests.test_reload_missing_loaderc
C sj t jd��V dd l}d |_| jj� | jj|�}| j|jd � | j||� | jt j
d |� W d Q R X d S )Nr1 r ) r rT r1 r3 r �invalidate_cachesrU � assertIsNotr] r4 r )r r1 rZ r r r �test_reload_loader_replaced� s
z'ReloadTests.test_reload_loader_replacedc C s� d}t jd ����}tjd���� t j|���� | jj� tjj ||d �}| j
j |�}|d||d d�}t j |� | jj
|�}t|�j� }|jd�}|jd�} |jdd � | j| j|� | j| j|� | j|j|� | j||� | jj� tjj ||d�}
| j
j |
�}|||
|tjj|
�gd d �}tj|� tj||
� | jj|�} t| �j� }|jd�}|jd�} |jdd � | j| j|� | j| j|� | j| |� | j|j|
� d | _| j||� W d Q R X W d Q R X W d Q R X d S )
NrY z.py� )r � __package__�__file__�
__cached__�__doc__r3 �__spec__� __builtins__z __init__.py)r rc rd re �__path__rf )r �temp_cwdr r0 �
DirsOnSysPathr r_ �osrC �joinr �cache_from_sourceZcreate_empty_filer
�vars�copy�popr r r: �dirname�mkdir�renamerU r] ZmaxDiff) r r �cwdrC �cached�expectedr �nsr: �spec� init_pathrZ r r r � test_reload_location_changed� sX
z(ReloadTests.test_reload_location_changedc + C s& d}t jd ��� }tjd���� t j|���� | jj� tjj ||d�}| j
j |�}||d d�}tj |� t
|d��}|jd� W d Q R X | jj|�}t|�j� }|jd�} |jd�}
|jd�} |jd d � | j| j|� | j| jd � | j| d � | jt|
�ttjj|�g�� | jt�� | j W d Q R X | j||� | jj� tjj ||d
�} | j
j | �}||| |tjj| �gd d d �}tj || � | jj |�}
t|
�j� }|jd�} |jd�} |jd d � | j| j|� | j| j| � | j|
|� | j| j| � | j||� W d Q R X W d Q R X W d Q R X d S ) NrY z __init.py)r rc rf �wz eggs = Noner3 ri rg rh z __init__.py)r rc rd re ri rf Zeggs) r rj r r0 rk r r_ rl rC rm r rn rs �open�writer
ro rp rq r r r] r: r` �setrr r" r? rt rU )r r ru Zbad_pathrv rw Z init_filer rx r: rC ry rz rZ r r r � test_reload_namespace_changed* s`
z)ReloadTests.test_reload_namespace_changedc C s\ d}d}t j|dd��<}t j|||�\}}| jj|�}| jj|�}| j||� W d Q R X d S )NrY �hamT)r )r Z temp_moduleZ submoduler r
rU r] )r r ZsubnameZpkg_dir�fullname�_r� rZ r r r �test_reload_submoduleb s z!ReloadTests.test_reload_submoduleN)
r r+ r, rV r\ r^ ra r{ r� r� r r r r rO � s 38rO )r r c @ s e Zd Zdd� Zdd� ZdS )�InvalidateCacheTestsc s� G dd� d�}d� |� �|� }t jjd�� | j� fdd�� |t j� < | j�fdd�� | jj� | j�j� | j|j� d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zGInvalidateCacheTests.test_method_called.<locals>.InvalidatingNullFinderc W s
d| _ d S )NF)�called)r Zignoredr r r �__init__w s zPInvalidateCacheTests.test_method_called.<locals>.InvalidatingNullFinder.__init__c W s d S )Nr )r �argsr r r rN y s zSInvalidateCacheTests.test_method_called.<locals>.InvalidatingNullFinder.find_modulec S s
d| _ d S )NT)r� )r r r r r_ { s zYInvalidateCacheTests.test_method_called.<locals>.InvalidatingNullFinder.invalidate_cachesN)r r+ r, r� rN r_ r r r r �InvalidatingNullFinderv s r� Z gobledeegookr c s t jj� �S )N)r4 �path_importer_cache� __delitem__r )�keyr r �<lambda>� s z9InvalidateCacheTests.test_method_called.<locals>.<lambda>c s t jj� �S )N)r4 r �remover )�meta_insr r r� � s ) r4 r �insert�
addCleanupr� r r_ �
assertTruer� )r r� Zpath_insr )r� r� r �test_method_calledt s
z'InvalidateCacheTests.test_method_calledc s. d� d t j� < | j� fdd�� | jj� d S )NZ
gobbledeegookc s t jj� �S )N)r4 r� r� r )r� r r r� � s z:InvalidateCacheTests.test_method_lacking.<locals>.<lambda>)r4 r� r� r r_ )r r )r� r �test_method_lacking� s
z(InvalidateCacheTests.test_method_lackingN)r r+ r, r� r� r r r r r� r s r� c @ s e Zd Zdd� ZdS )�FrozenImportlibTestsc C s t d }| j|jjjd� d S )NrG �FrozenImporter)r ZassertNotEqualr3 � __class__r )r Z source_initr r r �test_no_frozen_importlib� s z-FrozenImportlibTests.test_no_frozen_importlibN)r r+ r, r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� StartupTestsc C s� x�t jj� D ]~\}}t|tj�r | j|d��X | jt|d�dj |�� | j
j j |�rd| j
|jd � n | j
jj |�r�| j
|jd � W d Q R X q W d S )N)r r3 z!{!r} lacks a __loader__ attribute)r4 r �items�
isinstancer1 r2 rS r� �hasattrr rH �BuiltinImporterrN r` r3 r� )r r r r r r � test_everyone_has___loader__� s z)StartupTests.test_everyone_has___loader__c C s� x�t jj� D ]v\}}t|tj�r | j|d��P | jt|d�� | j j
j |�r\| j |j
d � n | j jj |�rx| j |j
d � W d Q R X q W d S )N)r rg )r4 r r� r� r1 r2 rS r� r� rH r� rN r` rg r� )r r r r r r �test_everyone_has___spec__� s z'StartupTests.test_everyone_has___spec__N)r r+ r, r� r� r r r r r� � s r� )rH �__main__)&rb r r Zimport_importlibr rH Zos.pathrl r4 Ztestr r1 Zunittestr5 r Z test_bothZFrozen_ImportModuleTestsZSource_ImportModuleTestsr- rF Z Frozen_FindLoaderPEP451TestsZ Source_FindLoaderPEP451TestsrM Z Frozen_FindLoaderPEP302TestsZ Source_FindLoaderPEP302TestsrO ZFrozen_ReloadTestsZSource_ReloadTestsr� ZFrozen_InvalidateCacheTestsZSource_InvalidateCacheTestsZTestCaser� r� ZFrozen_StartupTestsZSource_StartupTestsr �mainr r r r �<module> s4
RH
*!