a
    œcl_™  ã                   @   s0   d Z ddlmZ ddlmZ G dd„ deƒZdS )z Tests for samba.dcerpc.registry.é    )Úwinreg)ÚRpcInterfaceTestCasec                       s<   e Zd Z‡ fdd„Zdd„ Zdd„ Zdd„ Zd	d
„ Z‡  ZS )ÚWinregTestsc                    s*   t t| ƒ ¡  t d|  ¡ |  ¡ ¡| _d S )Nzncalrpc:)Úsuperr   ÚsetUpr   Zget_loadparmZget_credentialsÚconn©Úself©Ú	__class__© ú=/usr/lib/python3/dist-packages/samba/tests/dcerpc/registry.pyr      s    ÿzWinregTests.setUpc                 C   s   | j  d tjtjB ¡S ©N)r   ÚOpenHKLMr   ÚKEY_QUERY_VALUEÚKEY_ENUMERATE_SUB_KEYSr   r   r   r   Úget_hklm   s    
ÿzWinregTests.get_hklmc                 C   s&   | j  d tjtjB ¡}| j  |¡ d S r   )r   r   r   r   r   ÚCloseKey)r	   Úhandler   r   r   Ú	test_hklm#   s    
ÿzWinregTests.test_hklmc                 C   s2   |   ¡ }| j |¡}|  t|j¡ | j |¡ d S r   )r   r   Z
GetVersionÚassertEqualÚintr   r   )r	   r   Úversionr   r   r   Útest_getversion(   s    zWinregTests.test_getversionc                 C   sH   | j  d tjtjB ¡}| j  |t ¡ ¡}|  dt|ƒ¡ | j  	|¡ d S )Né	   )
r   r   r   r   r   ZQueryInfoKeyÚStringr   Úlenr   )r	   r   Úxr   r   r   Útest_getkeyinfo.   s    
ÿzWinregTests.test_getkeyinfo)	Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   Ú__classcell__r   r   r
   r   r      s
   r   N)Ú__doc__Zsamba.dcerpcr   Zsamba.testsr   r   r   r   r   r   Ú<module>   s   