
    b                     >    d Z ddlmZ ddlmZ  G d de          ZdS )z Tests for samba.dcerpc.registry.    )winreg)RpcInterfaceTestCasec                   6     e Zd Z fdZd Zd Zd Zd Z xZS )WinregTestsc                     t          t          |                                            t          j        d|                                 |                                           | _        d S )Nzncalrpc:)superr   setUpr   get_loadparmget_credentialsconn)self	__class__s    =/usr/lib/python3/dist-packages/samba/tests/dcerpc/registry.pyr	   zWinregTests.setUp   sU    k4  &&(((M*d.?.?.A.A"&"6"6"8"8: :			    c                 f    | j                             d t          j        t          j        z            S N)r   OpenHKLMr   KEY_QUERY_VALUEKEY_ENUMERATE_SUB_KEYS)r   s    r   get_hklmzWinregTests.get_hklm   s/    y!!$"("86;X"XZ Z 	Zr   c                     | j                             d t          j        t          j        z            }| j                             |           d S r   )r   r   r   r   r   CloseKey)r   handles     r   	test_hklmzWinregTests.test_hklm#   sG    ##D$*$:V=Z$Z\ \	6"""""r   c                     |                                  }| j                            |          }|                     t          |j                   | j                            |           d S r   )r   r   
GetVersionassertEqualintr   r   )r   r   versions      r   test_getversionzWinregTests.test_getversion(   sY    )&&v..g/000	6"""""r   c                 <   | j                             d t          j        t          j        z            }| j                             |t          j                              }|                     dt          |                     | j         	                    |           d S )N	   )
r   r   r   r   r   QueryInfoKeyStringr   lenr   )r   r   xs      r   test_getkeyinfozWinregTests.test_getkeyinfo.   s    ##D$*$:V=Z$Z\ \I""66=??;;CFF###	6"""""r   )	__name__
__module____qualname__r	   r   r   r    r'   __classcell__)r   s   @r   r   r      sw        : : : : :
Z Z Z# # #
# # ## # # # # # #r   r   N)__doc__samba.dcerpcr   samba.testsr   r    r   r   <module>r0      sf   $ ' &       , , , , , ,# # # # #& # # # # #r   