File: //opt/cppython/lib/python3.8/test/test_importlib/__pycache__/test_namespace_pkgs.cpython-38.pyc
U
/D�gE* � @ 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 5 | 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 �F/opt/cppython/lib/python3.8/test/test_importlib/test_namespace_pkgs.py�sys_modules_context s
r
c
k sT | � dtj� | � dtj� tjf | �}|� t� � dV W 5 Q R X W 5 Q 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 �portion1 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 5 Q R X d S �Nr �� assertRaises� ImportError�foo.twor. 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+ �reprr. r r r �test_module_reprS s z'SingleNamespacePackage.test_module_reprN)r r r! r r0 r6 r8 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 5 Q R X tj� t
j� | j d�� dd l}| �|j
jd� d S )Nr r( �portion2�portion2 foo two)r* r+ r, r- r3 r4 r5 r r �appendr r r �twor. r r r �test_dynamic_path[ s z-DynamicPathNamespacePackage.test_dynamic_pathN)r r r! r r> r r r r r9 X s r9 c @ s e Zd ZdgZdd� ZdS )�CombinedNamespacePackages�
both_portionsc C s4 dd l }dd l}| �|jjd� | �|jjd� d S �Nr zboth_portions foo one�both_portions foo two�r* r5 r+ r, r- r= r. r r r � test_importsn s z&CombinedNamespacePackages.test_importsN�r r r! r rD r r r r r? k s r? c @ s e Zd ZddgZdd� ZdS )�SeparatedNamespacePackagesr& r: c C s4 dd l }dd l}| �|jjd� | �|jjd� d S )Nr r( r; rC r. r r r rD x s z'SeparatedNamespacePackages.test_importsNrE r r r r rF u s rF c @ s, e Zd ZddgZdd� Zdd� Zdd� Zd S )
�%SeparatedOverlappingNamespacePackagesr& r@ c C s4 dd l }dd l}| �|jjd� | �|jjd� d S �Nr r( rB rC 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 rA ) r r �reverser* r5 r+ r, r- 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 rH )r5 r* r+ r, r- 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 rI rK rL r r r r rG s rG c @ s" e Zd ZdgZdd� Zdd� ZdS )�SingleZipNamespacePackage�top_level_portion1.zipc C s dd l }| �|jjd� d S r' r) r. r r r r0 � s z-SingleZipNamespacePackage.test_simple_packagec C s" | � t�� dd l}W 5 Q R X d S r1 r2 r. r r r r6 � s z0SingleZipNamespacePackage.test_cant_import_otherN�r r r! r r0 r6 r r r r rM � s rM c @ s e Zd ZddgZdd� ZdS )� SeparatedZipNamespacePackagesrN r: c C sT dd l }dd l}| �|jjd� | �|jjd� | �d|jj� | �d|jj� d S )Nr r( r; rN �.zip) r* r5 r+ r, r- r= �assertInr$ � assertNotInr. r r r rD � s z*SeparatedZipNamespacePackages.test_importsNrE r r r r rP � s rP c @ s" e Zd ZdgZdd� Zdd� ZdS )� SingleNestedZipNamespacePackage�#nested_portion1.zip/nested_portion1c C s dd l }| �|jjd� d S r' r) r. r r r r0 � s z3SingleNestedZipNamespacePackage.test_simple_packagec C s" | � t�� dd l}W 5 Q R X d S r1 r2 r. r r r r6 � s z6SingleNestedZipNamespacePackage.test_cant_import_otherNrO r r r r rT � s rT c @ s e Zd ZddgZdd� ZdS )�#SeparatedNestedZipNamespacePackagesrU r: c C sb dd l }dd l}| �|jjd� | �|jjd� tj�dd�}| � ||jj
� | � d|jj
� d S )Nr r( r; znested_portion1.zipZnested_portion1rQ ) r* r5 r+ r, r- r= r r r rR r$ rS )r r/ �fnr r r rD � s z0SeparatedNestedZipNamespacePackages.test_importsNrE r r r r rV � s rV c @ s e Zd ZddddgZdd� ZdS )�
LegacySupportZnot_a_namespace_pkgr& r: r@ c C sN dd l }| �t�� dd l}W 5 Q R X | �d|j� | �dt|j�� � � d S )Nr �__init__� namespace)
r* r3 r4 r5 rR r$ rS �str�
__loader__�lowerr. r r r �+test_non_namespace_package_takes_precedence� s
z9LegacySupport.test_non_namespace_package_takes_precedenceN)r r r! r r^ r r r r rX � s rX 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 5 Q R X | �t|j�d� | �t|jj�d� d S )Nr � �parent child one�parent child two) �parent.child.oner+ �len�__path__�child�parent.child.twor, r- r= r3 r4 �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 5 Q R X t j
�tj
�| jd�� dd l }| �t|j�d� | �t|jj�d� | �|jjjd� d S )Nr r` ra rb Zproject3� zparent child three)rc r+ rd re rf rg r, r- r= r3 r4 rh r r r< r r r Zthreeri r r r �test_project3_succeeds� s z-DynamicPathCalculation.test_project3_succeedsN)r r r! r rk rm r r r r r_ � s r_ c @ s( e Zd ZdgZejdd� �Zdd� ZdS )�ZipWithMissingDirectoryzmissing_directory.zipc C s dd l }d S r1 )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+ r= r- )r Zbarr r r �test_present_directory s z.ZipWithMissingDirectory.test_present_directoryN)r r r! r �unittestZexpectedFailurero rp r r r r rn � s
rn 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 rs r r r �$test_module_before_namespace_package s zGModuleAndNamespacePackageInSameDir.test_module_before_namespace_packageN)r r r! r rt r r r r rr s rr c @ s* e Zd ZdgZdd� Zdd� Zdd� ZdS ) � ReloadTestsr& c C s& dd l }t�|�}| �|jjd� d S r' )r* � importlib�reloadr+ r, r- r. r r r r0 % s
z ReloadTests.test_simple_packagec C sR dd l }| �t�� dd l}W 5 Q R X t�|�}| �t�� dd l}W 5 Q R X d S r1 )r/ r3 r4 r5 rv rw r. r r r r6 * s
z"ReloadTests.test_cant_import_otherc C sd dd l }| �t�� dd l}W 5 Q R X tj�tj�| j d�� t
� |�}dd l}| � |j
jd� d S )Nr r: r; )r* r3 r4 r5 r r r< r r r rv rw r+ r= r- r. r r r r> 2 s
z ReloadTests.test_dynamic_pathN)r r r! r r0 r6 r> r r r r ru " s ru c @ s* e Zd ZdgZdd� Zdd� Zdd� ZdS ) � LoaderTestsr& c C s* dd l }| �|j|jj� | �|j� d S r1 )r/ r+ r\ �__spec__�loaderZassertIsNotNoner. r r r �!test_namespace_loader_consistencyC s z-LoaderTests.test_namespace_loader_consistencyc C s& dd l }| �|jj� | �|j� d S r1 )r/ Z assertIsNonery �originr$ r. r r r �!test_namespace_origin_consistencyI s z-LoaderTests.test_namespace_origin_consistencyc C s0 dd l }tj�| jdd�}| �|jd |� d S )Nr r& r/ )r/ r r r r r+ re )r r/ Z
expected_pathr r r �test_path_indexableO s z LoaderTests.test_path_indexableN)r r r! r r{ r} r~ r r r r rx @ s rx �__main__) �
contextlibrv r r rq Ztest.test_importlibr �contextmanagerr
r ZTestCaser r% r9 r? rF rG rM rP rT rV rX r_ rn rr ru rx r �mainr r r r �<module> s6
-