File: //usr/lib/python3/dist-packages/samba/tests/blackbox/__pycache__/downgradedatabase.cpython-310.pyc
o
�/a� � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ ej �
ej � d�d�Z
G dd� de�ZG d d
� d
e�ZG d d � d e�ZdS )
� )�BlackboxTestCaseN)� check_output)�SamDB�
SRCDIR_ABSz(source4/scripting/bin/samba_downgrade_dbc s< e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd � Z� ZS )
�DowngradeTestBasez>Test that sambadowngradedatabase downgrades the samba databasec s� t t| ��� t| d�s| �d� d}|j| d�}| �|d� tj� | j
d�� tj� � d�| _ t j
| j dgd �| _ | j jd
t jd gd �}|d
d }dd� |D �}� fdd�|D �| _| j�| j � d S )N�backendz%Subclass this class and set 'backend'z�samba-tool domain provision --domain FOO --realm foo.example.com --targetdir {self.tempdir} --backend-store {self.backend} --host-name downgradetest --option="vfs objects=dfs_samba4 acl_xattr fake_acls xattr_tdb"��selfzProvisioning for downgrade�privatezsam.ldbzmodules:)�optionsz
@PARTITION� partition��base�scope�attrsr c S s g | ] }t |��d �d �qS )�:� )�str�split��.0�p� r �H/usr/lib/python3/dist-packages/samba/tests/blackbox/downgradedatabase.py�
<listcomp>6 � z+DowngradeTestBase.setUp.<locals>.<listcomp>c s g | ] }t j�� |��qS r )�os�path�joinr �� private_dirr r r 7 s �)�superr �setUp�hasattr�fail�format� check_runr r r �tempdir�sam_path�ldb�Ldb�search�
SCOPE_BASE�dbs�append)r �prov_cmd�
partitions�� __class__r r r" s&
�
�zDowngradeTestBase.setUpc s� t �tj�| jd�� t �tj�| jd�� t �tj�| jd�� t �tj�| jd�� t �tj�| jd�� t�tj�| jd�� t�tj�| jd�� tt| �� � d S )Nr
�etc�statezbind-dnszmsg.lockz names.tdbz gencache.tdb)
�shutil�rmtreer r r r' �unlinkr! r �tearDownr r1 r r r8 ; s zDowngradeTestBase.tearDownc s� g }| j D ] }tdd|g�}|�d�}|�d�}|dd� |D �7 }qd��fdd �|D �}d
� � fd d �|D �}d d � |D �}d
��fdd �|D �}dd� |D �}|||fS )Nz bin/ldbdumpz-izutf-8�
c S s g | ] }|� d �r|�qS )�#��
startswith�r �sr r r r P � z?DowngradeTestBase.ldbdump_keys_pack_formats.<locals>.<listcomp>z # key: GUID=c �& h | ]}|� � �r|t� �d � �qS �N�r< �len�r �c)� guid_key_tagr r � <setcomp>S � �z>DowngradeTestBase.ldbdump_keys_pack_formats.<locals>.<setcomp>z
# key: DN=c r@ rA rB rD )�
dn_key_tagr r rG W rH c S s h | ] }|� d �s|�qS )�@r; )r �dr r r rG [ r? z# pack format: c r@ rA rB rD )�pack_format_tagr r rG ^ rH c S s g | ]}t |d ��qS )� )�intr= r r r r ` s )r- r �decoder )r �comments�db�dump� guid_keys�dn_keys�dn_keys_no_at_attrs� pack_formatsr )rI rF rL r �ldbdump_keys_pack_formatsI s
z+DowngradeTestBase.ldbdump_keys_pack_formatsc sH | j jdt jdgd�� | �t� �d� � d �� }� fdd�|D �}|S )Nz @ATTRIBUTES�*r
r r c s h | ] }t � d �|���qS )r )r �get)r �k��
at_attributesr r rG k r z4DowngradeTestBase.attribute_types.<locals>.<setcomp>)r) r+ r, � assertEqualrC �keys)r r^ �attribute_typesr r[ r r_ e s � z!DowngradeTestBase.attribute_types) �__name__�
__module__� __qualname__�__doc__r" r8 rW r_ �
__classcell__r r r1 r r s r c @ � e Zd ZdZdd� ZdS )�DowngradeTestTDB�tdbc C s� d}t jt|�d � }| �� \}}}| �t|�d� | �t|�d� | �|| �� v � | �|t jg� t|�}| j dt
| j f dd� | �� \}}}| �t|�d� | �t|�|� | �|| �� v� | �|t j g� d S �N� LDB_SYNTAX_� r z%s -H %sz Running sambadowngradedatabase)�msg�
r) �SYNTAX_ORDERED_INTEGERrC rW �
assertGreaterr] �
assertTruer_ �PACKING_FORMAT_V2r& �COMMANDr( �PACKING_FORMAT�r � type_prefix�ordered_int_typerT rS rV � num_guid_keys_before_downgrader r r �test_downgrade_databasev � �z(DowngradeTestTDB.test_downgrade_databaseN)r` ra rb r rw r r r r rf o � rf c @ re )�DowngradeTestMDB�mdbc C s� d}t jt|�d � }| �� \}}}| �t|�d� | �t|�d� | �|| �� v � | �|t jg� t|�}| j dt
| j f dd� | �� \}}}| �t|�|� | �t|�d� | �|| �� v� | �|t j g� d S rh rl rs r r r �test_undo_guid� rx z DowngradeTestMDB.test_undo_guidN)r` ra rb r r| r r r r rz � ry rz )� samba.testsr r r) r5 �
subprocessr � samba.samdbr r r �environrY rq r rf rz r r r r �<module> s �R