File: //lib/python3/dist-packages/samba/tests/kcc/__pycache__/kcc_utils.cpython-310.pyc
o
�/a�6 � @ s� d Z ddlZddlZddlmZmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddlm
Z
G d d
� d
ejj�ZG d d � d ejj�ZdS )
z Tests for samba.kcc.kcc_utils� N)�new_connection_schedule�drsblobs)�uncovered_sites_to_cover)� Credentials)�system_session)�SamDB)� delete_forcec @ s e Zd Zdd� ZdS )�
ScheduleTestsc C s> t � }| �|tj� | �|jd� | �t|jd j�d� d S )N� r � ) r �assertIsInstancer �schedule� assertEqual�size�len� dataArray�slots)�selfr
� r �;/usr/lib/python3/dist-packages/samba/tests/kcc/kcc_utils.py� test_new_connection_schedule! s z*ScheduleTests.test_new_connection_scheduleN)�__name__�
__module__� __qualname__r r r r r r s r c @ sr e Zd Zdd� Zdd� Zdd� Zdd� Zg d fd
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�SiteCoverageTestsc C sV d| _ tj�� | _i | _i | _t� | _| j� | j� t
� | _ t | j | j| jd�| _
d S )N�kcc_)� session_info� credentials�lp)�prefix�samba�tests� env_loadparmr �sites�
site_linksr �creds�guessr �sessionr �samdb)r r r r �setUp. s �zSiteCoverageTests.setUpc C sN | j �� | jD ] }t| j |dgd� q| jD ]}t| j |� q| j �� d S )Nz
tree_delete:1)�controls)r( �transaction_startr# r r$ �transaction_commit)r �site� site_linkr r r �tearDown= s
zSiteCoverageTests.tearDownc C s* d� ||�}| j�|d| j�� d�� |S )NzCN={0},CN=Servers,{1}�server)�dn� objectClass�serverReference)�formatr( �add� domain_dn)r �namer- r1 r r r � _add_serverH s �z SiteCoverageTests._add_serverc C sR d� || j�� �}| j�|dd�� | j�d| dgd�� || j|< ||�� fS )NzCN={0},CN=Sites,{1}r- )r1 r2 z CN=Servers,�serversContainer)r4 r( �get_config_basednr5 r# �lower)r r7 r1 r r r � _add_siteQ s
���
zSiteCoverageTests._add_site�d c C s: d� || j�� �}| j�|dt|�|d�� || j|< |S )Nz2CN={0},CN=IP,CN=Inter-Site Transports,CN=Sites,{1}�siteLink)r1 r2 �cost�siteList)r4 r( r: r5 �strr$ )r r7 �linksr? r1 r r r �_add_site_linka s
��
z SiteCoverageTests._add_site_linkc C s� | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd d |� | �| jd d |� | �| jd |||g� | j �� t| j |�}|�� | � |g|� t| j |�}|�� | � g |� d S )N�ABCD�BCDE� uncovered�1�link�
r( r+ r< r r8 rC r, r �sortr �r �site1�name1�site2�name2� uncovered_dnrF �to_coverr r r �#test_single_site_link_same_dc_countn s
�
z5SiteCoverageTests.test_single_site_link_same_dc_countc C s | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd d |� | �| jd d |� | �| jd d |� | �| jd d |� | �| jd d |� | �| jd |||g� | j �� t| j |�}|�� | � g |� t| j |�}|�� | � |g|� d S )NrD rE rF rG �2�3rH rI rK r r r �(test_single_site_link_different_dc_count� s&
�
z:SiteCoverageTests.test_single_site_link_different_dc_countc C s2 | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd d |� | �| jd d |� | �| jd d |� | �| jd d |� | �| jd d |� | �| jd ||g� | �| jd ||g� | j �� t| j |�}|�� | � |g|� t| j |�}|�� | � |g|� d S ) NrD rE rF rG rS rT �link1�link2rI rK r r r � test_two_site_links_same_cost� s,
� �
z/SiteCoverageTests.test_two_site_links_same_costc C s | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd d |� | �| jd d |� | �| jd d |� | j| jd ||gdd� | j| jd ||gd
d� | j �� t| j |�}|�� | � |g|� t| j |�}|�� | � g |� d S ) NrD rE rF rG rS rV �2 �r? rW �K rI rK r r r �#test_two_site_links_different_costs� s,
� �
z5SiteCoverageTests.test_two_site_links_different_costsc
C sp | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd d |� | �| jd d |� | �| jd d |� | �| jd d |� | j| jd ||gdd � | j| jd
||gd d � | j| jd ||gd
d � | j �� t| j |�} | �� | � |g| � t| j |�} | �� | � g | � t| j |�} | �� | � g | � d S )NrD rE �CDEFrF rG rS rV rY rZ rW r[ �link3�< rI �
r rL rM rN rO �site3�name3rP rF rQ r r r �%test_three_site_links_different_costs� s>
� � �
z7SiteCoverageTests.test_three_site_links_different_costsc
C sr | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd d |� | �| jd d |� | �| jd d |� | �| jd d |� | j| jd ||gdd � | j| jd
||gd d � | j| jd ||gdd � | j �� t| j |�} | �� | � |g| � t| j |�} | �� | � g | � t| j |�} | �� | � |g| � d S )
NrD rE r] rF rG rS rV rY rZ rW r[ r^ rI r` r r r �%test_three_site_links_duplicate_costs s>
� � �
z7SiteCoverageTests.test_three_site_links_duplicate_costsc C sl | j �� | �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd �\}}| �| jd �\} }
| �| jd �\} } | �| jd �\}
}| �| jd �\}}| �| jd �\}}| �| jd d
|� | �| jd d
|� | �| jd d |� | �| jd d
|� | �| jd d |� | �| jd d |� | �| jd d
|� | �| jd d |� | �| jd d
| � | �| jd d | � | �| jd d
| � | j| jd
||||
gdd� | j| jd || | |
gdd� | j| jd ||||gdd� | j| jd || | |gdd� | j| jd ||||gdd� | j| jd || | |gdd� | j �� t| j |�}|�� | � g |� t| j |�}|�� | � g |� t| j |�}|�� | � ||g|� t| j |�}|�� | � ||g|� t| j |
�}|�� | � g |� t| j | �}|�� | � g |� |||fD ]}t| j |�}|�� | � g |� �q�d S )NrD rE r] �1234�2345�3456�
uncovered1�
uncovered2�
uncovered3rG rS rT �link1A�1 rZ �link2ArY �link1B�link2B�link1C�link2CrI )r rL rM rN rO ra rb �site4�name4�site5�name5�site6�name6�
uncovered_dn1rh �
uncovered_dn2ri �
uncovered_dn3rj rQ �to_checkr r r �0test_complex_setup_with_multiple_uncovered_sites* s�
�
�
�
�
�
�
�zBSiteCoverageTests.test_complex_setup_with_multiple_uncovered_sitesN)r r r r) r/ r8 r< rC rR rU rX r\ rc rd r| r r r r r , s
' (r )�__doc__r � samba.tests�samba.kcc.kcc_utilsr r r �samba.credentialsr �
samba.authr � samba.samdbr r r! �TestCaser r r r r r �<module> s