File: //usr/local/lib/python3.7/test/test_importlib/__pycache__/test_namespace_pkgs.cpython-37.pyc
B
��g~) � @ sh d dl Z d dlZd dlZd dlZd dlZd dlmZ e jdd� �Ze jdd� �Z G dd� dej
�Z G d d
� d
e �Z G d d � d e �Z
G d
d� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd � d e �ZG d d � d e �ZG d d � d e �ZG d!d"� d"e �ZG d#d$� d$e �ZG d%d&� d&e �Zed'k�rde� � dS )(� N)�utilc c s> t j} t j�� }z
dV W d| t _t j�� t j�|� X dS )z�
Make sure sys.modules is the same object and has the same content
when exiting the context as when entering.
Similar to importlib.test.util.uncache, but doesn't require explicit
names.
N)�sys�modules�copy�clear�update)Zsys_modules_savedZsys_modules_copy� r �C/usr/local/lib/python3.7/test/test_importlib/test_namespace_pkgs.py�sys_modules_context s
r
c
k sT | � dtj� | � dtj� tjf | �}|� t� � dV W dQ R X W dQ R X dS )z�
Save import state and sys.modules cache and restore it on exit.
Typical usage:
>>> with namespace_tree_context(path=['/tmp/xxyy/portion1',
... '/tmp/xxyy/portion2']):
... pass
� meta_path�
path_hooksN)�
setdefaultr r r r Z import_stater
)�kwargsZimport_contextr r r �namespace_tree_context$ s
r c @ s6 e Zd ZdZej�ej�e�d�Z dd� Z
dd� Z dS )�NamespacePackageTestzj
Subclasses should define self.root and self.paths (under that root)
to be added to sys.path.
Znamespace_pkgsc s2 � fdd�� j D �� _t� jd�� _� j�� d S )Nc s g | ]}t j�� j|��qS r )�os�path�join�root)�.0r )�selfr r �
<listcomp>> s z.NamespacePackageTest.setUp.<locals>.<listcomp>)r )�pathsZresolved_pathsr �ctx� __enter__)r r )r r �setUp< s zNamespacePackageTest.setUpc C s | j �d d d � d S )N)r �__exit__)r r r r �tearDownC s z NamespacePackageTest.tearDownN) �__name__�
__module__� __qualname__�__doc__r r r �dirname�__file__r r r r r r r r 5 s r c @ s* e Zd ZdgZdd� Zdd� Zdd� ZdS ) �SingleNamespacePackage�portion1c C s dd l }| �|jjd� d S )Nr zportion1 foo one)�foo.one� assertEqual�one�attr)r �foor r r �test_simple_packageK s z*SingleNamespacePackage.test_simple_packagec C s" | � t�� dd l}W d Q R X d S )Nr )� assertRaises� ImportError�foo.two)r r* r r r �test_cant_import_otherO s z-SingleNamespacePackage.test_cant_import_otherc C s dd l }| �t|�d� d S )Nr z<module 'foo' (namespace)>)r& r' �repr)r r* r r r �test_module_reprS s z'SingleNamespacePackage.test_module_reprN)r r r r r+ r/ r1 r r r r r$ H s r$ c @ s e Zd ZdgZdd� ZdS )�DynamicPathNamespacePackager% c C sj dd l }| �|jjd� | �t�� dd l}W d Q R X tj� t
j� | j d�� dd l}| �|j
jd� d S )Nr zportion1 foo one�portion2zportion2 foo two)r& r' r( r) r, r- r. r r �appendr r r �two)r r* r r r �test_dynamic_path[ s z-DynamicPathNamespacePackage.test_dynamic_pathN)r r r r r6 r r r r r2 X s r2 c @ s e Zd ZdgZdd� ZdS )�CombinedNamespacePackages�
both_portionsc C s4 dd l }dd l}| �|jjd� | �|jjd� d S )Nr zboth_portions foo onezboth_portions foo two)r& r. r' r( r) r5 )r r* r r r � test_importsn s z&CombinedNamespacePackages.test_importsN)r r r r r9 r r r r r7 k s r7 c @ s e Zd ZddgZdd� ZdS )�SeparatedNamespacePackagesr% r3 c C s4 dd l }dd l}| �|jjd� | �|jjd� d S )Nr zportion1 foo onezportion2 foo two)r& r. r' r( r) r5 )r r* r r r r9 x s z'SeparatedNamespacePackages.test_importsN)r r r r r9 r r r r r: u s r: c @ s, e Zd ZddgZdd� Zdd� Zdd� Zd S )
�%SeparatedOverlappingNamespacePackagesr% r8 c C s4 dd l }dd l}| �|jjd� | �|jjd� d S )Nr zportion1 foo onezboth_portions foo two)r& r. r' r( r) r5 )r r* r r r �test_first_path_wins� s z:SeparatedOverlappingNamespacePackages.test_first_path_winsc C s> t j�� dd l}dd l}| �|jjd� | �|jjd� d S )Nr zboth_portions foo onezboth_portions foo two) r r �reverser& r. r' r( r) r5 )r r* r r r �test_first_path_wins_again� s
z@SeparatedOverlappingNamespacePackages.test_first_path_wins_againc C s4 dd l }dd l}| �|jjd� | �|jjd� d S )Nr zportion1 foo onezboth_portions foo two)r. r& r' r( r) r5 )r r* r r r �+test_first_path_wins_importing_second_first� s zQSeparatedOverlappingNamespacePackages.test_first_path_wins_importing_second_firstN)r r r r r< r> r? r r r r r; s r; c @ s" e Zd ZdgZdd� Zdd� ZdS )�SingleZipNamespacePackageztop_level_portion1.zipc C s dd l }| �|jjd� d S )Nr zportion1 foo one)r& r' r( r) )r r* r r r r+ � s z-SingleZipNamespacePackage.test_simple_packagec C s" | � t�� dd l}W d Q R X d S )Nr )r, r- r. )r r* r r r r/ � s z0SingleZipNamespacePackage.test_cant_import_otherN)r r r r r+ r/ r r r r r@ � s r@ c @ s e Zd ZddgZdd� ZdS )� SeparatedZipNamespacePackagesztop_level_portion1.zipr3 c C sT dd l }dd l}| �|jjd� | �|jjd� | �d|jj� | �d|jj� d S )Nr zportion1 foo onezportion2 foo twoztop_level_portion1.zipz.zip) r& r. r' r( r) r5 �assertInr# � assertNotIn)r r* r r r r9 � s z*SeparatedZipNamespacePackages.test_importsN)r r r r r9 r r r r rA � s rA c @ s" e Zd ZdgZdd� Zdd� ZdS )� SingleNestedZipNamespacePackagez#nested_portion1.zip/nested_portion1c C s dd l }| �|jjd� d S )Nr zportion1 foo one)r& r' r( r) )r r* r r r r+ � s z3SingleNestedZipNamespacePackage.test_simple_packagec C s" | � t�� dd l}W d Q R X d S )Nr )r, r- r. )r r* r r r r/ � s z6SingleNestedZipNamespacePackage.test_cant_import_otherN)r r r r r+ r/ r r r r rD � s rD c @ s e Zd ZddgZdd� ZdS )�#SeparatedNestedZipNamespacePackagesz#nested_portion1.zip/nested_portion1r3 c C sb dd l }dd l}| �|jjd� | �|jjd� tj�dd�}| � ||jj
� | � d|jj
� d S )Nr zportion1 foo onezportion2 foo twoznested_portion1.zipZnested_portion1z.zip) r& r. r' r( r) r5 r r r rB r# rC )r r* �fnr r r r9 � s z0SeparatedNestedZipNamespacePackages.test_importsN)r r r r r9 r r r r rE � s rE c @ s e Zd ZddddgZdd� ZdS )�
LegacySupportZnot_a_namespace_pkgr% r3 r8 c C sN dd l }| �t�� dd l}W d Q R X | �d|j� | �dt|j�� � � d S )Nr �__init__� namespace)
r& r, r- r. rB r# rC �str�
__loader__�lower)r r* r r r �+test_non_namespace_package_takes_precedence� s
z9LegacySupport.test_non_namespace_package_takes_precedenceN)r r r r rM r r r r rG � s rG c @ s$ e Zd ZddgZdd� Zdd� ZdS )�DynamicPathCalculationZproject1Zproject2c C s� dd l }| �t|j�d� | �t|jj�d� dd l}| �t|j�d� | �t|jj�d� | �|jjjd� | �|jjjd� | � t
�� dd l }W d Q R X | �t|j�d� | �t|jj�d� d S )Nr � zparent child onezparent child two) �parent.child.oner' �len�__path__�child�parent.child.twor( r) r5 r, r- �parent.child.three)r �parentr r r �test_project3_fails� s z*DynamicPathCalculation.test_project3_failsc C s� dd l }| �t|j�d� | �t|jj�d� dd l}| �t|j�d� | �t|jj�d� | �|jjjd� | �|jjjd� | � t
�� dd l }W d Q R X t j
�tj
�| jd�� dd l }| �t|j�d� | �t|jj�d� | �|jjjd� d S )Nr rO zparent child onezparent child twoZproject3� zparent child three)rP r' rQ rR rS rT r( r) r5 r, r- rU r r r4 r r r Zthree)r rV r r r �test_project3_succeeds� s z-DynamicPathCalculation.test_project3_succeedsN)r r r r rW rY r r r r rN � s rN c @ s( e Zd ZdgZejdd� �Zdd� ZdS )�ZipWithMissingDirectoryzmissing_directory.zipc C s dd l }d S )Nr )r& )r r* r r r �test_missing_directory s z.ZipWithMissingDirectory.test_missing_directoryc C s dd l }| �|jjd� d S )Nr zmissing_directory foo two)Zbar.twor' r5 r) )r Zbarr r r �test_present_directory s z.ZipWithMissingDirectory.test_present_directoryN)r r r r �unittestZexpectedFailurer[ r\ r r r r rZ � s rZ c @ s e Zd ZdgZdd� ZdS )�"ModuleAndNamespacePackageInSameDirZ module_and_namespace_packagec C s dd l }| �|jd� d S )Nr z in module)�a_testr' r) )r r_ r r r �$test_module_before_namespace_package s zGModuleAndNamespacePackageInSameDir.test_module_before_namespace_packageN)r r r r r` r r r r r^ s r^ c @ s* e Zd ZdgZdd� Zdd� Zdd� ZdS ) � ReloadTestsr% c C s& dd l }t�|�}| �|jjd� d S )Nr zportion1 foo one)r& � importlib�reloadr' r( r) )r r* r r r r+ % s
z ReloadTests.test_simple_packagec C sR dd l }| �t�� dd l}W d Q R X t�|�}| �t�� dd l}W d Q R X d S )Nr )r* r, r- r. rb rc )r r* r r r r/ * s
z"ReloadTests.test_cant_import_otherc C sd dd l }| �t�� dd l}W d Q R X tj�tj�| j d�� t
� |�}dd l}| � |j
jd� d S )Nr r3 zportion2 foo two)r& r, r- r. r r r4 r r r rb rc r' r5 r) )r r* r r r r6 2 s
z ReloadTests.test_dynamic_pathN)r r r r r+ r/ r6 r r r r ra " s ra c @ s" e Zd ZdgZdd� Zdd� ZdS )� LoaderTestsr% c C s* dd l }| �|j|jj� | �|j� d S )Nr )r* r' rK �__spec__�loaderZassertIsNotNone)r r* r r r �!test_namespace_loader_consistencyC s z-LoaderTests.test_namespace_loader_consistencyc C s& dd l }| �|jj� | �|j� d S )Nr )r* Z assertIsNonere �originr# )r r* r r r �!test_namespace_origin_consistencyI s z-LoaderTests.test_namespace_origin_consistencyN)r r r r rg ri r r r r rd @ s rd �__main__) �
contextlibrb r r r] Ztest.test_importlibr �contextmanagerr
r ZTestCaser r$ r2 r7 r: r; r@ rA rD rE rG rN rZ r^ ra rd r �mainr r r r �<module> s2
-