File: //usr/local/lib/python3.7/test/test_importlib/extension/__pycache__/test_loader.cpython-37.pyc
B
��gy. � @ s� d dl mZ d dl mZ e�d�ZddlZddlZddlZddl Z ddl
Z ddl Z ddl m
Z
G dd� dej�Zejeed �\ZZG d
d � d ej�Zejeed �\ZZed kr�e �� dS )
� )�abc)�utilzimportlib.machinery� N)�assert_python_failurec @ sd e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dZ
dZ dd� Z dZ
dd� Zdd� ZdS )� LoaderTestsz)Test load_module() for extension modules.c C s | j �tjjtjj�| _d S )N)� machinery�ExtensionFileLoaderr �
EXTENSIONS�name� file_path�loader)�self� r �E/usr/local/lib/python3.7/test/test_importlib/extension/test_loader.py�setUp s zLoaderTests.setUpc C s | j �|�S )N)r � load_module)r
�fullnamer r r r s zLoaderTests.load_modulec C s: | j �� | j �d � | �t�� | �d� W d Q R X d S )NZXXX)r r � assertRaises� ImportError)r
r r r �test_load_module_API s
z LoaderTests.test_load_module_APIc C s( | j �tjjtjj�}| �| j|� d S )N)r r r r r
r � assertEqualr )r
�otherr r r �
test_equality s
zLoaderTests.test_equalityc C s, | j �dtjj tjj�}| �| j|� d S )N�_)r r r r r
r �assertNotEqualr )r
r r r r �test_inequality% s
zLoaderTests.test_inequalityc C s� t �t jj��r | �t jj�}x8dt jjfdt jjfdgD ]\}}| �t||�|� q:W | �t jjt j
� | � |j | j
j� W d Q R X d S )N�__name__�__file__)� __package__� )r �uncacher r
r r r �getattrZassertIn�sys�modules�assertIsInstance�
__loader__r r )r
�module�attr�valuer r r � test_module* s
zLoaderTests.test_moduleNc C sF t �t jj��. | �t jj�}| �t jj�}| �||� W d Q R X d S )N)r r r r
r �assertIs)r
�module1�module2r r r �test_module_reuse; s z LoaderTests.test_module_reusec C s8 d}| � t��}| �|� W d Q R X | �|jj|� d S )Nzasdfjkl;)r r r r � exceptionr
)r
r
�cmr r r �test_unloadableD s zLoaderTests.test_unloadablec C s` | � | j�tjj�� xD| jjD ]8}tj �
dddd| �}| j� d|�}| � |�d�� q W d S )NZsome�pathZpkg�__init__)
Z assertFalser �
is_packager r r
r �EXTENSION_SUFFIXES�osr1 �joinr Z
assertTrue)r
�suffixr1 r r r r �test_is_packageJ s
zLoaderTests.test_is_package)r �
__module__� __qualname__�__doc__r r r r r r) � test_package�test_lacking_parentr- �test_state_after_failurer0 r8 r r r r r s r )r c @ s� e Zd ZdZdd� ZdZdZdZdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Ze�eed%� d&�d'd(� �Z dS ))� MultiPhaseExtensionModuleTestszMTest loading extension modules with multi-phase initialization (PEP 489)
c C sF d| _ | j�d �}tj�| j �| _| js,t�| j�| j | jj �| _
d S )NZ_testmultiphase) r
r �
FileFinder� importlibr � find_spec�spec�AssertionErrorr �originr )r
�finderr r r r Y s
z$MultiPhaseExtensionModuleTests.setUpNc
C s� t �| j��� | �� }x6d| jfd| jjfdgD ]\}}| �t||�|� q0W | �t �� |j
W dQ R X | � |t j
| j � | �|j| jj� W dQ R X dS )z Test loading an extension moduler r )r r N)r r r
r rC rE r r! r �AttributeError�__path__r* r" r# r$ r% r r )r
r& r' r( r r r r) j s
z*MultiPhaseExtensionModuleTests.test_modulec
C s t �| j��� | �� }| �|tj� |�� }| �|� d�d� | �|� � d� | �
t �� |j W dQ R X d|_ | �|j d� | �|�
dd�d� | �|�� t� | �|�d�d d� | �
|j�� |�� �W dQ R X | �|jd � | �|jd
� W dQ R X dS ) z@Test basic functionality of stuff defined in an extension moduleZabcdNr � � � Z23Z123i� zsomething different)r r r
r r$ �types�
ModuleType�Exampler Zdemor rG r ZfooZStr�str�errorZ int_const� str_const)r
r& Zexr r r �test_functionalityx s" z1MultiPhaseExtensionModuleTests.test_functionalityc C sB t �| j��, | �� }|j}t�|� | �||j� W dQ R X dS )z6Test that reload didn't re-set the module's attributesN)r r r
r rN rA �reloadr* )r
r& Zex_classr r r � test_reload� s
z*MultiPhaseExtensionModuleTests.test_reloadc
C s� | � � }| �d�� | �|�d�d� W dQ R X | �d��& | �t�� |�d� W dQ R X W dQ R X | �d��& | �t�� |�d� W dQ R X W dQ R X dS )zBAssert that the PyState_{Find,Add,Remove}Module C API doesn't workZPyState_FindModuler NZPyState_AddModulerK ZPyState_RemoveModuler )r �subTestr Z call_state_registration_funcr � SystemError)r
r& r r r �test_try_registration� s z4MultiPhaseExtensionModuleTests.test_try_registrationc C s | j �| j�S )z'Load the module from the test extension)r r r
)r
r r r r � s z*MultiPhaseExtensionModuleTests.load_modulec C s> | j j}| j�||�}tj�||�}tj�|�}|�|� |S )z-Load a module from the test extension by name) rC rE r r rA r �spec_from_loader�module_from_spec� exec_module)r
r rE r rC r& r r r �load_module_by_name� s
z2MultiPhaseExtensionModuleTests.load_module_by_namec C sD | � d| j �}| �|tj� | �|jd| j � | �|jd� dS )z"Test loading a simulated submodulezpkg.zsomething differentN)r[ r
r$ rL rM r r rQ )r
r& r r r �test_load_submodule� s z2MultiPhaseExtensionModuleTests.test_load_submodulec C sF | � d�}| �|tj� | �|jd� | �|jd� | �dtj � dS )z-Test loading module with a one-character name�xzsomething differentN)
r[ r$ rL rM r r rQ Z assertNotInr" r# )r
r& r r r �test_load_short_name� s
z3MultiPhaseExtensionModuleTests.test_load_short_namec C s( | � | j�}| � | j�}| �||� dS )z,Test that 2 loads result in 2 module objectsN)r[ r
Z assertIsNot)r
r+ r, r r r �test_load_twice� s z.MultiPhaseExtensionModuleTests.test_load_twicec C s8 d}| � t��}| �|� W dQ R X | �|jj|� dS )zTest nonexistent modulezasdfjkl;N)r r r[ r r. r
)r
r
r/ r r r r0 � s z.MultiPhaseExtensionModuleTests.test_unloadablec C s8 d}| � t��}| �|� W dQ R X | �|jj|� dS )z9Test behavior with nonexistent module with non-ASCII nameu foóN)r r r[ r r. r
)r
r
r/ r r r �test_unloadable_nonascii� s z7MultiPhaseExtensionModuleTests.test_unloadable_nonasciic C s: | j d }| �|�}| �t|�tt�� | �|jd� dS )z4Test returning a non-module object from create worksZ
_nonmodule� N)r
r[ r �type�unittestr �three)r
r
�modr r r �test_nonmodule� s
z-MultiPhaseExtensionModuleTests.test_nonmodulec C sN | j d }| �|�}| �t|�tt�� | �|jd� | �|�dd�d� dS )z6Test creating a non-module object with methods definedZ_nonmodule_with_methodsra �
rK rI N)r
r[ r rb rc r rd Zbar)r
r
re r r r �test_nonmodule_with_methods� s
z:MultiPhaseExtensionModuleTests.test_nonmodule_with_methodsc C s4 | j d }| �|�}| �|tj� | �|j|� dS )z%Test that NULL slots aren't a problemZ _null_slotsN)r
r[ r$ rL rM r r )r
r
r& r r r �test_null_slots� s
z.MultiPhaseExtensionModuleTests.test_null_slotsc C sV xPdD ]H}| � |��4 | jd | }| �t�� | �|� W dQ R X W dQ R X qW dS )z5Test SystemError is raised for misbehaving extensions)Zbad_slot_largeZbad_slot_negativeZcreate_int_with_stateZ
negative_sizeZ export_nullZexport_uninitializedZ export_raiseZexport_unreported_exceptionZ create_nullZ create_raiseZcreate_unreported_exceptionZnonmodule_with_exec_slotsZexec_errZ
exec_raiseZexec_unreported_exceptionr N)rU r
r rV r[ )r
Z name_baser
r r r �test_bad_modules� s
z/MultiPhaseExtensionModuleTests.test_bad_modulesc
C sh | j d dfdg}xP|D ]H\}}| �|��0 | �|�}| �|j|� | �|jd| � W dQ R X qW dS )z4Test that modules with non-ASCII names can be loadedu _zkouška_načteníZCzech)u ＿インポートテストZJapanesezModule named in %sN)r
rU r[ r r r; )r
Zcasesr
�langr& r r r �
test_nonascii� s
z,MultiPhaseExtensionModuleTests.test_nonasciiZgettotalrefcountz.--with-pydebug has to be enabled for this testc C s d}t d|� dS )z� Issue #32374: Test that traverse fails when accessing per-module
state before Py_mod_exec was executed.
(Multiphase initialization modules only)
aY if True:
try:
from test import support
import importlib.util as util
spec = util.find_spec('_testmultiphase')
spec.name = '_testmultiphase_with_bad_traverse'
with support.SuppressCrashReport():
m = spec.loader.create_module(spec)
except:
# Prevent Python-level exceptions from
# ending the process with non-zero status
# (We are testing for a crash in C-code)
passz-cN)r )r
Zscriptr r r �test_bad_traverse s z0MultiPhaseExtensionModuleTests.test_bad_traverse) r r9 r: r; r r< r= r> r) rR rT rW r r[ r\ r^ r_ r0 r` rf rh ri rj rl rc ZskipIf�hasattrr" rm r r r r r? U s. r? �__main__)r r r Zimport_importlibr Zos.pathr5 r" rL rc �importlib.utilrA Ztest.support.script_helperr r Z test_bothZFrozen_LoaderTestsZSource_LoaderTestsr? Z%Frozen_MultiPhaseExtensionModuleTestsZ%Source_MultiPhaseExtensionModuleTestsr �mainr r r r �<module> s
E T