File: //lib64/python3.6/test/__pycache__/test_pkgutil.cpython-36.pyc
3
\E � @ s� d dl mZmZmZmZ d dlZd dlZd dlZd dlm Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZG dd� dej�ZG dd� dej�ZG dd � d ej�ZG d
d � d ej�ZG d d
� d
ej�Zdd� Zedkr�e� dS )� )� run_unittest�unload�check_warnings� CleanImportN)�spec_from_file_locationc @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� PkgutilTestsc C s. t j� | _| jtj| j� tjjd| j� d S )Nr ) �tempfile�mkdtemp�dirname�
addCleanup�shutil�rmtree�sys�path�insert)�self� r �$/usr/lib64/python3.6/test_pkgutil.py�setUp s
zPkgutilTests.setUpc C s t jd= d S )Nr )r r )r r r r �tearDown s zPkgutilTests.tearDownc C s� d}d}t jj| j|�}t j|� tt jj|d�d�}|j� tt jj|d�d�}|j|� |j� t jt jj|d�� tt jj|dd�d�}|j|� |j� tj |d�}| j
||� tj |d�}| j
||� t j |= d S )N�test_getdata_filesyss% Hello, world!
Second line
Third linez __init__.py�wbzres.txt�subz sub/res.txt)
�osr �joinr
�mkdir�open�close�write�pkgutil�get_data� assertEqualr �modules)r �pkg�
RESOURCE_DATAZ package_dir�f�res1�res2r r r r s$
z!PkgutilTests.test_getdata_filesysc
C s� d}d}d}t jj| j|�}tj|d�}|j|d d� |j|d |� |j|d |� |j� tjj d |� t
j |d
�}| j ||� t
j |d �}| j ||� g }x.t
j
|g�D ] } | j| t
j� |j| j� q�W | j |dg� tjd = tj|= d S ) Nztest_getdata_zipfile.zip�test_getdata_zipfiles% Hello, world!
Second line
Third line�wz /__init__.py� z/res.txtz /sub/res.txtr zres.txtz sub/res.txt)r r r r
�zipfile�ZipFile�writestrr r r r r r! Z iter_modules�assertIsInstanceZ
ModuleInfo�append�namer" )
r �zipr# r$ �zip_file�zr& r' �namesZ
moduleinfor r r r( : s* z!PkgutilTests.test_getdata_zipfilec C sV d}t jj| j|�}t j|d� | jt j|� x"tj| jgd�D ]}| j d� q@W d S )NZunreadable_packager )r zunexpected package found)
r r r r
r r �rmdirr �
walk_packagesZfail)r Z package_name�d�tr r r � test_unreadable_dir_on_syspath] s z+PkgutilTests.test_unreadable_dir_on_syspathc C st d}t jj| j|�}t j|� tt jj|d�d�}|j� t jt jj|d�� tt jj|dd�d�}|j� tt jj|dd�d�}|j� d}t jj| j|�}t j|� tt jj|d�d�}|j� t jt jj|d�� tt jj|dd�d�}|j� tt jj|dd�d�}|j� dddddd g}d
d � tj| jg�D �}| j ||� x&|D ] }|j
d ��rb�qNt j |= �qNW d S )
N�test_walkpackages_filesysz __init__.pyr r zmod.pyz sub.test_walkpackages_filesysz!sub.test_walkpackages_filesys.modz test_walkpackages_filesys.subz!test_walkpackages_filesys.sub.modc S s g | ] }|d �qS )� r )�.0�er r r �
<listcomp>� s z:PkgutilTests.test_walkpackages_filesys.<locals>.<listcomp>�mod)
r r r r
r r r r r6 r! �endswithr r" ) r �pkg1Zpkg1_dirr% �pkg2Zpkg2_dir�expected�actualr# r r r r: h s@
z&PkgutilTests.test_walkpackages_filesysc C s d}d}d}t jj| j|�}tj|d�}|j|d d� |j|d | d d� |j|d | d d� |j|d d� |j|d | d d� |j|d | d d� |j� tjj d |� dd
d dd d
g}dd� t
j |g�D �}| j ||� tjd = x"|D ]}|j
d��rq�tj|= q�W dS )zBTests the same as test_walkpackages_filesys, only with a zip file.z test_walkpackages_zipfile.zip�test_walkpackages_zipfiler r) z /__init__.pyr* �/z/mod.pyr z sub.test_walkpackages_zipfilez!sub.test_walkpackages_zipfile.modz test_walkpackages_zipfile.subz!test_walkpackages_zipfile.sub.modc S s g | ] }|d �qS )r; r )r< r= r r r r> � s z:PkgutilTests.test_walkpackages_zipfile.<locals>.<listcomp>r? N)r r r r
r+ r, r- r r r r r6 r! r@ r" ) r r1 rA rB r2 r3 rC rD r# r r r rE � s4
z&PkgutilTests.test_walkpackages_zipfileN)
�__name__�
__module__� __qualname__r r r r( r9 r: rE r r r r r s # (r c @ sL e Zd ZG dd� de�ZG dd� de�Zdd� Zdd� Zd d
� Zd d � Z d
S )�PkgutilPEP302Testsc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z PkgutilPEP302Tests.MyTestLoaderc C s d S )Nr )r �specr r r �
create_module� s z-PkgutilPEP302Tests.MyTestLoader.create_modulec C s |j jdd�d |j d< d S )N�loadsr r; )�__dict__�get)r r? r r r � exec_module� s z+PkgutilPEP302Tests.MyTestLoader.exec_modulec C s dS )Nz
Hello, world!r )r r r r r r � s z(PkgutilPEP302Tests.MyTestLoader.get_dataN)rG rH rI rL rP r r r r r � MyTestLoader� s rQ c @ s e Zd Zddd�ZdS )z!PkgutilPEP302Tests.MyTestImporterNc C s t j� }t|d|jj |g d�S )Nz<%s>)�loader�submodule_search_locations)rJ rQ r � __class__rG )r �fullnamer �targetrR r r r � find_spec� s
z+PkgutilPEP302Tests.MyTestImporter.find_spec)NN)rG rH rI rW r r r r �MyTestImporter� s rX c C s t jjd| j� � d S )Nr )r � meta_pathr rX )r r r r r � s zPkgutilPEP302Tests.setUpc C s t jd= d S )Nr )r rY )r r r r r � s zPkgutilPEP302Tests.tearDownc C s | j tjdd�d� tjd= d S )N�foo�dummyz
Hello, world!)r! r r r r" )r r r r �test_getdata_pep302� s z&PkgutilPEP302Tests.test_getdata_pep302c C sD dd l }| j|jd� | jtjdd�d� | j|jd� tjd= d S )Nr r; rZ r[ z
Hello, world!)rZ r! rM r r r r" )r rZ r r r �test_alreadyloaded� s
z%PkgutilPEP302Tests.test_alreadyloadedN)
rG rH rI �objectrQ rX r r r\ r] r r r r rJ � s rJ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ExtendPathTestsc C s\ t j� }tjjd|� tjj||�}tj|� ttjj|d�d��}|j d� W d Q R X |S )Nr z __init__.pyr) zKfrom pkgutil import extend_path
__path__ = extend_path(__path__, __name__)
)
r r r r r r r r r r )r �pkgnamer
Zpkgdir�flr r r � create_init� s
zExtendPathTests.create_initc C s@ t jj|||d �}t|d��}tdj|�|d� W d Q R X d S )Nz.pyr) zvalue={})�file)r r r r �print�format)r r
r` Zsubmodule_name�valueZ module_namera r r r �create_submodule� s z ExtendPathTests.create_submodulec C s� d}| j |�}| j |�}| j||dd� | j||dd� dd l}dd l}| j|jjd� | j|jjd� | jt|j �tt
j j ||�t
j j ||�g�� t
j|� t
j|� tj d= tj d= tjd= tjd= tjd= d S )NrZ �barr �bazr; zfoo.barzfoo.baz)rb rg �foo.bar�foo.bazr! rh rf ri �sorted�__path__r r r r r
r r" )r r` � dirname_0� dirname_1rZ r r r � test_simple� s&
zExtendPathTests.test_simplec
C s� t j}t j}d}d}| j|�}tjj||�}dj||�}tj j
|d � tj j
|d � z�| j |||d� t j
|� t||��}||�} x||D ]t}
t j|
|�} | j} y
| j} W n tk
r� Y nX | j|
t jj� | j|
| � | j| t jj� | jt j|
|�� q�W | jt�� t|d�� W d Q R X | jt�� t|d�� W d Q R X W d tj|� tjd= ytj d= tj d= W n t k
�r� Y nX X d S )N�spamZeggsz{}.{}r zinvalid.modulez.spamz spam.eggs) r �iter_importers� get_importerrb r r r re r r" �poprg � importlib�
import_module�list� _get_specrR �AttributeErrorr. � machinery�
FileFinderr! �SourceFileLoader� assertIsNoneZ assertRaises� ImportErrorr r
�KeyError)
r rr rs r` �modnamer
ZpathitemrU Z importersZexpected_importer�finderrK rR r r r �test_iter_importers sJ
z#ExtendPathTests.test_iter_importersc C s� d}| j |�}| j |�}| j||dd� tjtjj||d�� | j||dd� dd l}dd l}| j|j j
d� | j|j j
d� | jt |j
�t tjj||�tjj||�g�� tj|� tj|� tjd= tjd= tjd= tjd= tjd= d S ) NrZ rh r z __init__.pyri r; zfoo.barzfoo.baz)rb rg r �unlinkr r rj rk r! rh rf ri rl rm r r
r r" )r r` rn ro rZ r r r �test_mixed_namespace@ s(
z$ExtendPathTests.test_mixed_namespaceN)rG rH rI rb rg rp r� r� r r r r r_ � s
0r_ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�NestedNamespacePackageTestc C s t j� | _tjd d � | _d S )N)r r �basedirr r �old_path)r r r r r a s
z NestedNamespacePackageTest.setUpc C s | j tjd d �< tj| j� d S )N)r� r r r r
r� )r r r r r e s z#NestedNamespacePackageTest.tearDownc C sl |j dd�\}}tjj| j|jdtjj��}tj|dd� ttjj||d �d��}|j |� W d Q R X d S )N�.r; T)�exist_okz.pyr) )
�rsplitr r r r� �replace�sep�makedirsr r )r r0 �contents�base�final� base_pathr% r r r rL i s
z(NestedNamespacePackageTest.create_modulec C s d}| j d|� | j d|� | j d|� | j d|� | j dd� | j dd � tjjd
tjj| jd �� tjjd
tjj| jd �� d
d l}| jt d
� | j
t |j �d� d
d l
}| jt d� | j
t |jj �d� d
dlm} d
dlm} | j
|d� | j
|d� d S )NzBimport pkgutil; __path__ = pkgutil.extend_path(__path__, __name__)za.pkg.__init__zb.pkg.__init__za.pkg.subpkg.__init__zb.pkg.subpkg.__init__za.pkg.subpkg.czc = 1zb.pkg.subpkg.dzd = 2r �a�br# � z
pkg.subpkg)�c)r7 r; )rL r r r r r r� r# r r r! �lenrm Z
pkg.subpkgZsubpkgZ pkg.subpkg.cr� Z pkg.subpkg.dr7 )r Zpkgutil_boilerplater# r� r7 r r r � test_nestedp s& z&NestedNamespacePackageTest.test_nestedN)rG rH rI r r rL r� r r r r r� _ s r� c @ s| e Zd Zdd� Zdd� Zdd� Zdd� Zeje d kd
�d d � �Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS ) �ImportlibMigrationTestsc C s t dtf�S )Nz5This emulation is deprecated, use 'importlib' instead)r �DeprecationWarning)r r r r �check_deprecated� s z(ImportlibMigrationTests.check_deprecatedc C s" | j � � tjd� W d Q R X d S )Nr* )r� r Z ImpImporter)r r r r �test_importer_deprecated� s
z0ImportlibMigrationTests.test_importer_deprecatedc C s( | j � � tjdddd� W d Q R X d S )Nr* )r� r Z ImpLoader)r r r r �test_loader_deprecated� s
z.ImportlibMigrationTests.test_loader_deprecatedc
C sX t � �H}| jtjd�� | jtjd�� | jtjd�� | jt|j�d� W d Q R X d S )Nr r z test.supportr )r �assertIsNotNoner �
get_loaderr! r� �warnings)r r) r r r � test_get_loader_avoids_emulation� s
z8ImportlibMigrationTests.test_get_loader_avoids_emulation�__main__z not compatible with __main__c C sJ t }b z8t� �(}| jtjt�� | jt|j�d� W d Q R X W d |a X d S )Nr ) �
__loader__r r� r r� rG r! r� r� )r Z this_loaderr) r r r �0test_get_loader_handles_missing_loader_attribute� s zHImportlibMigrationTests.test_get_loader_handles_missing_loader_attributec
C sJ d}t t�|�}|`t|�� |tj|< tj|�}W d Q R X | j|� d S )Nrq )�typer �__spec__r r" r r� r} )r r0 r? rR r r r �.test_get_loader_handles_missing_spec_attribute� s
zFImportlibMigrationTests.test_get_loader_handles_missing_spec_attributec
C sL d}t t�|�}d |_t|�� |tj|< tj|�}W d Q R X | j|� d S )Nrq )r� r r� r r" r r� r} )r r0 r? rR r r r �+test_get_loader_handles_spec_attribute_none� s
zCImportlibMigrationTests.test_get_loader_handles_spec_attribute_nonec C s6 d}d t j|< ztj|�}W d t j|= X | j|� d S )Nz
totally bogus)r r" r r� r} )r r0 rR r r r �#test_get_loader_None_in_sys_modules� s
z;ImportlibMigrationTests.test_get_loader_None_in_sys_modulesc C s d}t j|�}| j|� d S )Nz
totally bogus)r � find_loaderr} )r r0 rR r r r � test_find_loader_missing_module� s
z7ImportlibMigrationTests.test_find_loader_missing_modulec
C sX t � �H}| jtjd�� | jtjd�� | jtjd�� | jt|j�d� W d Q R X d S )Nr r z test.supportr )r r� r r� r! r� r� )r r) r r r �!test_find_loader_avoids_emulation� s
z9ImportlibMigrationTests.test_find_loader_avoids_emulationc
C s8 t � �(}| jtjd�� | jt|j�d� W d Q R X d S )Nz*??r )r r} r rs r! r� r� )r r) r r r �"test_get_importer_avoids_emulation� s z:ImportlibMigrationTests.test_get_importer_avoids_emulationc
C s: t � �*}xtj� D ]}qW | jt|j�d� W d Q R X d S )Nr )r r rr r! r� r� )r r) Zimporterr r r �$test_iter_importers_avoids_emulation� s z<ImportlibMigrationTests.test_iter_importers_avoids_emulationN)rG rH rI r� r� r� r� �unittestZskipIfr� r� r� r� r� r� r� r� r r r r r� � s r� c C s6 t ttttt� dd l} dd l}| jj � |j
� d S )Nr ) r r rJ r_ r� r� � zipimportru �_zip_directory_cache�clear�invalidate_caches)r� ru r r r � test_main� s
r� r� )Z test.supportr r r r r� r ru �importlib.utilr r r Zos.pathr r r+ ZTestCaser rJ r_ r� r� r� rG r r r r �<module> s& #-})X