File: //lib/python3/dist-packages/samba/tests/__pycache__/upgradeprovisionneeddc.cpython-310.pyc
o
�/a� � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZmZmZmZmZmZmZmZmZmZ ddlmZmZ dd lm Z ddl Z dd
d �Z d ej d
ddf Z G dd� de�Z!G dd� de�Z"dS )z0Tests for samba.upgradeprovision that need a DC.� N)�param)� Credentials)�system_session)�
getpolicypath� find_provision_key_parameters)
� get_paths�get_ldbs�identic_rename�
updateOEMInfo�
getOEMInfo�
update_gpo�delta_update_basesamdb�update_dns_account_password� search_constructed_attrs_stored�&increment_calculated_keyversion_number)� env_loadparm�TestCaseInTempDir)�create_dummy_secretsdbc C s d S �N� )�a�br r �D/usr/lib/python3/dist-packages/samba/tests/upgradeprovisionneeddc.py� dummymessage( s r z%s/%s/%s�SELFTEST_PREFIX� ad_dc_ntvfsz etc/smb.confc @ s e Zd ZdZdd� Zdd� ZdS )�'UpgradeProvisionBasicLdbHelpersTestCasezISome simple tests for individual functions in the provisioning code.
c C s6 t td t�}t� }t� }|�|� t||t� |� d S r )r r �
smb_conf_pathr r �guessr r )�self�paths�creds�lpr r r �
test_get_ldbs3 s
z5UpgradeProvisionBasicLdbHelpersTestCase.test_get_ldbsc C s� t td t�}t� }t� }|�|� d}t||t� |�}t|j |j
|j |t|�}| � |j
d� | � t|j��� |�� � | �|jd � | �|jd� d S )Nzdc=samba,dc=example,dc=comzSAMBA.EXAMPLE.COM� )r r r r r r r r r �sam�secrets�idmap� assertEqual�realm�str�rootdn�lower�assertNotEquals� policyid_dc�ntdsguid)r r r! r" r+ �ldbs�namesr r r �test_find_key_param: s
�z;UpgradeProvisionBasicLdbHelpersTestCase.test_find_key_paramN)�__name__�
__module__� __qualname__�__doc__r# r2 r r r r r / s r c sx e Zd Zdd� Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
� fdd�Z� ZS )� UpgradeProvisionWithLdbTestCasec C s t j�| jd�S )N�sam.ldb)�os�path�join�tempdir�r r r r �_getEmptyDbNameK s z/UpgradeProvisionWithLdbTestCase._getEmptyDbNamec s� t t| ��� ttd t�}t� | _t� | _ | j�
| j � || _ t || jt
� | j �| _t| jj| jj| jj|t| j �| _ttj�| jd��| _d S )N�ref.ldb)�superr7 �setUpr r r r r! r r" r r r r r0 r r% r& r' r1 r r9 r: r; r< � referencedb)r r �� __class__r r rA N s �
�z%UpgradeProvisionWithLdbTestCase.setUpc C s( t | jj| jjdg�}| �d|v � d S )Nzmsds-KeyVersionNumber)r r0 r% r1 r+ � assertFalse)r �hashAttr r r �$test_search_constructed_attrs_stored\ s
�zDUpgradeProvisionWithLdbTestCase.test_search_constructed_attrs_storedc C s� d| j j }i }d||�� < t| jj| j j|� | �| jj�|d�d� d||�� < t| jj| j j|� | �| jj�|d�d� d S )Nz CN=Administrator,CN=Users,%s� �
unicodePwd� )r1 r+ r, r r0 r% r( �"get_attribute_replmetadata_version)r �dn�hashr r r �+test_increment_calculated_keyversion_numberb s, ��� ���zKUpgradeProvisionWithLdbTestCase.test_increment_calculated_keyversion_numberc C sp d}t �| jjd| �}t| jj|� | jjjd|t jdgd�}| �t|�d� | �t |d d �d| � d S )NzDC=samba,DC=example,DC=comzCN=Guest,CN=Users,%sz (name=Guest)rL )�
expression�base�scope�attrs� r )
�ldb�Dnr0 r% r �search�
SCOPE_SUBTREEr( �lenr* )r r+ �guestDN�resr r r �test_identic_renamey s � z3UpgradeProvisionWithLdbTestCase.test_identic_renamec C s( | � � }t| jj|| jt� | jt� d S r )r> r
r �samdbr! r r" r )r � dummysampathr r r �test_delta_update_basesamdb� s
�z;UpgradeProvisionWithLdbTestCase.test_delta_update_basesamdbc C sd t | jj| jj| jj�}t�|� | �t j
� |�� t | j| j
j| j| jt� | �t j
� |�� d S r )r r �sysvolr1 � dnsdomain�policyid�shutil�rmtreerE r9 r: �isdirr r0 r% r" r �
assertTrue)r �dirr r r �test_update_gpo_simple� s �
z6UpgradeProvisionWithLdbTestCase.test_update_gpo_simplec C s� t j�| jd�}| jj}|| j_t �|� t �t j�|| jj�� t �t j�t j�|| jj�d�� t | j| j
j | j| j t
� t�|� || j_d S )N�
testupdategpo�Policies)r9 r: r; r< r r_ �mkdirr1 r` r r0 r% r" r rb rc )r r: �saver r r �test_update_gpo_acl� s
�
z3UpgradeProvisionWithLdbTestCase.test_update_gpo_aclc C s: | j �d�}d|�dd� }t| jj|�}| �|d� d S )Nr) �DC=%s�.�, DC=r$ )r" �get�replacer r0 r% r- )r r) �basedn�oemr r r �test_getOEMInfo� s z/UpgradeProvisionWithLdbTestCase.test_getOEMInfoc C s t | jj| jj| j� d S r )r r0 r% r& r1 r= r r r �test_update_dns_account� s �z7UpgradeProvisionWithLdbTestCase.test_update_dns_accountc C st | j �d�}d|�dd� }t| jj|�}t| jj|� t| jj|�}| �t|�t|�� | � t
� dt|��� d S )Nr) rm rn ro z.*upgrade to.*) r" rp rq r r0 r% r
r- r* re �re�match)r r) rr rs �oem2r r r �test_updateOEMInfo� s z2UpgradeProvisionWithLdbTestCase.test_updateOEMInfoc sB dD ]}t j�| j|�}t j�|�rt �|� qtt| ��� d S )N)r? z secrets.ldbz secrets.tdbzsecrets.tdb.bakz secrets.ntdbr8 ) r9 r: r; r< �exists�unlinkr@ r7 �tearDown)r �namer: rC r r r| � s
�z(UpgradeProvisionWithLdbTestCase.tearDown)r3 r4 r5 r> rA rG rN r[ r^ rg rl rt ru ry r| �
__classcell__r r rC r r7 I s
r7 )NN)#r6 r9 rv rb �sambar �samba.credentialsr �
samba.authr �samba.provisionr r �samba.upgradehelpersr r r r
r r r
r r r � samba.testsr r �samba.tests.provisionr rT r �environr r r7 r r r r �<module> s 0