File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/encrypted_secrets.cpython-310.pyc
o
�/a � @ sx d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ G d d
� d
e�ZdS ) zySmoke test for encrypted secrets
A quick test to confirm that the secret attributes are being stored
encrypted on disk.
� N)�TestCase)� Credentials)�SamDB)�system_session)�
ndr_unpack)�drsblobsc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�EncryptedSecretsTestsc sZ t t| ��� tj�� | _t� | _t � | _
| j� | j� t � | _
t | j
| j| jd�| _
d S )N)� session_info� credentials�lp)�superr �setUp�samba�tests� env_loadparmr r �credsr �session�guessr �ldb)�self�� __class__� �?/usr/lib/python3/dist-packages/samba/tests/encrypted_secrets.pyr
' s �zEncryptedSecretsTests.setUpc C s� | j �� }d|�� }tj�d|�}| j�|�}t jd| t j d�}d| }|j
t j |dgd�}| � dt
|�d k� |d }|d d } | �t
| �d
k� ttj| �}
d } | �| |
jj� d S )
z8Test that secret attributes are stored encrypted on diskz%s.ldbz sam.ldb.dzldb://)�flagsz CN=Administrator,CN=Users,%s�
unicodePwd��scope�base�attrsTr � l �-� N)r � domain_dn�upper�os�path�joinr � private_path�Ldb�FLG_DONT_CREATE_DB�search�
SCOPE_BASE�assertIs�len�
assertTruer r �EncryptedSecret� assertEqual�header�magic) r �basedn�backend_filename�backend_subpath� backend_path� backenddb�dn�res�obj�blob� encryptedr1 r r r �test_encrypted_secrets2 s&
� � z,EncryptedSecretsTests.test_encrypted_secretsc C sX | j jt jddgd�}| �t|�dk� | �d|d v � |d d }| �d|v � dS )zcTest that databases are provisioned with encryptedSecrets as a
required feature
z @SAMBA_DSDB�requiredFeaturesr r s encryptedSecretsN)r r) r* r- r, )r r8 �required_featuresr r r �test_required_featuresI s
� z,EncryptedSecretsTests.test_required_features)�__name__�
__module__� __qualname__r
r< r? �
__classcell__r r r r r % s r )�__doc__r# r r � samba.testsr �samba.credentialsr � samba.samdbr �
samba.authr � samba.ndrr � samba.dcerpcr r r r r r �<module> s