File: //lib/python3/dist-packages/samba/tests/blackbox/__pycache__/samba_dnsupdate.cpython-310.pyc
o
�/a� � @ sx d dl Zd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlZd dlZG dd � d ejj�ZdS )
� N)�StringIO)�
get_string)�
cmd_sambatool)� Credentials)�system_session)�SamDBc s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
�SambaDnsUpdateTestsz'Blackbox test case for samba_dnsupdate.c sV t j�d�| _tt| ��� z| �d�}| �d|v |� W d S t jj y* Y d S w )N�
DNS_SERVER_IP�samba_dnsupdate --verboses Looking for DNS entry)
�samba�tests�env_get_var_value� server_ip�superr �setUp� check_output�
assertTrue�BlackboxProcessError)�self�out�� __class__� �F/usr/lib/python3/dist-packages/samba/tests/blackbox/samba_dnsupdate.pyr ! s
�zSambaDnsUpdateTests.setUpc
C sZ z| � d�}W n tjjy" } z
| �d| � W Y d }~nd }~ww | �d|v |� d S )Nr
�!Error calling samba_dnsupdate: %s� No DNS updates needed)r r r r �failr )r r �er r r � test_samba_dnsupate_no_change* s ��z1SambaDnsUpdateTests.test_samba_dnsupate_no_changec
C s* z| � d�}| �d|v |� | �d|v |� W n tjjy" Y nw z| � d�}W n tjjyE } z
| �d| � W Y d }~nd }~ww | �d|v |� z
| � d| j �}W n tjjys } z
| �d| � W Y d }~nd }~ww | �d|v |� | �d|v |� | � d�}| �d|v || � d S ) Nz/samba_dnsupdate --verbose --current-ip=10.0.0.1s DNS updates ands DNS deletes neededz>samba_dnsupdate --verbose --use-nsupdate --current-ip=10.0.0.1r r z=samba_dnsupdate --verbose --use-samba-tool --rpc-server-ip=%sr
)r r r r r r r )r r r �rpc_outr r r �test_samba_dnsupate_set_ip1 s0
�����
z.SambaDnsUpdateTests.test_samba_dnsupate_set_ipc
C s� d}t j| }t� |_t� |_d}|�d| d|� |�d| d|�}|d ur,| �d� tj� � | _
t � | _ | j �
| j
� t� | _| j
�d�}|d }t�||� t| j| j | j
d �| _t�� }t�| jd
�| j�� ��|_t�d �|| j�� �tj d �|d < d
�|�}t | � |��} | � |� � | � | j�!|� t�||� t | � |��} | � d| � | �"|� � | � |�d| d|�}|d ur�| �d� d S d S )N�sites�ABCDEFGHIJKLMNOPQRSTUVWXYZz
samba-tool %s�remove�createzError creating new site�dns_update_cache�_tmp)� session_info� credentials�lpz@CN=DEFAULTIPSITELINK,CN=IP,CN=Inter-Site Transports,CN=Sites,{0}zCN={0},CN=Sites,{1}�siteListz(samba_dnsupdate --verbose --use-file={0}zNo DNS updates neededzError deleting site)#r � subcommandsr �outf�errf�_runr r r � env_loadparmr) r �creds�guessr �session� private_path�shutil�copyfiler �samdb�ldb�Message�Dn�format�get_config_basedn�dn�MessageElement� FLAG_MOD_ADDr r � assertNotIn�lower�modify�assertIn)
r �name�cmd� site_name�result�uc_fn�tmp_uc�m�dns_cr r r r �test_add_new_uncovered_siteI sX
�
����
�z/SambaDnsUpdateTests.test_add_new_uncovered_site) �__name__�
__module__� __qualname__�__doc__r r r rK �
__classcell__r r r r r s r )� samba.testsr �ior � samba.commonr �samba.netcmd.mainr �samba.credentialsr �
samba.authr � samba.samdbr r7 r4 r �BlackboxTestCaser r r r r �<module> s