File: //lib/python3/dist-packages/samba/tests/dcerpc/__pycache__/registry.cpython-310.pyc
o
�/a� � @ s0 d Z ddlmZ ddlmZ G dd� de�ZdS )z Tests for samba.dcerpc.registry.� )�winreg)�RpcInterfaceTestCasec s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) � WinregTestsc s* t t| ��� t�d| �� | �� �| _d S )Nzncalrpc:)�superr �setUpr � get_loadparm�get_credentials�conn��self�� __class__� �=/usr/lib/python3/dist-packages/samba/tests/dcerpc/registry.pyr s
�zWinregTests.setUpc C s | j �d tjtjB �S �N)r �OpenHKLMr �KEY_QUERY_VALUE�KEY_ENUMERATE_SUB_KEYSr
r r r �get_hklm s
�zWinregTests.get_hklmc C s& | j �d tjtjB �}| j �|� d S r )r r r r r �CloseKey)r �handler r r � test_hklm# s
�zWinregTests.test_hklmc C s2 | � � }| j�|�}| �t|j� | j�|� d S r )r r �
GetVersion� assertEqual�intr
r )r r �versionr r r �test_getversion( s zWinregTests.test_getversionc C sH | j �d tjtjB �}| j �|t�� �}| �dt|�� | j � |� d S )N� )
r r r r r � QueryInfoKey�Stringr �lenr )r r �xr r r �test_getkeyinfo. s
�zWinregTests.test_getkeyinfo) �__name__�
__module__� __qualname__r r r r r" �
__classcell__r r r r r s r N)�__doc__� samba.dcerpcr � samba.testsr r r r r r �<module> s