File: //lib/python3/dist-packages/samba/tests/dcerpc/__pycache__/array.cpython-310.pyc
o
�/ae � @ sP d Z ddlmZ ddlZddlmZmZ ddlZddl Z G dd� dej
j �Z dS )zBTests for array handling in PIDL generated bindings samba.dcerpc.*� )�drsblobsN)�
ndr_unpack�ndr_packc sP e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�
ArrayTestsc s$ t t| ��� t�� t�� | _d S )N)�superr �setUp�talloc�enable_null_tracking� total_blocks�startup_blocks��self�� __class__� �:/usr/lib/python3/dist-packages/samba/tests/dcerpc/array.pyr s zArrayTests.setUpc s> t t| ��� t�� t�� | jkr t�� | � d� d S d S )Nz it appears we are leaking memory)
r r �tearDown�gc�collectr r
r � report_full�failr r r r r " s �zArrayTests.tearDownc C s� t �� }d|_t �� |_d|j_t �� }d|_d|_t �� }d|_d|_t �� }d|_d|_|||g|j_t �
� | � |� | � |jd� | � |j� | � |jjd� | � t
|jj�|jj� | � |jjd � | � |jjd jd� d S �N� � � r )r �replPropertyMetaDataBlob�version�replPropertyMetaDataCtr1�ctr�count�replPropertyMetaData1�attid�arrayr r �assertIsNotNone� assertEqual�len�r
�rmd�rmd1�rmd2�rmd3r r r �test_array_from_python) s,
z!ArrayTests.test_array_from_pythonc C s� zvt �� }d|_t �� |_d|j_t �� }d|_d|_t �� }d|_d|_t �� }d|_d|_|||g|j_t �
� | � |� | � |jd� | � |j� | � |jjd� | � t
|jj�|jj� | � |jjd � | � |jjd jd� t� � Y d S r )r r r r r r r r! r" r r r# r$ r% � Exceptionr& r r r �test_array_with_exceptionG s4
z$ArrayTests.test_array_with_exceptionc C s� dd� }|� }t �� | �|� | �|jd� | �|j� | �|jjd� | �t|jj�|jj� | �|jjd � | �|jjd j d� d S )Nc S sn t �� } d| _t �� | _d| j_t �� }d|_d|_t �� }d|_d|_t �� }d|_d|_|||g| j_| S )Nr r r ) r r r r r r r r! r" )r' r( r) r* r r r �get_rmdl s
z;ArrayTests.test_array_from_python_function.<locals>.get_rmdr r r )
r r r# r$ r r r r% r" r! )r
r. r' r r r � test_array_from_python_functionk s
z*ArrayTests.test_array_from_python_functionc C s t �� }d|_t �� |_d|j_t �� }d|_d|_t �� }d|_d|_t �� }d|_d|_|||g|j_t |�}t
� � t t j|�}| �
|� | �|jd� | �
|j� | �|jjd� | �t|jj�|jj� | �
|jjd � | �|jjd jd� | �|jjd j|jjd j� d S r )r r r r r r r r! r" r r r r r# r$ r% )r
r' r( r) r* �packed� rmd_unpackedr r r �test_array_from_ndr� s6
�z ArrayTests.test_array_from_ndrc
C s� t �� }d|_t �� |_d|j_t �� }d|_d|_t �� }d|_d|_t �� }d|_d|_|||g|j_z |`| � d� W n t
yP } zW Y d }~nd }~ww z |j`| � d� W d S t
yo } zW Y d }~d S d }~ww )Nr r r z"succeeded in deleting rmd1.versionz#succeeded in deleting rmd.ctr.array) r r r r r r r r! r" r �AttributeError)r
r' r( r) r* �er r r �test_array_delete� s8
����z ArrayTests.test_array_delete) �__name__�
__module__� __qualname__r r r+ r- r/ r2 r5 �
__classcell__r r r r r s $!$r )
�__doc__� samba.dcerpcr � samba.tests�samba� samba.ndrr r r r �tests�TestCaser r r r r �<module> s