Your IP : 216.73.217.32
��YfR � @ s` d d l Z d d l Z d d l Z d d l Z d d l m Z e j d � Z d d l m Z e j d d d g �d d l
Ty! e j e j j d � d
Z
Wn e e f k
r� d Z
Yn Xe j � d d � Z e � d k r� d
n d Z e d6 k rd
n d Z d e j � f Z d e Z d e Z d d e f d d e f d d e f d d d d d g e f d d e f d d! d8 e f d# d$ d9 e f d% d& e f g Z Gd' d( � d( e j � Z Gd) d* � d* e � Z e j! e
d+ � Gd, d- � d- e � � Z" e j! e d. � Gd/ d0 � d0 e � � Z# d1 d2 � Z$ e% d3 k r\e
sUe& d4 d5 � e$ � d S):� N)�support� threading)�machine�winregZ required_on�win)�*z--remote� � ZAMD64TF� z Python Test Key [%d] - Delete Mez SOFTWARE\zSOFTWARE\Classes\z Int Value�- z
String ValzA string valueZ StringExpandzThe path is %path%z Multi-stringZLotsZof�string�valueszRaw Datas binary dataz
Big String�x� z
Big Binary� xu Japanese 日本u 日本語c @ sv e Z d Z d d � Z d d � Z d e d d � Z d e d d � Z d d
d � Z d d d
� Z
d d � Z d S)�BaseWinregTestsc C s | j t t � d S)N)� delete_tree�HKEY_CURRENT_USER�
test_key_name)�self� r �5/opt/alt/python35/lib64/python3.5/test/test_winreg.py�setUp4 s zBaseWinregTests.setUpc C s� y t | | t � } Wn t k
r. d SYn Xx= y t | d � } Wn t k
rZ PYn X| j | | � q2 Wt | � t | | � d S)Nr )�OpenKey�KEY_ALL_ACCESS�OSError�EnumKeyr �CloseKey� DeleteKey)r �rootZsubkey�hkeyZ subsubkeyr r r r 9 s
zBaseWinregTests.delete_tree�sub_keyc C s� t | t t d � | | t � } | j | j d k � | | | � } x- t D]% \ } } } t | | d | | � qN Wt | � \ } }
} | j | d d � | j |
d d � t | � \ } }
} | j | d d � | j |
t t � d � t
| � } t | � y t | � | j d � Wn t
k
r8Yn Xt
| � }
| j � y t |
� | j d � Wn t
k
r~Yn Xd S)Nz
Default valuer r z"Not the correct number of sub keysz Not the correct number of valueszAIt appears the CloseKey() function does not close the actual key!zBIt appears the key.Close() function does not close the actual key!)�SetValuer �REG_SZ�
assertTrue�handle� test_data�
SetValueEx� QueryInfoKey� assertEqual�len�intr �failr �Close)r �root_key� subkeystr� CreateKey�keyr! �
value_name�
value_data�
value_type�nkeys�nvalues� since_modZ int_sub_keyZint_keyr r r �_write_test_dataI s6
z BaseWinregTests._write_test_datac C s~ t | t � } | j | d d � | | t � } | | | � �� } d } xP y t | | � } Wn t k
rt PYn X| j | t k d d � | d } qL W| j | t t � d � xR t D]J \ } }
} t | | � \ } }
| j | |
d � | j |
| d � q� WWd QRX| j � t | d � } | j | | d � y t | d � | j
d
� Wn t k
roYn X| j � d S) Nz
Default valuez+Registry didn't give back the correct valuer Tz&Didn't read back the correct test datar z'Didn't read the correct number of itemsz!Could not directly read the valuezRead subkey value wrongz2Was able to get a second key when I only have one!) �
QueryValuer r) � EnumValuer r& r* � QueryValueExr- r r, )r r. r/ r �valr1 r! �index�datar2 r3 r4 Zread_valZread_typr r r �_read_test_datas s>
z BaseWinregTests._read_test_datac C s& t | t d t � } t | | d t � } x$ t D] \ } } } t | | � q1 Wt | � \ } } }
| j | d d � | j | d d � | j � t | | � y t | | � | j d � Wn t
k
r� Yn X| j � t | t � y t | t � } | j d � Wn t
k
r!Yn Xd S)Nr z subkey not empty before deletez Deleting the key twice succeededz Could open the non-existent key) r r r r& Z DeleteValuer( r) r- r r, r ) r r. r/ r1 r! r2 r3 r4 r5 r6 r7 r r r �_delete_test_data� s*
z!BaseWinregTests._delete_test_datac C s4 | j | | � | j | | � | j | | � d S)N)r8 r? r@ )r r. r/ r r r � _test_all� s zBaseWinregTests._test_allc C s� t d | d | d d d t � � } | j | j d k � Wd QRXt d | d | d d d t � � } | j | j d k � Wd QRXd S)Nr1 r! �reservedr �access)� CreateKeyExr r$ r% � OpenKeyEx)r r1 r! ZckeyZokeyr r r �_test_named_args� s z BaseWinregTests._test_named_argsN) �__name__�
__module__� __qualname__r r r0 r8 r r? r@ rA rF r r r r r 2 s )( r c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z e
j e d! k d � d d � � Z d d � Z d d � Z d d � Z d S)"�LocalWinregTestsc C s! | j t � | j t d � d S)Nu
日本-subkey)rA r )r r r r �test_registry_works� s
z$LocalWinregTests.test_registry_worksc C sO d d � } | j t d | �d d � } | j t d | �| j t � d S)Nc S s t | | d t � S)Nr )rD r )r1 r! r r r �<lambda>� s zILocalWinregTests.test_registry_works_extended_functions.<locals>.<lambda>r0 c S s t | | d t � S)Nr )rE �KEY_READ)r1 r! r r r rL � s r )r8 r r? r@ )r ZckeZoker r r �&test_registry_works_extended_functions� s
z7LocalWinregTests.test_registry_works_extended_functionsc C s! | j t t � t t t � d S)N)rF r r r )r r r r �test_named_arguments� s z%LocalWinregTests.test_named_argumentsc C sC t d t � } | j | j d � | j � | j | j d � d S)Nr )�ConnectRegistry�HKEY_LOCAL_MACHINE�assertNotEqualr% r- r) )r �hr r r �,test_connect_registry_to_local_machine_works� s
z=LocalWinregTests.test_connect_registry_to_local_machine_worksc C s d d � } | j t | � d S)Nc S s
t d t � S)NZ abcdefghijkl)rP r r r r r rL � s zBLocalWinregTests.test_inexistant_remote_registry.<locals>.<lambda>)� assertRaisesr )r Zconnectr r r � test_inexistant_remote_registry� s z0LocalWinregTests.test_inexistant_remote_registryc C sA t d � } | j t | � t � | j | t j d d � d S)Nz
%windir%\testZwindirz\test)ZExpandEnvironmentStringsr) �type�str�os�environ)r �rr r r � testExpandEnvironmentStrings� s z-LocalWinregTests.testExpandEnvironmentStringsc C sb y6 t d t � � } | j | j d � t � Wd QRXWn% t k
r] | j | j d � Yn Xd S)Nr )rP rQ rR r% r r) )r rS r r r �test_context_manager� s
z%LocalWinregTests.test_context_managerc
s� d � G� f d d � d t j � } | � } | j � z� t t t d � �j } x` t d � D]R } t | � \ } } } x4 t | � D]& } t | | � } t | | d � q� Wq^ WWd QRXWd d � | j
� t t t d � t t t � Xd S)NFc s" e Z d Z � f d d � Z d S)z>LocalWinregTests.test_changing_value.<locals>.VeryActiveThreadc sf t t t � �P } d } d d } x6 � sZ | r7 d n | } | } t | d t | � q% WWd QRXd S)NTr i� Zchanging_value)r0 r r r" r# )r r1 Z use_shortZ long_string�s)�doner r �run s
zBLocalWinregTests.test_changing_value.<locals>.VeryActiveThread.runN)rG rH rI r` r )r_ r r �VeryActiveThread� s ra z\changing_valuei� r T) r ZThread�startr0 r r �ranger( r: r9 �joinr )
r ra Zthreadr1 �_� num_subkeys�
num_values�t�i�namer )r_ r �test_changing_value� s
$
z$LocalWinregTests.test_changing_valuec C s� d d } zR t t t � �< } t | | t d � t | � \ } } } t | d � Wd QRXWd t t d j t | f � � t t t � Xd S)Nr � r �\) r0 r r r" r# r( r r rd )r rj r1 rf rg rh r r r �
test_long_key s
z LocalWinregTests.test_long_keyc C su y t t d � WnP t k
rc } z0 | j t j t j f k rN | j d � � WYd d } ~ Xn Xt t d � d S)Nr zMaccess denied to registry key (are you running in a non-interactive session?)� )r: ZHKEY_PERFORMANCE_DATAr �errnoZEPERMZEACCESZskipTestr; )r �er r r �test_dynamic_key' s
z!LocalWinregTests.test_dynamic_key� r zRequires Windows XPc - C s� z� t t t � � } | j | j d � Wd QRXt t t � } | j | j d � | j t � � t | � Wd QRX| j t � � t | � Wd QRX| j t � � t
| � Wd QRX| j t � � t t t � Wd QRXWd t t t � Xd S)Nr )
r0 r r rR r% r rU �NotImplementedError�DisableReflectionKey�EnableReflectionKey�QueryReflectionKey� DeleteKeyExr )r �ckr1 r r r �test_reflection_unsupported5 s z,LocalWinregTests.test_reflection_unsupportedc C s[ zF t t t � �0 } | j | j d � t | d d t d � Wd QRXWd t t t � Xd S)Nr � test_namel )r0 r r rR r% r' � REG_DWORDr )r ry r r r �test_setvalueex_value_rangeI s
!z,LocalWinregTests.test_setvalueex_value_rangec C s� z� t t t � �k } | j | j d � d } t | d d t | � t | d � \ } } | j | t � | j | | � Wd QRXWd t t t � Xd S)Nr l r{ )
r0 r r rR r% r' r| r; r) r )r ry �test_val�ret_val�ret_typer r r � test_queryvalueex_return_valueU s z/LocalWinregTests.test_queryvalueex_return_valuec C s� z� t t t � �k } | j | j d � d } t | d d t | � t | d � \ } } | j | t � | j | | � Wd QRXWd t t t � Xd S)Nr r{ )
r0 r r rR r% r' �
REG_BINARYr; r) r )r ry r~ r r� r r r �#test_setvalueex_crash_with_none_arge s z4LocalWinregTests.test_setvalueex_crash_with_none_argN)rs r )rG rH rI rK rN rO rT rV r\ r] rk rn rr �unittest�
skipUnless�WIN_VERrz r} r� r� r r r r rJ � s $ rJ z Skipping remote registry testsc @ s e Z d Z d d � Z d S)�RemoteWinregTestsc C s t t t � } | j | � d S)N)rP � REMOTE_NAMEr rA )r Z
remote_keyr r r �test_remote_registry_worksw s z,RemoteWinregTests.test_remote_registry_worksN)rG rH rI r� r r r r r� t s r� zx64 specific registry testsc @ sp e Z d Z d d � Z d d � Z e j e d � d d � � Z e j e d � d d � � Z d
d � Z
d S)
�Win64WinregTestsc C s3 | j t t � t d t d t d t d d � d S)Nr1 r! rC rB r )rF r r rx r )r r r r rO s z%Win64WinregTests.test_named_argumentsc
C si t t d � �S } | j t | � � | j t | � � | j t | � � | j t | � � Wd QRXd S)NZSoftware)r rQ r$ rw Z assertIsNonerv ru )r r1 r r r �test_reflection_functions� s
z*Win64WinregTests.test_reflection_functionsz OS doesn't support reflectionc ( C s^ zCt t t d t t B� �� } | j | j d � t t t d t t B� � } | j | j d � Wd QRXt | d d t d � d d � } | j
t | � Wd QRXt t t d t t B� �I } | j | j d � | j
d t | d � � t | d d t d � Wd QRXt t t d t t B� � } | j
d t | d � � Wd QRXWd t t t t d � Xd S)Nr ro Z32KEYc S s t t t d t t B� S)Nr )r r �test_reflect_key_namerM �KEY_WOW64_64KEYr r r r rL � s z2Win64WinregTests.test_reflection.<locals>.<lambda>Z64KEY)rD r r� r �KEY_WOW64_32KEYrR r% r r' r# rU r r� r) r9 rM rx )r � created_keyr1 � open_failr r r �test_reflection� s( $ z Win64WinregTests.test_reflectionc C s� z� t t t d t t B� �S } t | � } | j t | � t � | j | � t
| � | j t | � � Wd QRXd d � } | j t
| � t t t d t t B� � } | j | j d � Wd QRXWd t t t t d � Xd S)Nr c S s t t t d t t B� S)Nr )rE r r� rM r� r r r r rL � s z:Win64WinregTests.test_disable_reflection.<locals>.<lambda>)rD r r� r r� rw r) rW �boolZ assertFalseru r$ rU r rE rM rR r% rx )r r� Zdisabledr� r1 r r r �test_disable_reflection� s
z(Win64WinregTests.test_disable_reflectionc
C s* | j t � � } t t d � Wd QRXd S)NZ some_value_that_does_not_exist)rU �FileNotFoundErrorr9 ZHKEY_CLASSES_ROOT)r Zctxr r r �test_exception_numbers� s z'Win64WinregTests.test_exception_numbersN) rG rH rI rO r� r� r� �HAS_REFLECTIONr� r� r� r r r r r� | s
, r� c C s t j t t t � d S)N)r Z run_unittestrJ r� r� r r r r � test_main� s r� �__main__z)Remote registry calls can be tested usingz('test_winreg.py --remote \\machine_name')r
r i @ i�? i @ )'rY �sysrp r� Ztestr �
import_moduler �platformr r �argvr= r� �
IndexError�
ValueErrorZgetwindowsversionr� Z
WIN64_MACHINEr� �getpidZ
test_key_baser r� r| r# Z
REG_EXPAND_SZZ REG_MULTI_SZr� r& ZTestCaser rJ r� r� r� r� rG �printr r r r �<module> sH $
!
�� `