File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/libsmb.cpython-310.pyc
o
�/a � @ s� d Z ddlmZ ddlmZ ddlmZ ddlm Z m
Z
ddl m Z ddl
mZ ddlZddlZddlZddlZddlZG d d
� d
ejj�Zed krTddlZe�� dS dS ) z Tests for samba.samba3.libsmb.� )�libsmb_samba_internal)�security)�param)� credentials�
NTSTATUSError)�NT_STATUS_DELETE_PENDING)�SMB_ENCRYPTION_REQUIREDNc @ sF e Zd ZG dd� dej�Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
S )�LibsmbTestCasec @ s e Zd Zdd� Zdd� ZdS )zLibsmbTestCase.OpenClosec C s( t j�| � || _|| _|| _d| _d S )NF)� threading�Thread�__init__�conn�filename�num_ops�exc)�selfr
r r � r �4/usr/lib/python3/dist-packages/samba/tests/libsmb.pyr % s
z!LibsmbTestCase.OpenClose.__init__c C sf | j }z t| j�D ]}|j| jdtjd�}|�|d� |�|� q W d S t y2 t
� � | _ Y d S w )N� )�CreateDisposition�
DesiredAccessT)
r
�ranger �creater r �SEC_STD_DELETE�delete_on_close�close� Exception�sys�exc_infor )r �c�i�fr r r �run, s
� � �z LibsmbTestCase.OpenClose.runN)�__name__�
__module__� __qualname__r r" r r r r � OpenClose# s r& c C sR t �� }|�t�d�� t�� }|�|� |�t�d�� |� t�d�� ||fS )N�
SMB_CONF_PATH�USERNAME�PASSWORD)
�s3param� get_context�load�os�getenvr � Credentials�guess� set_username� set_password)r �lp�credsr r r �
prep_creds7 s
zLibsmbTestCase.prep_credsc C s� | � � \}}tjt�d�d||ddd�}g }td�D ]}t�|dt|� d�}|� |� q|D ]}|�
� q/|D ]}|� � |j rK|j d |j d ��q8d S )
N� SERVER_IP�tmpT)�multi_threaded�
force_smb1r �test�
r � )
r5 �libsmb�Connr- r. r r r& �str�append�start�joinr )r r3 r4 r � mythreadsr �tr r r �test_OpenCloseB s" �
��z LibsmbTestCase.test_OpenClosec C sT dt �dd� }| �� \}}|�t� t�t�d�d||�}|� |� |�
|� d S )N�
testing_%dr � r6 r7 � �random�randintr5 �set_smb_encryptionr r= r> r- r. �mkdir�rmdir�r �test_dirr3 r4 r r r r �test_SMB3EncryptionRequiredW s
�
z*LibsmbTestCase.test_SMB3EncryptionRequiredc C sX dt �dd� }| �� \}}|�t� tjt�d�d||dd�}|� |� |�
|� d S )NrF r rG r6 r7 T)r9 rH rN r r r �test_SMB1EncryptionRequiredc s
�
z*LibsmbTestCase.test_SMB1EncryptionRequiredc C s | � � \}}d}t�t�d�d||�}t�t�d�d||�}z|�|� W n Y |�|� |j|tj d�}|�
|d� |� dd� | � t
��}|�d|d � W d � n1 s[w Y |jjd tkrltd
��|�
|d � |�|� z
|�|� |�d� W d S Y d S ) Nz \dst-subdirr6 r7 )r r< z\src.txts Contentz\dst.txtr z$Rename must fail with DELETE_PENDING)r5 r= r> r- r. �deltreerL r r r r �savefile� assertRaisesr �rename� exception�argsr �AssertionErrorr �unlink)r r3 r4 �dstdir�c1�c2�dnum�cmr r r �test_RenameDstDelOnCloseo s0
�
z'LibsmbTestCase.test_RenameDstDelOnCloseN) r# r$ r% r
r r& r5 rE rP rQ r_ r r r r r ! s r �__main__)�__doc__� samba.samba3r r= � samba.dcerpcr r r* �sambar r �samba.ntstatusr �samba.credentialsr � samba.testsr
r r- rI �tests�TestCaser r# �unittest�mainr r r r �<module> s" n �