File: //lib/python3/dist-packages/twisted/test/__pycache__/test_modules.cpython-38.pyc
U
W[�E � @ s d Z ddlmZmZ ddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ dd lmZ dd
lmZ G d d � d ee�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z dddddgZ dS )zV
Tests for twisted.python.modules, abstract access to imported or importable
objects.
� )�division�absolute_importN)�modules)�
networkString)�FilePath)�namedAny)�TestCase)�TwistedModulesMixin)�zipitc @ s e Zd ZdZedfdd�ZdS )�TwistedModulesTestCasez/
Base class for L{modules} test cases.
Fc C s8 |j |d�D ]}|j|kr | S q | �d|f � dS )z�
You don't ever actually want to do this, so it's not in the public
API, but sometimes we want to compare the result of an iterative call
with a lookup call and make sure they're the same for test purposes.
��importPackagesz+Unable to find module %r through iteration.N)� walkModules�nameZfail)�self�modname�wherer
�modinfo� r �;/usr/lib/python3/dist-packages/twisted/test/test_modules.py�findByIteration! s
z&TwistedModulesTestCase.findByIterationN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
BasicTestsc
C s0 t d� d}| �� }|�d�}|�d��|� |�d�}|�� |�d��|� |�d��d� | �� }|�d�}|�d��|� |�d�}|�� |�d��|� |�d��d� | �|j|jg� t�d�}zd d
� |j
d d �D �} W 5 t t
j� � �D ]}|� d�r�t
j|= q�X dd
ddg}
| �| |
� dS )zd
Duplicate packages are not yielded when iterating over namespace
packages.
�pkgutilsB import pkgutil; __path__ = pkgutil.extend_path(__path__, __name__)� test_package� __init__.pyZnested_packagez module.py� z
module2.pyc S s g | ]
}|j �qS r �r )�.0�modr r r �
<listcomp>m s z6BasicTests.test_namespacedPackages.<locals>.<listcomp>Tr ztest_package.nested_packagez"test_package.nested_package.modulez#test_package.nested_package.module2N)�
__import__�pathEntryWithOnePackage�child�
setContent�makedirs�replaceSysPath�pathr � getModule�list�sys�keys�
startswithr � assertEqual) r ZnamespaceBoilerplate�entryZtestPackagePathZ nestedEntryZ anotherEntryZanotherPackagePathZanotherNestedEntry�moduleZ walkedNames�expectedr r r �test_namespacedPackages0 s@ �
�
�z"BasicTests.test_namespacedPackagesc C sH g }t j| �� jg|ji i ddid�}| �|g � | �|d �� � dS )a
If a package has been explicitly forbidden from importing by setting a
L{None} key in sys.modules under its name,
L{modules.PythonPath.__getitem__} should still be able to retrieve an
unloaded L{modules.PythonModule} for that package.
r N)�sysPathZ moduleLoaderZ
importerCacheZ sysPathHooksZ
moduleDict)r �
PythonPathr% r* �appendr0 � assertFalse�isLoaded)r Z
shouldNotLoadr* r r r � test_unimportablePackageGetItem~ s � z*BasicTests.test_unimportablePackageGetItemc C s^ | � � }| �|jg� | �ddi� tt�� �}| �dd� |D �dg� | �|d � � � dS )a
If a package has been explicitly forbidden from importing by setting a
L{None} key in sys.modules under its name, L{modules.walkModules} should
still be able to retrieve an unloaded L{modules.PythonModule} for that
package.
r Nc S s g | ]
}|j �qS r r )r! �mr r r r# � s zBBasicTests.test_unimportablePackageWalkModules.<locals>.<listcomp>r )
r% r) r* ZreplaceSysModulesr, r r r0 r8 r9 )r � existentPathZwalkedr r r �#test_unimportablePackageWalkModules� s �z.BasicTests.test_unimportablePackageWalkModulesc C s~ | � � }t| �� �}| �|�� � | �|jg� t�d�g}t t�
� �}t j� |j� t t�
� �}| �
||� | �
||� dS )z~
Verify that L{modules.walkModules} ignores entries in sys.path which
do not exist in the filesystem.
r N)r% r �mktempr8 �existsr) r* r r+ r, r r- r7 r0 )r r< ZnonexistentPathr3 �
beforeModules� afterModulesr r r �test_nonexistentPaths� s z BasicTests.test_nonexistentPathsc C sp | � � }t| �� �}| �|�� � |�d� | �|jg� tt �
� �}t j� |j� tt �
� �}| �
||� dS )z�
Verify that L{modules.walkModules} ignores entries in sys.path which
refer to regular files in the filesystem.
s zip file or whatever
N)r% r r> r8 r? r' r) r* r, r r r- r7 r0 )r r< ZnonDirectoryPathr@ rA r r r �test_nonDirectoryPaths� s
z!BasicTests.test_nonDirectoryPathsc C s | � t�d�| �d�� dS )z�
Scrounge around in the top-level module namespace and make sure that
Twisted shows up, and that the module thusly obtained is the same as
the module that we find when we look for it explicitly by name.
�twistedN�r0 r r+ r �r r r r �test_twistedShowsUp� s �z BasicTests.test_twistedShowsUpc C s& | � t�d�| jdt�d�d�� dS )zo
Verify that the walkModules APIs will give us back subpackages, not just
subpackages.
�twisted.pythonrD )r NrE rF r r r �test_dottedNames� s ��zBasicTests.test_dottedNamesc C s* t �� D ] }| �d|jkd|j � qdS )z�
Verify that the iterModules API will only return top-level modules and
packages, not submodules or subpackages.
�.z9no nested modules should be returned from iterModules: %rN)r � iterModulesr8 r �filePath)r r2 r r r �test_onlyTopModules� s ��z BasicTests.test_onlyTopModulesc C sB dD ]8}t |�}| �t�|��� |� | �| �|��� |� qdS )zl
Verify that we can locate and load packages, modules, submodules, and
subpackages.
��osrD rH ztwisted.python.reflectN)r ZfailUnlessIdenticalr r+ �loadr )r �nr; r r r �test_loadPackagesAndModules� s � �z&BasicTests.test_loadPackagesAndModulesc C s( dD ] }| � t�|�jjjtj� qdS )zu
Verify that path entries discovered via module loading are, in fact, on
sys.path somewhere.
rN N)Z failUnlessInr r+ Z pathEntryrL r* r- )r rQ r r r �test_pathEntriesOnPath� s
�z!BasicTests.test_pathEntriesOnPathc s� t | �� �}|�� tj|jgd�}|j� � fdd�}|�d��d� t j
|jdd� | � t t
|�� ��d� ||_| � |d j|�d�� d
S ) z~
Verify that .py files will always be preferred to .pyc files, regardless of
directory listing order.
�r5 c s$ �| �}|j � � fdd�}||_ |S )Nc s t � � �} | �� | �� | S �N)r, �sort�reverse)�x�ZoriginalChildrenr r � evilChildren s
zLBasicTests.test_alwaysPreferPy.<locals>._evilSmartPath.<locals>.evilChildren)�children)ZpathName�orZ �ZoriginalSmartPathrY r �_evilSmartPath s
z6BasicTests.test_alwaysPreferPy.<locals>._evilSmartPathzabcd.py�
T)�quiet� �abcdN)r r> �createDirectoryr r6 r* Z
_smartPathr& r' �
compileallZ compile_dirr0 �lenr, r[ rL )r �mypath�ppr^ r r] r �test_alwaysPreferPy s �z BasicTests.test_alwaysPreferPyc C s� t | �� �}|�� tj|jgd�}|�d�}|�� |�d��d� tj� |j� t
d� z2t |�
� �}| �t|�d� | �|d jd� W 5 tjd= tj� |j� X dS )z�
A package can delete its __path__ for some reasons,
C{modules.PythonPath} should be able to deal with it.
rT rb r s
del __path__
� r N)r r> rc r r6 r* r& r' r- r7 r$ �remover, r r0 re r )r rf rg Zsubpath�lr r r �test_packageMissingPath) s
z"BasicTests.test_packageMissingPathN)r r r r4 r: r= rB rC rG rI rM rR rS rh rl r r r r r . s N
r c @ sZ e Zd ZdZe�� Zdd� Zdd� Zddd�Z d d
� Z
d d � Z d
d� Z dd� Z
dd� ZdS )�PathModificationTestszp
These tests share setup/cleanup behavior of creating a dummy package and
stuffing some code in it.
c C s� | � � | _t| j�| _| j�� dt| j�f | _| j�| j�| _ | j �� | j �d��
d� | j �d��
d� | j �d��
d� | j �d��
d� d| _ d S )Nzpyspacetests%dr r za.pyzb.pyz c__init__.pyF) r> �pathExtensionNamer Z
pathExtensionrc �next�
_serialnum� packageNamer& � packagePathr' � pathSetUprF r r r �setUpI s
zPathModificationTests.setUpc C s" | j r
t�d| _ tj�| j� d S �NT)rs �AssertionErrorr- r* r7 rn rF r r r �
_setupSysPathW s
z#PathModificationTests._setupSysPathTc C s� | � � }t|�}|�� |�d��d� | j�d��td�t|���� | � � t
� | j �}| �
| j| j d ||d�|d � | �
|d �� jd� d S ) Nz foozle.pys x = 123
r z__path__.append({0})
z.foozler Zfoozle�{ )r> r rc r& r' rr r �format�reprrw r r+ rq r0 r rP rX )r ZdoImportZmoddir2Zfpmdr r r r �_underUnderPathTest] s � ��z)PathModificationTests._underUnderPathTestc C s | � � dS )z]
Verify that iterModules will honor the __path__ of already-loaded packages.
N)r{ rF r r r �"test_underUnderPathAlreadyImportedn s z8PathModificationTests.test_underUnderPathAlreadyImportedc C s< t �| j�}dd� |�� D �}|�� | �|dddg� d S )Nc S s g | ]}|j �d �d �qS )rJ ���)r �split)r! r r r r r# w s z6PathModificationTests._listModules.<locals>.<listcomp>�a�bZ c__init__)r r+ rq rK rV r0 )r ZpkginfoZnfnir r r � _listModulesu s �z"PathModificationTests._listModulesc C s | � � | �� dS )z�
Make sure the module list comes back as we expect from iterModules on a
package, whether zipped or not.
N)rw r� rF r r r �test_listingModules} s z)PathModificationTests.test_listingModulesc C s | � � t| j� | �� dS )z�
Make sure the module list comes back as we expect from iterModules on a
package, whether zipped or not, even if the package has already been
imported.
N)rw r rq r� rF r r r �"test_listingModulesAlreadyImported� s
z8PathModificationTests.test_listingModulesAlreadyImportedc C sB | j r>d}tj�� | jks&t|d ��| jtjks>t|d ��d S )Nz<path cleanup failed: don't be surprised if other tests breakz, 1z, 2)rs r- r* �poprn rv )r ZHORKr r r �tearDown� s z PathModificationTests.tearDownN)T)r r r r � itertools�countrp rt rw r{ r| r� r� r� r� r r r r rm @ s
rm c @ s e Zd ZdZdd� Zdd� ZdS )�RebindingTestsz�
These tests verify that the default path interrogation API works properly
even when sys.path has been rebound to a different object.
c C s: | j r
t�d| _ tj| _tjd d � t_tj�| j� d S ru )rs rv r- r* � savedSysPathr7 rn rF r r r rw � s
z RebindingTests._setupSysPathc C s | j r| jt_dS )zF
Clean up sys.path by re-binding our original object.
N)rs r� r- r* rF r r r r� � s zRebindingTests.tearDownN)r r r r rw r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�ZipPathModificationTestsc C sH | j r
t�t| j| jd � | jd7 _t�| j�s:t�t�| � d S )Nz.zip)rs rv r
rn �zipfileZ
is_zipfilerm rw rF r r r rw � s
z&ZipPathModificationTests._setupSysPathN)r r r rw r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �PythonPathTestsz�
Tests for the class which provides the implementation for all of the
public API of L{twisted.python.modules}, L{PythonPath}.
c s~ G dd� dt �� � fdd�}dg}i }|g}i }dd� }t�|||||�}t|�� ��| �t��d� | �t�fd d
�� d S ) z}
Make sure that the behavior when encountering an unknown importer
type is not catastrophic failure.
c @ s e Zd ZdS )z>PythonPathTests.test_unhandledImporter.<locals>.SecretImporterN)r r r r r r r �SecretImporter� s r� c s � � S rU r r )r� r r �hook� s z4PythonPathTests.test_unhandledImporter.<locals>.hookz example/pathc S s d S rU r r r r r � sysloader� s z9PythonPathTests.test_unhandledImporter.<locals>.sysloaderri c s � d d S )Nr r2 r r )�entriesr r �<lambda>� r z8PythonPathTests.test_unhandledImporter.<locals>.<lambda>N) �objectr r6 r, Z iterEntriesr0 re Z assertRaises�KeyError)r r� ZsyspathZ
sysmodulesZsyshooksZsyscacher� �spacer )r� r� r �test_unhandledImporter� s" � z&PythonPathTests.test_unhandledImporterc C s� t �g tj g i �}|t }| �| jg�}| �|d d t� | �|d d tt j
�� � � � d t d � | �t
|�d� | �|jt� dS )z�
If the path a module loaded with L{PythonPath.__getitem__} is not
present in the path importer cache, a warning is emitted, but the
L{PythonModule} is returned as usual.
r �category�messagez
(for module zR) not in path importer cache (PEP 302 violation - check your local configuration).ri N)r r6 r- r Z
flushWarnings� test_inconsistentImporterCacher0 � UserWarningr rD �__file__�parent�dirnamere r )r r� Z
thisModule�warningsr r r r� � s
����z.PythonPathTests.test_inconsistentImporterCachec C s t �� }| �d|� dS )z�
L{PythonPath} implements the C{in} operator so that when it is the
right-hand argument and the name of a module which exists on that
L{PythonPath} is the left-hand argument, the result is C{True}.
rO N)r r6 ZassertIn�r ZthePathr r r �test_containsModule� s z#PythonPathTests.test_containsModulec C s t �� }| �d|� dS )z�
L{PythonPath} implements the C{in} operator so that when it is the
right-hand argument and the name of a module which does not exist on
that L{PythonPath} is the left-hand argument, the result is C{False}.
Z bogusModuleN)r r6 Z assertNotInr� r r r �test_doesntContainModule� s z(PythonPathTests.test_doesntContainModuleN)r r r r r� r� r� r� r r r r r� � s
r� ) r Z
__future__r r r- r� rd r� rD Ztwisted.pythonr Ztwisted.python.compatr Ztwisted.python.filepathr Ztwisted.python.reflectr Ztwisted.trial.unittestr Z#twisted.python.test.modules_helpersr Z twisted.python.test.test_zippathr
r r rm r� r� r� �__all__r r r r �<module> s2 [ D �