File: //usr/local/lib/python3.7/test/__pycache__/test_pkgutil.cpython-37.opt-2.pyc
B
��gsF � @ 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 @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )� PkgutilTestsc C s. t �� | _| �tj| j� tj�d| j� d S )Nr ) �tempfile�mkdtemp�dirname�
addCleanup�shutil�rmtree�sys�path�insert)�self� r �-/usr/local/lib/python3.7/test/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 j�| j|�}t �|� tt j�|d�d�}|�� tt j�|d�d�}|�|� |�� t �t j�|d�� tt j�|dd�d�}|�|� |�� t� |d�}| �
||� t� |d�}| �
||� 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 j�| j|�}t�|d�}|�|d d� |�|d |� |�|d |� |�� tj� d |� t
� |d
�}| � ||� t
� |d �}| � ||� g }x.t
�
|g�D ] } | �| t
j� |�| j� q�W | � |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 j�| j|�}t �|d� | �t j|� x"tj| jgd�D ]}| � 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 j�| j|�}t �|� tt j�|d�d�}|�� t �t j�|d�� tt j�|dd�d�}|�� tt j�|dd�d�}|�� d}t j�| j|�}t �|� tt j�|d�d�}|�� t �t j�|d�� tt j�|dd�d�}|�� tt j�|dd�d�}|�� dddddd g}d
d � t�| jg�D �}| � ||� x&|D ] }|�
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 j�| j|�}t�|d�}|�|d d� |�|d | d d� |�|d | d d� |�|d d� |�|d | d d� |�|d | d d� |�� tj� d |� dd
d dd d
g}dd� t
� |g�D �}| � ||� tjd = x"|D ]}|�
d��rq�tj|= q�W d S )Nz 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? )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_zipfilec C s\ d}| � ttf�� tt�|�� W d Q R X d}| � ttf�� tt�|�� W d Q R X d S )NZtest_dirs test_dir)� assertRaises� TypeError�
ValueError�listr r6 )r Z str_inputZ bytes_inputr r r �2test_walk_packages_raises_on_string_or_bytes_input� s z?PkgutilTests.test_walk_packages_raises_on_string_or_bytes_inputN) �__name__�
__module__� __qualname__r r r r( r9 r: rE rK 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 �dd�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)rL rM rN rQ rU r r r r r � MyTestLoader� s rV c @ s e Zd Zddd�ZdS )z!PkgutilPEP302Tests.MyTestImporterNc C s t �� }t|d|jj |g d�S )Nz<%s>)�loader�submodule_search_locations)rO rV r � __class__rL )r �fullnamer �targetrW r r r � find_spec� s
z+PkgutilPEP302Tests.MyTestImporter.find_spec)NN)rL rM rN r\ r r r r �MyTestImporter� s r] c C s t j�d| �� � d S )Nr )r � meta_pathr r] )r r r r r � s zPkgutilPEP302Tests.setUpc C s t jd= d S )Nr )r r^ )r r r r r � s zPkgutilPEP302Tests.tearDownc C s | � t�dd�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 }| �|jd� | �t�dd�d� | �|jd� tjd= d S )Nr r; r_ r` z
Hello, world!)r_ r! rR r r r r" )r r_ r r r �test_alreadyloaded� s
z%PkgutilPEP302Tests.test_alreadyloadedN)
rL rM rN �objectrV r] r r ra rb r r r r rO � s rO c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ExtendPathTestsc C s\ t �� }tj�d|� tj�||�}t�|� ttj�|d�d��}|� 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 j�|||d �}t|d��}td�|�|d� W d Q R X d S )Nz.pyr) zvalue={})�file)r r r r �print�format)r r
re Zsubmodule_name�valueZ module_namerf r r r �create_submodule� s z ExtendPathTests.create_submodulec C s� d}| � |�}| � |�}| �||dd� | �||dd� dd l}dd l}| �|jjd� | �|jjd� | �t|j �tt
j � ||�t
j � ||�g�� t
�|� t
�|� tj d= tj d= tjd= tjd= tjd= d S )Nr_ �barr �bazr; zfoo.barzfoo.baz)rg rl �foo.bar�foo.bazr! rm rk rn �sorted�__path__r r r r r
r r" )r re � dirname_0� dirname_1r_ r r r � test_simple� s&
zExtendPathTests.test_simplec
C s� t j}t j}d}d}| �|�}tj�||�}d�||�}tj �
|d � tj �
|d � z�| � |||d� t �
|� t||��}||�} x||D ]t}
t �|
|�} | j} y
| j} W n tk
r� Y nX | �|
t jj� | �|
| � | �| t jj� | �t �|
|�� q�W | �t�� t|d�� W d Q R X | �t�� t|d�� W d Q R X W d t�|� 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_importerrg r r r rj r r" �poprl � importlib�
import_modulerJ � _get_specrW �AttributeErrorr. � machinery�
FileFinderr! �SourceFileLoader� assertIsNonerG � ImportErrorr r
�KeyError)
r rw rx re �modnamer
ZpathitemrZ Z importersZexpected_importer�finderrP rW r r r �test_iter_importers sJ
z#ExtendPathTests.test_iter_importersc C s� d}| � |�}| � |�}| �||dd� t�tj�||d�� | �||dd� dd l}dd l}| �|j j
d� | �|j j
d� | �t |j
�t tj�||�tj�||�g�� t�|� t�|� tjd= tjd= tjd= tjd= tjd= d S ) Nr_ rm r z __init__.pyrn r; zfoo.barzfoo.baz)rg rl r �unlinkr r ro rp r! rm rk rn rq rr r r
r r" )r re rs rt r_ r r r �test_mixed_namespaceI s(
z$ExtendPathTests.test_mixed_namespaceN)rL rM rN rg rl ru r� r� r r r r rd � s
0rd c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�NestedNamespacePackageTestc C s t �� | _tjd d � | _d S )N)r r �basedirr r �old_path)r r r r r j s
z NestedNamespacePackageTest.setUpc C s | j tjd d �< t�| j� d S )N)r� r r r r
r� )r r r r r n s z#NestedNamespacePackageTest.tearDownc C sl |� dd�\}}tj�| j|�dtjj��}tj|dd� ttj�||d �d��}|� |� 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 rQ r s
z(NestedNamespacePackageTest.create_modulec C s d}| � d|� | � d|� | � d|� | � d|� | � dd� | � dd � tj�d
tj�| jd �� tj�d
tj�| jd �� d
d l}| �t d
� | �
t |j �d� d
d l
}| �t d� | �
t |jj �d� d
dlm} d
dlm} | �
|d� | �
|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; )rQ r r r r r r� r# r r r! �lenrr Z
pkg.subpkgZsubpkgZ pkg.subpkg.cr� Z pkg.subpkg.dr7 )r Zpkgutil_boilerplater# r� r7 r r r � test_nestedy s& z&NestedNamespacePackageTest.test_nestedN)rL rM rN r r rQ r� r r r r r� h s r� c @ s| e Zd Zdd� Zdd� Zdd� Zdd� Ze�e 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" | � � � t�d� 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( | � � � t�dddd� 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}| �t�d�� | �t�d�� | �t�d�� | �t|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� �(}| �t�t�� | �t|j�d� W d Q R X W d |a X d S )Nr ) �
__loader__r r� r r� rL 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|< t�|�}W d Q R X | �|� d S )Nrv )�typer �__spec__r r" r r� r� )r r0 r? rW 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|< t�|�}W d Q R X | �|� d S )Nrv )r� r r� r r" r r� r� )r r0 r? rW 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|< zt�|�}W d t j|= X | �|� d S )Nz
totally bogus)r r" r r� r� )r r0 rW r r r �#test_get_loader_None_in_sys_modules� s
z;ImportlibMigrationTests.test_get_loader_None_in_sys_modulesc C s d}t �|�}| �|� d S )Nz
totally bogus)r � find_loaderr� )r r0 rW r r r � test_find_loader_missing_module� s
z7ImportlibMigrationTests.test_find_loader_missing_modulec C sX t � �H}| �t�d�� | �t�d�� | �t�d�� | �t|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 � �(}| �t�d�� | �t|j�d� W d Q R X d S )Nz*??r )r r� r rx r! r� r� )r r) r r r �"test_get_importer_avoids_emulation� s z:ImportlibMigrationTests.test_get_importer_avoids_emulationc C s: t � �*}xt�� D ]}qW | �t|j�d� W d Q R X d S )Nr )r r rw r! r� r� )r r) Zimporterr r r �$test_iter_importers_avoids_emulation� s z<ImportlibMigrationTests.test_iter_importers_avoids_emulationN)rL rM rN 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}| j� � |�
� d S )Nr ) r r rO rd r� r� � zipimportrz �_zip_directory_cache�clear�invalidate_caches)r� rz r r r � test_main� s
r� r� )Z test.supportr r r r r� r rz �importlib.utilr r r Zos.pathr r r+ ZTestCaser rO rd r� r� r� rL r r r r �<module> s& ,-})X