File: //opt/cppython/lib/python3.8/test/test_importlib/extension/__pycache__/test_loader.cpython-38.pyc
U
/D�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 �H/opt/cppython/lib/python3.8/test/test_importlib/extension/test_loader.py�setUp s �zLoaderTests.setUpc C s | j �|�S r )r � load_module)r �fullnamer r r r s zLoaderTests.load_modulec C s: | j �� | j �d � | �t�� | �d� W 5 Q R X d S )NZXXX)r r � assertRaises� ImportErrorr r r r �test_load_module_API s
z LoaderTests.test_load_module_APIc C s( | j �tjjtjj�}| �| j|� d S r )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 �test_inequality% s �zLoaderTests.test_inequalityc C s� t �t jj��n | �t jj�}dt jjfdt jjfdfD ]\}}| �t||�|� q8| �t jjt j
� | � |j | j
j� W 5 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 5 Q R X d S r )r r% r r r �assertIs�r Zmodule1Zmodule2r r r �test_module_reuse; s z LoaderTests.test_module_reusec C s8 d}| � t��}| �|� W 5 Q R X | �|jj|� d S )N�asdfjkl;)r r r r � exceptionr �r r �cmr r r �test_unloadableD s zLoaderTests.test_unloadablec C s\ | � | j�tjj�� | jjD ]8}tj �
dddd| �}| j� d|�}| � |�d�� q d S )NZsome�pathZpkg�__init__)
Z assertFalser �
is_packager r r r �EXTENSION_SUFFIXES�osr8 �joinr
Z
assertTrue)r �suffixr8 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_parentr2 �test_state_after_failurer7 r? 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��� | �� }d| jfd| jjfdfD ]\}}| �t||�|� q.| �t �� |j
W 5 Q R X | � |t j
| j � | �|j| jj� W 5 Q R X dS )z Test loading an extension moduler r! r" N)r r% r r rJ rL r r& r �AttributeError�__path__r0 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 5 Q R X d|_ | �|j d� | �|�
dd�d� | �|�� t� | �|�d�d d� | �
|j�� |�� �W 5 Q R X | �|jd � | �|jd
� W 5 Q R X dS ) z@Test basic functionality of stuff defined in an extension moduleZabcdNr � � � Z23Z123i� �something different)r r% r r r) �types�
ModuleType�Exampler Zdemor rN r ZfooZStr�str�errorZ int_const� str_const)r r, �exr r r �test_functionalityx s" z1MultiPhaseExtensionModuleTests.test_functionalityc C sB t �| j��, | �� }|j}t�|� | �||j� W 5 Q R X dS )z6Test that reload didn't re-set the module's attributesN)r r% r r rV rH �reloadr0 )r r, Zex_classr r r � test_reload� s
z*MultiPhaseExtensionModuleTests.test_reloadc
C s� | � � }| �d�� | �|�d�d� W 5 Q R X | �d��& | �t�� |�d� W 5 Q R X W 5 Q R X | �d��& | �t�� |�d� W 5 Q R X W 5 Q R X dS )zBAssert that the PyState_{Find,Add,Remove}Module C API doesn't workZPyState_FindModuler NZPyState_AddModulerR 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) rJ rL r r
rH r �spec_from_loader�module_from_spec� exec_module)r r rL r rJ 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.rS N)re r r) rT rU r r rY 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�xrS N)
re r) rT rU r r rY Z assertNotInr' 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)re r Z assertIsNotr1 r r r �test_load_twice� s z.MultiPhaseExtensionModuleTests.test_load_twicec C s8 d}| � t��}| �|� W 5 Q R X | �|jj|� dS )zTest nonexistent moduler3 N�r r re r r4 r r5 r r r r7 � s z.MultiPhaseExtensionModuleTests.test_unloadablec C s8 d}| � t��}| �|� W 5 Q R X | �|jj|� dS )z9Test behavior with nonexistent module with non-ASCII nameu foóNrj r5 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 re 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_methodsrl �
rR rP N)r re r rm rn r ro Zbarrp 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 re r) rT rU r r )r r r, r r r �test_null_slots� s
z.MultiPhaseExtensionModuleTests.test_null_slotsc C sR dD ]H}| � |��4 | jd | }| �t�� | �|� W 5 Q R X W 5 Q R X qdS )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)r^ r r r_ re )r Z name_baser r r r �test_bad_modules� s
z/MultiPhaseExtensionModuleTests.test_bad_modulesc
C sd | j d dfdg}|D ]H\}}| �|��0 | �|�}| �|j|� | �|jd| � W 5 Q R X qdS )z4Test that modules with non-ASCII names can be loadedu _zkouška_načteníZCzech)u ＿インポートテストZJapanesezModule named in %sN)r r^ re r r rB )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-cNr )r Zscriptr r r �test_bad_traverse s z0MultiPhaseExtensionModuleTests.test_bad_traverse) r r@ rA rB r rC rD rE r/ r[ r] ra r re rf rh ri r7 rk rr rt ru rv rx rn ZskipIf�hasattrr' ry r r r r rF U s2 �rF �__main__)r$ r r Zimport_importlibr Zos.pathr< r' rT rn �importlib.utilrH Ztest.support.script_helperr r Z test_bothZFrozen_LoaderTestsZSource_LoaderTestsrF Z%Frozen_MultiPhaseExtensionModuleTestsZ%Source_MultiPhaseExtensionModuleTestsr �mainr r r r �<module> s(
E � T �