File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/domain_backup_offline.cpython-310.pyc
o
�/a( � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ G dd � d e�ZdS )
� N)�BlackboxTestCase�BlackboxProcessError)� CommandError)�LoadParm)�join_DC)� Credentials)�get_samba_loggerc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zg fd d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S ),�DomainBackupOfflineCmpc C � | � d� d S �N�tdb��nested_testcase��self� r �C/usr/lib/python3/dist-packages/samba/tests/domain_backup_offline.py�%test_domain_backup_offline_nested_tdb& � z<DomainBackupOfflineCmp.test_domain_backup_offline_nested_tdbc C r
�N�mdbr
r r r r �%test_domain_backup_offline_nested_mdb) r z<DomainBackupOfflineCmp.test_domain_backup_offline_nested_mdbc C s� | � |�| _d | _tj�| jd�}tj�| jdd�}t�||� tj�| jdd�}t|d�}|� d|� |�
d|� | � | j�}t �
|�}|�� }| �t|�tt |��� d S )N�private�state�etczsmb.conf)�filename_for_non_global_lpz private dirF)� provision�prov_dir� extract_dir�os�path�join�shutil�mover �set�dump�backup�tarfile�open�getnames� assertEqual�len) r �backend�src�dst�smbconf�lp� backup_file�tf�namesr r r r , s
z&DomainBackupOfflineCmp.nested_testcasec C r
r ��join_restore_testcaser r r r �+test_domain_backup_offline_join_restore_tdbD r zBDomainBackupOfflineCmp.test_domain_backup_offline_join_restore_tdbc C r
r r4 r r r r �+test_domain_backup_offline_join_restore_mdbG r zBDomainBackupOfflineCmp.test_domain_backup_offline_join_restore_mdbc
C sb | � |�| _d | _z| �| j�}W n ty( } z | �|� W Y d }~nd }~ww | �|�| _d S �N)r! r r r& r �fail�restore)r r, r1 �er r r r5 J s ��z,DomainBackupOfflineCmp.join_restore_testcasec C r
r ��hard_link_testcaser r r r �(test_domain_backup_offline_hard_link_tdbU r z?DomainBackupOfflineCmp.test_domain_backup_offline_hard_link_tdbc C r
r r< r r r r �(test_domain_backup_offline_hard_link_mdbX r z?DomainBackupOfflineCmp.test_domain_backup_offline_hard_link_mdbc C s� | � |�| _d | _t�tj�| jdd�tj�| jdd�� | �| j�}tj | j
d�| _t � |�}|�
| j� | �tj�tj�| jdd��� | �tj�tj�| jdd��� d S )Nr z krb5.confr ��dir�statedir)r r r r �linkr r! r& �tempfile�mkdtemp�tempdirr' r( �
extractall�
assertTrue�exists� assertFalse�r r, r1 r2 r r r r= [ s �
� �z)DomainBackupOfflineCmp.hard_link_testcasec C r
r ��untar_testcaser r r r �$test_domain_backup_offline_untar_tdbq r z;DomainBackupOfflineCmp.test_domain_backup_offline_untar_tdbc C r
r rL r r r r �$test_domain_backup_offline_untar_mdbt r z;DomainBackupOfflineCmp.test_domain_backup_offline_untar_mdbc C r
r ��restore_testcaser r r r �&test_domain_backup_offline_restore_tdbw r z=DomainBackupOfflineCmp.test_domain_backup_offline_restore_tdbc C r
r rP r r r r �&test_domain_backup_offline_restore_mdbz r z=DomainBackupOfflineCmp.test_domain_backup_offline_restore_mdbc C s^ | � |�| _d | _| �| j�}| �|�| _g d�}dd�|� }d|g}| �| j| j|� d S )N)
�servicePrincipalName�lastLogonTimestamp�rIDAllocationPool�rIDAvailablePool� rIDUsedPool�localPolicyFlags�operatingSystem� displayName� dnsRecord�
dNSTombstonedzmsDS-NC-Replica-LocationszmsDS-HasInstantiatedNCs�interSiteTopologyGeneratorz --filter=�,z--two)r r r r& r: r! �ldapcmp)r r, r1 � ignore_attrs�
filter_arg�argsr r r rQ } s z'DomainBackupOfflineCmp.restore_testcasec C sX | � |�| _d | _| �| j�}tj| jd�| _t�|�}|� | j� | �
| j| j� d S )Nr@ ) r r r r& rD rE rF r' r( rG r` rK r r r rM � s
z%DomainBackupOfflineCmp.untar_testcasec C sr t j�dd�}dt j�t j�|�|� }dt j�t j�|�|� }dD ]}dd�|||g| � }| �|� q#d S )Nr zsam.ldbztdb://)�domain�
configuration�schema� dnsdomain� dnsforestzsamba-tool ldapcmp � )r r r! �realpath� check_output) r r �ex_dirrc �sam_fn�url1�url2� partition�cmdr r r r` � s �z DomainBackupOfflineCmp.ldapcmpc C s. t j| jd�}d}|j||d�}| �|� |S )Nr@ z�samba-tool domain provision --domain FOO --realm foo.example.com --targetdir {target} --backend-store {backend} --host-name OLDSERVER --option="vfs objects=dfs_samba4 acl_xattr fake_acls xattr_tdb")�targetr, �rD rE rF �formatrk )r r, rr �prov_cmdr r r r � s
z DomainBackupOfflineCmp.provisionc
C s� t j| jd�}d}d}|jtjd tjd tjd tjd tjd |||d �}| �|� d
}|jtjd tjd tjd |d �}| �|� |S ) Nr@ �offlinebackupdcz�samba-tool domain join {domain} DC --server {server} --realm {realm} --username {username}%{password} --targetdir {target} --backend-store {backend} --option='netbios name = {new_dc_name}' --option="vfs objects=dfs_samba4 acl_xattr fake_acls xattr_tdb"� DC_SERVER�DOMAIN�REALM�USERNAME�PASSWORD)�serverrd �realm�username�passwordrr r, � new_dc_nameztsamba-tool domain demote --server {server} --username {username}%{password} --remove-other-dead-server={new_dc_name})r| r~ r r� )rD rE rF rt r �environrk )r r, rr r� �join_cmd�
demote_cmdr r r r! � s* �
�
zDomainBackupOfflineCmp.joinc C sb dj |d�}| �|� dd� t�|�D �}t|�dkr&tdd� t|�� ��tj�||d �}|S ) Nz\samba-tool domain backup offline --targetdir={prov_dir} --configfile={prov_dir}/etc/smb.conf)r c S s$ g | ]}|� d �r|�d�r|�qS )z
samba-backup-z.tar.bz2)�
startswith�endswith)�.0�fnr r r �
<listcomp>� s
�
�z1DomainBackupOfflineCmp.backup.<locals>.<listcomp>� z(expected domain backup to create one tarz file but got {0}r )rt rk r �listdirr+ r r r! )r r rq � tar_filesr1 r r r r& � s �
�z DomainBackupOfflineCmp.backupc C s* t j| jd�}dj||d�}| �|� |S )Nr@ z\samba-tool domain backup restore --backup-file={f} --targetdir={d} --newservername=NEWSERVER)�f�drs )r r1 r rq r r r r: � s �
z DomainBackupOfflineCmp.restorec C s& t �| j� | jrt �| j� d S d S r8 )r" �rmtreer r r r r r �tearDown� s �z DomainBackupOfflineCmp.tearDownN)�__name__�
__module__� __qualname__r r r r6 r7 r5 r> r? r= rN rO rR rS rQ rM r` r r! r& r: r� r r r r r $ s, $ r )r' r r" rD � samba.testsr r � samba.netcmdr � samba.paramr �
samba.joinr �samba.credentialsr � samba.loggerr r r r r r �<module> s