File: //usr/local/lib/python3.7/test/test_importlib/__pycache__/test_windows.cpython-37.pyc
B
��gW � @ s$ 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&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 st yt | |td�}W n tk
r& d S X x4yt|d�}W n tk
rN P Y nX t||� q*W t|� t| |� d S )N)�accessr )r r �OSErrorr �delete_registry_treer r )�root�subkey�hkeyZ subsubkey� r �</usr/local/lib/python3.7/test/test_importlib/test_windows.pyr s r c c s� | j jr| j j}n| j j}|j|dtjd d� d�}zHt|d��4}tt |�}|d kr^|d }t
|dt |� d V W d Q R X W d | j jr�t j
�|�}tt |� X d S )Nz%d.%d� )�fullname� sys_versionza = 1z.py� )�WindowsRegistryFinder� DEBUG_BUILD�REGISTRY_KEY_DEBUG� REGISTRY_KEY�format�sys� version_infor r
r r r
�os�path�dirnamer )� machinery�namer' r �key�locationr r r r � setup_module$ s
r- 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 )N�spam)r) r � find_spec�assertIs)�self�specr r r �test_find_spec_missing? s z1WindowsRegistryFinderTests.test_find_spec_missingc C s | j j�d�}| �|d � d S )Nr/ )r) r � find_moduler1 )r2 �loaderr r r �test_find_module_missingC s z3WindowsRegistryFinderTests.test_find_module_missingc C sV t | j| j��> | jj�| j�}| jj�| j�}| �|d � | �|d � W d Q R X d S )N)r- r) � test_moduler r5 r0 Z assertIsNot)r2 r6 r3 r r r �test_module_foundG s
z,WindowsRegistryFinderTests.test_module_foundc C sV t | j| jdd��: | jj�| j�}| jj�| j�}| �|� | �|� W d Q R X d S )N�.)r' )r- r) r8 r r5 r0 Z assertIsNone)r2 r6 r3 r r r �test_module_not_foundN s
z0WindowsRegistryFinderTests.test_module_not_foundN) �__name__�
__module__� __qualname__r# r& �getpidr8 r4 r7 r9 r; r r r r r. 9 s
r. )r) c @ s e Zd Zdd� ZdS )�WindowsExtensionSuffixTestsc C s~ | j j}d�tjt�ddt� ��}y|�d�}W n& t k
rV |�d�}d| }Y nX | �
||� |�|�}| � ||� 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)r2 �suffixesZ expected_tagZ
untagged_iZtagged_ir r r �test_tagged_suffix[ s
z.WindowsExtensionSuffixTests.test_tagged_suffixN)r< r= r> rH r r r r r@ Y s r@ )N)'r r Z test_utilZimport_importlibr) r& rC r$ ZunittestZtestr Zdistutils.utilr �
contextlibr r �
import_moduler r
r r r
r r r r r r r- Z
skipUnless�platform�
startswithr. Z test_bothZ!Frozen_WindowsRegistryFinderTestsZ!Source_WindowsRegistryFinderTestsr@ Z"Frozen_WindowsExtensionSuffixTestsZ"Source_WindowsExtensionSuffixTestsr r r r �<module> s$
, " "