File: //lib/python3/dist-packages/samba/tests/kcc/__pycache__/ldif_import_export.cpython-310.pyc
o
�/a�& � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZm Z ddlm
Z
ddl m Z ddl
mZ ddlmZ ddlmZ ee�� �Zej�ejd d
�Zej�ejd d �Zd ZG d
d� dejj �Z G dd� dejj �Z dS )z&Tests for samba.kcc.ldif_import_export� N)�ldif_import_export�KCC)�ldb)�misc)�LoadParm)� Credentials)�SamDB�
SRCDIR_ABSz'testdata/ldif-utils-test-multisite.ldifz#testdata/unconnected-intrasite.ldif)
)zXCN=WIN08,CN=Servers,CN=Site-4,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=com�Site-4)zXCN=WIN07,CN=Servers,CN=Site-4,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comr
)zXCN=WIN06,CN=Servers,CN=Site-3,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comzSite-3)zXCN=WIN09,CN=Servers,CN=Site-5,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=com�Site-5)zXCN=WIN10,CN=Servers,CN=Site-5,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comr )zXCN=WIN02,CN=Servers,CN=Site-2,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=com�Site-2)zXCN=WIN04,CN=Servers,CN=Site-2,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comr )zXCN=WIN03,CN=Servers,CN=Site-2,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comr )zXCN=WIN05,CN=Servers,CN=Site-2,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comr )�iCN=WIN01,CN=Servers,CN=Default-First-Site-Name,CN=Sites,CN=Configuration,DC=ad,DC=samba,DC=example,DC=comzDefault-First-Site-Namec sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�LdifImportExportTestsc �0 t t| ��� t� | _t� | _| j�| j� d S �N)�superr �setUpr �lpr �creds�guess��self�� __class__� �D/usr/lib/python3/dist-packages/samba/tests/kcc/ldif_import_export.pyr I � zLdifImportExportTests.setUpc G �( |D ]}|� | j�s J �t�|� qd S r ��
startswith�tempdir�os�unlink�r �files�fr r r � remove_filesO � �z"LdifImportExportTests.remove_filesc C s d S r r r r r r �test_write_search_urlT s z+LdifImportExportTests.test_write_search_urlc C s� t j�| jd�}t�|| jt�}| �|t � d}|j
t � |d| �t j
dgd�}t�|�� �}| �t�|d d d �|� |j
dt j
dgd �}t � ||d d d �d
��}| �|t � |d| �� | �|� d S ) N�ldapr
�CN=NTDS Settings,�
objectGUID��scope�attrsr � �
dsServiceName��baser- r. �utf8)r! �path�joinr r �
ldif_to_samdbr �MULTISITE_LDIF�assertIsInstancer �searchr �Dn�
SCOPE_BASEr �GUID�
get_ntds_GUID� assertEqual�decoder& )r �dburl�samdb�dsa�res� ntds_guid�service_name_res�dnr r r �test_ldif_to_samdbW s(
� � ��z(LdifImportExportTests.test_ldif_to_samdbc C s� t D ]v\}}tj�| jd| �}tj|| jt|d�}| � |t
� | � |� � |� |j
t�|d| �tjdgd�}t�|�� �}| � t�|d d d �|� |j
dtjdgd �}t�||d d d �d
��}| � |t�|d| �� | �|� qd S ) Nz!ldif-to-samba-forced-local-dsa-%s)�forced_local_dsar* r+ r, r r/ r0 r1 r3 )�MULTISITE_LDIF_DSASr! r4 r5 r r r6 r r7 r8 r r> �server_site_namer9 r r: r; r r< r= r? r& ) r rB �siter@ rA rC rD rE rF r r r �#test_ldif_to_samdb_forced_local_dsam s2 �
� � �� �z9LdifImportExportTests.test_ldif_to_samdb_forced_local_dsac C s� t j�| jd�}t j�| jd�}t j�| jd�}t�|| jt�}| �|t � tj
||| jd |d� | � t j� |�dd� t�|| j|�}| �|t � d}|j
t�|d| �tjd gd
�}| �|� | �|� | �|� d S ) Nr) �ldap_roundtrip�ldif)r r � ldif_filei� z%LDIF should be larger than 1000 bytesr
r* r+ r, )r! r4 r5 r r r6 r r7 r8 r �samdb_to_ldif_file�
assertGreater�getsizer9 r r: r; r& )r r@ �dburl2rO rA rB rC r r r �test_samdb_to_ldif_file� s2
� ��
� �
z-LdifImportExportTests.test_samdb_to_ldif_file)
�__name__�
__module__� __qualname__r r& r( rG rL rT �
__classcell__r r r r r H s r c sN e Zd Z� fdd�Zdd� Zddd�Zd d
� Zd d � Zd
d� Zdd� Z � Z
S )�KCCMultisiteLdifTestsc r r )r rY r r r r r r r r r r r � r zKCCMultisiteLdifTests.setUpc G r r r r# r r r r& � r' z"KCCMultisiteLdifTests.remove_filesFNc C s> t t|||d�}tj�| jd�}|�|| jt� | � |� |S )N)�readonly�verify� dot_file_dir�tmpdb)
r �unix_nowr! r4 r5 r � import_ldifr r7 r& )r �namerZ r[ r\ �my_kccr] r r r �_get_kcc� s �
z KCCMultisiteLdifTests._get_kccc C s8 | � d�}t|�� �}tdd� tD ��}| �||� d S )Nz test-listc s s � | ]}|d V qdS )r Nr )�.0�xr r r � <genexpr>� s � z7KCCMultisiteLdifTests.test_list_dsas.<locals>.<genexpr>)rb �set� list_dsasrI r> )r ra �dsas�
expected_dsasr r r �test_list_dsas� s
z$KCCMultisiteLdifTests.test_list_dsasc C sR | j ddd�}tj�| jd�}|�|| jt� |jd| j| j dd� | �
|� dS )zUCheck that the KCC generates graphs that pass its own verify
option.
� test-verifyT�r[ � verify-tmpdbNF��attempt_live_connections) rb r! r4 r5 r r_ r r7 �runr r& �r ra r] r r r � test_verify� s �z!KCCMultisiteLdifTests.test_verifyc C s� | j ddd�}tj�| jd�}|�|| jt� z?z
|jd| j| j dd� W n t
j j j
y1 Y n ty> | �d� Y nw W | �|� dS W | �|� dS W | �|� dS | �|� w ) z@Check that the KCC generates errors on a unconnected db
rk Trl rm NFrn zDid not expect this error.)rb r! r4 r5 r r_ r �UNCONNECTED_LDIFrp r �samba�kcc� graph_utils�
GraphError� Exception�failr& rq r r r �test_unconnected_db� s&
� ����z)KCCMultisiteLdifTests.test_unconnected_dbc C s� | j d| jd�}tj�| jd�}|g}|�|| jt� |jd| j| j dd� d}t�
| j�D ]=}|� d�rktj�| j|�}tj� |�rZt
�|d g�d
krZt
�|d |g�}| �|d
� t�|�j}| �|d
� |�|� q.| j|� dS ) z4Check that KCC writes dot_files when asked.
z
test-dotfiles)r\ z
dotfile-tmpdbNFrn z /usr/bin/dotz.dotz-?r z-Tcanon)rb r r! r4 r5 r_ r r7 rp r �listdir�endswith�exists�
subprocess�callr> �stat�st_size�assertNotEqual�appendr& ) r ra r] r$ �dot�fn�ffn�r�sizer r r �
test_dotfiles� s( �
�z#KCCMultisiteLdifTests.test_dotfiles)FFN) rU rV rW r r& rb rj rr rz r� rX r r r r rY � s
rY ) �__doc__rt r! �timer~ �logging� samba.tests� samba.kccr r r � samba.dcerpcr � samba.paramr �samba.credentialsr � samba.samdbr �intr^ r4 r5 �environr7 rs rI �tests�TestCaseInTempDirr rY r r r r �<module> s. ��U