File: //lib/python3/dist-packages/samba/tests/samba_tool/__pycache__/ou.cpython-310.pyc
o
�/at1 � @ s0 d dl Z d dlZd dlmZ G dd� de�ZdS )� N)�SambaToolCmdTestc s� e Zd ZdZg ZdZ� fdd�Z� fdd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zi fdd�Zdd� Zdd� Z� ZS )�
OUCmdTestCasez#Tests for samba-tool ou subcommandsNc sF t t| ��� | �ddtjd dtjd tjd f �| _g | _| j�| � ddi�� | j�| � dd i�� | j�| � dd
i�� | j�| � dd i�� | jD ]P}| �
|�\}}}| � |||� | � |d d
� | j�
d|d �}| �d| |� | �|d �}| �|� | � d|�d� |d � | � d|�d� |d � qPd S )Nz-Hz ldap://%s� DC_SERVERz-U%s%%%s� DC_USERNAME� DC_PASSWORD�name�testou1�testou2�testou3�testou4� �$There shouldn't be any error message�OU=%s�
Added ou "%s"�%s� description)�superr �setUp�getSamDB�os�environ�samdb�ous�append� _randomOU�
_create_ou�assertCmdSuccess� assertEqual�normalize_dn_in_domain�assertIn�_find_ou�assertIsNotNone�get)�self�ou�result�out�err�
full_ou_dn�found�� __class__� �;/usr/lib/python3/dist-packages/samba/tests/samba_tool/ou.pyr s, �
��zOUCmdTestCase.setUpc s` t t| ��� | jD ]#}| �|d �r-| �ddd|d �\}}}| �|||d|d � q
d S )Nr r$ �deleter �Failed to delete ou '%s')r r �tearDownr r � runsubcmdr )r# r$ r% r& r' r* r, r- r0 ; s
�
���zOUCmdTestCase.tearDownc C sX | j D ]}| �|�\}}}| �|d� | �d|� q| j D ]-}| �ddd|d �\}}}| �|||d|d � | �|d �}| �|d|d � q | j D ]K}| �dd d|d d
|d �\}}}| �|||� | �|d d
� | j �
d|d �}| �d| |� | �|d �}| �d|� d� d|d � qN| j D ]5}| j �
d|d �}| �ddt |��\}}}| �|||d|d � | �|d �}| �|d|d � q�| j D ]S}| j �
d|d �}| �dd t |�d
|d �\}}}| �|||� | �|d d
� | j �
d|d �}| �d| |� | �|d �}| �d|� d� d|d � q�dS )z3This tests the "ou create" and "ou delete" commandsz Succeeded to add existing ouzalready existsr$ r. r r r/ z Deleted ou '%s' still exists�add�--description=%sr r r
r r N)
r r �
assertCmdFailr r1 r r � assertIsNoner r r r" �str)r# r$ r% r& r' r) r( r, r, r- �
test_newouE sj
�
�
�
�
�
�
�
�
��zOUCmdTestCase.test_newouc C s� | � dd�\}}}| �|||d� d}| jj| j�� tj|dgd�}| �t|�dkd� |D ]}|j ddd �}| �
|t |�d
| �}q.d S ) Nr$ �list�Error running list� (objectClass=organizationalUnit)r ��base�scope�
expression�attrsr �no ous found in samdb��idx�ou '%s' not found) r1 r r �search� domain_dn�ldb�
SCOPE_SUBTREE�
assertTrue�lenr" � assertMatchr6 ) r# r% r& r' �
search_filter�oulist�ouobjr r) r, r, r- � test_list� s � ��zOUCmdTestCase.test_listc
C s� t | j�� �}| �ddd|�\}}}| �|||d� d}| jj|tj|dgd�}| �t |�dkd � |D ]}|j
ddd
�}| � |t |�d | �} q4d S ) Nr$ r8 z-br9 r: r r; r r@ rA rC ) r6 r rE r1 r rD rF rG rH rI r" rJ )
r# �base_dnr% r& r' rK rL rM r r) r, r, r- �test_list_base_dn� s � ��z OUCmdTestCase.test_list_base_dnc C s� | j D ]`}d}|d | }| �ddd|d d| �\}}}| �|||d|d � | �|d �}| �|d|d � | �|�}| �|d| � | �ddd| d|d �\}}}| �|||d| � qd S ) N�
RenameTestr r$ �renamer zFailed to rename ou '%s'z Renamed ou '%s' still existsz Renamed ou '%s' does not exist)r r1 r r r5 r! )r# r$ �ousuffix� newounamer% r& r' r) r, r, r- � test_rename� s4
�
�
�
�
�
��zOUCmdTestCase.test_renamec
C s� | � ddi�}| �|�\}}}| �|||� | jD ]�}| �|d ��d�}| �ddd|d d|d �\}}}| �|||d|d � | �|dd � | j� d|d �}| �
d
| |� | �|d �}| � |�d�|d |d � t �
| jd |d |d | j�� f �} | �|�d�| d
|d � | �ddd| d|�� �\}}}| �|||d|d � q| �ddd|d �\}}}| �|||d|d � d S )Nr �parentOU�dnr$ �mover zFailed to move ou '%s'r r
z
Moved ou "%s"z+Moved ou '%s' still exists with the same dnzOU=%s,OU=%s,%sz Moved ou '%s' does not existr r. r/ )r r r r r r" r1 r r r r �assertNotEqualsrF �DnrE �parent)
r# �parentour% r& r' r$ �olddnr( r) �
newexpecteddnr, r, r- � test_move� s\
�
��� �����
�
�
�
�zOUCmdTestCase.test_movec C s� | � ddd| j�� d�\}}}| �|||d� | �|dd� | jj| j�� tjg d�}| �t |�d kd
� |D ]}| �
|t |j �d |j �}q8d S ) Nr$ � listobjectsr � --full-dnzFailed to list ou's objectsr r
)r< r= r? r zno objects found�object '%s' not found)
r1 r rE r r rD rF �SCOPE_ONELEVELrH rI rJ r6 rW )r# r% r& r' �objlist�objr) r, r, r- �test_listobjects� s$
�
����z OUCmdTestCase.test_listobjectsc C s� | � ddd�\}}}| �|||d� | �|dd� d}| jj| j�� tj|g d�}| �t |�d kd
� |D ]}| �
|t |j �d |j �}q5d S ) Nr$ r8 ra zFailed to list ousr r
r: r; r zno ou objects foundrb )
r1 r r r rD rE rF rG rH rI rJ r6 rW )r# r% r& r' �filterrd re r) r, r, r- �test_list_full_dn� s&
�
����z OUCmdTestCase.test_list_full_dnc C s$ | � � | j dd�d�}|�|� |S )zRcreate an ou with random attribute values, you can specify base
attributes�d )�count)r r )�
randomName�update)r# r< r$ r, r, r- r s
�
zOUCmdTestCase._randomOUc C s | � ddd|d d|d �S )Nr$ r2 r r r3 r )r1 )r# r$ r, r, r- r s
�zOUCmdTestCase._create_ouc C sD dt �|�d| j�� f }| jj| j�� t j|d�}|r |d S d S )Nz"(&(name=%s)(objectCategory=%s,%s))z1CN=Organizational-Unit,CN=Schema,CN=Configuration)r< r= r> r )rF �
binary_encoder rE rD rG )r# r rK rL r, r, r- r s ���zOUCmdTestCase._find_ou)�__name__�
__module__� __qualname__�__doc__r r r r0 r7 rN rP rU r_ rf rh r r r �
__classcell__r, r, r* r- r s
=( r )r rF �samba.tests.samba_tool.baser r r, r, r, r- �<module> s