File: //lib/python3/dist-packages/samba/tests/__pycache__/complex_expressions.cpython-310.pyc
o
�/a|I � @ s` d dl Z d dlZd dlmZ d dlZd dlZd dlZd dlm Z d dl
m Z d dl Z dZ
dZdZdZe �d�ejd ��Ze�e�Ze�e�Ze�e� ejd d
d d d
� e�� \Z Z e e �dk rje� � e� d� e�!� Z"e�#e"�Z$dZ%d a&G dd� de �Z'e(dkr�d dl)m*Z* e d Z%de%vr�ej+�,e%�r�de% Z%nde% Z%e*e(d� dS de�-d� Z%dS )� N)�system_session)�TestCase�D � � �A z
{0} <host>z-v�
store_true�verbosez#print successful expression outputs)�action�dest�helpc s e Zd Ze� fdd��Zedd� �Zdd� Zdd� Zd d
� Zd8d d
�Z dd� Z
dd� Z dd� Z dd� Z
d8dd�Zdd� Zdd� Zd d � Zd8d d �Zd d!� Zd"d#� Zd$d%� Zd8d&d'�Zd(d)� Zd*d+� Zd8d,d-�Zd.d/� Zd0d1� Zd9d2d3�Z d4d5� Z d6d7� Z � Z S ):�ComplexExpressionTestsc
s� t t| ��� tjjttt� t d�| _d}d�
|| j� � �| _ z | j�
| j dg� W n Y z | j�| j � W n tjyV } z|jd tkrPtdj
|d�� |�d }~ww d| _d | _d
| _d | _d | _d
| _d| _d| _d S )N)�lp� session_info� credentials�ComplexExprTestz
OU={0},{1}�
tree_delete:1r zatest ou {ou} already exists. Delete with "samba-tool group delete OU={ou} --force-subtree-delete")�ouz testuser{0}�
�accountExpires�accountNameHistory�flags�preferredDeliveryMethod�msTSExpireDate� countryCode) �superr
�
setUpClass�samba�samdb�SamDB�hostr r �creds�format� domain_dn�base_dn�delete� create_ou�ldb�LdbError�args�ERRCODE_ENTRY_EXISTS�print�
name_template� default_n�
largeint_f�str_f�int_f�enum_f�time_f� ranged_int_f)�cls�ou_name�e�� __class__� �A/usr/lib/python3/dist-packages/samba/tests/complex_expressions.pyr = s:
����
z!ComplexExpressionTests.setUpClassc C s | j �| jdg� d S )Nr )r r% r$ )r4 r9 r9 r: �
tearDownClassa s z$ComplexExpressionTests.tearDownClassc s� t d7 a d�t �j���j��� � ��fdd�|D �}|D ](}dd� |�� D �}z�j�|� W q tjyF } zt d| � |�d }~ww �|fS )Nr zOU=testou{0},{1}c
s0 g | ]}d d� |��d�j� |�dd� |i�qS )�dnzCN=testuser{0},{1}�name� objectClass�user)r" r, ��.0�n��field�ou_dn�selfr9 r: �
<listcomp>l s � �z<ComplexExpressionTests.make_test_objects.<locals>.<listcomp>c S s i | ] \}}|t |��qS r9 ��str)rA �k�vr9 r9 r: �
<dictcomp>u s �z<ComplexExpressionTests.make_test_objects.<locals>.<dictcomp>zfailed to add %s)
�ou_countr" r$ r r& �items�addr' r( r+ )rF rD �vals� ldap_objects� ldap_object�stringed_ldap_objectr6 r9 rC r: �make_test_objectsf s$ �� ��z(ComplexExpressionTests.make_test_objectsc
C sr d}zt � � }| jj|tj|d�}t � � | }W n ty, } ztd| � |�d }~ww td�||�� ||fS )Nr ��base�scope�
expressionz failed expr z
{0} took {1}s)�timer �searchr' �
SCOPE_SUBTREE� Exceptionr+ r" )rF �exprr< �
time_taken�
start_time�resr6 r9 r9 r: �time_ldap_search� s � ��z'ComplexExpressionTests.time_ldap_searchc C s� | � ||�\}}dd� |D �}t� }|D ]#} z
|jdi | ��}
W n ty) Y qw t|
�r7|�t| d �� q| �||� tj rfdd� |D �} | | } d�
| �pRd} d�
|�pYd}
t d�|||
| �� d S d S ) Nc S s h | ] }t |�d �d ��qS �r= r �rI �get)rA �rowr9 r9 r: � <setcomp>� � z9ComplexExpressionTests.assertLDAPQuery.<locals>.<setcomp>r= c S s h | ]}|d �qS )r= r9 �rA �lr9 r9 r: rf � � z
z [NOTHING]zCPASS: Expression {0} took {1}s and returned:
{2}
Excluded:
{3}
r9 ) ra �setr" �KeyError�evalrO rI � assertEqual�optsr �joinr+ )rF � ldap_exprrE �py_exprrQ r` r^ �results�expected_resultsrR �
final_expr�ldap_object_names�excluded�returnedr9 r9 r: �assertLDAPQuery� s4 �� ���z&ComplexExpressionTests.assertLDAPQueryNc C s� | j }|p| j}| �|t|��\}}d||d ||d f }d|d ||d f }| �||||� t|d �}d||f }d||f }| �||||� d||f }d||f }| �||||� d S ) N�(&(%s>=%s)(%s<=%s))r �%d <= {%s} <= %d� �(%s<=%s)�
{%s} <= %dz(%s>=%s)�
{%s} >= %d)r- r0 rT �rangery �int)rF rD rB rE rQ r] rr �half_nr9 r9 r: �test_int_range� s
z%ComplexExpressionTests.test_int_rangec C � | � | j� d S �N)r� r. �rF r9 r9 r: �test_largeint_range� � z*ComplexExpressionTests.test_largeint_rangec C r� r� )r� r1 r� r9 r9 r: �test_enum_range� r� z&ComplexExpressionTests.test_enum_rangec C sd | j }d}d}tt|| |��}| �||�\}}d||d f }d||d f }| �||||� d S )Ni � �(%s<=%d)� r~ )r3 �listr� rT ry ) rF rD �ubound�widthrP rE rQ r] rr r9 r9 r: �test_ranged_int_range� s z,ComplexExpressionTests.test_ranged_int_rangec s� | j }| j}| j }t|d �}d� � fdd�t| |�D �}dd� |D �}| �||�\}}d|t� �d f }d|� f }| �||||� d |t� d
�d |t� d
�d f }d � d
|� d
f }| �||||� d S ) Nr| l �f�Hc s g | ]}� | �qS r9 r9 �rA �t�� base_timer9 r: rG � rj z:ComplexExpressionTests.test_time_range.<locals>.<listcomp>c S s g | ]}t |�d �qS )�.0ZrH r� r9 r9 r: rG � � r} r� zint("{%s}"[:-3]) <= %drz r z %d <= int("{%s}"[:-3]) <= %d)r- r2 r� r� rT rI ry ) rF rB rD r� �
time_rangerE rQ r] rr r9 r� r: �test_time_range� s& ��z&ComplexExpressionTests.test_time_rangec
s� | j �� p| j� | �� t���\}}g d�}g d�}� �fdd�|D �}� �fdd�|D �}t||�D ] \}} | �||| |� q3d S )N)�=�<=�>=)z==r� r� c � g | ] }d � |�f �qS )z(%s%s%d)r9 �rA �c�rD rB r9 r: rG � � zHComplexExpressionTests.test_int_single_cmp_op_speeds.<locals>.<listcomp>c r� )z{%s}%s%dr9 r� r� r9 r: rG r� )r- r0 rT r� �zipry )
rF rD rE rQ �comp_ops� py_comp_ops�exprs�py_exprsr] rr r9 r� r: � test_int_single_cmp_op_speeds� s
�z4ComplexExpressionTests.test_int_single_cmp_op_speedsc C r� r� )r� r. r� r9 r9 r: �"test_largeint_single_cmp_op_speeds r� z9ComplexExpressionTests.test_largeint_single_cmp_op_speedsc C r� r� )r� r1 r� r9 r9 r: � test_enum_single_cmp_op_speeds r� z5ComplexExpressionTests.test_enum_single_cmp_op_speedsc
C s | j }td�}d}dd� t||| �D �}| �||�\}}t||d �}d|t|�f }d|t|�f } | �||| |� d|t|�f }d |t|�f } | �||| |� d
|t|d �|t|d �f }d t|d �|t|d �f } | �||| |� d | }d
| } | �||| |� d S )N�Ar c S s g | ] }d � t|���qS )zabc{0}d)r" �chrr� r9 r9 r: rG r� z<ComplexExpressionTests.test_str_ordering.<locals>.<listcomp>r| z (%s>=abc%s)z'{%s}' >= 'abc%s'z (%s<=abc%s)z'{%s}' <= 'abc%s'z(&(%s>=abc%s)(%s<=abc%s))z 'abc%s' <= '{%s}' <= 'abc%s'z(%s>=1)z
'{%s}' >= '1')r/ �ordr� rT r� r� ry )
rF rD �a_ordrB � str_rangerE rQ r� r] rr r9 r9 r: �test_str_ordering s. �
�z(ComplexExpressionTests.test_str_orderingc C sV |p| j }| j}| �|tt|���\}}g d�}|D ]}|| }| �||d|� qd S )N)z (%s>=abc)z (%s<=abc)z(%s=abc)�False)r0 r- rT r� r� ry )rF rD rB rE rQ �int_expressionsr] r9 r9 r: �test_invalid_expressions+ s
�z/ComplexExpressionTests.test_invalid_expressionsc C r� r� )r� r. r� r9 r9 r: �!test_largeint_invalid_expressions7 r� z8ComplexExpressionTests.test_largeint_invalid_expressionsc C r� r� )r� r1 r� r9 r9 r: � test_enum_invalid_expressions: r� z4ComplexExpressionTests.test_enum_invalid_expressionsc C sf dd� t d�D �}| �| j|�\}}d| j }d| j }| �||||� d| j }| �||||� d S )Nc S � g | ]}d t |� �qS )u äbcrH r@ r9 r9 r: rG > r� z@ComplexExpressionTests.test_case_insensitive.<locals>.<listcomp>r u
(%s=äbc1)u "{%s}"=="äbc1"u
(%s=ÄbC1))r� rT r/ ry )rF r� rE rQ r] �pyexprr9 r9 r: �test_case_insensitive= s
z,ComplexExpressionTests.test_case_insensitivec C sR |p| j }d}tt| |��}| �||�\}}d| }d| }| �||||� d S )N� z(%s>=-3)z
{%s} >= -3)r0 r� r� rT ry )rF rD r� � around_zerorE rQ r] rr r9 r9 r: �test_negative_cmpI s
z(ComplexExpressionTests.test_negative_cmpc C r� r� )r� r. r� r9 r9 r: �test_negative_cmp_largeintS r� z1ComplexExpressionTests.test_negative_cmp_largeintc C r� r� )r� r1 r� r9 r9 r: �test_negative_cmp_enumV r� z-ComplexExpressionTests.test_negative_cmp_enumc
sp � p| j � | j}| �� tt|���\}}d� |d f }d� |d f }| �||||� | j�|dg� dd� t|�D �}z
| �� |�\}}W n tj yd } z|j
d t krY|�W Y d }~d S d }~ww t d� � | jj
|tjd d
�} � fd d�| D �}
d d� t|�D �} | �t|
�t| �� d
� |d f }d� |d f }|D ]
} t| � �| � < q�| �||||� d S )Nz
(%s>=00%d)r| r r c S r� )�00rH )rA �numr9 r9 r: rG j r� z;ComplexExpressionTests.test_zero_prefix.<locals>.<listcomp>r z4LDAP allowed insertion of 0-prefixed nums for field �(objectClass=user)rU c � g | ] }t |�� �d ��qS �r rc �rA �r�rD r9 r: rG z rg c S � g | ]}t |��qS r9 rH r@ r9 r9 r: rG { rj �(%s>=%d))r0 r- rT r� r� ry r r% r' r( r) �ERRCODE_INVALID_VALUEr+ rZ r[ rn rk r� )
rF rD rB rE rQ r] rr �zero_pref_numsr6 r` �
returned_nums�expectrR r9 r� r: �test_zero_prefix[ s:
�� �z'ComplexExpressionTests.test_zero_prefixc C r� r� )r� r. r� r9 r9 r: �test_zero_prefix_largeint� r� z0ComplexExpressionTests.test_zero_prefix_largeintc C r� r� )r� r1 r� r9 r9 r: �test_zero_prefix_enum� r� z,ComplexExpressionTests.test_zero_prefix_enumc sd � p| j � |pd}d}tt|| || ��}| �� |�\}}d� |d f }d|d � |f }| �||||� d� |f }d}| �||||� d� |� f }d � |f }| �||||� d
� }d � |d f }| �||||� | jj|tjd
d�} � fdd�| D �}
tt|| |d ��} tt| d | | d ��} dd� | | D �}
| � t
|
�t
|
�� d S )N��r� r� � r{ r� �Truez(&(%s<=%d)(%s>=0))r~ z(%s<=0)r r r� rU c r� r� rc r� r� r9 r: rG � rg z<ComplexExpressionTests.test_int_overflow.<locals>.<listcomp>r| c S r� r9 rH r@ r9 r9 r: rG � rj ) r0 r� r� rT ry r rZ r' r[ rn rk )rF rD �ofr� rP rE rQ r] rr r` r� �up_to_overflow� negativesr� r9 r� r: �test_int_overflow� s4
� z(ComplexExpressionTests.test_int_overflowc C s | � | jd� d S )Nr� )r� r1 r� r9 r9 r: �test_enum_overflow� s z)ComplexExpressionTests.test_enum_overflowc
s� d� d}| � � tt|���\}}| jj|tjdd�}� fdd�|D �}� fdd�|D �}tt|��}|tt |�d � }d � |f }d
� |f } | �
||| |� d � |f }d � |f } | �
||| |� d S )
N�
uSNChangedr r� rU c s0 g | ]}d t |d d �� t|� d �i�qS rb )rI r� r� r� r9 r: rG � s
��z:ComplexExpressionTests.test_usnchanged.<locals>.<listcomp>c s g | ]}|� �qS r9 r9 rh r� r9 r: rG � rj r| z (&(%s<=%d)(objectClass=user))r~ z (&(%s>=%d)(objectClass=user))r ) rT r� r� r rZ r' r[ �sortedr� �lenry )
rF rB rE �_r` rQ �nums�
search_numr] rr r9 r� r: �test_usnchanged� s& �
� z&ComplexExpressionTests.test_usnchangedr� )NN) �__name__�
__module__� __qualname__� classmethodr r; rT ra ry r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r9 r9 r7 r: r
: s< #
$
*
'r
�__main__)� TestProgramz://ztdb://%sz ldap://%s)�modulezldap://�SERVER).�optparser � samba.getopt�getopt�options�sys�osrY �
samba.authr � samba.testsr r' r* �ERRCODE_OPERATIONS_ERRORr� �ERRCODE_CLASS_VIOLATION� OptionParserr" �argv�parser� SambaOptions� sambaopts�CredentialsOptions�credopts�add_option_group�
add_option�
parse_argsro r) r� � print_usage�exit� get_loadparmr �get_credentialsr! r rM r
r� �samba.tests.subunitrunr� �path�isfile�getenvr9 r9 r9 r: �<module> sR
�
%