Your IP : 216.73.216.85
a
XC?hA � @ sJ d dl mZ e�d�ZddlZddlZddlZddlZddl m
Z
ddl m Z ddl
mZ d dlmZ e
jd d
gd � dd lmZmZmZmZmZmZmZmZmZ d
d� Zeddd��Z e� ej � d
�d�G dd� d��Z ej!e ed�\Z"Z#e� ej � d
�d�G dd� d��Z$ej!e$ed�\Z%Z&e� ej � d
�d�G dd� dej'��Z(dS )� )�utilzimportlib.machinery� N)�support)� get_platform)�contextmanager)� temp_module�winreg�win)Z required_on) � CreateKey�HKEY_CURRENT_USER�SetValue�REG_SZ�KEY_ALL_ACCESS�EnumKey�CloseKey� DeleteKey�OpenKeyc C sp zt | |td�}W n ty& Y d S 0 zt|d�}W n tyL Y qZY n0 t||� q(t|� t| |� d S )N)�accessr )r r �OSErrorr �delete_registry_treer r )�root�subkey�hkeyZ subsubkey� r �E/opt/alt/python39/lib64/python3.9/test/test_importlib/test_windows.pyr s
r c c s0 | j jr| j j}n| j j}|j|dtjd d� d�}d�tjjtjj�}|� � �
|� � �slJ d�||���z�t |d��~}z(t t
|� | j jr�tj�|�}n|}W n ty� |}Y n0 tt
|�}|d u r�|d }t|dt|� d V W d � n1 s�0 Y W |�r,tt
|� n|�r*tt
|� 0 d S ) Nz%d.%d� )�fullname� sys_versionz Software\Python\PythonCore\{}.{}z$expected key '{}' to start with '{}'za = 1z.py� )�WindowsRegistryFinder� DEBUG_BUILD�REGISTRY_KEY_DEBUG� REGISTRY_KEY�format�sys� version_info�major�minor�casefold�
startswithr r r �os�path�dirnamer r
r r
r ) � machinery�namer+ r �keyZbase_key�locationZ
delete_keyr r r r � setup_module$ s<
� �
�
& �r1 zrequires Windowsc @ s: e Zd Zd�e�� �Zdd� Zdd� Zdd� Z dd � Z
d
S ) �WindowsRegistryFinderTestsz spamham{}c C s | j j�d�}| �|d � d S �NZspam)r- r � find_spec�assertIs)�self�specr r r �test_find_spec_missingJ s z1WindowsRegistryFinderTests.test_find_spec_missingc C s | j j�d�}| �|d � d S r3 )r- r � find_moduler5 )r6 �loaderr r r �test_find_module_missingN s z3WindowsRegistryFinderTests.test_find_module_missingc C sj t | j| j��H | jj�| j�}| jj�| j�}| �|d � | �|d � W d � n1 s\0 Y d S )N)r1 r- � test_moduler r9 r4 Z assertIsNot�r6 r: r7 r r r �test_module_foundR s
z,WindowsRegistryFinderTests.test_module_foundc C sj t | j| jdd��D | jj�| j�}| jj�| j�}| �|� | �|� W d � n1 s\0 Y d S )N�.)r+ )r1 r- r< r r9 r4 Z assertIsNoner= r r r �test_module_not_foundY s
z0WindowsRegistryFinderTests.test_module_not_foundN) �__name__�
__module__� __qualname__r# r* �getpidr< r8 r; r> r@ r r r r r2 D s
r2 )r- c @ s e Zd Zdd� ZdS )�WindowsExtensionSuffixTestsc C s| | j j}d�tjt�ddt� ��}z|�d�}W n$ t yT |�d�}d| }Y n0 | �
||� |�|�}| � ||� d S )Nz .cp{0.major}{0.minor}-{1}.pydz [^a-zA-Z0-9]�_z.pydz_d.pydZ_d) r- �EXTENSION_SUFFIXESr# r$ r% �re�subr �index�
ValueErrorZassertInZ
assertLess)r6 �suffixesZ expected_tagZ
untagged_iZtagged_ir r r �test_tagged_suffixf s �
z.WindowsExtensionSuffixTests.test_tagged_suffixN)rA rB rC rM r r r r rE d s rE c @ s e Zd Zdd� Zdd� ZdS )�WindowsBootstrapPathTestsc G s8 ddl m} ||� }|�� |�� kr(d S | �||� d S )Nr )�
_path_join)Z importlib._bootstrap_externalrO r( Z assertEqual)r6 ZexpectedZinputsrO Zactualr r r �
check_join} s
z$WindowsBootstrapPathTests.check_joinc C s� | � dddd� | � dddddd� | � dddd� | � ddd� | � dd� | � d d
d� | � d d d
d� | � d d d � | � d ddd
� | � d dd� | � ddd� | � d ddd
� | � ddd� | � dddd� | � ddddd
� | � dddddd
� | � ddddddd
� | � dd d ddd
� | � dd� | � d d � | � d d d!� | � dd"d!� | � d#d#d!� | � d$d%d!� | � d$d&d!� d S )'NzC:\A\BzC:\�A�BzD:\�DzC:BzA\BzD:AzD:zC:\B\CzA\B\C�CzB\CzA\B/CzB/CzA/zB\zA\../Cz../CzA.\.\BzA.r? z\\Server\Share\A\B\Cz\\Server\Sharez\Az\\Server2\Share2z\\Serverz\Sharez\\Server\Share\\zC:\AzC:/Ar zC:/zC:z//Server/Share\z//Server/Share/z//Server/Share)rP )r6 r r r �test_path_join� s6
� z(WindowsBootstrapPathTests.test_path_joinN)rA rB rC rP rU r r r r rN { s rN )N))r r Z test_utilZimport_importlibr- r* rH r$ Zunittest�testr Zdistutils.utilr �
contextlibr r �
import_moduler r
r r r
r r r r r r r1 Z
skipUnless�platformr) r2 Z test_bothZ!Frozen_WindowsRegistryFinderTestsZ!Source_WindowsRegistryFinderTestsrE Z"Frozen_WindowsExtensionSuffixTestsZ"Source_WindowsExtensionSuffixTestsZTestCaserN r r r r �<module> s4
, � �