File: //lib/python3/dist-packages/samba/tests/samba_tool/__pycache__/contact.cpython-310.pyc
o
�/a]T � @ 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� Zdd� Zi fdd�Zi fdd�Zd!dd �Zd d � Zd d � Z� ZS )"�ContactCmdTestCasez(Tests for samba-tool contact subcommandsNc
sl t t| ��� dtjd tjd f | _| �ddtjd | j�| _d }g | _| � ddd��}| j�
|� | � d d d
��}| j�
|� | � d d d d
dd
dd��}| j�
|� | � ddddd��}| j�
|� | jD ]J}| � |�\}}}| � |||� | �
d|d� | �d|d |� | �|d �}| �|� |d }| �d|�d� |� | �d|�d � |d � qid S ) Nz-U%s%%%s� DC_USERNAME� DC_PASSWORDz-Hz ldap://%s� DC_SERVER�contact1)� expectedname�name�contact2)r � givenName�contact3�contact3displayname� not_contact3�I�12345)r r � displayNamer �initials�sn�mobilez
James T. Kirk�Kirk�T�James)r r r r �ERROR�$There shouldn't be any error message� Contact '%s' added successfullyr �%sr � description)�superr �setUp�os�environ�creds�getSamDB�samdb�contacts�_randomContact�append�_create_contact�assertCmdSuccess� assertNotIn�assertIn�
_find_contact�assertIsNotNone� assertEqual�get)�self�contact�result�out�err�found� contactname�� __class__� �@/usr/lib/python3/dist-packages/samba/tests/samba_tool/contact.pyr sh
� �� � � �
���
��zContactCmdTestCase.setUpc s` t t| ��� | jD ]#}| �|d �r-| �ddd|d �\}}}| �|||d|d � q
d S )Nr r0 �deleter � Failed to delete contact '%s')r r �tearDownr$ r+ � runsubcmdr( )r/ r0 r1 r2 r3 r6 r8 r9 r< U s
�
����zContactCmdTestCase.tearDownc C s� | j D ]}| �|�\}}}| �|d� | �d|� q| j D ]-}| �ddd|d �\}}}| �|||d|d � | �|d �}| �|d|d � q | �d d
i�}| � |�\}}}| �|||� | j D ]G}| j|d d �\}}}| �|||� | �
|d
d� | �d|d |� | �|d �}|d }| �
d|� d � |� | �
d|� d� |d � qd| j D ]>}t �
| jd|d |d | j�� f �}| �ddd| �\}}}| �|||d|d � | �|d �}| �|d|d � q�| �ddd|d �\}}}| �|||d|d � | j D ]F}| �|�\}}}| �|||� | �
|d
d� | �d|d |� | �|d �}|d }| �
d|� d � |� | �
d|� d� |d � �q dS )z=This tests the "contact create" and "contact delete" commandsz$Succeeded to create existing contactzalready existsr0 r: r r r; z!Deleted contact '%s' still existsr �testOUz OU=testOU)�ou� r r r �CN=%s,OU=%s,%sr? �OU=%s�Failed to delete ou '%s'N)r$ r' �
assertCmdFailr* r= r( r+ � assertIsNone� _randomOU�
_create_our- r. �ldb�Dnr# � domain_dn) r/ r0 r1 r2 r3 r4 �parentour5 �
expecteddnr8 r8 r9 �test_newcontact` s�
�
����
���
���
�
����
�
�
����z"ContactCmdTestCase.test_newcontactc C s� | � dd�\}}}| �|||d� d}| jj| j�� tj|dgd�}| �t|�dkd� |D ]}|j ddd �}| �
|t |�d
| � q.d S ) Nr0 �list�Error running list�(objectClass=contact)r ��base�scope�
expression�attrsr �no contacts found in samdb��idx�contact '%s' not found) r= r( r# �searchrJ rH �
SCOPE_SUBTREE�
assertTrue�lenr. � assertMatch�str)r/ r1 r2 r3 �
search_filter� contactlist�
contactobjr r8 r8 r9 � test_list� s � ��z ContactCmdTestCase.test_listc C s� | � ddd�\}}}| �|||d� d}| jj| j�� tj|dgd�}| �t|�dkd � |D ]}| � |t
|j �d
t
|j � � q/d S ) Nr0 rN z --full-dnrO rP �dnrQ r rV rY ) r= r( r# rZ rJ rH r[ r\ r] r^ r_ rd )r/ r1 r2 r3 r` ra rb r8 r8 r9 �test_list_full_dn� s � ��z$ContactCmdTestCase.test_list_full_dnc C s� t | j�� �}| �ddd|�\}}}| �|||d� d}| jj|tj|dgd�}| �t |�dkd � |D ]}|j
ddd
�}| � |t |�d | � q4d S ) Nr0 rN z-brO rP r rQ r rV rW rY ) r_ r# rJ r= r( rZ rH r[ r\ r] r. r^ ) r/ �base_dnr1 r2 r3 r` ra rb r r8 r8 r9 �test_list_base_dn� s$
�� ��z$ContactCmdTestCase.test_list_base_dnc
C s� | � ddi�}| �|�\}}}| �|||� | jD ]�}| �|d ��d�}| �ddd|d d|d �\}}}| �|||d |d � | �|d
d � | �d |d |� | �|d �}| � |�d�|d
|d � |d }t
� | j d||d | j �
� f �} | �|�d�| d|d � | �ddd|d d|�� �\}}}| �|||d |d � q| �ddd|d �\}}}| �|||d|d � d S )Nr �parentOUr rd r0 �mover rB zFailed to move contact '%s'r@ r zMoved contact "%s"z0Moved contact '%s' still exists with the same dnrA z!Moved contact '%s' does not existr? r: rC )rF rG r( r$ r+ r. r= r- r* �assertNotEqualsrH rI r# rJ �parent)
r/ rK r1 r2 r3 r0 �olddnr4 r5 �
newexpecteddnr8 r8 r9 � test_move� sf
�
���������
�
��
�
�z ContactCmdTestCase.test_movec C s. | j D �]}d|v r|d n|d }d| }d}d| }d| }d|||f }| �dd |d
d | d | d
| �\}} }
| �|| |
� | �|
dd� | �d| � | �|�} | �d| �d� |� | �d| �d� |� | �d| �d� |� | �d| �d� |� | �d| �d� |� | �dd |d| ddd�\}} }
| �|| |
� | �|
dd� | �d| � | �|�} | �| �d�d� | �| �d�d� | �| �d�d� d|v r�|d nd} d|v r�|d nd}
d|v r�|d nd}d|v r�|d n|}| �dd |d| d | d |
d
| �\}} }
| �|| |
� qdS )zCrename and remove given name, initials and surname for all contactsr r �new_given_name_of_�A�new_surname_of_�
new_cn_of_z %s %s. %sr0 �renamez
--reset-cnz --surname=%sz
--initials=%sz--given-name=%sr@ � Shouldn't be any error messages� successfullyr r r r �cn�--force-new-cn=%sz
--surname=z --initials=z
--given-name=N�r$ r= r( r- r* r+ r. )r/ r0 r �
new_givenname� new_initials� new_surname�new_cn� expected_cnr1 r2 r3 r4 � old_surname� old_initials�
old_givenname�old_cnr8 r8 r9 �&test_rename_givenname_initials_surname
s^
�
�
��z9ContactCmdTestCase.test_rename_givenname_initials_surnamec C s� | j D ]s}d|v r
|d n|d }d| }| �dd|d| �\}}}| �|||� | �|dd� | �d |� | �|�}| �d
|�d � |� | �dd|d �\}}}| �|� | �d
|� | �d|� | �dd|d| �\}}}| �|||� qdS )z/rename and try to remove the cn of all contactsr r rr r0 rs rw r@ rt ru r rv z--force-new-cn=zFailed to rename contactzdelete protected attributeN)r$ r= r( r- r* r+ r. rD )r/ r0 r r| r1 r2 r3 r4 r8 r8 r9 �test_rename_cnE s,
�
�
��z!ContactCmdTestCase.test_rename_cnc C sz | j D ]�}d|v r
|d n|d }d| }d| }| �dd|d| d| �\}}}| �|||� | �|d d
� | �d |� | �|�}| �d |�d
� |� | �d |�d� |� | �dd|dd�\}}}| �|||� | �|d d
� | �d |� | �|�}| �|�d
�d� | �|�d�d� d|v r�|d nd } d|v r�|d nd }
| �dd|d| d|
�\}}}| �|||� qdS )zHrename and remove the mail and the displayname attribute of all contactsr r �new_mailaddress_of_znew displayname of r0 rs z--mail-address=%sz--display-name=%sr@ rt ru r �mailr z--mail-address=z--display-name=Nr rv rx ) r/ r0 r �new_mail�new_displaynamer1 r2 r3 r4 �old_mail�old_displaynamer8 r8 r9 �#test_rename_mailaddress_displaynameb sH
��
�
�
��z6ContactCmdTestCase.test_rename_mailaddress_displaynamec C s d| j dd�i}|�|� |S )zVCreate a contact with random attribute values, you can specify base
attributesr �d ��count��
randomName�update)r/ rR r0 r8 r8 r9 r% � s �
z!ContactCmdTestCase._randomContactc C s$ | � � | j dd�d�}|�|� |S )zSCreate an ou with random attribute values, you can specify base
attributes.r� r� )r r r� )r/ rR r? r8 r8 r9 rF � s
�
z ContactCmdTestCase._randomOUc C s� d}d|v r|d� |d �7 }|d� | j�7 }|d ur"|d� |�7 }d|v r/|d� |d �7 }d|v r<|d � |d �7 }d
|v rI|d � |d
�7 }d |v rV|d
� |d �7 }d|v rc|d� |d �7 }d|v rp|d� |d �7 }|�� }| jddg|�R � S )Nr@ r z{0}z {0}z --ou={0}r z --description={0}r z --surname={0}r z --initials={0}r z --given-name={0}r z --display-name={0}r z --mobile-number={0}r0 �create)�formatr! �splitr= )r/ r0 r? �argsr8 r8 r9 r' � s( z"ContactCmdTestCase._create_contactc C s | � ddd|d d|d �S )Nr? r� rB r z--description=%sr )r= )r/ r? r8 r8 r9 rG � s
�z ContactCmdTestCase._create_ouc C s> |}dt �|� }| jj| j�� t j|g d�}|r |d S d S )Nz!(&(objectClass=contact)(name=%s))rQ r )rH �
binary_encoder# rZ rJ r[ )r/ r r5 r` ra r8 r8 r9 r+ � s ��z ContactCmdTestCase._find_contact)N)�__name__�
__module__� __qualname__�__doc__r$ r# r r< rM rc re rg rn r� r� r� r% rF r' rG r+ �
__classcell__r8 r8 r6 r9 r s& 7 N+8 )
r )r rH �samba.tests.samba_tool.baser r r8 r8 r8 r9 �<module> s