File: //opt/cppython/lib/python3.8/test/test_importlib/__pycache__/test_windows.cpython-38.opt-2.pyc
U
/D�gM � @ 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 st zt | |td�}W n tk
r( Y d S X zt|d�}W n tk
rP Y q^Y nX t||� q*t|� t| |� d S )N)�accessr )r r �OSErrorr �delete_registry_treer r )�root�subkey�hkeyZ subsubkey� r �?/opt/cppython/lib/python3.8/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}t
t |�}|d kr^|d }t|dt|� d V W 5 Q R X W 5 | j jr�tj� |�}t
t |� X d S )Nz%d.%d� )�fullname� sys_versionza = 1z.py� )�WindowsRegistryFinder� DEBUG_BUILD�REGISTRY_KEY_DEBUG� REGISTRY_KEY�format�sys� version_info�os�path�dirnamer r r r
r r
)� 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 �NZspam)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 r/ )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 5 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 5 Q R X d S )N�.)r' )r- r) r8 r r5 r0 Z assertIsNoner9 r r r �test_module_not_foundN s
z0WindowsRegistryFinderTests.test_module_not_foundN) �__name__�
__module__� __qualname__r# r& �getpidr8 r4 r7 r: r< r r r r r. 9 s
r. )r) c @ s e Zd Zdd� ZdS )�WindowsExtensionSuffixTestsc C s~ | j j}d�tjt�ddt� ��}z|�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? rI r r r r rA Y s rA c @ s e Zd Zdd� Zdd� ZdS )�WindowsBootstrapPathTestsc G s8 ddl m} ||� }|�� |�� kr(d S | �||� d S )Nr )�
_path_join)Z importlib._bootstrap_externalrK �casefoldZ assertEqual)r2 ZexpectedZinputsrK �actualr r r �
check_joinr 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)rN )r2 r r r �test_path_joiny s<
� z(WindowsBootstrapPathTests.test_path_joinN)r= r> r? rN rS r r r r rJ p s rJ )N))r r Z test_utilZimport_importlibr) r& rD r$ Zunittest�testr 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_WindowsRegistryFinderTestsrA Z"Frozen_WindowsExtensionSuffixTestsZ"Source_WindowsExtensionSuffixTestsZTestCaserJ r r r r �<module> s4
, � �