File: //usr/local/lib/python3.10/test/test_importlib/__pycache__/test_namespace_pkgs.cpython-310.pyc
o
�it0 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ e j dd� �Z
e j dd� �Z G dd� dej �Z
G d d
� d
e
�ZG d d � d e
�ZG 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
�Z G d'd(� d(e
�Z e d)kr�e� � dS dS )*� N)�utilc c sZ � t j} t j�� }zdV W | t _t j�� t j�|� dS | t _t j�� t j�|� w )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 �D/usr/local/lib/python3.10/test/test_importlib/test_namespace_pkgs.py�sys_modules_context s �
�
r
c k s� � | � dtj� | � dtj� tjdi | ��}|�' t� � dV W d � n1 s+w Y W d � dS W d � dS 1 sCw Y 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_hooksNr )�
setdefaultr r r r Z import_stater
)�kwargsZimport_contextr r r �namespace_tree_context& s � P�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 �tearDownE s z NamespacePackageTest.tearDownN) �__name__�
__module__� __qualname__�__doc__r r r �dirname�__file__r r r r r r r r 7 s
r c @ �* e Zd ZdgZdd� Zdd� Zdd� ZdS ) �SingleNamespacePackage�portion1c C � dd l }| �|jjd� d S �Nr �portion1 foo one��foo.one� assertEqual�one�attr�r �foor r r �test_simple_packageM � z*SingleNamespacePackage.test_simple_packagec C �8 | � t��
dd l}W d � d S 1 sw Y d S �Nr �� assertRaises� ImportError�foo.twor0 r r r �test_cant_import_otherQ �
"�z-SingleNamespacePackage.test_cant_import_otherc C sV dd l }t�� � t�d� | �|jj�|�d� W d � d S 1 s$w Y d S )Nr �ignorez<module 'foo' (namespace)>)r, �warnings�catch_warnings� simplefilterr- �__spec__�loader� module_reprr0 r r r �test_module_reprU s
�"�z'SingleNamespacePackage.test_module_reprN)r r r! r r2 r: rC r r r r r& J s
r& c @ � e Zd ZdgZdd� ZdS )�DynamicPathNamespacePackager' c C s~ dd l }| �|jjd� | �t�� dd l}W d � n1 s w Y tj� t
j� | j d�� dd l}| �|j
jd� d S )Nr r* �portion2�portion2 foo two)r, r- r. r/ r7 r8 r9 r r �appendr r r �twor0 r r r �test_dynamic_path` s
�z-DynamicPathNamespacePackage.test_dynamic_pathN)r r r! r rJ r r r r rE ] � rE c @ rD )�CombinedNamespacePackages�
both_portionsc C �4 dd l }dd l}| �|jjd� | �|jjd� d S �Nr zboth_portions foo one�both_portions foo two�r, r9 r- r. r/ rI r0 r r r � test_importss � z&CombinedNamespacePackages.test_importsN�r r r! r rR r r r r rL p rK rL c @ � e Zd ZddgZdd� ZdS )�SeparatedNamespacePackagesr' rF c C rN )Nr r* rG rQ r0 r r r rR } rS z'SeparatedNamespacePackages.test_importsNrT r r r r rV z � rV c @ rD )�-SeparatedNamespacePackagesCreatedWhileRunningr' c C s4 t �� ��}tj�|� dd l}| �|jjd� | � t
�� dd l }W d � n1 s+w Y t j�
|d�}t �|� t j�
|d�}t|ddd��
}|�d� W d � n1 sZw Y | � t
�� dd l }W d � n1 ssw Y t�� dd l }| �|jjd � W d � d S 1 s�w Y d S )
Nr r* r1 zjust_created.py�wzutf-8)�encodingzattr = "just_created foo"zjust_created foo)�tempfile�TemporaryDirectoryr r rH r, r- r. r/ r7 r8 Zfoo.just_createdr r �mkdir�open�write� importlib�invalidate_cachesZ just_created)r Ztemp_dirr1 �namespace_pathZ module_path�filer r r �test_invalidate_caches� s(
�
�
�"�zDSeparatedNamespacePackagesCreatedWhileRunning.test_invalidate_cachesN)r r r! r rd r r r r rX � rK rX c @ s, e Zd ZddgZdd� Zdd� Zdd� Zd S )
�%SeparatedOverlappingNamespacePackagesr' rM c C rN �Nr r* rP rQ r0 r r r �test_first_path_wins� rS z:SeparatedOverlappingNamespacePackages.test_first_path_winsc C s> t j�� dd l}dd l}| �|jjd� | �|jjd� d S rO ) r r �reverser, r9 r- r. r/ rI r0 r r r �test_first_path_wins_again� s
z@SeparatedOverlappingNamespacePackages.test_first_path_wins_againc C rN rf )r9 r, r- r. r/ rI r0 r r r �+test_first_path_wins_importing_second_first� rS zQSeparatedOverlappingNamespacePackages.test_first_path_wins_importing_second_firstN)r r r! r rg ri rj r r r r re � s
re c @ �" e Zd ZdgZdd� Zdd� ZdS )�SingleZipNamespacePackage�top_level_portion1.zipc C r( r) r+ r0 r r r r2 � r3 z-SingleZipNamespacePackage.test_simple_packagec C r4 r5 r6 r0 r r r r: � r; z0SingleZipNamespacePackage.test_cant_import_otherN�r r r! r r2 r: r r r r rl � � rl c @ rU )� SeparatedZipNamespacePackagesrm rF c C sT dd l }dd l}| �|jjd� | �|jjd� | �d|jj� | �d|jj� d S )Nr r* rG rm �.zip) r, r9 r- r. r/ rI �assertInr$ � assertNotInr0 r r r rR � s z*SeparatedZipNamespacePackages.test_importsNrT r r r r rp � rW rp c @ rk )� SingleNestedZipNamespacePackage�#nested_portion1.zip/nested_portion1c C r( r) r+ r0 r r r r2 � r3 z3SingleNestedZipNamespacePackage.test_simple_packagec C r4 r5 r6 r0 r r r r: � r; z6SingleNestedZipNamespacePackage.test_cant_import_otherNrn r r r r rt � ro rt c @ rU )�#SeparatedNestedZipNamespacePackagesru rF c C sb dd l }dd l}| �|jjd� | �|jjd� tj�dd�}| � ||jj
� | � d|jj
� d S )Nr r* rG znested_portion1.zipZnested_portion1rq ) r, r9 r- r. r/ rI r r r rr r$ rs )r r1 �fnr r r rR � s z0SeparatedNestedZipNamespacePackages.test_importsNrT r r r r rv � rW rv c @ s e Zd Zg d�Zdd� ZdS )�
LegacySupport)Znot_a_namespace_pkgr' rF rM c C sb dd l }| �t�� dd l}W d � n1 sw Y | �d|j� | �dt|j�� � � d S )Nr �__init__� namespace)
r, r7 r8 r9 rr r$ rs �str�
__loader__�lowerr0 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 � rW 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 d � n1 sTw Y | �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/ rI r7 r8 �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 � n1 sTw Y t j
�tj
�| jd�� dd l }| �t|j�d� | �t|jj�d� | �|jjjd� d S )Nr r� r� r� Zproject3� zparent child three)r� r- r� r� r� r� r. r/ rI r7 r8 r� r r rH r r r Zthreer� r r r �test_project3_succeeds
s
�z-DynamicPathCalculation.test_project3_succeedsN)r r r! r r� r� r r r r r � s r c @ s( e Zd ZdgZejdd� �Zdd� ZdS )�ZipWithMissingDirectoryzmissing_directory.zipc C s dd l }d S r5 )r, r0 r r r �test_missing_directory) s z.ZipWithMissingDirectory.test_missing_directoryc C r( )Nr zmissing_directory foo two)Zbar.twor- rI r/ )r Zbarr r r �test_present_directory9 s z.ZipWithMissingDirectory.test_present_directoryN)r r r! r �unittestZexpectedFailurer� r� r r r r r� & s
r� c @ rD )�"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_packageB s zGModuleAndNamespacePackageInSameDir.test_module_before_namespace_packageN)r r r! r r� r r r r r� ? rK r� c @ r% ) � ReloadTestsr' c C s& dd l }t�|�}| �|jjd� d S r) )r, r` �reloadr- r. r/ r0 r r r r2 L s
z ReloadTests.test_simple_packagec C s| dd l }| �t�� dd l}W d � n1 sw Y t�|�}| �t��
dd l}W d � d S 1 s7w Y d S r5 )r1 r7 r8 r9 r` r� r0 r r r r: Q s
�
"�z"ReloadTests.test_cant_import_otherc C sx dd l }| �t�� dd l}W d � n1 sw Y tj�tj�| j d�� t
� |�}dd l}| � |j
jd� d S )Nr rF rG )r, r7 r8 r9 r r rH r r r r` r� r- rI r/ r0 r r r rJ Y s
�
z ReloadTests.test_dynamic_pathN)r r r! r r2 r: rJ r r r r r� I s
r� c @ r% ) � LoaderTestsr' c C s* dd l }| �|j|jj� | �|j� d S r5 )r1 r- r| r@ rA ZassertIsNotNoner0 r r r �!test_namespace_loader_consistencyj s z-LoaderTests.test_namespace_loader_consistencyc C s& dd l }| �|jj� | �|j� d S r5 )r1 Z assertIsNoner@ �originr$ r0 r r r �!test_namespace_origin_consistencyp s z-LoaderTests.test_namespace_origin_consistencyc C s0 dd l }tj�| jdd�}| �|jd |� d S )Nr r' r1 )r1 r r r r r- r� )r r1 Z
expected_pathr r r �test_path_indexablev s z LoaderTests.test_path_indexableN)r r r! r r� r� r� r r r r r� g s
r� �__main__) �
contextlibr` r r r[ r� r= Ztest.test_importlibr �contextmanagerr
r ZTestCaser r& rE rL rV rX re rl rp rt rv rx r r� r� r� r� r �mainr r r r �<module> s@
"
-
�