File: //lib/python3/dist-packages/samba/tests/__pycache__/dsdb_schema_attributes.cpython-310.pyc
o
��9a�# � @ sH d dl Z d dlZd dlZd dlZd dlmZmZ G dd� dejj�Z dS )� N)�
SCOPE_BASE�LdbErrorc sr e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zdd
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )�SchemaAttributesTestCasec s� t t| ��� tj�� | _tj�| j�� �| _ | j j
ddt dgd�}| � t
|�d� |d d d | _|d d d | _t|d d d �| _d S ) N� �*)�base�
expression�scope�attrs� r �schemaNamingContext�defaultNamingContext�forestFunctionality)�superr �setUp�samba�tests� env_loadparm�lp�
connect_samdb� samdb_url�samdb�searchr � assertEqual�len� schema_dn�base_dn�int� forest_level)�self�res�� __class__� �D/usr/lib/python3/dist-packages/samba/tests/dsdb_schema_attributes.pyr $ s z SchemaAttributesTestCase.setUpc s t t| ��� d S �N)r r �tearDown)r r! r# r$ r& 1 s z!SchemaAttributesTestCase.tearDownc C s d}| j �|� d S )Nz@
dn:
changetype: modify
add: schemaUpdateNow
schemaUpdateNow: 1
)r � modify_ldif)r �ldifr# r# r$ �_ldap_schemaUpdateNow4 s z.SchemaAttributesTestCase._ldap_schemaUpdateNowc C s8 |t �dt �� � }|�dd�}d|| jf }|||fS )Nz%s�-r zCN=%s,%s)�time�strftime�gmtime�replacer )r �prefix�obj_name�
obj_ldap_name�obj_dnr# r# r$ �_make_obj_names= s
z(SchemaAttributesTestCase._make_obj_namesNc C sT d| d | d | d | d| t t�dd�� d }|d ur(||d 7 }|S )
Nz
dn: zA
objectClass: top
objectClass: attributeSchema
adminDescription: z
adminDisplayName: z
cn: z*
attributeId: 1.3.6.1.4.1.7165.4.6.1.8.%d.r i�� z_
attributeSyntax: 2.5.5.12
omSyntax: 64
instanceType: 4
isSingleValued: TRUE
systemOnly: FALSE
�
)�str�random�randint)r � attr_name�attr_dn�sub_oid�extrar( r# r# r$ �_make_attr_ldifC s4 ����������� z(SchemaAttributesTestCase._make_attr_ldifc C s� | � d�\}}}| �||ddtjj �}| j�|� | �� | jjdt j
d�}| � ||d � | � t
|d | �d� | � t|d | d �d� | jjdt j
d�}| � |d d
� |d d D �� d S ) NzschemaAttributes-IdxAttr-r �searchFlags: %d� @ATTRIBUTES�r r r �CASE_INSENSITIVE�
@INDEXLISTc S � g | ]}t |��qS r# �r5 ��.0�xr# r# r$ �
<listcomp>n � zESchemaAttributesTestCase.test_AddIndexedAttribute.<locals>.<listcomp>�@IDXATTR�r3 r< r �dsdb�SEARCH_FLAG_ATTINDEXr �add_ldifr) r �ldbr �assertInr r r5 �r r8 �attr_ldap_namer9 r( �attr_res�idx_resr# r# r$ �test_AddIndexedAttributeX s
� "z1SchemaAttributesTestCase.test_AddIndexedAttributec C s� | � d�\}}}| �||d�}| j�|� | �� | jjdtjd�}| �||d � | � t
|d | �d� | � t |d | d �d� | jjdtjd�}| � |d d
� |d d D �� d S ) NzschemaAttributes-UnIdxAttr-� r> r? r r r@ rA c S rB r# rC rD r# r# r$ rG � rH zGSchemaAttributesTestCase.test_AddUnIndexedAttribute.<locals>.<listcomp>rI )
r3 r< r rM r) r rN r rO r r r5 � assertNotInrP r# r# r$ �test_AddUnIndexedAttributep s "z3SchemaAttributesTestCase.test_AddUnIndexedAttributec
C sd | � d�\}}}| �||ddtjj �}| j�|� | �� | � d�\}}}| �||ddtjj �}| j�|� | �� | jjdt j
d�}| � ||d � | � t
|d | �d � | � t|d | d �d
� | � ||d � | � t
|d | �d � | � t|d | d �d
� | jjd t j
d�} | � |d d
� | d d D �� | � |dd
� | d d D �� d S )NzschemaAttributes-2IdxAttr-� r= zschemaAttributes-Attr-� r> r? r r r@ rA c S rB r# rC rD r# r# r$ rG � rH zISchemaAttributesTestCase.test_AddTwoIndexedAttributes.<locals>.<listcomp>rI c S rB r# rC rD r# r# r$ rG � rH rJ )
r r8 rQ r9 r( �
attr_name2�attr_ldap_name2�attr_dn2rR rS r# r# r$ � test_AddTwoIndexedAttributes� s,
�
� "z5SchemaAttributesTestCase.test_AddTwoIndexedAttributesc C �J ddgd�}t j�| j|t j�}| j�|� | jjdt jdgd�}| �t |�d� | �t
|d j �d� | �t |d �d� | � d|d v � | �t |d d �d� | �t
|d d d �d� t
j�| j�� �}|�� |�� | jjdt jdgd�}| �t |�d� | �t
|d j �d� | �t |d �d� | �d|d v � d S )Nr> �HIDDEN��dn� @TEST_EXTRArb �r r r
r r �rN �Message� from_dictr � FLAG_MOD_ADD�modifyr r r r r5 ra �
assertTruer r r r r �transaction_start�transaction_commit� assertFalse�r �m�msgr �samdb2r# r# r$ �test_modify_at_attributes� �0 � � �z2SchemaAttributesTestCase.test_modify_at_attributesc C r^ )NrA �1r` rb rc r r rd rm r# r# r$ �test_modify_at_indexlist� rr z1SchemaAttributesTestCase.test_modify_at_indexlistc
C s| ddgd�}t j�| j|t j�}z| j�|� | �d� W d S ty= } z|jd }| � |t j
� W Y d }~d S d }~ww )NrA rs )ra z@TEST_NOT_EXTRAz9modify of @INDEXLIST with a failed constraint should failr ) rN re rf r �FLAG_MOD_DELETErh �failr �argsr �ERR_NO_SUCH_ATTRIBUTE)r rn ro �err�enumr# r# r$ � test_modify_fail_of_at_indexlist� s �
��z9SchemaAttributesTestCase.test_modify_fail_of_at_indexlistr% )�__name__�
__module__� __qualname__r r& r) r3 r< rT rW r] rq rt r{ �
__classcell__r# r# r! r$ r " s
&""r )
r+ r6 � samba.testsr rN r r r �TestCaser r# r# r# r$ �<module> s