File: //lib/python3/dist-packages/samba/tests/samba_tool/__pycache__/computer.cpython-310.pyc
o
��a�? � @ sX d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dl m
Z
G dd� de�Z dS )� N)�SambaToolCmdTest)�dsdb)�
ndr_unpack�ndr_pack)�dnspc 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�Zi fdd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Z� ZS )#�ComputerCmdTestCasez)Tests for samba-tool computer subcommandsNc
s� t t� ��� dtjd tjd f � _� �ddtjd � j�� _d� _d� _ � �
� � � }|d � jgd
�|d � j gd | d
gd�|d � j� j gd | d d | d gd�d|d ig}� fdd�|D �� _ � j D ]i}� �
|�\}}}� �|||� � �d|d� � �d|d |� � �|d �}� �|� |d �d�}|d } |d �d�s�d|d } � �d|�d� |� � �d|�d � | � � �d|�d � |d � qmd S ) Nz-U%s%%%s� DC_USERNAME� DC_PASSWORDz-Hz ldap://%s� DC_SERVERz 10.10.10.10z'2001:0db8:0a0b:12f0:0000:0000:0000:0001�cmp1)�name�ip_address_list�cmp2zhost/�SPN20)r r
�service_principal_name_listzcmp3$�SPN30�SPN31r zcmp4$c s g | ]}� j |d ��qS ))�base)�_randomComputer)�.0�item��self� �A/usr/lib/python3/dist-packages/samba/tests/samba_tool/computer.py�
<listcomp>C s z-ComputerCmdTestCase.setUp.<locals>.<listcomp>�ERROR�$There shouldn't be any error message� Computer '%s' added successfully�$�%s$�%s�sAMAccountName� description)�superr �setUp�os�environ�creds�getSamDB�samdb�ipv4�ipv6�
randomName�lower� computers�_create_computer�assertCmdSuccess� assertNotIn�assertIn�_find_computer�assertIsNotNone�rstrip�endswith� assertEqual�get)
r �computer_basename�data�computer�result�out�err�found� expectedname�expectedsamaccountname�� __class__r r r% # sb �
��
�� ��
���
���zComputerCmdTestCase.setUpc s` t t| ��� | jD ]#}| �|d �r-| �ddd|d �\}}}| �|||d|d � q
d S )Nr r< �deleter! � Failed to delete computer '%s')r$ r �tearDownr/ r4 � runsubcmdr1 )r r< r= r> r? rC r r rG ] s
�
����z ComputerCmdTestCase.tearDownc C s6 | j D ]}|�dg �}| �|d |�}| �|� qd S )Nr r )r/ r9 � _find_service_principal_name�
assertTrue)r r< �expected_namesr@ r r r �,test_newcomputer_with_service_principal_nameh s
�z@ComputerCmdTestCase.test_newcomputer_with_service_principal_namec C s� | j D ]}|�dg �D ]}| �|d |�}| �|� q q| j D ]-}| �ddd|d �\}}}| �|||d|d � | �|d �}| �|d|d � q | j D ]}|�dg �D ]}| �|d |�}| �|� qXqPd S )Nr
r r< rE r! rF �"Deleted computer '%s' still exists) r/ r9 �_find_dns_recordrJ rH r1 r4 � assertIsNone� assertFalse)r r< �
ip_addressr@ r= r> r? r r r �!test_newcomputer_with_dns_recordso s4
�
�
����
��z5ComputerCmdTestCase.test_newcomputer_with_dns_recordsc C s� | j D ]}| �|�\}}}| �|d� | �d|� q| j D ]-}| �ddd|d �\}}}| �|||d|d � | �|d �}| �|d|d � q | j D ]o}| �dd d|d d
|d �\}}}| �|||� | �|d d
� | �d|d |� | �|d �}|d � d�}|d }|d �
d�s�d|d }| �d|� d� |� | �d|� d� |� | �d|� d � |d � qNdS )z<This tests the "computer add" and "computer delete" commandsz"Succeeded to add existing computerzalready existsr< rE r! r rF rM �add�--description=%sr# � r r r r r" N) r/ r0 �
assertCmdFailr3 rH r1 r4 rO r8 r6 r7 r9 )r r< r= r> r? r@ rA rB r r r �test_newcomputer� sX
�
����
��� ���z$ComputerCmdTestCase.test_newcomputerc C s� | � dd�\}}}| �|||d� dtj }| jj| j�� tj|dgd�}| � t
|�dkd� |D ]}|j ddd �}| � |t
|�d
| �}q1d S ) Nr< �list�Error running list�(sAMAccountType=%u)�samaccountname�r �scope�
expression�attrsr �no computers found in samdb��idx�computer '%s' not found�rH r1 r �ATYPE_WORKSTATION_TRUSTr* �search� domain_dn�ldb�
SCOPE_SUBTREErJ �lenr9 � assertMatch�str� r r= r> r? �
search_filter� computerlist� computerobjr r@ r r r � test_list� s" �� ��z ComputerCmdTestCase.test_listc C s� | � ddd�\}}}| �|||d� dtj }| jj| j�� tj|g d�}| � t
|�dkd� |D ]}|j d dd
�}| � |t
|�d | �}q1d S ) Nr< rX z --full-dnrY rZ r\ r r` �dnra rc rd rm r r r �test_list_full_dn� s" �� ��z%ComputerCmdTestCase.test_list_full_dnc
C s� t | j�� �}| �ddd|�\}}}| �|||d� dtj }| jj|tj |dgd�}| �
t |�dkd � |D ]}|j ddd
�}| �
|t |�d | �} q7d S ) Nr< rX z-brY rZ r r\ r r` ra rc )rl r* rg rH r1 r re rf rh ri rJ rj r9 rk )
r �base_dnr= r> r? rn ro rp r r@ r r r �test_list_base_dn� s$ �� ��z%ComputerCmdTestCase.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 �
d
�}t � | j
d||d | j
�� f �} | �|�d�| d|d � | �ddd|d d|�� �\}}}| �|||d|d � q| �ddd|d �\}}}| �|||d|d � d S )Nr �parentOUrr r< �mover! �OU=%sz Failed to move computer '%s'rU r zMoved computer "%s"z1Moved computer '%s' still exists with the same dnr zCN=%s,OU=%s,%sz"Moved computer '%s' does not exist�ourE zFailed to delete ou '%s')� _randomOU�
_create_our1 r/ r4 r9 rH r8 r3 �assertNotEqualsr6 rh �Dnr* rg �parent)
r �parentour= r> r? r< �olddnr@ � computername�
newexpecteddnr r r � test_move� sd
�
���������
�
��
�
�z ComputerCmdTestCase.test_movec C �$ | � � | j dd�d�}|�|� |S )zWcreate a computer with random attribute values, you can specify base
attributes�d ��count�r r# �r- �update)r r r< r r r r �
�
z#ComputerCmdTestCase._randomComputerc C r� )zRcreate an ou with random attribute values, you can specify base
attributesr� r� r� r� )r r ry r r r rz # r� z ComputerCmdTestCase._randomOUc C st d� |d | j|d �}|�dg �D ] }|d� |�7 }q|�dg �D ] }|d� |�7 }q"|�� }| jdd g|�R � S )
Nz{0} {1} --description={2}r r# r
z --ip-address={0}r z --service-principal-name={0}r< rS )�formatr( r9 �splitrH )r r< �argsrQ �service_principal_namer r r r0 . s �z$ComputerCmdTestCase._create_computerc C s | � ddd|d d|d �S )Nry rS rx r rT r# )rH )r ry r r r r{ < s
�z ComputerCmdTestCase._create_ouc C sZ |}|� d�s d| }dt�|�d| j�� f }| jj| j�� tj|d�}|r+|d S d S )Nr r �,(&(sAMAccountName=%s)(objectCategory=%s,%s))�&CN=Computer,CN=Schema,CN=Configuration)r r] r^ r )r7 rh �
binary_encoder* rg rf ri )r r r[ rn ro r r r r4 @ s
���z"ComputerCmdTestCase._find_computerc C s� |� d�}| jjd�| j�� �tjd�|�ddgd�}|D ]-}d|v r,t|d �dkr,q |d D ]}tt j
|�}t|j �}t|�t|�krI dS q0q d S )
Nr zDC=DomainDnsZones,{0}z"(&(objectClass=dnsNode)(name={0}))� dnsRecord�
dNSTombstonedr\ �TRUETF) r6 r* rf r� �get_default_basednrh ri rl r r �DnssrvRpcRecordr; )r r rQ �records�record�dns_record_bin�dns_record_obj�ipr r r rN P s$
�
��z$ComputerCmdTestCase._find_dns_recordc C s� |� d�d }dt�|�d| j�� f }| jj| j�� tj|dgd�}t� }|D ]}|�dg �D ] }|� t
|�� q0q(|t|�kS )zVFind all servicePrincipalName values and compare with expected_service_principal_namesr r� r� �servicePrincipalNamer\ ) �striprh r� r* rg rf ri �setr9 rS rl )r r � expected_service_principal_namesr[ rn �
computer_list�namesr< r r r rI f s& ���� z0ComputerCmdTestCase._find_service_principal_name)�__name__�
__module__� __qualname__�__doc__r/ r* r% rG rL rR rW rq rs ru r� r rz r0 r{ r4 rN rI �
__classcell__r r rC r r s( : + * r ) r& rh �samba.tests.samba_tool.baser �sambar � samba.ndrr r � samba.dcerpcr r r r r r �<module> s